ڎj2dZddlZddlZddlmZddlmZmZddlm Z m Z ddl m Z m Z dd lmZmZmZmZmZdd lmZdd lmZmZdd lmZd dlmZGdde ZdS)zThe write tool in agentscope.N)Path)AnyList)ToolBaseToolMiddlewareBase)DEFAULT_DANGEROUS_FILESDEFAULT_DANGEROUS_DIRECTORIES)PermissionContextPermissionDecisionPermissionBehaviorPermissionModePermissionRule) ToolChunk) TextBlockToolResultState) AgentState) BackendBasec eZdZUdZdZeed< dZeed< dddddd dd d d gd Ze ee fed<dZ e ed<dZ e ed<dZe ed<dZe ed<dZe ed<eeddfdeedeedeedzdedzddf fd Zde ee fdedefdZd edzde ee fde fd!Zde ee fdeefd"Z d%d ed ed#edzdefd$Z xZ!S)&WritezThe write tool.nameanWrites a file to the local filesystem. Usage: - This tool will overwrite the existing file if there is one at the provided path. - If this is an existing file, you MUST use the Read tool first to read the file's contents. This tool will fail if you did not read the file first. - ALWAYS prefer editing existing files in the codebase. NEVER write new files unless explicitly required. - NEVER proactively create documentation files (*.md) or README files. Only create documentation files if explicitly requested by the User. - Only use emojis if the user explicitly requests it. Avoid writing emojis to files unless asked. descriptionobjectstringzGThe absolute path to the file to write (must be absolute, not relative))typerz The content to write to the file) file_pathcontentrr)r propertiesrequired input_schemaFis_mcp is_read_onlyis_concurrency_safeis_external_toolTis_state_injectedNdangerous_filesdangerous_directories middlewaresbackendreturncddlm}t|t ||_t ||_|p ||_dS)aInitialize the write tool. Args: dangerous_files (`list[str]`, optional): Sensitive files that require explicit user confirmation, even in BYPASS mode. Matched by basename (case-insensitive). Defaults to `DEFAULT_DANGEROUS_FILES`. Pass a custom list to fully replace the defaults, or `[]` to disable the filename check. dangerous_directories (`list[str]`, optional): Sensitive directories that require explicit user confirmation. Matched when any path segment equals an entry (case-insensitive). Defaults to `DEFAULT_DANGEROUS_DIRECTORIES`. Pass a custom list to fully replace the defaults, or `[]` to disable the directory check. middlewares (`List[ToolMiddlewareBase] | None`, optional): Tool middlewares wrapping the tool execution. backend (`BackendBase | None`, optional): The sandbox backend to use for file I/O. When ``None``, a :class:`LocalBackend` is created. r) LocalBackend)r*N)_backendr.super__init__listr(r))selfr(r)r*r+r. __class__s MC:\Projects\DSCons\.venv\Lib\site-packages\agentscope\tool\_builtin\_write.pyr1zWrite.__init__Asi: +***** [111#O44%)*?%@%@"1<<>> tool_inputcontextcK|d}|sttjdS||r ttjd|ddS|jtjtj fvr6| ||r ttj d|d d Sttjd S) a}Check permissions for file writing. This method implements Write-specific permission checks: 1. Dangerous path check (safety check, bypass-immune) 2. ACCEPT_EDITS mode check for files in working directories Args: tool_input (`dict[str, Any]`): The tool input containing "file_path" key context (`PermissionContext`): The permission context with mode and rules Returns: `PermissionDecision`: ASK for dangerous paths, ALLOW for safe operations in ACCEPT_EDITS mode, PASSTHROUGH otherwise rzNo file path provided)behaviormessagez7Permission required: Write operation on sensitive file z)Safety check: dangerous file or directoryT)r:r;decision_reason bypass_immunezPermission granted for writing z (in working directory)z5File is in working directory and not a dangerous path)r:r;r<) getr r PASSTHROUGH_is_dangerous_pathASKmoder ACCEPT_EDITSDONT_ASK_path_in_allowed_working_pathALLOW)r3r7r8rs r5check_permissionszWrite.check_permissionsfs$.NN;//  %+7/   " "9 - - %+/."+.. K"   <  '  #   11)WEE )/5.i...%' "'3    r6 rule_contentclK|dS|dd}|sdStj||S)aCheck if a permission rule matches the file path. Matches rule_content as a glob pattern against the "file_path" parameter using fnmatch. If rule_content is None, matches all invocations (tool-name-level rule). Args: rule_content (`str | None`): Glob pattern to match against the file path (e.g., "src/**"), or None to match all invocations tool_input (`dict[str, Any]`): The tool input data containing "file_path" key Returns: `bool`: True if the glob pattern matches the file path, False otherwise NTrr>F)r?fnmatch)r3rIr7rs r5 match_rulezWrite.match_rulesE,  4NN;33  5y,777r6cK|dd}|sgS|j|}|r|ddznd}t |j|t jdgS)a Generate suggested permission rules for the file path. Suggests a glob pattern covering the parent directory of the file, allowing the user to grant permission for the entire directory at once. Args: tool_input (`dict[str, Any]`): The tool input data containing "file_path" key Returns: `List[PermissionRule]`: A single suggested rule covering the parent directory (e.g., file "/src/main.py" -> rule "src/**") rr>z/\z/**z** suggested) tool_namerIr:source)r?r/dirnamerstriprrrrG)r3r7rparentpatterns r5generate_suggestionszWrite.generate_suggestionss$NN;33  I&&y115;D6==''%// )$+1"      r6 _agent_statec K|j|s.ttd|gtjdS|j|d{VrS|Q|j|d{V}|/ttd|dgtjdS|j|d{V}d}|rG |j |d{V d }n#t$rd}YnwxYwt|j }|jd d t|gd{V|j||d d{Vt%|d }dt+j|d |d |sdnd|d|d} ttd|d|dgtjd| |ddS)z.Write content to a file and return the result.z0Error: file_path must be an absolute path, got: )textT)rstateis_lastNz Error: File zU exists but has not been read yet. You must read the file first before writing to it.r>zutf-8mkdirz-p )keependsz /dev/nullza/zb/r )fromfiletofilenz The file z has been written successfully (z lines).r)diffr occurrences)rrYrZmetadata)r/isabsrrrERROR file_exists tool_context get_cache read_filedecode ExceptionrrS exec_shellstr write_fileencodelensplitjoindifflib unified_diff splitlinesRUNNING) r3rrrVcache file_existedprevious_content parent_dir line_count diff_texts r5callz Write.calls]}""9-- , ),, &+    -++I66 6 6 6 6 6 6 (&3==iHHHHHHHHE} !"5 "5"5"5*/     $"]66yAAAAAAAA   & &-11)<<<<<<<<&//!  & & &$&    & )__+ m&& dC OO ,         m&&  NN7 # #         t,,-- GG   ++T+::""D"11'3IKK9Ii9I9I'I''      -Y--"--- ")!&     s#3D D&%D&)N)"__name__ __module__ __qualname____doc__rrm__annotations__rr"dictrr#boolr$r%r&r'r r r2rrrr1r r rHrLrrUrrr} __classcell__)r4s@r5rrsD#/eKeee2! 3 !A   !),$$L$sCx. FDL$ %%%%"d""""t"""&=+H7;&* #2#2c#2 $Cy#2,-4 #2 t# #2  #2#2#2#2#2#2J> cN> #>   > > > > @8Dj8cN8  8888<" cN"  n " " " " P+/ f f f f !4' f  f f f f f f f f r6r)rrsrKpathlibrtypingrr_baserr _constantsr r permissionr r rrr _responserr;rrrYrr/rrr6r5rs[##00000000"!!!!!11111111!!!!!!t t t t t Ht t t t t r6