feat: add CLI operation hook (#22)

Author: mtth

src/git_draft/__main__.py

@@ -5,7 +5,7 @@
 import optparse
 import sys
 
-from .bots import load_bot
+from .bots import Operation, load_bot
 from .common import Config, PROGRAM, Store, ensure_state_home, open_editor
 from .manager import Manager
 
@@ -23,6 +23,11 @@ def new_parser() -> optparse.OptionParser:
         help="show log path and exit",
         action="store_true",
     )
+    parser.add_option(
+        "--root",
+        help="path used to locate repository",
+        dest="root",
+    )
 
     def add_command(name: str, **kwargs) -> None:
         def callback(_option, _opt, _value, parser) -> None:
@@ -81,6 +86,10 @@ def callback(_option, _opt, _value, parser) -> None:
     return parser
 
 
+def print_operation(op: Operation) -> None:
+    print(op)
+
+
 def main() -> None:
     config = Config.load()
     (opts, _args) = new_parser().parse_args()
@@ -91,7 +100,11 @@ def main() -> None:
         return
     logging.basicConfig(level=config.log_level, filename=str(log_path))
 
-    manager = Manager.create(Store.persistent())
+    manager = Manager.create(
+        store=Store.persistent(),
+        path=opts.root,
+        operation_hook=print_operation,
+    )
     command = getattr(opts, "command", "generate")
     if command == "generate":
         bot = load_bot(opts.bot, {})

src/git_draft/bots/__init__.py

@@ -5,11 +5,13 @@
 
 from typing import Any, Mapping
 
-from .common import Action, Bot, Toolbox
+from .common import Action, Bot, Operation, OperationHook, Toolbox
 
 __all__ = [
     "Action",
     "Bot",
+    "Operation",
+    "OperationHook",
     "Toolbox",
 ]
 

src/git_draft/bots/common.py

@@ -19,6 +19,9 @@ class Toolbox:
     # including inferring the most important files, and allowing returning
     # signature-only versions.
 
+    # TODO: Support a diff-based edit method.
+    # https://gist.github.com/noporpoise/16e731849eb1231e86d78f9dfeca3abc
+
     def __init__(self, hook: OperationHook | None = None) -> None:
         self.operations = list[Operation]()
         self._operation_hook = hook

src/git_draft/manager.py

@@ -11,7 +11,7 @@
 import time
 from typing import Match, Sequence, override
 
-from .bots import Bot, Toolbox
+from .bots import Bot, OperationHook, Toolbox
 from .common import Store, random_id, sql
 
 
@@ -54,8 +54,8 @@ class _Toolbox(Toolbox):
     concurrent editing without interference.
     """
 
-    def __init__(self, repo: git.Repo) -> None:
-        super().__init__()
+    def __init__(self, repo: git.Repo, hook: OperationHook | None) -> None:
+        super().__init__(hook)
         self._repo = repo
 
     @override
@@ -85,15 +85,27 @@ def _write(self, path: PurePosixPath, contents: str) -> None:
 class Manager:
     """Draft state manager"""
 
-    def __init__(self, store: Store, repo: git.Repo) -> None:
+    def __init__(
+        self, store: Store, repo: git.Repo, hook: OperationHook | None = None
+    ) -> None:
         with store.cursor() as cursor:
             cursor.executescript(sql("create-tables"))
         self._store = store
         self._repo = repo
+        self._operation_hook = hook
 
     @classmethod
-    def create(cls, store: Store, path: str | None = None) -> Manager:
-        return cls(store, git.Repo(path, search_parent_directories=True))
+    def create(
+        cls,
+        store: Store,
+        path: str | None = None,
+        operation_hook: OperationHook | None = None,
+    ) -> Manager:
+        return cls(
+            store,
+            git.Repo(path, search_parent_directories=True),
+            operation_hook,
+        )
 
     def _create_branch(self, sync: bool) -> _Branch:
         if not self._repo.active_branch:
@@ -155,7 +167,7 @@ def generate_draft(
             )
 
         start_time = time.perf_counter()
-        toolbox = _Toolbox(self._repo)
+        toolbox = _Toolbox(self._repo, self._operation_hook)
         action = bot.act(prompt, toolbox)
         end_time = time.perf_counter()