feat: --workdir

1 files changed, 10 insertions, 11 deletions

diff --git a/actions.py b/actions.py
index 9146e53..765fe3d 100644
--- a/actions.py
+++ b/actions.py
@@ -4,12 +4,12 @@ import subprocess
 
 def _path_checked(path: Path, cwd=Path.cwd()) -> Optional[Path]:
     """Resolves `path`, returning `None` if it's not a subdirectory of `cwd`."""
-    path = path.resolve()
+    path = (path if path.is_absolute() else (cwd / path)).resolve()
     cwd = cwd.resolve()
     return path if path.is_relative_to(cwd) else None
 
-def _path_helper(path_str: str, callback):
-    path = _path_checked(Path(path_str))
+def _path_helper(path_str: str, callback, cwd=Path.cwd()):
+    path = _path_checked(Path(path_str), cwd=cwd)
     if path is None:
         return {"status": "error", "reason": f"Permission denied: {path_str}"}
     else:
@@ -18,33 +18,33 @@ def _path_helper(path_str: str, callback):
         except Exception as e:
             return {"status": "error", "reason": f"Failure: {e}"}
 
-def ls(args):
+def ls(args, workdir=Path.cwd()):
     '{"action": "ls", "path": PATH} - list files at PATH'
     def inner(path: Path):
         return {"content": list(map(lambda s: str(s), path.iterdir()))}
     match args:
         case {"action": "ls", "path": path_str}:
-            return _path_helper(path_str, inner)
+            return _path_helper(path_str, inner, cwd=workdir)
 
-def read(args):
+def read(args, workdir=Path.cwd()):
     '{"action": "read", "path": PATH} - read file at PATH'
     def inner(path: Path):
         contents = path.read_text(encoding="utf-8")
         return {"content": contents}
     match args:
         case {"action": "read", "path": path_str}:
-            return _path_helper(path_str, inner)
+            return _path_helper(path_str, inner, cwd=workdir)
 
-def write(args):
+def write(args, workdir=Path.cwd()):
     '{"action": "write", "path": PATH, "content": CONTENT} - write CONTENT into file at PATH'
     def inner(path: Path, text: str):
         n_bytes = path.write_text(text)
         return {"bytes": n_bytes}
     match args:
         case {"action": "write", "path": path_str, "content": content}:
-            return _path_helper(path_str, lambda path: inner(path, content))
+            return _path_helper(path_str, lambda path: inner(path, content), cwd=workdir)
 
-def tldr(args):
+def tldr(args, workdir=Path.cwd()):
     '{"action": "tldr", "program": PROGRAM} - read tldr docs for PROGRAM'
     match args:
         case {"action": "tldr", "program": program_str}:
@@ -61,4 +61,3 @@ def tldr(args):
             except Exception as e:
                 return {"status": "error", "reason": f"An unexpected error occurred: {e}"}
 
-ALL_ACTIONS = [ls, read, write, tldr]