Andrew/refactor

.github/workflows/publish_testpypi.yml

@@ -28,7 +28,7 @@ jobs:
       # don't modify sync file! messes up version!
       - run: uv sync --all-extras --frozen # does order matter?
       - run: uv build
-      - run: uv run --no-sync choreo_get_browser -v --i ${{ matrix.chrome_v }}
+      - run: uv run --no-sync choreo_get_chrome -v --i ${{ matrix.chrome_v }}
       - name: Reinstall from wheel
         run: >
           uv pip install dist/choreographer-$(uv

.github/workflows/test.yml

@@ -26,7 +26,7 @@ jobs:
       - name: Install choreographer
         run:  uv sync --all-extras
       - name: Install google-chrome-for-testing
-        run: uv run choreo_get_browser
+        run: uv run choreo_get_chrome
       - name: Diagnose
         run:  uv run choreo_diagnose --no-run
         timeout-minutes: 1

.pre-commit-config.yaml

@@ -2,6 +2,7 @@
 # See https://pre-commit.com/hooks.html for more hooks
 %YAML 1.2
 ---
+exclude: "site/.*"
 repos:
   - repo: https://github.com/pre-commit/pre-commit-hooks
     rev: v3.2.0

choreographer/__init__.py

@@ -1,24 +1,19 @@
-"""choreographer is a browser controller for python."""
+"""
+choreographer is a browser controller for python.
 
-import choreographer._devtools_protocol_layer as protocol
+choreographer is natively async, so while there are two main entrypoints:
+classes `Browser` and `BrowserSync`, the sync version is very limited, functioning
+as a building block for more featureful implementations.
 
-from ._browser import Browser, BrowserClosedError, browser_which, get_browser_path
-from ._cli_utils import get_browser, get_browser_sync
-from ._pipe import BlockWarning, PipeClosedError
-from ._system_utils._tempfile import TempDirectory, TempDirWarning
-from ._tab import Tab
+See the main README for a quickstart.
+"""
+
+from .browser_sync import (
+    BrowserSync,
+    TabSync,
+)
 
 __all__ = [
-    "BlockWarning",
-    "Browser",
-    "BrowserClosedError",
-    "PipeClosedError",
-    "Tab",
-    "TempDirWarning",
-    "TempDirectory",
-    "browser_which",
-    "get_browser",
-    "get_browser_path",
-    "get_browser_sync",
-    "protocol",
+    "BrowserSync",
+    "TabSync",
 ]

choreographer/_brokers/__init__.py

@@ -0,0 +1,5 @@
+from ._sync import BrokerSync
+
+__all__ = [
+    "BrokerSync",
+]

choreographer/_brokers/_sync.py

@@ -0,0 +1,38 @@
+import json
+from threading import Thread
+
+import logistro
+
+from choreographer import protocol
+from choreographer.channels import ChannelClosedError
+
+logger = logistro.getLogger(__name__)
+
+
+class BrokerSync:
+    def __init__(self, browser, channel):
+        self.browser = browser
+        self.channel = channel
+
+    # This is the broker
+    def run_output_thread(self, **kwargs):
+        def run_print():
+            try:
+                while True:
+                    responses = self.channel.read_jsons()
+                    for response in responses:
+                        print(json.dumps(response, indent=4), **kwargs)
+            except ChannelClosedError:
+                print("ChannelClosedError caught.", **kwargs)
+
+        logger.info("Starting thread to dump output to stdout.")
+        Thread(target=run_print).start()
+
+    def send_json(self, obj):
+        protocol.verify_params(obj)
+        key = protocol.calculate_message_key(obj)
+        self.channel.write_json(obj)
+        return key
+
+    def clean(self):
+        pass