claude-nb 0.3.0

package/lib/board_msg.py

@@ -0,0 +1,230 @@
+"""board_msg — send / inbox / ack / status / log commands."""
+
+import hashlib
+import shutil
+import subprocess
+from pathlib import Path
+
+from lib.board_db import BoardDB, ts
+from lib.common import parse_flags
+
+
+def _ack_marker_path(db: BoardDB, name: str) -> Path:
+    """Path to file recording max message_id seen by last inbox call."""
+    return db.env.sessions_dir / f".{name}.ack_max_id"
+
+
+def _nudge_session(db: BoardDB, recipient: str) -> None:
+    """Inject a fixed prompt into the recipient's tmux session to check inbox.
+
+    Only sends a hardcoded command — never user-controlled content — to avoid
+    shell metacharacter injection via tmux send-keys.
+    """
+    if recipient == "all":
+        sessions = [r[0] for r in db.query("SELECT name FROM sessions WHERE name != 'all'")]
+    else:
+        sessions = [recipient]
+    prefix = db.env.prefix
+    board = db.env.install_home / "bin" / "board"
+    for name in sessions:
+        if not name.isalnum():
+            continue
+        sess = f"{prefix}-{name}"
+        r = subprocess.run(["tmux", "has-session", "-t", sess], capture_output=True)
+        if r.returncode == 0:
+            cmd = f"{board} --as {name} inbox"
+            subprocess.run(["tmux", "send-keys", "-t", sess, "-l", cmd], capture_output=True)
+            subprocess.run(["tmux", "send-keys", "-t", sess, "Enter"], capture_output=True)
+
+
+def cmd_send(db: BoardDB, identity: str, args: list[str]) -> None:
+    name = identity.lower()
+    flags, send_args = parse_flags(args, value_flags={"attach": ["--attach", "-a"]})
+    attach_file = flags.get("attach")
+
+    if not send_args:
+        print("Usage: ./board --as <name> send <to> <message> [--attach <file>]")
+        raise SystemExit(1)
+
+    to = send_args[0].lower()
+
+    if to != "all" and not db.scalar("SELECT COUNT(*) FROM sessions WHERE name=?", (to,)):
+        print(f"ERROR: 收件人 '{to}' 不存在")
+        raise SystemExit(1)
+
+    msg = " ".join(send_args[1:]) if len(send_args) > 1 else ""
+
+    if not msg and not attach_file:
+        print("ERROR: 消息不能为空")
+        raise SystemExit(1)
+
+    attach_ref = ""
+    stored_path = ""
+    h = ""
+    if attach_file:
+        path = Path(attach_file)
+        if not path.is_file():
+            print(f"ERROR: file not found: {attach_file}")
+            raise SystemExit(1)
+        data = path.read_bytes()
+        h = hashlib.sha256(data).hexdigest()[:12]
+        ext = path.suffix.lstrip(".")
+        orig = path.name
+        stored_path = f"files/{h}.{ext}" if ext else f"files/{h}"
+        files_dir = db.env.claudes_dir / "files"
+        files_dir.mkdir(exist_ok=True)
+        dest = db.env.claudes_dir / stored_path
+        if not dest.exists():
+            shutil.copy2(str(path), str(dest))
+        attach_ref = f" [附件: {orig} → {stored_path}]"
+        if not msg:
+            msg = f"分享文件: {orig}"
+
+    full_msg = msg + attach_ref
+    now = ts()
+    attach_val = h if attach_file else None
+
+    with db.conn() as c:
+        if attach_file:
+            db.execute(
+                "INSERT OR IGNORE INTO files(hash, original_name, extension, sender, stored_path) "
+                "VALUES (?, ?, ?, ?, ?)",
+                (h, orig, ext, name, stored_path),
+                c=c,
+            )
+        msg_id = db.execute(
+            "INSERT INTO messages(ts, sender, recipient, body, attachment) VALUES (?, ?, ?, ?, ?)",
+            (now, name, to, full_msg, attach_val),
+            c=c,
+        )
+        db.deliver_to_inbox(name, to, msg_id, c=c)
+
+    print("OK sent")
+    if attach_ref:
+        print(f"  附件已存储: {stored_path}")
+
+    _nudge_session(db, to)
+
+
+def cmd_status(db: BoardDB, identity: str, args: list[str]) -> None:
+    name = identity.lower()
+    if not args:
+        print("Usage: ./board --as <name> status <description>")
+        raise SystemExit(1)
+    desc = " ".join(args)
+    now = ts()
+    full_status = f"{desc} — {now}"
+    db.execute(
+        "UPDATE sessions SET status=?, updated_at=? WHERE name=?",
+        (full_status, now, name),
+    )
+    db.sync_status_to_file(name, full_status)
+    print("OK status updated")
+
+
+def cmd_inbox(db: BoardDB, identity: str) -> None:
+    name = identity.lower()
+    if not db.scalar("SELECT COUNT(*) FROM sessions WHERE name=?", (name,)):
+        print(f"ERROR: 会话 '{name}' 未注册")
+        raise SystemExit(1)
+    count = db.scalar("SELECT COUNT(*) FROM inbox WHERE session=? AND read=0", (name,))
+    if not count:
+        print("收件箱为空")
+    else:
+        print(f"你有 {count} 条未读:")
+        rows = db.query(
+            "SELECT i.message_id, '- [' || m.ts || '] **' || m.sender || '**: ' || m.body "
+            "FROM inbox i JOIN messages m ON i.message_id=m.id "
+            "WHERE i.session=? AND i.read=0 ORDER BY m.ts",
+            (name,),
+        )
+        max_id = 0
+        for msg_id, line in rows:
+            print(line)
+            if msg_id > max_id:
+                max_id = msg_id
+        if max_id > 0:
+            _ack_marker_path(db, name).write_text(str(max_id))
+        print(f"\n(运行 ack 清除: board --as {identity} ack)")
+
+    _task_print_queue_short(db, name)
+
+
+def cmd_ack(db: BoardDB, identity: str) -> None:
+    name = identity.lower()
+    marker = _ack_marker_path(db, name)
+    max_id = None
+    if marker.exists():
+        try:
+            max_id = int(marker.read_text().strip())
+        except ValueError:
+            pass
+
+    if max_id:
+        count = db.scalar(
+            "SELECT COUNT(*) FROM inbox WHERE session=? AND read=0 AND message_id<=?",
+            (name, max_id),
+        )
+    else:
+        count = db.scalar("SELECT COUNT(*) FROM inbox WHERE session=? AND read=0", (name,))
+
+    if count == 0:
+        print("收件箱已经是空的")
+        db.clear_inbox_file(name)
+        marker.unlink(missing_ok=True)
+        return
+
+    if max_id:
+        db.execute(
+            "UPDATE inbox SET read=1 WHERE session=? AND read=0 AND message_id<=?",
+            (name, max_id),
+        )
+    else:
+        db.execute("UPDATE inbox SET read=1 WHERE session=? AND read=0", (name,))
+
+    print(f"OK {count} 条已清空（完整记录在 messages.log）")
+    db.clear_inbox_file(name)
+    marker.unlink(missing_ok=True)
+
+
+def cmd_log(db: BoardDB, identity: str, args: list[str]) -> None:
+    flags, positional = parse_flags(args, bool_flags={"mine": ["--mine"]})
+    filter_name = identity.lower() if flags.get("mine") and identity else ""
+    n = 20
+    for a in positional:
+        try:
+            n = int(a)
+        except ValueError:
+            pass
+
+    if filter_name:
+        rows = db.query(
+            "SELECT '[' || ts || '] ' || sender || ' → ' || recipient || ': ' || body "
+            "FROM messages WHERE sender=? OR recipient=? OR recipient='all' "
+            "ORDER BY id DESC LIMIT ?",
+            (filter_name, filter_name, n),
+        )
+    else:
+        rows = db.query(
+            "SELECT '[' || ts || '] ' || sender || ' → ' || recipient || ': ' || body "
+            "FROM messages ORDER BY id DESC LIMIT ?",
+            (n,),
+        )
+    for (line,) in reversed(rows):
+        print(line)
+
+
+def _task_print_queue_short(db: BoardDB, target: str) -> None:
+    rows = db.query(
+        "SELECT id, status, priority, description FROM tasks "
+        "WHERE session=? AND status != 'done' "
+        "ORDER BY CASE status WHEN 'active' THEN 0 ELSE 1 END, priority DESC, id ASC",
+        (target,),
+    )
+    print("\n任务队列:")
+    if not rows:
+        print("  (无待办任务)")
+        return
+    for tid, status, priority, desc in rows:
+        marker = "*" if status == "active" else " "
+        print(f"  {marker} #{tid} [{status} p{priority}] {desc}")

