incomplete - add an edit command

Author: AkshajSinghal

README.md

@@ -2,7 +2,6 @@
 
 AtOffice Shell is a polished open-source Python CLI that combines jokes, todos, alarms, world clocks, and a productivity-focused terminal shell.
 
-![AtOffice Shell banner](docs/screenshot-placeholder.png)
 
 ## Key Features
 
@@ -42,6 +41,7 @@ addtask "Review PR" "Work"
 time
 world
 settings
+edit
 ```
 
 ## CLI Commands

atoffice_shell/project.py

@@ -4,8 +4,12 @@
 import re
 import shlex
 import subprocess
+from pathlib import Path
 
 import typer
+from textual.app import App, ComposeResult
+from textual.binding import Binding
+from textual.widgets import Footer, Header, TextArea
 
 from .pyfunny import joke, joke_trex
 from .settings import launch_settings
@@ -34,6 +38,56 @@ def _prompt_command() -> tuple[str, str, str]:
     return raw_command, command, argument
 
 
+class AtonEditor(App):
+    """Simple full-screen text editor for ATON shell files."""
+
+    inherit_bindings = True
+
+    CSS = """
+    Screen { padding: 0; }
+    #editor { height: 1fr; }
+    Footer { height: 1; }
+    """
+
+    BINDINGS = [
+        ("ctrl+shift+s", "save_file", "Ctrl+Shift+S Save"),
+        ("ctrl+shift+q", "quit_app", "Ctrl+Shift+Q Quit"),
+    ]
+
+    def __init__(self, file_path: str, **kwargs) -> None:
+        super().__init__(**kwargs)
+        self.file_path = file_path
+        self.file_content = ""
+
+        import os
+
+        if os.path.exists(file_path):
+            try:
+                with open(file_path, "r", encoding="utf-8") as handle:
+                    self.file_content = handle.read()
+            except Exception as error:
+                self.file_content = f"Error reading file: {error}"
+
+    def compose(self) -> ComposeResult:
+        yield Header()
+        yield TextArea(self.file_content, id="editor_text_area")
+        yield Footer()
+
+    def on_mount(self) -> None:
+        self.text_area.focus()
+
+    def action_save_file(self) -> None:
+        text_area = self.query_one("#editor_text_area", TextArea)
+        try:
+            with open(self.file_path, "w", encoding="utf-8") as handle:
+                handle.write(text_area.text)
+        except Exception:
+            pass
+
+    def action_quit_app(self) -> None:
+        self.exit()
+
+
 def _handle_joke_command(command: str) -> bool:
     if command in {"joke", "joke_trex", "joke-trex"}:
         if command == "joke":
@@ -72,6 +126,26 @@ def _handle_todo_command(command: str) -> bool:
     return False
 
 
+def _handle_edit_command(raw_command: str) -> bool:
+    command, argument = _split_command(raw_command)
+    if command != "edit":
+        return False
+
+    if not argument.strip():
+        typer.secho("⚠️ Syntax: edit <filename>", fg=typer.colors.YELLOW)
+        return True
+
+    file_path = Path(argument.strip())
+    initial_text = file_path.read_text(encoding="utf-8") if file_path.exists() else ""
+
+    try:
+        AtonEditor(str(file_path), initial_text).run()
+    except Exception as error:
+        typer.secho(f"Editor error: {error}", fg=typer.colors.RED)
+
+    return True
+
+
 def _handle_local_command(command: str, argument: str) -> str:
     if command == "addtask" and not argument:
         typer.secho(
@@ -167,6 +241,8 @@ def run_interactive_shell() -> None:
             continue
         if _handle_cd_command(raw_command):
             continue
+        if _handle_edit_command(raw_command):
+            continue
         if _handle_os_fallback(raw_command):
             continue
 

atoffice_shell/pyfunny.py

@@ -12,6 +12,8 @@
 DEFAULT_JOKE_CHARACTER = "cow"
 DEFAULT_JOKE_SOUND = "cow-sound.mp3"
 
+# Hi! This commit was made through the `edit` command!
+# This one too!
 
 def _sound_path(filename: str) -> Path:
     return Path(__file__).resolve().parent / "chronoterm" / "sounds" / filename

pyproject.toml

@@ -28,6 +28,7 @@ dependencies = [
   "playsound==1.2.2",
   "pytz>=2024.1",
   "platformdirs>=3.0.0",
+  "textual>=0.86.0",
 ]
 
 [project.urls]

tests/test_cli.py

@@ -4,7 +4,7 @@
 from typer.testing import CliRunner
 
 from atoffice_shell.cli import app
-from atoffice_shell.project import _handle_cd_command, _handle_local_command, _handle_os_fallback
+from atoffice_shell.project import _handle_cd_command, _handle_edit_command, _handle_local_command, _handle_os_fallback
 
 
 runner = CliRunner()
@@ -86,4 +86,35 @@ def test_addtask_missing_arguments_is_blocked(monkeypatch) -> None:
     )
 
     assert _handle_local_command("addtask", "") == "handled"
-    assert messages and "Missing arguments" in messages[0][0]
+
+
+def test_edit_requires_filename(monkeypatch) -> None:
+    messages = []
+
+    monkeypatch.setattr(
+        "atoffice_shell.project.typer.secho",
+        lambda message, fg=None: messages.append((message, fg)),
+    )
+
+    assert _handle_edit_command("edit") is True
+    assert messages and "Syntax: edit <filename>" in messages[0][0]
+
+
+def test_edit_launches_editor_for_existing_file(monkeypatch, tmp_path) -> None:
+    file_path = tmp_path / "note.txt"
+    file_path.write_text("hello", encoding="utf-8")
+
+    calls = {}
+
+    class FakeEditor:
+        def __init__(self, filename: str, initial_text: str) -> None:
+            calls["filename"] = filename
+            calls["initial_text"] = initial_text
+
+        def run(self) -> None:
+            calls["ran"] = True
+
+    monkeypatch.setattr("atoffice_shell.project.AtonEditor", FakeEditor)
+
+    assert _handle_edit_command(f"edit {file_path}") is True
+    assert calls == {"filename": str(file_path), "initial_text": "hello", "ran": True}