remote-claude 0.1.0

package/stats/query.py

@@ -0,0 +1,151 @@
+"""
+StatsQuery：本地统计查询 + CLI 格式化输出
+"""
+
+import sqlite3
+import time
+from pathlib import Path
+from typing import Optional
+
+_DB_PATH = Path.home() / ".local" / "share" / "remote-claude" / "stats.db"
+
+
+def _get_conn() -> Optional[sqlite3.Connection]:
+    if not _DB_PATH.exists():
+        return None
+    try:
+        return sqlite3.connect(str(_DB_PATH))
+    except Exception:
+        return None
+
+
+def _date_range(range_str: str) -> tuple[str, str]:
+    """解析 range 字符串，返回 (start_date, end_date)"""
+    end_date = time.strftime('%Y-%m-%d')
+    if range_str == 'today' or range_str is None:
+        start_date = end_date
+    elif range_str.endswith('d'):
+        days = int(range_str[:-1])
+        start_date = time.strftime(
+            '%Y-%m-%d', time.localtime(time.time() - days * 86400)
+        )
+    elif range_str.endswith('m'):
+        months = int(range_str[:-1])
+        start_date = time.strftime(
+            '%Y-%m-%d', time.localtime(time.time() - months * 30 * 86400)
+        )
+    else:
+        start_date = end_date
+    return start_date, end_date
+
+
+def _fmt_num(n: int) -> str:
+    """格式化数字（千分位逗号）"""
+    return f"{n:,}"
+
+
+def query_summary(range_str: str = 'today', session_name: str = '',
+                  detail: bool = False) -> str:
+    """生成统计摘要字符串"""
+    conn = _get_conn()
+    if conn is None:
+        return "暂无统计数据（数据库未初始化）"
+
+    start_date, end_date = _date_range(range_str)
+
+    # 构建查询条件
+    where = "WHERE date >= ? AND date <= ?"
+    params: list = [start_date, end_date]
+    if session_name:
+        where += " AND session_name = ?"
+        params.append(session_name)
+
+    def _sum(category: str, event: str) -> int:
+        row = conn.execute(
+            f"SELECT COALESCE(SUM(value), 0) FROM events "
+            f"{where} AND category=? AND event=?",
+            params + [category, event]
+        ).fetchone()
+        return row[0] if row else 0
+
+    def _count(category: str, event: str) -> int:
+        row = conn.execute(
+            f"SELECT COUNT(*) FROM events "
+            f"{where} AND category=? AND event=?",
+            params + [category, event]
+        ).fetchone()
+        return row[0] if row else 0
+
+    # 收集各类指标
+    sess_start = _count('session', 'start')
+    sess_attach = _count('session', 'attach')
+    sess_end = _count('session', 'end')
+
+    term_connect = _count('terminal', 'connect')
+    term_input = _count('terminal', 'input')
+    term_disconnect = _count('terminal', 'disconnect')
+
+    lark_msg = _count('lark', 'message')
+    lark_opt = _count('lark', 'option_select')
+    lark_key = _count('lark', 'raw_key')
+    lark_cmd = _count('lark', 'cmd')
+    lark_attach = _count('lark', 'attach')
+    lark_cmd_start = _count('lark', 'cmd_start')
+
+    card_create = _count('card', 'create')
+    card_update = _count('card', 'update')
+    card_freeze = _count('card', 'freeze')
+    card_fallback = _count('card', 'fallback')
+
+    token_total = _sum('token', 'usage')
+
+    err_card = _count('error', 'card_api')
+    err_bridge = _count('error', 'bridge_send')
+    err_total = err_card + err_bridge
+    card_ops = card_create + card_update
+    err_rate = f"{err_total / card_ops * 100:.2f}%" if card_ops else "0%"
+
+    # 构建输出
+    if start_date == end_date:
+        date_label = start_date
+    else:
+        date_label = f"{start_date} ~ {end_date}"
+
+    lines = [
+        f"━━━━ Remote Claude 使用统计 ({date_label}) ━━━━",
+        "",
+        f"会话:  启动 {_fmt_num(sess_start)} | 连接 {_fmt_num(sess_attach + term_connect + lark_attach)} | 结束 {_fmt_num(sess_end)}",
+        f"终端:  输入 {_fmt_num(term_input)} 次 | 连接 {_fmt_num(term_connect)} 次",
+        f"飞书:  消息 {_fmt_num(lark_msg)} | 选项 {_fmt_num(lark_opt)} | 快捷键 {_fmt_num(lark_key)} | 命令 {_fmt_num(lark_cmd)}",
+        f"卡片:  创建 {_fmt_num(card_create)} | 更新 {_fmt_num(card_update)} | 冻结 {_fmt_num(card_freeze)}",
+        f"Token: ~{token_total / 1000:.1f}k",
+        f"错误:  API {_fmt_num(err_card)} ({err_rate})",
+    ]
+
+    if detail:
+        lines += [
+            "",
+            "── 详细分类 ──",
+            f"飞书会话:  cmd_start {_fmt_num(lark_cmd_start)} | attach {_fmt_num(lark_attach)}",
+            f"卡片降级:  {_fmt_num(card_fallback)}",
+            f"桥接错误:  {_fmt_num(err_bridge)}",
+        ]
+
+    conn.close()
+    return "\n".join(lines)
+
+
+def reset_stats() -> str:
+    """清空所有统计数据"""
+    conn = _get_conn()
+    if conn is None:
+        return "暂无统计数据"
+    try:
+        conn.execute("DELETE FROM events")
+        conn.execute("DELETE FROM daily_summary")
+        conn.execute("DELETE FROM meta WHERE key != 'first_run'")
+        conn.commit()
+        conn.close()
+        return "统计数据已清空"
+    except Exception as e:
+        return f"清空失败: {e}"