package/lib/board_task.py

@@ -0,0 +1,200 @@
+"""board_task — task queue: add / done / list / next."""
+
+from lib.board_db import BoardDB, ts
+from lib.common import is_privileged, is_terminal_task_status, parse_flags
+
+
+def _promote_next(db: BoardDB, target: str) -> None:
+    active = db.scalar("SELECT COUNT(*) FROM tasks WHERE session=? AND status='active'", (target,))
+    if active:
+        return
+    next_id = db.scalar(
+        "SELECT id FROM tasks WHERE session=? AND status='pending' ORDER BY priority DESC, id ASC LIMIT 1",
+        (target,),
+    )
+    if next_id:
+        db.execute("UPDATE tasks SET status='active' WHERE id=?", (next_id,))
+
+
+def _print_queue(db: BoardDB, target: str, include_done: bool = False) -> None:
+    if include_done:
+        rows = db.query(
+            "SELECT id, status, priority, description, created_at, COALESCE(done_at, '') "
+            "FROM tasks WHERE session=? "
+            "ORDER BY CASE status WHEN 'active' THEN 0 WHEN 'pending' THEN 1 ELSE 2 END, "
+            "priority DESC, id ASC",
+            (target,),
+        )
+    else:
+        rows = db.query(
+            "SELECT id, status, priority, description, created_at, '' "
+            "FROM tasks WHERE session=? AND status NOT IN ('done') "
+            "ORDER BY CASE status WHEN 'active' THEN 0 ELSE 1 END, "
+            "priority DESC, id ASC",
+            (target,),
+        )
+    print("\n任务队列:")
+    if not rows:
+        print("  (无待办任务)")
+        return
+    for tid, status, priority, desc, _created, done_at in rows:
+        marker = "*" if status == "active" else " "
+        if status == "done":
+            print(f"  {marker} #{tid} [{status} p{priority}] {desc} (done {done_at})")
+        else:
+            print(f"  {marker} #{tid} [{status} p{priority}] {desc}")
+
+
+def cmd_task(db: BoardDB, identity: str, args: list[str]) -> None:
+    subcmd = args[0] if args else "list"
+    rest = args[1:] if len(args) > 1 else []
+
+    if subcmd == "add":
+        _task_add(db, identity, rest)
+    elif subcmd == "done":
+        _task_done(db, identity, rest)
+    elif subcmd == "list":
+        _task_list(db, identity, rest)
+    elif subcmd == "next":
+        _task_next(db, identity)
+    else:
+        print("Usage: ./board --as <name> task {add|done|list|next}")
+        raise SystemExit(1)
+
+
+def _task_add(db: BoardDB, identity: str, args: list[str]) -> None:
+    name = identity.lower()
+    flags, desc_parts = parse_flags(
+        args,
+        value_flags={"to": ["--to"], "priority": ["--priority", "-p"]},
+    )
+    target = str(flags.get("to", name)).lower()
+    priority = int(flags["priority"]) if "priority" in flags else 0
+
+    if not desc_parts:
+        print("Usage: ./board --as <name> task add [--to session] [--priority N] <description>")
+        raise SystemExit(1)
+
+    desc = " ".join(desc_parts)
+
+    db.ensure_session(target)
+
+    active = db.scalar("SELECT COUNT(*) FROM tasks WHERE session=? AND status='active'", (target,))
+    status = "active" if not active else "pending"
+
+    task_id = db.execute(
+        "INSERT INTO tasks(session, description, status, priority) VALUES (?, ?, ?, ?)",
+        (target, desc, status, priority),
+    )
+    print(f"OK task #{task_id} added to {target} ({status})")
+
+    if target != name:
+        now = ts()
+        msg_id = db.execute(
+            "INSERT INTO messages(ts, sender, recipient, body) VALUES (?, ?, ?, ?)",
+            (now, name, target, f"[TASK #{task_id}] {desc}"),
+        )
+        db.execute("INSERT INTO inbox(session, message_id) VALUES (?, ?)", (target, msg_id))
+        db.sync_inbox_to_file(target)
+        print(f"OK notified {target}")
+    _print_queue(db, target)
+
+
+def _task_done(db: BoardDB, identity: str, args: list[str]) -> None:
+    name = identity.lower()
+
+    task_id = args[0] if args else None
+
+    if not task_id:
+        _promote_next(db, name)
+        task_id = db.scalar(
+            "SELECT id FROM tasks WHERE session=? AND status='active' ORDER BY id ASC LIMIT 1",
+            (name,),
+        )
+    if not task_id:
+        print(f"No active task for {name}.")
+        _print_queue(db, name)
+        return
+
+    task_id = int(task_id)
+    row = db.query_one("SELECT session, status, description FROM tasks WHERE id=?", (task_id,))
+    if not row:
+        print(f"ERROR: task #{task_id} not found")
+        raise SystemExit(1)
+
+    assignee, status, desc = row
+    if assignee != name and not is_privileged(name):
+        print(f"ERROR: task #{task_id} belongs to {assignee}; only owner, Orca, or Coral can mark it done")
+        raise SystemExit(1)
+
+    if is_terminal_task_status(status):
+        print(f"Task #{task_id} is already done.")
+        _print_queue(db, assignee)
+        return
+
+    now = ts()
+    db.execute("UPDATE tasks SET status='done', done_at=? WHERE id=?", (now, task_id))
+    print(f"OK task #{task_id} done: {desc}")
+
+    _promote_next(db, assignee)
+    nxt = db.query_one(
+        "SELECT id, description FROM tasks WHERE session=? AND status='active' ORDER BY id ASC LIMIT 1",
+        (assignee,),
+    )
+    if nxt:
+        print(f"Next: #{nxt[0]} {nxt[1]}")
+    else:
+        print(f"No remaining active/pending tasks for {assignee}.")
+    _print_queue(db, assignee)
+
+
+def _task_list(db: BoardDB, identity: str, args: list[str]) -> None:
+    flags, positional = parse_flags(
+        args,
+        value_flags={"session": ["--session"]},
+        bool_flags={"all": ["--all"], "done": ["--done", "--include-done"]},
+    )
+    all_sessions = bool(flags.get("all"))
+    include_done = bool(flags.get("done"))
+    target = str(flags["session"]).lower() if "session" in flags else ""
+
+    if not target and positional:
+        target = positional[0].lower()
+        if len(positional) > 1:
+            print("Usage: ./board task list [session|--all] [--done]")
+            raise SystemExit(1)
+
+    if all_sessions:
+        print("=== Task Queue ===")
+        if include_done:
+            rows = db.query(
+                "SELECT session, id, status, priority, description FROM tasks "
+                "ORDER BY session, CASE status WHEN 'active' THEN 0 WHEN 'pending' THEN 1 ELSE 2 END, "
+                "priority DESC, id ASC"
+            )
+        else:
+            rows = db.query(
+                "SELECT session, id, status, priority, description FROM tasks WHERE status != 'done' "
+                "ORDER BY session, CASE status WHEN 'active' THEN 0 ELSE 1 END, "
+                "priority DESC, id ASC"
+            )
+        if not rows:
+            print("  (no tasks)")
+            return
+        for session, tid, status, priority, desc in rows:
+            print(f"  {session:<8s} #{tid} [{status} p{priority}] {desc}")
+        return
+
+    if not target:
+        if not identity:
+            print("Usage: ./board task list [session|--all] [--done]")
+            raise SystemExit(1)
+        target = identity.lower()
+    _print_queue(db, target, include_done)
+
+
+def _task_next(db: BoardDB, identity: str) -> None:
+    name = identity.lower()
+
+    _promote_next(db, name)
+    _print_queue(db, name)