feedback-mcp 1.0.64__py3-none-any.whl

session_manager.py

@@ -0,0 +1,368 @@
+#!/usr/bin/env python3
+"""
+会话状态管理器
+用于管理stop hook和feedback UI之间的会话状态，避免死循环
+"""
+import json
+import os
+import sys
+from datetime import datetime, timedelta
+from pathlib import Path
+from typing import Optional, Dict, Any
+
+class SessionManager:
+    """管理会话状态，避免stop hook死循环
+
+    状态保存在 .workspace/chat_history/{session_id}.json 中
+    """
+
+    def __init__(self, session_id: Optional[str] = None, project_path: Optional[str] = None):
+        """初始化会话管理器
+
+        Args:
+            session_id: 会话ID
+            project_path: 项目路径，用于定位.workspace目录
+        """
+        self.session_id = session_id
+        self.project_path = project_path or os.getcwd()
+
+        # 不再加载所有sessions，改为按需加载单个session
+        # self.sessions = self._load_sessions()
+        # self._cleanup_old_sessions()
+
+    def _get_chat_history_path(self, session_id: str) -> Path:
+        """获取指定session的chat_history文件路径
+
+        Args:
+            session_id: 会话ID
+
+        Returns:
+            Path: chat_history文件路径
+        """
+        workspace_dir = Path(self.project_path) / '.workspace' / 'chat_history'
+        workspace_dir.mkdir(parents=True, exist_ok=True)
+        return workspace_dir / f"{session_id}.json"
+
+    def _load_session_data(self, session_id: str) -> list:
+        """加载指定session的chat_history数据（兼容旧格式）
+
+        Args:
+            session_id: 会话ID
+
+        Returns:
+            list: chat_history数据数组（内部格式，用于兼容现有代码）
+        """
+        chat_file = self._get_chat_history_path(session_id)
+        if not chat_file.exists():
+            return []
+
+        try:
+            with open(chat_file, 'r', encoding='utf-8') as f:
+                data = json.load(f)
+
+                # 新格式：{'dialogues': [...], 'control': {...}}
+                if isinstance(data, dict) and 'dialogues' in data:
+                    # 转换为内部数组格式
+                    result = []
+
+                    # 添加对话记录（自动添加type字段以兼容现有代码）
+                    for dialogue in data.get('dialogues', []):
+                        dialogue_with_type = dialogue.copy()
+                        dialogue_with_type['type'] = 'dialogue'
+                        result.append(dialogue_with_type)
+
+                    # 添加control记录
+                    if 'control' in data and data['control']:
+                        result.append(data['control'])
+
+                    return result
+
+                # 旧格式：直接返回数组
+                return data if isinstance(data, list) else []
+        except (json.JSONDecodeError, IOError):
+            return []
+
+    def _save_session_data(self, session_id: str, data: list):
+        """保存指定session的chat_history数据（新格式）
+
+        Args:
+            session_id: 会话ID
+            data: chat_history数据数组
+        """
+        chat_file = self._get_chat_history_path(session_id)
+        try:
+            # 转换为新格式：dialogues数组
+            new_format_data = {
+                'dialogues': []
+            }
+
+            # 过滤出对话项
+            for item in data:
+                if isinstance(item, dict):
+                    # 保留 agent 记录（直接添加）
+                    if item.get('role') == 'agent':
+                        new_format_data['dialogues'].append(item)
+                        continue
+
+                    item_type = item.get('type')
+                    if item_type == 'dialogue':
+                        # 创建新格式的对话项（移除type字段）
+                        dialogue = {
+                            'timestamp': item.get('timestamp', ''),
+                            'time_display': item.get('time_display', ''),
+                            'messages': item.get('messages', [])
+                        }
+                        new_format_data['dialogues'].append(dialogue)
+                    elif item_type == 'stop_hook_status':
+                        # 保留control字段（如果存在）
+                        if 'control' not in new_format_data:
+                            new_format_data['control'] = item
+
+            with open(chat_file, 'w', encoding='utf-8') as f:
+                json.dump(new_format_data, f, indent=2, ensure_ascii=False)
+        except IOError as e:
+            print(f"Error saving session data: {e}", file=sys.stderr)
+
+    def _get_stop_hook_status(self, session_id: str) -> Optional[Dict[str, Any]]:
+        """获取session的stop_hook状态
+
+        Args:
+            session_id: 会话ID
+
+        Returns:
+            Optional[Dict]: stop_hook状态数据，如果不存在则返回None
+        """
+        data = self._load_session_data(session_id)
+        # 在数组末尾查找stop_hook_status元素
+        for item in reversed(data):
+            if isinstance(item, dict) and item.get('type') == 'stop_hook_status':
+                return item
+        return None
+
+    def _set_stop_hook_status(self, session_id: str, status: str, action: Optional[str] = None):
+        """设置session的stop_hook状态
+
+        Args:
+            session_id: 会话ID
+            status: 状态值
+            action: 动作描述
+        """
+        data = self._load_session_data(session_id)
+
+        # 移除旧的stop_hook_status元素
+        data = [item for item in data if not (isinstance(item, dict) and item.get('type') == 'stop_hook_status')]
+
+        # 添加新的stop_hook_status元素
+        data.append({
+            'type': 'stop_hook_status',
+            'status': status,
+            'timestamp': datetime.now().isoformat(),
+            'last_action': action or status
+        })
+
+        self._save_session_data(session_id, data)
+
+    def _clear_stop_hook_status(self, session_id: str):
+        """清除session的stop_hook状态
+
+        Args:
+            session_id: 会话ID
+        """
+        data = self._load_session_data(session_id)
+        # 移除stop_hook_status元素
+        data = [item for item in data if not (isinstance(item, dict) and item.get('type') == 'stop_hook_status')]
+        self._save_session_data(session_id, data)
+
+    def get_session_status(self, session_id: str) -> Optional[str]:
+        """获取会话状态
+
+        Args:
+            session_id: 会话ID
+
+        Returns:
+            会话状态: "user_closed_by_button", "timeout_closed", "active" 或 None
+        """
+        status_data = self._get_stop_hook_status(session_id)
+        if status_data:
+            return status_data.get("status")
+        return None
+
+    def set_session_status(self, session_id: str, status: str, action: Optional[str] = None):
+        """设置会话状态
+
+        Args:
+            session_id: 会话ID
+            status: 状态 ("user_closed_by_button", "timeout_closed", "active")
+            action: 最后的动作描述
+        """
+        self._set_stop_hook_status(session_id, status, action)
+
+    def is_feedback_closed(self, session_id: str) -> bool:
+        """检查会话是否因用户关闭feedback而结束
+
+        Args:
+            session_id: 会话ID
+
+        Returns:
+            True如果用户主动关闭了feedback窗口
+        """
+        status = self.get_session_status(session_id)
+        return status == "user_closed"
+
+    def mark_feedback_closed(self, session_id: str):
+        """标记feedback被用户关闭（通用方法，保留向后兼容）
+
+        Args:
+            session_id: 会话ID
+        """
+        self.set_session_status(session_id, "user_closed", "feedback_window_closed_by_user")
+
+    def mark_user_closed_by_button(self, session_id: str):
+        """标记用户点击关闭按钮
+
+        Args:
+            session_id: 会话ID
+        """
+        self.set_session_status(session_id, "user_closed_by_button", "user_clicked_close_button")
+
+    def mark_timeout_closed(self, session_id: str):
+        """标记超时自动关闭
+
+        Args:
+            session_id: 会话ID
+        """
+        self.set_session_status(session_id, "timeout_closed", "feedback_timeout_auto_closed")
+
+    def is_user_closed_by_button(self, session_id: str) -> bool:
+        """检查是否用户点击关闭按钮
+
+        Args:
+            session_id: 会话ID
+
+        Returns:
+            True如果用户点击关闭按钮
+        """
+        status = self.get_session_status(session_id)
+        return status == "user_closed_by_button"
+
+    def is_timeout_closed(self, session_id: str) -> bool:
+        """检查是否超时关闭
+
+        Args:
+            session_id: 会话ID
+
+        Returns:
+            True如果超时自动关闭
+        """
+        status = self.get_session_status(session_id)
+        return status == "timeout_closed"
+
+    def reset_on_feedback_show(self, session_id: str):
+        """feedback展示时重置状态
+
+        Args:
+            session_id: 会话ID
+        """
+        # 清除所有状态
+        self.clear_session(session_id)
+
+    def get_block_count(self, session_id: str) -> int:
+        """获取会话的阻止次数
+
+        Args:
+            session_id: 会话ID
+
+        Returns:
+            阻止次数
+        """
+        status_data = self._get_stop_hook_status(session_id)
+        if status_data:
+            return status_data.get("block_count", 0)
+        return 0
+
+    def increment_block_count(self, session_id: str) -> int:
+        """增加会话的阻止次数
+
+        Args:
+            session_id: 会话ID
+
+        Returns:
+            更新后的阻止次数
+        """
+        status_data = self._get_stop_hook_status(session_id)
+        if status_data:
+            new_count = status_data.get("block_count", 0) + 1
+            status_data["block_count"] = new_count
+            status_data["timestamp"] = datetime.now().isoformat()
+
+            # 更新到chat_history
+            data = self._load_session_data(session_id)
+            # 移除旧的stop_hook_status
+            data = [item for item in data if not (isinstance(item, dict) and item.get('type') == 'stop_hook_status')]
+            # 添加更新后的status
+            data.append(status_data)
+            self._save_session_data(session_id, data)
+            return new_count
+        else:
+            # 首次创建
+            self._set_stop_hook_status(session_id, "active", "block_count_increment")
+            data = self._load_session_data(session_id)
+            # 找到刚创建的status并设置block_count
+            for item in reversed(data):
+                if isinstance(item, dict) and item.get('type') == 'stop_hook_status':
+                    item["block_count"] = 1
+                    break
+            self._save_session_data(session_id, data)
+            return 1
+
+    def clear_session(self, session_id: str):
+        """清除会话状态
+
+        Args:
+            session_id: 会话ID
+        """
+        self._clear_stop_hook_status(session_id)
+
+
+def main():
+    """命令行接口"""
+    import argparse
+    
+    parser = argparse.ArgumentParser(description='管理会话状态')
+    parser.add_argument('action', choices=['get', 'set', 'check', 'mark_closed', 'clear'],
+                        help='要执行的操作')
+    parser.add_argument('session_id', help='会话ID')
+    parser.add_argument('--status', help='设置的状态 (用于set操作)')
+    parser.add_argument('--state-file', help='状态文件路径')
+    
+    args = parser.parse_args()
+    
+    manager = SessionManager(args.state_file)
+    
+    if args.action == 'get':
+        status = manager.get_session_status(args.session_id)
+        print(status if status else "none")
+        
+    elif args.action == 'set':
+        if not args.status:
+            print("Error: --status required for set action", file=sys.stderr)
+            sys.exit(1)
+        manager.set_session_status(args.session_id, args.status)
+        print(f"Session {args.session_id} status set to {args.status}")
+        
+    elif args.action == 'check':
+        is_closed = manager.is_feedback_closed(args.session_id)
+        print("closed" if is_closed else "active")
+        sys.exit(0 if not is_closed else 1)
+        
+    elif args.action == 'mark_closed':
+        manager.mark_feedback_closed(args.session_id)
+        print(f"Session {args.session_id} marked as closed by user")
+        
+    elif args.action == 'clear':
+        manager.clear_session(args.session_id)
+        print(f"Session {args.session_id} cleared")
+
+
+if __name__ == "__main__":
+    main()

