Merge pull request #6 from OpenAdaptAI/feat/wormhole-share

feat(share): add Magic Wormhole sharing for recordings

Author: abrichr

openadapt_capture/cli.py

@@ -340,6 +340,29 @@ def _save_transcript(
         print(f"[{mins}:{secs:05.2f}] {seg['text']}")
 
 
+def share(action: str, path_or_code: str, output_dir: str = ".") -> None:
+    """Share recordings via Magic Wormhole.
+
+    Args:
+        action: Either "send" or "receive".
+        path_or_code: Recording path (for send) or wormhole code (for receive).
+        output_dir: Output directory for receive (default: current dir).
+
+    Examples:
+        capture share send ./my_recording
+        capture share receive 7-guitarist-revenge
+        capture share receive 7-guitarist-revenge ./recordings
+    """
+    from openadapt_capture.share import send, receive
+
+    if action == "send":
+        send(path_or_code)
+    elif action == "receive":
+        receive(path_or_code, output_dir)
+    else:
+        print(f"Unknown action: {action}. Use 'send' or 'receive'.")
+
+
 def main() -> None:
     """CLI entry point."""
     import fire
@@ -348,6 +371,7 @@ def main() -> None:
         "visualize": visualize,
         "info": info,
         "transcribe": transcribe,
+        "share": share,
     })
 
 

openadapt_capture/share.py

@@ -0,0 +1,177 @@
+"""Share recordings between computers using Magic Wormhole.
+
+Usage:
+    capture share send ./my_recording
+    capture share receive 7-guitarist-revenge
+"""
+
+from __future__ import annotations
+
+import shutil
+import subprocess
+import sys
+import tempfile
+from pathlib import Path
+from zipfile import ZIP_DEFLATED, ZipFile
+
+
+def _check_wormhole_installed() -> bool:
+    """Check if magic-wormhole is installed."""
+    return shutil.which("wormhole") is not None
+
+
+def _install_wormhole() -> bool:
+    """Attempt to install magic-wormhole."""
+    print("Installing magic-wormhole...")
+    try:
+        subprocess.run(
+            [sys.executable, "-m", "pip", "install", "magic-wormhole"],
+            check=True,
+            capture_output=True,
+        )
+        print("✓ magic-wormhole installed")
+        return True
+    except subprocess.CalledProcessError as e:
+        print(f"✗ Failed to install magic-wormhole: {e}")
+        return False
+
+
+def _ensure_wormhole() -> bool:
+    """Ensure magic-wormhole is available, install if needed."""
+    if _check_wormhole_installed():
+        return True
+    return _install_wormhole()
+
+
+def send(recording_dir: str) -> str | None:
+    """Send a recording via Magic Wormhole.
+
+    Args:
+        recording_dir: Path to the recording directory.
+
+    Returns:
+        The wormhole code if successful, None otherwise.
+    """
+    recording_path = Path(recording_dir)
+
+    if not recording_path.exists():
+        print(f"✗ Recording not found: {recording_path}")
+        return None
+
+    if not recording_path.is_dir():
+        print(f"✗ Not a directory: {recording_path}")
+        return None
+
+    if not _ensure_wormhole():
+        return None
+
+    # Create a temporary zip file
+    zip_name = f"{recording_path.name}.zip"
+
+    with tempfile.TemporaryDirectory() as tmpdir:
+        zip_path = Path(tmpdir) / zip_name
+
+        print(f"Compressing {recording_path.name}...")
+        with ZipFile(zip_path, "w", ZIP_DEFLATED, compresslevel=6) as zf:
+            for file in recording_path.rglob("*"):
+                if file.is_file():
+                    arcname = file.relative_to(recording_path.parent)
+                    zf.write(file, arcname)
+
+        size_mb = zip_path.stat().st_size / (1024 * 1024)
+        print(f"✓ Compressed to {size_mb:.1f} MB")
+
+        print("Sending via Magic Wormhole...")
+        print("(Keep this window open until transfer completes)")
+        print()
+
+        try:
+            # Run wormhole send
+            result = subprocess.run(
+                ["wormhole", "send", str(zip_path)],
+                check=True,
+            )
+            return "sent"  # Code is printed by wormhole itself
+        except subprocess.CalledProcessError as e:
+            print(f"✗ Wormhole send failed: {e}")
+            return None
+        except KeyboardInterrupt:
+            print("\n✗ Cancelled")
+            return None
+
+
+def receive(code: str, output_dir: str = ".") -> Path | None:
+    """Receive a recording via Magic Wormhole.
+
+    Args:
+        code: The wormhole code (e.g., "7-guitarist-revenge").
+        output_dir: Directory to save the recording (default: current dir).
+
+    Returns:
+        Path to the received recording directory, or None on failure.
+    """
+    if not _ensure_wormhole():
+        return None
+
+    output_path = Path(output_dir)
+    output_path.mkdir(parents=True, exist_ok=True)
+
+    with tempfile.TemporaryDirectory() as tmpdir:
+        tmpdir = Path(tmpdir)
+
+        print(f"Receiving from wormhole code: {code}")
+
+        try:
+            # Run wormhole receive
+            result = subprocess.run(
+                ["wormhole", "receive", "--accept-file", "-o", str(tmpdir), code],
+                check=True,
+            )
+
+            # Find the received zip file
+            zip_files = list(tmpdir.glob("*.zip"))
+            if not zip_files:
+                print("✗ No zip file received")
+                return None
+
+            zip_path = zip_files[0]
+            print(f"✓ Received {zip_path.name}")
+
+            # Extract
+            print("Extracting...")
+            with ZipFile(zip_path, "r") as zf:
+                zf.extractall(output_path)
+
+            # Find the extracted directory
+            extracted = [
+                p for p in output_path.iterdir()
+                if p.is_dir() and p.name != "__MACOSX"
+            ]
+
+            if extracted:
+                recording_dir = extracted[0]
+                print(f"✓ Saved to: {recording_dir}")
+                return recording_dir
+            else:
+                print(f"✓ Extracted to: {output_path}")
+                return output_path
+
+        except subprocess.CalledProcessError as e:
+            print(f"✗ Wormhole receive failed: {e}")
+            return None
+        except KeyboardInterrupt:
+            print("\n✗ Cancelled")
+            return None
+
+
+def main() -> None:
+    """CLI entry point for share commands."""
+    import fire
+    fire.Fire({
+        "send": send,
+        "receive": receive,
+    })
+
+
+if __name__ == "__main__":
+    main()

pyproject.toml

@@ -51,9 +51,14 @@ privacy = [
     "openadapt-privacy>=0.1.0",
 ]
 
+# Sharing via Magic Wormhole
+share = [
+    "magic-wormhole>=0.17.0",
+]
+
 # Everything
 all = [
-    "openadapt-capture[transcribe-fast,transcribe,privacy]",
+    "openadapt-capture[transcribe-fast,transcribe,privacy,share]",
 ]
 
 dev = [