webbee 0.1.3__tar.gz → 0.1.4__tar.gz

{webbee-0.1.3 → webbee-0.1.4}/CHANGELOG.md

@@ -1,5 +1,11 @@
 # Changelog
 
+## 0.1.4
+
+- **Copy on select.** Drag to select text in the dock and it's copied to your
+  clipboard automatically (via OSC 52 — works locally and over SSH), with a
+  brief “✓ copied” confirmation. Mouse-wheel scrolling still works.
+
 ## 0.1.3
 
 - **Sessions & security.** `/sessions` lists everywhere your Imperal account is

{webbee-0.1.3 → webbee-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: webbee
-Version: 0.1.3
+Version: 0.1.4
 Summary: Webbee 🐝 — the Imperal Cloud coding agent in your terminal
 Project-URL: Homepage, https://imperal.io
 Project-URL: Documentation, https://docs.imperal.io

{webbee-0.1.3 → webbee-0.1.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "webbee"
-version = "0.1.3"
+version = "0.1.4"
 description = "Webbee 🐝 — the Imperal Cloud coding agent in your terminal"
 readme = "README.md"
 license = "MIT"

webbee-0.1.4/src/webbee/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.4"

{webbee-0.1.3 → webbee-0.1.4}/src/webbee/tui.py

@@ -60,6 +60,7 @@ class OutputPane:
         from prompt_toolkit.formatted_text import ANSI
         from prompt_toolkit.layout.containers import Window
         from prompt_toolkit.layout.controls import FormattedTextControl
+        from prompt_toolkit.mouse_events import MouseEventType, MouseButton
         from rich.console import Console
 
         self._io = io.StringIO()
@@ -67,7 +68,35 @@ class OutputPane:
                                color_system="truecolor", width=width, highlight=False)
         self._ANSI = ANSI
         self._cache = (None, None)   # (text, ANSI) memo — bounds re-parse cost
-        self.control = FormattedTextControl(
+        self.copy_flash = ""         # transient toolbar note after a copy
+        self._flash_until = 0.0
+        pane = self
+
+        # Copy-on-select: left-drag over the pane copies the covered text to the
+        # system clipboard via OSC 52 (works locally + over SSH). The Window
+        # gives us CONTENT coordinates (Point(x=col, y=row)); with wrap_lines
+        # False, row = content line, col = char index. SCROLL events return
+        # NotImplemented so the Window keeps handling the wheel.
+        class _SelectControl(FormattedTextControl):
+            def __init__(self, **kw):
+                super().__init__(**kw)
+                self._down = None
+
+            def mouse_handler(self, ev):
+                et = ev.event_type
+                if et == MouseEventType.MOUSE_DOWN and ev.button == MouseButton.LEFT:
+                    self._down = ev.position
+                    return None
+                if et == MouseEventType.MOUSE_MOVE:
+                    return None if self._down is not None else NotImplemented
+                if et == MouseEventType.MOUSE_UP:
+                    down, self._down = self._down, None
+                    if down is not None and (down.x, down.y) != (ev.position.x, ev.position.y):
+                        pane._copy_selection(down, ev.position)  # real drag, not a click
+                    return None
+                return NotImplemented   # SCROLL_UP/DOWN → Window scrolls
+
+        self.control = _SelectControl(
             text=self._formatted, focusable=False, show_cursor=False,
             get_cursor_position=lambda: Point(0, self.window.vertical_scroll))
         self.window = Window(content=self.control, wrap_lines=False,
@@ -109,6 +138,57 @@ class OutputPane:
         the conversation survives leaving the alternate screen."""
         return self._io.getvalue()
 
+    # ---- copy-on-select --------------------------------------------------
+    def _selected_text(self, start, end) -> str:
+        """Plain text (ANSI stripped) covered by a drag from `start` to `end`
+        (both Points with content col=x / line=y). wrap_lines is False so a
+        content line maps 1:1 to a `\\n`-split line and col = char index."""
+        import re
+        plain = re.sub(r"\x1b\[[0-9;]*m", "", self._io.getvalue())
+        lines = plain.split("\n")
+        p1, p2 = (start.y, start.x), (end.y, end.x)
+        if p1 > p2:
+            p1, p2 = p2, p1
+        (y1, x1), (y2, x2) = p1, p2
+        n = len(lines)
+        if not (0 <= y1 < n):
+            return ""
+        y2 = min(y2, n - 1)
+        if y1 == y2:
+            return lines[y1][x1:x2 + 1]
+        out = [lines[y1][x1:]]
+        out.extend(lines[y1 + 1:y2])
+        out.append(lines[y2][:x2 + 1])
+        return "\n".join(out)
+
+    def _osc52_copy(self, text: str) -> None:
+        import base64
+        try:
+            from prompt_toolkit.application import get_app_or_none
+            app = get_app_or_none()
+            if app is None:
+                return
+            b64 = base64.b64encode(text.encode("utf-8", "replace")).decode("ascii")
+            app.output.write_raw("\x1b]52;c;" + b64 + "\x07")  # OSC 52 — set clipboard
+            app.output.flush()
+        except Exception:
+            pass
+
+    def _copy_selection(self, start, end) -> None:
+        import time as _t
+        text = self._selected_text(start, end)
+        if not text.strip():
+            return
+        self._osc52_copy(text)
+        self.copy_flash = f"✓ copied {len(text)} char{'s' if len(text) != 1 else ''}"
+        self._flash_until = _t.monotonic() + 1.8
+        self.notify()
+
+    def flash(self) -> str:
+        """The transient 'copied' note, while still fresh (else empty)."""
+        import time as _t
+        return self.copy_flash if _t.monotonic() < self._flash_until else ""
+
 
 async def run_session(*, pane, on_line, mode_getter, on_cycle, status,
                       is_busy, consent_pending, resolve_consent) -> bool:
@@ -177,6 +257,9 @@ async def run_session(*, pane, on_line, mode_getter, on_cycle, status,
         pane.window.vertical_scroll += 5
 
     def _toolbar():
+        f = pane.flash()
+        if f:
+            return [("class:tb.working", "  " + f)]   # transient copy confirmation
         st = status()
         return build_toolbar(mode_getter(), st["tokens"], st["cost"], busy=st["busy"],
                              current=st["current"], elapsed=st["elapsed"],
@@ -206,7 +289,7 @@ async def run_session(*, pane, on_line, mode_getter, on_cycle, status,
         # animate the spinner + tick the elapsed clock while a turn runs
         while True:
             await asyncio.sleep(0.25)
-            if is_busy():
+            if is_busy() or pane.flash():
                 app.invalidate()
 
     tick = asyncio.ensure_future(_ticker())

{webbee-0.1.3 → webbee-0.1.4}/tests/test_tui.py

@@ -70,3 +70,39 @@ def test_output_pane_captures_colored_text():
     out = pane.dump()
     assert "Error" in out and "ok" in out
     assert "\x1b[" in out          # ANSI colour escapes preserved for the pane
+
+
+# ── copy-on-select (drag → OSC 52) ────────────────────────────────────────────
+
+def test_selected_text_single_line():
+    from webbee.tui import OutputPane
+    from prompt_toolkit.data_structures import Point
+    pane = OutputPane(width=80)
+    pane.console.print("hello world")
+    assert pane._selected_text(Point(6, 0), Point(10, 0)) == "world"
+
+
+def test_selected_text_multi_line_strips_ansi():
+    from webbee.tui import OutputPane
+    from prompt_toolkit.data_structures import Point
+    pane = OutputPane(width=80)
+    pane.console.print("abcdef")
+    pane.console.print("[bold]ghijkl[/]")   # coloured — must be stripped
+    pane.console.print("mnopqr")
+    assert pane._selected_text(Point(3, 0), Point(2, 2)) == "def\nghijkl\nmno"
+
+
+def test_selected_text_reversed_order_normalizes():
+    from webbee.tui import OutputPane
+    from prompt_toolkit.data_structures import Point
+    pane = OutputPane(width=80)
+    pane.console.print("hello")
+    assert pane._selected_text(Point(4, 0), Point(0, 0)) == "hello"
+
+
+def test_copy_flash_expires():
+    from webbee.tui import OutputPane
+    pane = OutputPane(width=80)
+    pane.copy_flash = "✓ copied 5 chars"
+    pane._flash_until = 0.0            # already in the past
+    assert pane.flash() == ""

webbee-0.1.3/src/webbee/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.1.3"