refactor(cmd[hg]): Pass-through progress_callback

tony · tony · commit ec17adabf18f · 2022-10-23T09:09:35.000-05:00
diff --git a/src/libvcs/cmd/hg.py b/src/libvcs/cmd/hg.py
@@ -12,7 +12,7 @@
 from collections.abc import Sequence
 from typing import Any, Optional, Union
 
-from libvcs._internal.run import run
+from libvcs._internal.run import ProgressCallbackProtocol, run
 from libvcs._internal.types import StrOrBytesPath, StrPath
 
 _CMD = Union[StrOrBytesPath, Sequence[StrOrBytesPath]]
@@ -34,7 +34,14 @@ class HgPagerType(enum.Enum):
 
 
 class Hg:
-    def __init__(self, *, dir: StrPath) -> None:
+    progress_callback: Optional[ProgressCallbackProtocol] = None
+
+    def __init__(
+        self,
+        *,
+        dir: StrPath,
+        progress_callback: Optional[ProgressCallbackProtocol] = None,
+    ) -> None:
         """Lite, typed, pythonic wrapper for hg(1).
 
         Parameters
@@ -54,6 +61,8 @@ def __init__(self, *, dir: StrPath) -> None:
         else:
             self.dir = pathlib.Path(dir)
 
+        self.progress_callback = progress_callback
+
     def __repr__(self) -> str:
         return f"<Hg dir={self.dir}>"
 
@@ -171,6 +180,9 @@ def run(
         if help is True:
             cli_args.append("--help")
 
+        if self.progress_callback is not None:
+            kwargs["callback"] = self.progress_callback
+
         return run(
             args=cli_args,
             check_returncode=True if check_returncode is None else check_returncode,
