auto-coder 0.1.363__py3-none-any.whl → 0.1.365__py3-none-any.whl

autocoder/common/file_checkpoint/conversation_checkpoint.py

@@ -0,0 +1,182 @@
+import os
+import json
+import time
+import uuid
+from typing import List, Dict, Any, Optional
+from pydantic import BaseModel
+from loguru import logger
+
+
+class ConversationCheckpoint(BaseModel):
+    """对话检查点，用于保存特定时刻的对话状态"""
+    
+    checkpoint_id: str  # 检查点ID，与变更组ID对应
+    timestamp: float  # 创建时间戳
+    conversations: List[Dict[str, Any]]  # 对话历史
+    metadata: Optional[Dict[str, Any]] = None  # 元数据，可包含额外信息
+
+
+class ConversationCheckpointStore:
+    """对话检查点存储管理器"""
+    
+    def __init__(self, store_dir: Optional[str] = None, max_history: int = 50):
+        """
+        初始化对话检查点存储
+        
+        Args:
+            store_dir: 存储目录，默认为用户主目录下的.autocoder/conversation_checkpoints
+            max_history: 最大保存的历史版本数量
+        """
+        if store_dir is None:
+            home_dir = os.path.expanduser("~")
+            store_dir = os.path.join(home_dir, ".autocoder", "conversation_checkpoints")
+        
+        self.store_dir = os.path.abspath(store_dir)
+        self.max_history = max_history
+        
+        # 确保存储目录存在
+        os.makedirs(self.store_dir, exist_ok=True)
+        logger.info(f"对话检查点存储目录: {self.store_dir}")
+    
+    def save_checkpoint(self, checkpoint: ConversationCheckpoint) -> str:
+        """
+        保存对话检查点
+        
+        Args:
+            checkpoint: 对话检查点对象
+            
+        Returns:
+            str: 检查点ID
+        """
+        # 确保检查点有ID
+        if not checkpoint.checkpoint_id:
+            checkpoint.checkpoint_id = str(uuid.uuid4())
+        
+        # 确保时间戳存在
+        if not checkpoint.timestamp:
+            checkpoint.timestamp = time.time()
+        
+        # 构建文件路径
+        file_path = os.path.join(self.store_dir, f"{checkpoint.checkpoint_id}.json")
+        
+        # 保存为JSON文件
+        try:
+            with open(file_path, 'w', encoding='utf-8') as f:
+                f.write(checkpoint.json(ensure_ascii=False, indent=2))
+            
+            logger.info(f"已保存对话检查点: {checkpoint.checkpoint_id}")
+            
+            # 检查并清理过期的检查点
+            self._cleanup_old_checkpoints()
+            
+            return checkpoint.checkpoint_id
+        except Exception as e:
+            logger.error(f"保存对话检查点失败: {str(e)}")
+            return ""
+    
+    def get_checkpoint(self, checkpoint_id: str) -> Optional[ConversationCheckpoint]:
+        """
+        获取指定ID的对话检查点
+        
+        Args:
+            checkpoint_id: 检查点ID
+            
+        Returns:
+            Optional[ConversationCheckpoint]: 对话检查点对象，如果不存在则返回None
+        """
+        file_path = os.path.join(self.store_dir, f"{checkpoint_id}.json")
+        
+        if not os.path.exists(file_path):
+            logger.warning(f"对话检查点不存在: {checkpoint_id}")
+            return None
+        
+        try:
+            with open(file_path, 'r', encoding='utf-8') as f:
+                data = json.load(f)
+            
+            return ConversationCheckpoint(**data)
+        except Exception as e:
+            logger.error(f"读取对话检查点失败: {str(e)}")
+            return None
+    
+    def get_latest_checkpoint(self) -> Optional[ConversationCheckpoint]:
+        """
+        获取最新的对话检查点
+        
+        Returns:
+            Optional[ConversationCheckpoint]: 最新的对话检查点对象，如果不存在则返回None
+        """
+        checkpoints = self._get_all_checkpoints()
+        
+        if not checkpoints:
+            return None
+        
+        # 按时间戳降序排序
+        checkpoints.sort(key=lambda x: x.timestamp, reverse=True)
+        
+        return checkpoints[0] if checkpoints else None
+    
+    def delete_checkpoint(self, checkpoint_id: str) -> bool:
+        """
+        删除指定的对话检查点
+        
+        Args:
+            checkpoint_id: 检查点ID
+            
+        Returns:
+            bool: 是否成功删除
+        """
+        file_path = os.path.join(self.store_dir, f"{checkpoint_id}.json")
+        
+        if not os.path.exists(file_path):
+            logger.warning(f"要删除的对话检查点不存在: {checkpoint_id}")
+            return False
+        
+        try:
+            os.remove(file_path)
+            logger.info(f"已删除对话检查点: {checkpoint_id}")
+            return True
+        except Exception as e:
+            logger.error(f"删除对话检查点失败: {str(e)}")
+            return False
+    
+    def _get_all_checkpoints(self) -> List[ConversationCheckpoint]:
+        """
+        获取所有对话检查点
+        
+        Returns:
+            List[ConversationCheckpoint]: 对话检查点列表
+        """
+        checkpoints = []
+        
+        try:
+            for filename in os.listdir(self.store_dir):
+                if filename.endswith('.json'):
+                    file_path = os.path.join(self.store_dir, filename)
+                    
+                    try:
+                        with open(file_path, 'r', encoding='utf-8') as f:
+                            data = json.load(f)
+                        
+                        checkpoint = ConversationCheckpoint(**data)
+                        checkpoints.append(checkpoint)
+                    except Exception as e:
+                        logger.error(f"读取对话检查点文件失败 {filename}: {str(e)}")
+        except Exception as e:
+            logger.error(f"获取所有对话检查点失败: {str(e)}")
+        
+        return checkpoints
+    
+    def _cleanup_old_checkpoints(self):
+        """清理过期的检查点，保持历史记录数量在限制范围内"""
+        checkpoints = self._get_all_checkpoints()
+        
+        if len(checkpoints) <= self.max_history:
+            return
+        
+        # 按时间戳降序排序
+        checkpoints.sort(key=lambda x: x.timestamp, reverse=True)
+        
+        # 删除超出限制的旧检查点
+        for checkpoint in checkpoints[self.max_history:]:
+            self.delete_checkpoint(checkpoint.checkpoint_id)