stop_hook.py

@@ -0,0 +1,87 @@
+#!/usr/bin/env python3
+"""
+Stop Hook处理脚本
+智能处理stop事件，避免死循环
+"""
+import sys
+import json
+import os
+from pathlib import Path
+
+# 添加当前目录到Python路径
+current_dir = Path(__file__).parent
+sys.path.insert(0, str(current_dir))
+
+from session_manager import SessionManager
+from context_formatter import format_for_stop_hook
+
+
+def main():
+    """主函数"""
+    try:
+        # 从stdin读取JSON输入
+        input_data = json.load(sys.stdin)
+
+        # 提取关键信息
+        session_id = input_data.get('session_id', '')
+        stop_hook_active = input_data.get('stop_hook_active', False)
+
+        # 获取项目路径
+        project_path = os.getcwd()
+
+        # 创建会话管理器(传入project_path)
+        manager = SessionManager(session_id=session_id, project_path=project_path)
+
+        # 决策逻辑
+        if session_id:
+            # 1. 检查用户是否点击关闭按钮
+            if manager.is_user_closed_by_button(session_id):
+                # 用户主动点击关闭，完全不提示（静默允许停止）
+                # 🔧 立即清除状态,避免死循环
+                manager.clear_session(session_id)
+                return 0
+
+            # 2. 检查是否超时关闭
+            if manager.is_timeout_closed(session_id):
+                # 超时关闭场景，最多提示2次
+                current_block_count = manager.get_block_count(session_id)
+                MAX_BLOCK_COUNT = 2
+
+                if current_block_count >= MAX_BLOCK_COUNT:
+                    # 超过最大阻止次数，允许停止以避免死循环
+                    manager.clear_session(session_id)
+                    return 0
+
+                # 未超过2次，继续提示并增加计数
+                manager.increment_block_count(session_id)
+            else:
+                # 3. 正常场景（非关闭状态），重置计数
+                # 这样每次正常的stop hook都会重新开始计数
+                if manager.get_block_count(session_id) > 0:
+                    manager.clear_session(session_id)
+
+        # 4. 默认行为：阻止停止并提示使用feedback工具
+        # 使用新的格式化上下文信息
+        if session_id:
+            reason_text = format_for_stop_hook(session_id, project_path)
+        else:
+            reason_text = "请你调用 feedback mcp tool 向用户反馈/请示。示例：使用 mcp__feedback__feedback 工具向用户汇报当前工作进度、完成状态或请求下一步指示。"
+
+        result = {
+            "decision": "block",
+            "reason": reason_text
+        }
+        print(json.dumps(result, ensure_ascii=False))
+        return 0
+
+    except Exception as e:
+        # 发生错误时，默认允许停止（避免卡死）
+        error_result = {
+            "decision": "approve",
+            "reason": f"Hook处理出错: {str(e)}"
+        }
+        print(json.dumps(error_result, ensure_ascii=False), file=sys.stderr)
+        return 1
+
+if __name__ == "__main__":
+    sys.exit(main())