package/utils/components.py

@@ -0,0 +1,165 @@
+"""
+Claude CLI 输出组件模型
+
+将 Claude 终端输出解析为结构化组件，供 CardBuilder 渲染为飞书卡片。
+"""
+
+from dataclasses import dataclass, field
+from typing import List, Optional, Union
+
+
+@dataclass
+class OutputBlock:
+    """统一输出块（文本回复、工具调用、Agent/Plan 块的统一表示）
+
+    parser 不区分三者，统一产出此类型。content 为圆点字符之后的内容（已去掉首列圆点）。
+    card_builder 可根据首行内容格式决定渲染方式。
+    """
+    content: str          # 完整块内容（首行去掉圆点，后续行原样，\n 连接）
+    is_streaming: bool = False  # 首行首列圆点是否 blink
+    start_row: int = -1   # 在终端屏幕中的起始行号（用于帧间同一 Block 识别，-1 表示未知）
+    ansi_content: str = ""     # 对应 content，带 ANSI 转义码
+    indicator: str = ""        # 首列圆点字符原文（如 ●）
+    ansi_indicator: str = ""   # 带 ANSI 颜色的圆点字符
+
+
+@dataclass
+class TextBlock:
+    """文本回复（保留作为向后兼容，新代码请使用 OutputBlock）"""
+    content: str
+    is_streaming: bool = False
+
+
+@dataclass
+class UserInput:
+    """用户输入（❯ 行）"""
+    text: str
+    ansi_text: str = ""        # 对应 text，带 ANSI 转义码
+    indicator: str = ""        # 首列提示符原文（❯）
+    ansi_indicator: str = ""   # 带 ANSI 颜色的提示符
+
+
+@dataclass
+class ToolCall:
+    """工具调用块（保留作为向后兼容，新代码请使用 OutputBlock）"""
+    tool_name: str
+    args_summary: str
+    status: str = "running"
+    status_detail: str = ""
+    output: str = ""
+    is_streaming: bool = False
+
+
+@dataclass
+class AgentBlock:
+    """Agent/Plan 块（保留作为向后兼容，新代码请使用 OutputBlock）"""
+    agent_type: str
+    description: str
+    status: str = "running"
+    status_detail: str = ""
+    stats: str = ""
+    sub_calls: List[str] = field(default_factory=list)
+    is_streaming: bool = False
+
+
+@dataclass
+class OptionBlock:
+    """选项交互块（统一 option + permission 两种场景）
+
+    状态型组件：全局唯一，存储在 ClaudeWindow.option_block，不进入 blocks 累积列表。
+    - sub_type="option"：AskUserQuestion 选项（2 分割线 input_rows 中检测到编号选项）
+    - sub_type="permission"：权限确认（1 分割线 bottom_rows 中检测到编号选项）
+    """
+    sub_type: str = "option"   # "option" | "permission"
+    tag: str = ""              # 分类标签（option 场景）
+    title: str = ""            # 工具名称（permission 场景）
+    content: str = ""          # 详细内容（permission 场景）
+    question: str = ""         # 问题文本
+    options: List[dict] = field(default_factory=list)  # [{"label": str, "value": str}]
+    ansi_raw: str = ""         # 整个选项区域的 ANSI 原始文本
+    indicator: str = ""        # 首列字符原文
+    ansi_indicator: str = ""   # 带 ANSI 颜色的首列字符
+
+
+# 向后兼容别名
+PermissionBlock = OptionBlock
+
+
+@dataclass
+class StatusLine:
+    """状态行（✱ 动作... (时间 · tokens)）"""
+    action: str          # 动作动词（如 "Germinating..."）
+    elapsed: str = ""    # 时长（如 "16m 33s"）
+    tokens: str = ""     # token 消耗（如 "↓ 4.3k tokens"）
+    raw: str = ""        # 原始文本
+    ansi_raw: str = ""         # 对应 raw，带 ANSI 转义码
+    indicator: str = ""        # 首列星星字符原文（如 ✱）
+    ansi_indicator: str = ""   # 带 ANSI 颜色的星星字符
+
+
+@dataclass
+class Divider:
+    """水平分割线（区域边界标记，不渲染）
+
+    Claude CLI 终端有 2 条由 ─ 组成的分割线，将屏幕分为 3 个区域：
+    输出区（上）| ─── | 用户输入框（中）| ─── | 底部栏（下）
+    解析时用于定位区域边界，不作为卡片内容渲染。
+    """
+    pass
+
+
+@dataclass
+class BottomBar:
+    """底部栏（权限模式、后台任务等状态信息）
+
+    固定在终端最底部的状态栏，内容如：
+    - ▶▶ bypass permissions on (shift+tab to cycle) · esc to interrupt
+    - 2 bashes · ↓ to manage
+    - 4 local agents · ↓ to manage · ctrl+f to kill agents
+    """
+    text: str  # 原始文本内容
+    ansi_text: str = ""  # 对应 text，带 ANSI 转义码
+    has_background_agents: bool = False  # 底部栏是否包含后台 agent 信息
+    agent_count: int = 0                 # 后台 agent 数量
+    agent_summary: str = ""              # agent 摘要文本（如 "4 local agents"）
+
+
+@dataclass
+class AgentPanelBlock:
+    """Agent 管理面板（用户按 ↓ 展开的列表或 Enter 查看的详情）
+
+    出现在只有 1 条分割线的特殊布局中（输入区消失）。
+    列表模式：显示所有后台 agent 及其状态
+    详情模式：显示单个 agent 的详细信息
+    """
+    panel_type: str = "list"  # "list" | "detail"
+    # 列表模式字段
+    agent_count: int = 0
+    agents: List[dict] = field(default_factory=list)  # [{"name": str, "status": str, "is_selected": bool}]
+    # 详情模式字段
+    agent_name: str = ""
+    agent_type: str = ""
+    stats: str = ""
+    progress: str = ""
+    prompt: str = ""
+    # 通用字段
+    raw_text: str = ""
+    ansi_raw: str = ""
+
+
+@dataclass
+class PlanBlock:
+    """计划块（╭│╰ box-drawing 框线包裹的结构化计划内容）
+
+    累积型 Block：出现在输出区，保留在 blocks 历史列表中。
+    Claude CLI Plan Mode 显示的计划内容。
+    """
+    title: str           # 框内第一行非空文本（如 "Plan to implement"）
+    content: str         # 框内完整内容（去掉边框字符，\\n 连接）
+    is_streaming: bool = False  # 始终为 False（╭ 不 blink）
+    start_row: int = -1
+    ansi_content: str = ""
+
+
+# 所有组件类型的联合类型
+Component = Union[OutputBlock, TextBlock, UserInput, ToolCall, AgentBlock, OptionBlock, PermissionBlock, StatusLine, Divider, BottomBar, AgentPanelBlock, PlanBlock]

