remote-claude 0.1.0

package/lark_client/main.py

@@ -0,0 +1,306 @@
+#!/usr/bin/env python3
+"""
+Remote Claude 飞书客户端
+
+通过飞书聊天控制 remote_claude 会话
+"""
+
+import asyncio
+import json
+import logging
+import signal
+import sys
+from pathlib import Path
+
+import lark_oapi as lark
+
+# 在 SDK 配置 logging 之前，先设置根 logger 和我们自己模块的 DEBUG 级别
+logging.basicConfig(
+    level=logging.DEBUG,
+    format='[%(name)s] %(message)s',
+)
+# 将噪音较大的第三方库保持 INFO 级别
+for _noisy in ('urllib3', 'websockets', 'asyncio'):
+    logging.getLogger(_noisy).setLevel(logging.INFO)
+from lark_oapi.api.im.v1 import P2ImMessageReceiveV1
+from lark_oapi.event.callback.model.p2_card_action_trigger import (
+    P2CardActionTrigger, P2CardActionTriggerResponse, CallBackToast
+)
+
+from . import config
+from .lark_handler import handler
+
+def check_user_allowed(user_id: str) -> bool:
+    """检查用户是否在白名单中"""
+    if not config.ENABLE_USER_WHITELIST:
+        return True
+    return user_id in config.ALLOWED_USERS
+
+
+def handle_message_receive(data: P2ImMessageReceiveV1) -> None:
+    """处理收到的消息"""
+    try:
+        event = data.event
+        message = event.message
+        sender = event.sender
+
+        # 获取基本信息
+        user_id = sender.sender_id.open_id
+        chat_id = message.chat_id
+        message_type = message.message_type
+        chat_type = message.chat_type
+
+        # 检查用户白名单
+        if not check_user_allowed(user_id):
+            print(f"[Lark] 用户 {user_id} 不在白名单中")
+            return
+
+        # 只处理文本消息
+        if message_type != "text":
+            print(f"[Lark] 忽略非文本消息: {message_type}")
+            return
+
+        # 解析消息内容
+        content = json.loads(message.content)
+        text = content.get("text", "").strip()
+
+        # 移除 @ 提及
+        if message.mentions:
+            for mention in message.mentions:
+                text = text.replace(f"@_{mention.key}", "").strip()
+                text = text.replace(mention.key, "").strip()
+
+        if not text:
+            return
+
+        print(f"[Lark] 收到消息: {user_id[:8]}... -> {text[:50]}...")
+
+        # 异步处理消息（传入 chat_type 以支持群聊路由）
+        asyncio.create_task(handler.handle_message(user_id, chat_id, text, chat_type=chat_type))
+
+    except Exception as e:
+        print(f"[Lark] 处理消息异常: {e}")
+        import traceback
+        traceback.print_exc()
+
+
+def handle_card_action(event: P2CardActionTrigger) -> P2CardActionTriggerResponse:
+    """处理卡片按钮点击"""
+    try:
+        action = event.event.action
+        operator = event.event.operator
+        context = event.event.context
+
+        user_id = operator.open_id
+        chat_id = context.open_chat_id
+        message_id = context.open_message_id  # 原始卡片 message_id，用于就地更新
+        action_value = action.value or {}
+
+        print(f"[Lark] 收到卡片动作: user={user_id[:8]}..., action={action_value}")
+
+        # 检查用户白名单
+        if not check_user_allowed(user_id):
+            print(f"[Lark] 用户 {user_id} 不在白名单中")
+            toast = CallBackToast()
+            toast.type = "error"
+            toast.content = "您没有权限操作"
+            response = P2CardActionTriggerResponse()
+            response.toast = toast
+            return response
+
+        # 检测 form 提交（输入框 Enter ↵ 按钮）
+        form_value = getattr(action, 'form_value', None)
+        if form_value is not None:
+            command_text = (form_value.get("command") or "").strip()
+            print(f"[Lark] form 提交: user={user_id[:8]}..., command={command_text!r}")
+            if command_text:
+                # 有输入内容 → 直通 Claude
+                asyncio.create_task(handler.forward_to_claude(user_id, chat_id, command_text))
+            else:
+                # 空输入 → 发送原始 Enter 键（用于确认默认选项等场景）
+                asyncio.create_task(handler.send_raw_key(user_id, chat_id, "enter"))
+            return None
+
+        action_type = action_value.get("action", "")
+
+        # 处理选项选择动作
+        if action_type == "select_option":
+            option_value = action_value.get("value", "")
+            option_total = int(action_value.get("total", "0"))
+            print(f"[Lark] 用户选择了选项: {option_value} (total={option_total})")
+            asyncio.create_task(handler.handle_option_select(user_id, chat_id, option_value, option_total))
+            return None
+
+        # 列表卡片：进入会话
+        if action_type == "list_attach":
+            session_name = action_value.get("session", "")
+            print(f"[Lark] list_attach: session={session_name}")
+            asyncio.create_task(handler._cmd_attach(user_id, chat_id, session_name, message_id=message_id))
+            return None
+
+        # 列表卡片：断开连接
+        if action_type == "list_detach":
+            print(f"[Lark] list_detach: chat={chat_id[:8]}...")
+            asyncio.create_task(handler._handle_list_detach(user_id, chat_id, message_id=message_id))
+            return None
+
+        # 列表卡片：创建群聊
+        if action_type == "list_new_group":
+            session_name = action_value.get("session", "")
+            print(f"[Lark] list_new_group: session={session_name}")
+            asyncio.create_task(handler._cmd_new_group(user_id, chat_id, session_name))
+            return None
+
+        # 列表卡片：解散群聊
+        if action_type == "list_disband_group":
+            session_name = action_value.get("session", "")
+            print(f"[Lark] list_disband_group: session={session_name}")
+            asyncio.create_task(handler._cmd_disband_group(user_id, chat_id, session_name, message_id=message_id))
+            return None
+
+        # 目录卡片：进入子目录（继续浏览，就地更新原卡片）
+        if action_type == "dir_browse":
+            path = action_value.get("path", "")
+            print(f"[Lark] dir_browse: path={path}")
+            asyncio.create_task(handler._cmd_ls(user_id, chat_id, path, message_id=message_id))
+            return None
+
+        # 目录卡片：在该目录创建新 Claude 会话
+        if action_type == "dir_start":
+            path = action_value.get("path", "")
+            session_name = action_value.get("session_name", "")
+            print(f"[Lark] dir_start: path={path}, session={session_name}")
+            asyncio.create_task(handler._cmd_start(user_id, chat_id, f"{session_name} {path}"))
+            return None
+
+        # 目录卡片：在该目录启动会话并创建专属群聊
+        if action_type == "dir_new_group":
+            path = action_value.get("path", "")
+            session_name = action_value.get("session_name", "")
+            print(f"[Lark] dir_new_group: path={path}, session={session_name}")
+            asyncio.create_task(handler._cmd_start_and_new_group(user_id, chat_id, session_name, path))
+            return None
+
+        # /menu 卡片按钮
+        if action_type == "menu_detach":
+            asyncio.create_task(handler._cmd_detach(user_id, chat_id, message_id=message_id))
+            return None
+
+        if action_type == "menu_list":
+            asyncio.create_task(handler._cmd_list(user_id, chat_id, message_id=message_id))
+            return None
+
+        if action_type == "menu_help":
+            asyncio.create_task(handler._cmd_help(user_id, chat_id, message_id=message_id))
+            return None
+
+        if action_type == "menu_ls":
+            asyncio.create_task(handler._cmd_ls(user_id, chat_id, "", message_id=message_id))
+            return None
+
+        if action_type == "menu_tree":
+            asyncio.create_task(handler._cmd_ls(user_id, chat_id, "", tree=True, message_id=message_id))
+            return None
+
+        # 流式卡片：断开连接
+        if action_type == "stream_detach":
+            session_name = action_value.get("session", "")
+            print(f"[Lark] stream_detach: session={session_name}")
+            asyncio.create_task(handler._handle_stream_detach(user_id, chat_id, session_name, message_id=message_id))
+            return None
+
+        # 流式卡片：重新连接
+        if action_type == "stream_reconnect":
+            session_name = action_value.get("session", "")
+            print(f"[Lark] stream_reconnect: session={session_name}")
+            asyncio.create_task(handler._handle_stream_reconnect(user_id, chat_id, session_name, message_id=message_id))
+            return None
+
+        # 快捷键按钮（callback 模式）
+        if action_type == "send_key":
+            key_name = action_value.get("key", "")
+            print(f"[Lark] send_key: key={key_name}")
+            asyncio.create_task(handler.send_raw_key(user_id, chat_id, key_name))
+            return None
+
+        # 各卡片底部菜单按钮：辅助卡片就地→菜单，流式卡片降级新卡
+        if action_type == "menu_open":
+            asyncio.create_task(handler._cmd_menu(user_id, chat_id, message_id=message_id))
+            return None
+
+        # 默认响应
+        return P2CardActionTriggerResponse()
+
+    except Exception as e:
+        print(f"[Lark] 处理卡片动作异常: {e}")
+        import traceback
+        traceback.print_exc()
+        return P2CardActionTriggerResponse()
+
+
+class LarkBot:
+    """飞书机器人"""
+
+    def __init__(self):
+        self.ws_client = None
+        self.running = False
+
+    def start(self):
+        """启动机器人"""
+        # 检查配置
+        if not config.FEISHU_APP_ID or not config.FEISHU_APP_SECRET:
+            print("错误: 请配置 FEISHU_APP_ID 和 FEISHU_APP_SECRET")
+            print("在 .env 文件中添加:")
+            print("  FEISHU_APP_ID=your_app_id")
+            print("  FEISHU_APP_SECRET=your_app_secret")
+            return
+
+        print("=" * 50)
+        print("Remote Claude 飞书客户端")
+        print("=" * 50)
+        print(f"App ID: {config.FEISHU_APP_ID[:8]}...")
+        print(f"白名单: {'启用' if config.ENABLE_USER_WHITELIST else '禁用'}")
+        print("=" * 50)
+
+        # 设置信号处理
+        signal.signal(signal.SIGINT, self._signal_handler)
+        signal.signal(signal.SIGTERM, self._signal_handler)
+
+        # 创建事件处理器
+        event_handler = lark.EventDispatcherHandler.builder("", "") \
+            .register_p2_im_message_receive_v1(handle_message_receive) \
+            .register_p2_card_action_trigger(handle_card_action) \
+            .build()
+
+        # 创建 WebSocket 客户端
+        self.ws_client = lark.ws.Client(
+            config.FEISHU_APP_ID,
+            config.FEISHU_APP_SECRET,
+            event_handler=event_handler,
+            log_level=lark.LogLevel.INFO,
+        )
+
+        self.running = True
+        print("\n机器人已启动，等待消息...")
+        print("在飞书中发送 /help 查看使用说明\n")
+
+        # 启动 WebSocket（阻塞）
+        self.ws_client.start()
+
+    def _signal_handler(self, signum, frame):
+        """处理退出信号"""
+        print("\n正在关闭...")
+        self.running = False
+        if self.ws_client:
+            # WebSocket 客户端没有 stop 方法，直接退出
+            sys.exit(0)
+
+
+def main():
+    """入口函数"""
+    bot = LarkBot()
+    bot.start()
+
+
+if __name__ == "__main__":
+    main()

