Claude Bridge Worker - one-click setup for remote Claude Code workers

1 files changed, 125 insertions, 0 deletions

diff --git a/check_tasks.py b/check_tasks.py
new file mode 100755
index 0000000..c1377b3
--- /dev/null
+++ b/check_tasks.py
@@ -0,0 +1,125 @@
+#!/usr/bin/env python3
+"""
+Claude Bridge - Lab Hook Script
+支持两种 hook 事件:
+  - Stop:        干完活检查新任务，有 task 就阻止停止
+  - PostToolUse: 工作中检查，message 注入 context 不打断
+
+任务类型:
+  - task:    需要领取执行（Stop 时阻塞）
+  - message: 只注入 context（btw 模式，不打断）
+"""
+
+import json
+import os
+import sys
+import urllib.request
+
+BROKER_URL = os.environ.get("BROKER_URL", "")
+API_SECRET = os.environ.get("API_SECRET", "")
+SESSION_NAME = os.environ.get("SESSION_NAME", "")
+
+if not BROKER_URL or not API_SECRET or not SESSION_NAME:
+    sys.exit(1)
+
+try:
+    hook_input = json.loads(sys.stdin.read())
+except Exception:
+    hook_input = {}
+
+event = hook_input.get("hook_event_name", "Stop")
+
+if event == "Stop" and hook_input.get("stop_hook_active", False):
+    sys.exit(0)
+
+
+def consume_task(task_id):
+    """Claim + mark done (for messages that just need to be seen)"""
+    try:
+        req = urllib.request.Request(
+            f"{BROKER_URL}/tasks/{task_id}/claim",
+            method="POST",
+            headers={"Authorization": f"Bearer {API_SECRET}"},
+        )
+        urllib.request.urlopen(req, timeout=5)
+        req2 = urllib.request.Request(
+            f"{BROKER_URL}/tasks/{task_id}/result",
+            data=json.dumps({"result": "seen"}).encode(),
+            headers={
+                "Authorization": f"Bearer {API_SECRET}",
+                "Content-Type": "application/json",
+            },
+            method="POST",
+        )
+        urllib.request.urlopen(req2, timeout=5)
+    except Exception:
+        pass
+
+
+# 发心跳（静默，失败不影响主流程）
+try:
+    import socket
+    hb_data = json.dumps({"session": SESSION_NAME, "host": socket.gethostname()}).encode()
+    hb_req = urllib.request.Request(
+        f"{BROKER_URL}/heartbeat",
+        data=hb_data,
+        headers={"Authorization": f"Bearer {API_SECRET}", "Content-Type": "application/json"},
+        method="POST",
+    )
+    urllib.request.urlopen(hb_req, timeout=3)
+except Exception:
+    pass
+
+try:
+    req = urllib.request.Request(
+        f"{BROKER_URL}/tasks/pending?target={SESSION_NAME}",
+        headers={"Authorization": f"Bearer {API_SECRET}"},
+    )
+    with urllib.request.urlopen(req, timeout=10) as resp:
+        data = json.loads(resp.read())
+
+    items = data.get("tasks", [])
+    if not items:
+        sys.exit(0)
+
+    tasks = [i for i in items if i.get("type", "task") == "task"]
+    messages = [i for i in items if i.get("type") == "message"]
+
+    output_lines = []
+
+    # Hook 被触发了 = Claude 在活跃状态，可以消费 messages
+    for m in messages:
+        output_lines.append(f"[Bridge btw 请用reply_to_dispatcher回复] {m['content']}")
+        consume_task(m["id"])
+
+    if event == "Stop" and tasks:
+        task_list = "\n".join(
+            [f"  - [{t['id']}] {t['content'][:120]}" for t in tasks]
+        )
+        if output_lines:
+            output_lines.append("")
+        output_lines.append(f"有 {len(tasks)} 个新任务等待执行:")
+        output_lines.append(task_list)
+        output_lines.append("")
+        output_lines.append("请用 fetch_pending_tasks 获取完整信息，claim_task 领取并执行。")
+        output = {
+            "decision": "block",
+            "reason": "\n".join(output_lines),
+        }
+        print(json.dumps(output))
+    elif event == "PostToolUse" and tasks:
+        task_list = "\n".join(
+            [f"  - [{t['id']}] {t['content'][:120]}" for t in tasks]
+        )
+        output_lines.append(
+            f"\n[Bridge 提示] 有 {len(tasks)} 个新任务，当前工作完成后请处理:\n{task_list}"
+        )
+        print("\n".join(output_lines))
+    elif output_lines:
+        print("\n".join(output_lines))
+
+    sys.exit(0)
+
+except Exception as e:
+    print(f"check_tasks error: {e}", file=sys.stderr)
+    sys.exit(1)