package/utils/protocol.py

@@ -0,0 +1,164 @@
+"""
+通信协议定义
+
+消息格式：JSON + 换行符分隔
+二进制数据使用 base64 编码
+"""
+
+import json
+import base64
+from dataclasses import dataclass, asdict
+from typing import Optional, List
+from enum import Enum
+
+
+class MessageType(str, Enum):
+    """消息类型"""
+    INPUT = "input"          # 客户端 -> 服务端：用户输入
+    OUTPUT = "output"        # 服务端 -> 客户端：Claude 输出
+    HISTORY = "history"      # 历史输出（重连时）
+    ERROR = "error"          # 错误消息
+    RESIZE = "resize"        # 终端大小变化
+
+
+@dataclass
+class Message:
+    """基础消息"""
+    type: str
+
+    def to_json(self) -> str:
+        return json.dumps(asdict(self), ensure_ascii=False)
+
+    @classmethod
+    def from_json(cls, data: str) -> "Message":
+        obj = json.loads(data)
+        msg_type = obj.get("type")
+
+        if msg_type == MessageType.INPUT:
+            return InputMessage.from_dict(obj)
+        elif msg_type == MessageType.OUTPUT:
+            return OutputMessage.from_dict(obj)
+        elif msg_type == MessageType.HISTORY:
+            return HistoryMessage.from_dict(obj)
+        elif msg_type == MessageType.ERROR:
+            return ErrorMessage.from_dict(obj)
+        elif msg_type == MessageType.RESIZE:
+            return ResizeMessage.from_dict(obj)
+        else:
+            raise ValueError(f"Unknown message type: {msg_type}")
+
+
+@dataclass
+class InputMessage(Message):
+    """用户输入消息"""
+    data: str  # base64 编码的输入
+    client_id: str
+
+    def __init__(self, data: bytes, client_id: str):
+        super().__init__(type=MessageType.INPUT)
+        self.data = base64.b64encode(data).decode('ascii')
+        self.client_id = client_id
+
+    def get_data(self) -> bytes:
+        return base64.b64decode(self.data)
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> "InputMessage":
+        msg = object.__new__(cls)
+        msg.type = obj["type"]
+        msg.data = obj["data"]
+        msg.client_id = obj["client_id"]
+        return msg
+
+
+@dataclass
+class OutputMessage(Message):
+    """Claude 输出消息"""
+    data: str  # base64 编码的输出
+
+    def __init__(self, data: bytes):
+        super().__init__(type=MessageType.OUTPUT)
+        self.data = base64.b64encode(data).decode('ascii')
+
+    def get_data(self) -> bytes:
+        return base64.b64decode(self.data)
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> "OutputMessage":
+        msg = object.__new__(cls)
+        msg.type = obj["type"]
+        msg.data = obj["data"]
+        return msg
+
+
+@dataclass
+class HistoryMessage(Message):
+    """历史输出消息（重连时发送）"""
+    data: str  # base64 编码的历史输出
+
+    def __init__(self, data: bytes):
+        super().__init__(type=MessageType.HISTORY)
+        self.data = base64.b64encode(data).decode('ascii')
+
+    def get_data(self) -> bytes:
+        return base64.b64decode(self.data)
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> "HistoryMessage":
+        msg = object.__new__(cls)
+        msg.type = obj["type"]
+        msg.data = obj["data"]
+        return msg
+
+
+@dataclass
+class ErrorMessage(Message):
+    """错误消息"""
+    message: str
+    code: Optional[str] = None
+
+    def __init__(self, message: str, code: Optional[str] = None):
+        super().__init__(type=MessageType.ERROR)
+        self.message = message
+        self.code = code
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> "ErrorMessage":
+        msg = object.__new__(cls)
+        msg.type = obj["type"]
+        msg.message = obj["message"]
+        msg.code = obj.get("code")
+        return msg
+
+
+@dataclass
+class ResizeMessage(Message):
+    """终端大小变化消息"""
+    rows: int
+    cols: int
+    client_id: str
+
+    def __init__(self, rows: int, cols: int, client_id: str):
+        super().__init__(type=MessageType.RESIZE)
+        self.rows = rows
+        self.cols = cols
+        self.client_id = client_id
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> "ResizeMessage":
+        msg = object.__new__(cls)
+        msg.type = obj["type"]
+        msg.rows = obj["rows"]
+        msg.cols = obj["cols"]
+        msg.client_id = obj["client_id"]
+        return msg
+
+
+def encode_message(msg: Message) -> bytes:
+    """编码消息为字节流（JSON + 换行符）"""
+    return (msg.to_json() + "\n").encode('utf-8')
+
+
+def decode_message(data: bytes) -> Message:
+    """解码消息"""
+    return Message.from_json(data.decode('utf-8').strip())