D,j dZddlZddlZddlZddlZddlmZddlmZdZ dZ dZ eZ ddZ dd Zdd Zdd Zdd ZdS)a1 Editor module for handling system text editor interactions. This module provides functionality to: - Discover and launch the system's configured text editor - Create and manage temporary files for editing - Handle editor preferences from environment variables - Support cross-platform editor operations N)Console)dumpvivimnotepadcl||rdnd}t||tddS)z Print a status message with appropriate styling. :param success: Whether the operation was successful :param message: The message to display :param style: Optional style override. If None, uses green for success and red for failure Nz bold greenzbold red)style)consoleprint)successmessager s [/home/rurouni/.local/share/uv/tools/aider-chat/lib/python3.11/site-packages/aider/editor.pyprint_status_messagers< } '7 Z MM'M''' "IIIIIr c||d}|rd||d<tjdi|\}} tj|d5}||dddn #1swxYwYn##t $rtj|wxYw|S)a Create a temporary file with the given input data. :param input_data: Content to write to the temporary file :param suffix: Optional file extension (without the dot) :param prefix: Optional prefix for the temporary filename :param dir: Optional directory to create the file in :return: Path to the created temporary file :raises: OSError if file creation or writing fails )prefixdir.suffixwN)tempfilemkstemposfdopenwrite Exceptionclose) input_datarrrkwargsfdfilepathfs rwrite_temp_filer%)s s + +F ('v<<x#--f--LB Yr3   1 GGJ                      Os.A(A A(A  A(#A $A(( Bctjdtjd|}|S)a Fetches the preferred editor from the environment variables. This function checks the following environment variables in order to determine the user's preferred editor: - VISUAL - EDITOR :param default: The default editor to return if no environment variable is set. :type default: str or None :return: The preferred editor as specified by environment variables or the default value. :rtype: str or None VISUALEDITOR)renvironget)defaulteditors rget_environment_editorr-Fs.Z^^HbjnnXw&G&G H HF Mrctj}|dkrt}n|dkrt}nt}|r|}nt |}|S)a  Discovers and returns the appropriate editor command. Handles cases where the editor command includes arguments, including quoted arguments with spaces (e.g. 'vim -c "set noswapfile"'). :return: The editor command as a string :rtype: str WindowsDarwin)platformsystemDEFAULT_EDITOR_WINDOWSDEFAULT_EDITOR_OS_XDEFAULT_EDITOR_NIXr-)editor_overrider2default_editorr,s rdiscover_editorr8Ys[_  F / 8  ,+8 '77 Mrclt||}t|}|d|zz }tj|dt |d5}|}dddn #1swxYwY t j|n$#t$rtdd|dYnwxYw|S) ay Opens the system editor with optional input data and returns the edited content. This function creates a temporary file with the provided input data, opens it in the system editor, waits for the user to make changes and close the editor, then reads and returns the modified content. The temporary file is deleted afterwards. :param input_data: Initial content to populate the editor with :type input_data: str :param suffix: Optional file extension for the temporary file (e.g. '.txt', '.md') :type suffix: str or None :return: The edited content after the editor is closed :rtype: str  T)shellrNFz)WARNING: Unable to delete temporary file z%. You may need to delete it manually.) r%r8 subprocesscallopenreadrremovePermissionErrorr)r rr,r# command_strr$ output_datas r pipe_editorrEss"z622H!&))K3>!KOKt,,,, h  ffhh    (     H       s$A//A36A3;BB10B1)N)r NNN)r NN)__doc__rr1r=r rich.consoler aider.dumprr5r4r3r rr%r-r8rErrrrIs  " '))       :&4      r