autocoder/common/file_checkpoint/examples.py

@@ -0,0 +1,217 @@
+"""
+文件变更管理模块的使用示例
+
+展示如何使用文件变更管理模块进行文件变更的应用和撤销。
+"""
+
+import os
+import sys
+import json
+from typing import Dict, Any
+
+from autocoder.common.file_checkpoint.models import FileChange
+from autocoder.common.file_checkpoint.manager import FileChangeManager
+
+
+def example_apply_changes(project_dir: str):
+    """
+    示例：应用文件变更
+    
+    Args:
+        project_dir: 项目目录
+    """
+    print(f"示例：应用文件变更到项目 {project_dir}")
+    
+    # 创建文件变更管理器
+    manager = FileChangeManager(project_dir)
+    
+    # 准备文件变更
+    changes = {
+        "example.txt": FileChange(
+            file_path="example.txt",
+            content="这是一个示例文件\n用于演示文件变更管理模块的功能\n",
+            is_new=True
+        ),
+        "README.md": FileChange(
+            file_path="README.md",
+            content="# 示例项目\n\n这是一个用于演示文件变更管理模块的示例项目。\n",
+            is_new=True
+        )
+    }
+    
+    # 应用变更
+    result = manager.apply_changes(changes)
+    
+    # 输出结果
+    if result.success:
+        print(f"成功应用了 {len(result.change_ids)} 个文件变更")
+        for change_id in result.change_ids:
+            print(f"  - 变更ID: {change_id}")
+    else:
+        print("应用变更失败")
+        for file_path, error in result.errors.items():
+            print(f"  - {file_path}: {error}")
+
+
+def example_preview_changes(project_dir: str):
+    """
+    示例：预览文件变更
+    
+    Args:
+        project_dir: 项目目录
+    """
+    print(f"示例：预览文件变更")
+    
+    # 创建文件变更管理器
+    manager = FileChangeManager(project_dir)
+    
+    # 准备文件变更
+    changes = {
+        "example.txt": FileChange(
+            file_path="example.txt",
+            content="这是一个修改后的示例文件\n用于演示文件变更管理模块的功能\n新增的一行\n",
+            is_new=False
+        )
+    }
+    
+    # 预览变更
+    diff_results = manager.preview_changes(changes)
+    
+    # 输出差异
+    for file_path, diff_result in diff_results.items():
+        print(f"\n文件: {file_path}")
+        print(diff_result.get_diff_summary())
+        if diff_result.old_content is not None and not diff_result.is_new and not diff_result.is_deletion:
+            diff_text = manager.get_diff_text(diff_result.old_content, diff_result.new_content)
+            print("\n差异:")
+            print(diff_text)
+
+
+def example_undo_changes(project_dir: str):
+    """
+    示例：撤销文件变更
+    
+    Args:
+        project_dir: 项目目录
+    """
+    print(f"示例：撤销文件变更")
+    
+    # 创建文件变更管理器
+    manager = FileChangeManager(project_dir)
+    
+    # 撤销最近的变更
+    result = manager.undo_last_change()
+    
+    # 输出结果
+    if result.success:
+        print(f"成功撤销了变更，恢复了 {len(result.restored_files)} 个文件")
+        for file_path in result.restored_files:
+            print(f"  - {file_path}")
+    else:
+        print("撤销变更失败")
+        for file_path, error in result.errors.items():
+            print(f"  - {file_path}: {error}")
+
+
+def example_get_history(project_dir: str):
+    """
+    示例：获取变更历史
+    
+    Args:
+        project_dir: 项目目录
+    """
+    print(f"示例：获取变更历史")
+    
+    # 创建文件变更管理器
+    manager = FileChangeManager(project_dir)
+    
+    # 获取变更历史
+    changes = manager.get_change_history(limit=5)
+    
+    # 输出历史记录
+    print(f"最近 {len(changes)} 条变更记录:")
+    for change in changes:
+        timestamp = change.timestamp
+        file_path = change.file_path
+        change_type = "新建" if change.is_new else "删除" if change.is_deletion else "修改"
+        print(f"  - [{timestamp}] {change_type} {file_path} (ID: {change.change_id})")
+
+
+def example_integration_with_agentic_edit():
+    """
+    示例：与 AgenticEdit 集成
+    
+    展示如何将文件变更管理模块集成到 AgenticEdit 中
+    """
+    print("示例：与 AgenticEdit 集成")
+    
+    # 这是一个伪代码示例，展示如何修改 AgenticEdit.apply_changes 方法
+    code = """
+def apply_changes(self):
+    \"\"\"
+    Apply all tracked file changes to the original project directory.
+    \"\"\"
+    from autocoder.common.file_checkpoint.models import FileChange
+    from autocoder.common.file_checkpoint.manager import FileChangeManager
+    
+    # 创建文件变更管理器
+    manager = FileChangeManager(self.args.source_dir)
+    
+    # 将影子系统的变更转换为 FileChange 对象
+    changes = {}
+    for file_path, change in self.get_all_file_changes().items():
+        changes[file_path] = FileChange(
+            file_path=file_path,
+            content=change.content,
+            is_new=not os.path.exists(file_path)
+        )
+    
+    # 应用变更
+    result = manager.apply_changes(changes)
+    
+    # 处理结果
+    if result.success:
+        # 继续执行原有的 Git 提交等逻辑
+        if not self.args.skip_commit:
+            try:
+                # ... 原有的 Git 提交代码 ...
+            except Exception as e:
+                # ... 原有的错误处理 ...
+    else:
+        # 处理应用变更失败的情况
+        error_messages = "\\n".join([f"{path}: {error}" for path, error in result.errors.items()])
+        self.printer.print_str_in_terminal(
+            f"Failed to apply changes:\\n{error_messages}",
+            style="red"
+        )
+    """
+    
+    print(code)
+
+
+def main():
+    """主函数"""
+    if len(sys.argv) < 2:
+        print("用法: python examples.py <项目目录>")
+        return
+    
+    project_dir = sys.argv[1]
+    
+    # 运行示例
+    example_apply_changes(project_dir)
+    print("\n" + "-" * 50 + "\n")
+    
+    example_preview_changes(project_dir)
+    print("\n" + "-" * 50 + "\n")
+    
+    example_get_history(project_dir)
+    print("\n" + "-" * 50 + "\n")
+    
+    example_undo_changes(project_dir)
+    print("\n" + "-" * 50 + "\n")
+    
+    example_integration_with_agentic_edit()
+
+
+if __name__ == "__main__":
+    main()