tabs/__init__.py

@@ -0,0 +1,87 @@
+"""
+标签页模块 - 导出所有标签页组件
+"""
+
+# 基础标签页
+try:
+    from .base_tab import BaseTab
+except ImportError:
+    BaseTab = None
+
+# 简单标签页
+try:
+    from .workflow_tab import WorkflowTab
+except ImportError:
+    WorkflowTab = None
+
+try:
+    from .taskflow_tab import TaskflowTab
+except ImportError:
+    TaskflowTab = None
+
+try:
+    from .new_work_tab import NewWorkTab
+except ImportError:
+    NewWorkTab = None
+
+# 复杂标签页 - 分别导入，避免一个失败影响其他
+try:
+    from .chat_tab import ChatTab
+except ImportError:
+    ChatTab = None
+
+try:
+    from .stats_tab import StatsTab
+except ImportError:
+    StatsTab = None
+
+try:
+    from .memory_tab import MemoryTab
+except ImportError:
+    MemoryTab = None
+
+try:
+    from .rules_tab import RulesTab
+except ImportError:
+    RulesTab = None
+
+try:
+    from .todos_tab import TodosTab
+except ImportError:
+    TodosTab = None
+
+try:
+    from .checkpoints_tab import CheckpointsTab
+except ImportError:
+    CheckpointsTab = None
+
+try:
+    from .new_project_tab import NewProjectTab
+except ImportError:
+    NewProjectTab = None
+
+try:
+    from .workspace_tab import WorkspaceTab
+except ImportError:
+    WorkspaceTab = None
+
+try:
+    from .chat_history_tab import ChatHistoryTab
+except ImportError:
+    ChatHistoryTab = None
+
+__all__ = [
+    'BaseTab',
+    'WorkflowTab',
+    'TaskflowTab',
+    'NewWorkTab',
+    'ChatTab',
+    'ChatHistoryTab',
+    'StatsTab',
+    'MemoryTab',
+    'RulesTab',
+    'TodosTab',
+    'CheckpointsTab',
+    'NewProjectTab',
+    'WorkspaceTab'
+] 