package/lark_client/output_cleaner.py

@@ -0,0 +1,222 @@
+"""
+Claude CLI 输出清理器
+
+策略：不模拟终端，直接从原始输出中提取有意义的文本
+"""
+
+import re
+from typing import List
+
+
+class OutputCleaner:
+    """Claude CLI 输出清理器"""
+
+    def __init__(self):
+        self._buffer = b""
+        self._last_meaningful_content = ""
+        self._user_input = ""
+
+    def feed(self, data: bytes) -> None:
+        """喂入原始输出数据"""
+        self._buffer += data
+
+    def set_user_input(self, text: str) -> None:
+        """设置用户输入（用于过滤回显）"""
+        self._user_input = text.strip()
+
+    def get_response(self) -> str:
+        """获取清理后的响应内容"""
+        text = self._buffer.decode('utf-8', errors='replace')
+        return self._extract_response(text)
+
+    def clear(self) -> None:
+        """清空缓冲区"""
+        self._buffer = b""
+        self._last_meaningful_content = ""
+
+    def _extract_response(self, text: str) -> str:
+        """从原始输出中提取 Claude 的回复"""
+        # 1. 移除所有 ANSI 转义序列
+        text = self._strip_ansi(text)
+
+        # 2. 移除所有控制字符（保留换行和空格）
+        text = re.sub(r'[\x00-\x08\x0b\x0c\x0e-\x1f\x7f]', '', text)
+
+        # 3. 处理换行
+        text = text.replace('\r\n', '\n').replace('\r', '\n')
+
+        # 4. 提取有意义的内容
+        lines = text.split('\n')
+        meaningful_lines = []
+
+        for line in lines:
+            cleaned = self._clean_line(line)
+            if cleaned:
+                meaningful_lines.append(cleaned)
+
+        # 5. 去重并合并
+        result = self._deduplicate_lines(meaningful_lines)
+
+        return result
+
+    def _strip_ansi(self, text: str) -> str:
+        """移除所有 ANSI 转义序列"""
+        # CSI 序列: ESC [ ... 字母
+        text = re.sub(r'\x1b\[[0-9;?]*[a-zA-Z]', '', text)
+        # OSC 序列: ESC ] ... BEL
+        text = re.sub(r'\x1b\][^\x07]*\x07', '', text)
+        # 其他转义序列
+        text = re.sub(r'\x1b[^[\]a-zA-Z]*[a-zA-Z]', '', text)
+        return text
+
+    def _clean_line(self, line: str) -> str:
+        """清理单行内容"""
+        stripped = line.strip()
+
+        if not stripped:
+            return ""
+
+        # 跳过边框和分隔线
+        if all(c in '─━═-–—│╭╮╰╯┌┐└┘├┤┬┴┼ ' for c in stripped):
+            return ""
+        if stripped.startswith(('╭', '╰', '│', '┌', '└', '├')):
+            return ""
+
+        # 跳过界面元素
+        skip_patterns = [
+            r'^Welcome to',
+            r'^Welcome back',
+            r'^Try\s*"',
+            r'esc to',
+            r'for shortcuts',
+            r'^\?.*shortcuts',
+            r'· thinking',
+            r'· Thinking',
+        ]
+        for pattern in skip_patterns:
+            if re.search(pattern, stripped, re.IGNORECASE):
+                return ""
+
+        # 跳过动画文本
+        animation_words = ['evaporating', 'seasoning', 'fiddling', 'symbioting', 'thinking']
+        stripped_lower = stripped.lower()
+        for word in animation_words:
+            # 只跳过完整包含动画词的行
+            if word in stripped_lower and len(stripped) < 50:
+                return ""
+
+        # 跳过特殊符号行
+        if stripped in ['❯', '>', '$', '⏺', '·', '( )', ';', '()', '( · )', '( ·', '✻', '✳']:
+            return ""
+        if all(c in '⠋⠙⠹⠸⠼⠴⠦⠧⠇⠏⠐⠂✻✳ ' for c in stripped):
+            return ""
+
+        # 跳过状态标签
+        if stripped in ['问候', 'Basic Math', 'Greeting', 'Code', 'Analysis', 'Initial Greeting']:
+            return ""
+
+        # 跳过命令提示符行
+        if stripped.startswith('❯') or stripped.startswith('>'):
+            return ""
+
+        # 跳过包含大量 ─ 的行（分隔线）
+        if '─' in stripped and stripped.count('─') > 3:
+            return ""
+
+        # 跳过短的无意义行（动画碎片）
+        if len(stripped) <= 3 and not self._has_cjk(stripped):
+            return ""
+
+        # 跳过只有 ... 或省略号的行
+        if stripped in ['…', '...', '..', '.']:
+            return ""
+
+        # 跳过用户输入回显
+        if self._user_input and stripped == self._user_input:
+            return ""
+
+        # 提取⏺符号后的内容（这通常是 Claude 的实际回复）
+        if '⏺' in stripped:
+            parts = stripped.split('⏺', 1)
+            if len(parts) > 1:
+                return parts[1].strip()
+
+        return stripped
+
+    def _has_cjk(self, text: str) -> bool:
+        """检查是否包含中日韩字符"""
+        for char in text:
+            if '\u4e00' <= char <= '\u9fff':
+                return True
+            if '\u3040' <= char <= '\u30ff':
+                return True
+            if '\uac00' <= char <= '\ud7a3':
+                return True
+        return False
+
+    def _deduplicate_lines(self, lines: List[str]) -> str:
+        """去除重复行"""
+        if not lines:
+            return ""
+
+        # 去除连续重复
+        deduped = [lines[0]]
+        for line in lines[1:]:
+            if line != deduped[-1]:
+                deduped.append(line)
+
+        # 合并结果
+        result = '\n'.join(deduped)
+
+        # 清理多余空白
+        result = re.sub(r'\n{3,}', '\n\n', result)
+        result = re.sub(r' {2,}', ' ', result)
+
+        return result.strip()
+
+
+def test_cleaner():
+    """测试清理器"""
+    cleaner = OutputCleaner()
+    cleaner.set_user_input("hello")
+
+    # 模拟真实捕获的输出
+    raw_outputs = [
+        b'\x1b[?2026h\x1b[2K\x1b[G\x1b[1A\r\x1b[2C\x1b[2Ahello\r\x1b[2B',
+        b'\x1b]0;\xe2\x9c\xb3 Greeting\x07',
+        b'\x1b[?2026h\r\x1b[3A\x1b[48;2;55;55;55m\x1b[38;2;80;80;80m\xe2\x9d\xaf \x1b[38;2;255;255;255mhello \x1b[39m\x1b[49m',
+        b'\xe2\x94\x80\xe2\x94\x80\xe2\x94\x80\xe2\x94\x80\r\n\x1b[39m\x1b[22m  \x1b[38;2;153;153;153m? for shortcuts\x1b[39m',
+        b'\x1b[?2026h\r\x1b[6A\x1b[38;2;255;255;255m\xe2\x8f\xba\x1b[1C\x1b[39m\xe4\xbd\xa0\xe5\xa5\xbd\xef\xbc\x81\xe6\x9c\x89\xe4\xbb\x80\xe4\xb9\x88\xe5\x8f\xaf\xe4\xbb\xa5\xe5\xb8\xae\xe4\xbd\xa0\xe7\x9a\x84\xe5\x90\x97\xef\xbc\x9f',
+    ]
+
+    for data in raw_outputs:
+        cleaner.feed(data)
+
+    result = cleaner.get_response()
+    print(f"清理后的响应:\n'{result}'")
+    print()
+
+    # 验证
+    errors = []
+    if "你好！有什么可以帮你的吗？" not in result:
+        errors.append(f"期望包含回复")
+    if "shortcuts" in result:
+        errors.append(f"不应包含 'shortcuts'")
+    if "hello" in result.lower():
+        errors.append(f"不应包含用户输入 'hello'")
+    if "❯" in result:
+        errors.append(f"不应包含命令提示符 '❯'")
+    if "─" in result:
+        errors.append(f"不应包含分隔线")
+
+    if errors:
+        print("✗ 测试失败:")
+        for e in errors:
+            print(f"  - {e}")
+        print(f"实际结果: '{result}'")
+    else:
+        print("✓ 测试通过")
+
+
+if __name__ == "__main__":
+    test_cleaner()