wcgw 2.1.0__py3-none-any.whl → 2.1.1__py3-none-any.whl

wcgw/client/tools.py

@@ -7,6 +7,7 @@ import json
 import mimetypes
 from pathlib import Path
 import re
+import shlex
 import sys
 import threading
 import importlib.metadata
@@ -518,7 +519,7 @@ def read_image_from_shell(file_path: str) -> ImageData:
     else:
         with TemporaryDirectory() as tmpdir:
             rcode = os.system(
-                f"docker cp {BASH_STATE.is_in_docker}:{file_path} {tmpdir}"
+                f"docker cp {BASH_STATE.is_in_docker}:{shlex.quote(file_path)} {tmpdir}"
             )
             if rcode != 0:
                 raise Exception(f"Error: Read failed with code {rcode}")
@@ -553,7 +554,7 @@ def write_file(writefile: CreateFileNew, error_on_exist: bool) -> str:
     else:
         if error_on_exist:
             # Check if it exists using os.system
-            cmd = f"test -f {path_}"
+            cmd = f"test -f {shlex.quote(path_)}"
             status = os.system(f'docker exec {BASH_STATE.is_in_docker} bash -c "{cmd}"')
             if status == 0:
                 return f"Error: can't write to existing file {path_}, use other functions to edit the file"
@@ -570,7 +571,9 @@ def write_file(writefile: CreateFileNew, error_on_exist: bool) -> str:
             if rcode != 0:
                 return f"Error: Write failed with code while creating dirs {rcode}"
 
-            rcode = os.system(f"docker cp {tmppath} {BASH_STATE.is_in_docker}:{path_}")
+            rcode = os.system(
+                f"docker cp {shlex.quote(tmppath)} {BASH_STATE.is_in_docker}:{shlex.quote(path_)}"
+            )
             if rcode != 0:
                 return f"Error: Write failed with code {rcode}"
 
@@ -661,7 +664,9 @@ def do_diff_edit(fedit: FileEdit) -> str:
     else:
         # Copy from docker
         with TemporaryDirectory() as tmpdir:
-            rcode = os.system(f"docker cp {BASH_STATE.is_in_docker}:{path_} {tmpdir}")
+            rcode = os.system(
+                f"docker cp {BASH_STATE.is_in_docker}:{shlex.quote(path_)} {tmpdir}"
+            )
             if rcode != 0:
                 raise Exception(f"Error: Read failed with code {rcode}")
             path_tmp = os.path.join(tmpdir, os.path.basename(path_))
@@ -724,7 +729,9 @@ def do_diff_edit(fedit: FileEdit) -> str:
                 f.write(apply_diff_to)
             os.chmod(path_tmp, 0o777)
             # Copy to docker using docker cp
-            rcode = os.system(f"docker cp {path_tmp} {BASH_STATE.is_in_docker}:{path_}")
+            rcode = os.system(
+                f"docker cp {shlex.quote(path_tmp)} {BASH_STATE.is_in_docker}:{shlex.quote(path_)}"
+            )
             if rcode != 0:
                 raise Exception(f"Error: Write failed with code {rcode}")
 
@@ -1061,7 +1068,8 @@ def read_file(readfile: ReadFile, max_tokens: Optional[int]) -> str:
 
     else:
         return_code, content, stderr = command_run(
-            f"cat {readfile.file_path}", timeout=TIMEOUT
+            f"docker exec {BASH_STATE.is_in_docker} cat {shlex.quote(readfile.file_path)}",
+            timeout=TIMEOUT,
         )
         if return_code != 0:
             raise Exception(

{wcgw-2.1.0.dist-info → wcgw-2.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: wcgw
-Version: 2.1.0
+Version: 2.1.1
 Summary: Shell and coding agent on claude and chatgpt
 Project-URL: Homepage, https://github.com/rusiaaman/wcgw
 Author-email: Aman Rusia <gapypi@arcfu.com>

{wcgw-2.1.0.dist-info → wcgw-2.1.1.dist-info}/RECORD

@@ -10,13 +10,13 @@ wcgw/client/diff-instructions.txt,sha256=s5AJKG23JsjwRYhFZFQVvwDpF67vElawrmdXwvu
 wcgw/client/openai_client.py,sha256=qhntpGZANTyI2-vAWaI1pyiJsiOwEjhXc4lCvPpebiM,17752
 wcgw/client/openai_utils.py,sha256=YNwCsA-Wqq7jWrxP0rfQmBTb1dI0s7dWXzQqyTzOZT4,2629
 wcgw/client/sys_utils.py,sha256=GajPntKhaTUMn6EOmopENWZNR2G_BJyuVbuot0x6veI,1376
-wcgw/client/tools.py,sha256=QO_IkOugFrdz1pnX4VBvFvxAjpIDL0wmYmGjU_CkjRc,35504
+wcgw/client/tools.py,sha256=kokbgxxBikIK39moSeg5uGvtBUzLXHnP5-d4jywZYRQ,35762
 wcgw/client/mcp_server/Readme.md,sha256=I8N4dHkTUVGNQ63BQkBMBhCCBTgqGOSF_pUR6iOEiUk,2495
 wcgw/client/mcp_server/__init__.py,sha256=hyPPwO9cabAJsOMWhKyat9yl7OlSmIobaoAZKHu3DMc,381
 wcgw/client/mcp_server/server.py,sha256=XWZCMlL5HOpvRQjY1qoZnmFAe9x_rpRb_udZz9k8ks4,10815
 wcgw/relay/serve.py,sha256=RUcUeyL4Xt0EEo12Ul6VQjb4tRle4uIdsa85v7XXxEw,8771
 wcgw/relay/static/privacy.txt,sha256=s9qBdbx2SexCpC_z33sg16TptmAwDEehMCLz4L50JLc,529
-wcgw-2.1.0.dist-info/METADATA,sha256=43N0-Y_Ab6UhhwcLqM5UWaW1gmjo-BuBiZYLn-kDhC4,7421
-wcgw-2.1.0.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-wcgw-2.1.0.dist-info/entry_points.txt,sha256=eKo1omwbAggWlQ0l7GKoR7uV1-j16nk9tK0BhC2Oz_E,120
-wcgw-2.1.0.dist-info/RECORD,,
+wcgw-2.1.1.dist-info/METADATA,sha256=T6tzEXw6jTT0KpDoJmYa6QQkRl1xRB2E4VJg0YwITv4,7421
+wcgw-2.1.1.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+wcgw-2.1.1.dist-info/entry_points.txt,sha256=eKo1omwbAggWlQ0l7GKoR7uV1-j16nk9tK0BhC2Oz_E,120
+wcgw-2.1.1.dist-info/RECORD,,