tabs/base_tab.py

@@ -0,0 +1,34 @@
+"""
+选项卡基础类
+
+定义所有选项卡的通用接口和行为。
+"""
+
+from abc import ABC, abstractmethod, ABCMeta
+from PySide6.QtWidgets import QWidget
+from PySide6.QtCore import QObject
+
+
+class BaseTabMeta(type(QWidget), ABCMeta):
+    """解决QWidget和ABC的元类冲突"""
+    pass
+
+
+class BaseTab(QWidget, ABC, metaclass=BaseTabMeta):
+    """选项卡基础类
+    
+    所有选项卡都应该继承此类，并实现_setup_ui方法。
+    """
+    
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self._setup_ui()
+    
+    @abstractmethod
+    def _setup_ui(self):
+        """子类必须实现的UI创建方法"""
+        pass
+    
+    def refresh_data(self):
+        """刷新数据的通用方法，子类可以重写"""
+        pass 

tabs/chat_history_style.qss

@@ -0,0 +1,66 @@
+/* Chat History Tab Modern Styles */
+
+/* Global Scroll Area */
+QScrollArea {
+    border: none;
+    background-color: #1e1e1e;
+}
+
+QWidget#messagesContainer {
+    background-color: #1e1e1e;
+}
+
+/* -------------------------------------------------------------------------
+   Bubbles
+   ------------------------------------------------------------------------- */
+QFrame {
+    border: none;
+}
+
+QFrame#userBubble {
+    background-color: #0e639c; /* Blue */
+    border-radius: 12px;
+    border-top-right-radius: 2px;
+}
+
+QFrame#aiBubble {
+    background-color: #252526; /* Dark Gray */
+    border: 1px solid #454545;
+    border-radius: 12px;
+    border-top-left-radius: 2px;
+}
+
+QFrame#agentBubble {
+    background-color: #2d2b38; /* Slight Purple/Dark */
+    border: 1px solid #453555;
+    border-radius: 12px;
+    border-top-left-radius: 2px;
+}
+
+/* -------------------------------------------------------------------------
+   Avatars
+   ------------------------------------------------------------------------- */
+QLabel#avatarLabel {
+    font-size: 18px;
+    background-color: transparent;
+    padding: 4px;
+}
+
+/* -------------------------------------------------------------------------
+   Role Labels
+   ------------------------------------------------------------------------- */
+QLabel#roleLabel {
+    font-size: 11px;
+    font-weight: bold;
+    color: #cccccc;
+    margin-bottom: 2px;
+}
+
+/* -------------------------------------------------------------------------
+   Empty State
+   ------------------------------------------------------------------------- */
+QLabel#emptyStateLabel {
+    color: #6e6e6e;
+    font-size: 14px;
+    padding: 40px;
+}