aceflow-pateoas-framework-main/.aceflow/scripts/cli/init.py

import os
import json
from datetime import datetime
from ..core.state_engine import PATEOASStateEngine
from ..core.memory_pool import GlobalMemoryPool
from ..utils.logger import get_logger

# 初始化日志
logger = get_logger("aceflow.init")

def initialize_project(project_root='.', reset_state=False):
    """
    初始化AceFlow-PATEOAS项目结构
    
    参数:
        project_root (str): 项目根目录路径，默认为当前目录
        reset_state (bool): 是否重置现有状态，默认为False
    """
    try:
        # 1. 创建核心目录结构
        directories = [
            ".aceflow/config",
            ".aceflow/scripts/core",
            ".aceflow/scripts/cli",
            ".aceflow/scripts/utils",
            ".aceflow/scripts/migrations",
            ".aceflow/memory_pool/REQ",
            ".aceflow/memory_pool/CON",
            ".aceflow/memory_pool/TASK",
            ".aceflow/memory_pool/CODE",
            ".aceflow/memory_pool/TEST",
            ".aceflow/memory_pool/DEFECT",
            ".aceflow/memory_pool/FDBK",
            ".aceflow/templates/stage_templates",
            ".aceflow/templates/auxiliary_templates",
            ".aceflow/logs",
            "aceflow_result",
            ".vscode"
        ]
        
        for dir_path in directories:
            full_path = os.path.join(project_root, dir_path)
            os.makedirs(full_path, exist_ok=True)
            logger.info(f"创建目录: {full_path}")

        # 2. 初始化状态引擎
        state_engine = PATEOASStateEngine(project_root)
        if reset_state:
            logger.info("重置现有状态...")
            state_engine.initialize_state()

        # 3. 初始化记忆池
        memory_pool = GlobalMemoryPool()
        logger.info("全局记忆池初始化完成")

        # 4. 创建基础配置文件（如不存在）
        create_default_configs(project_root)

        # 5. 创建.gitignore文件
        create_gitignore(project_root)

        logger.info("项目初始化完成！")
        print("✅ AceFlow-PATEOAS项目初始化成功")
        print("📁 项目结构已创建")
        print("🔧 状态引擎和记忆池已初始化")
        print("📝 基础配置文件已生成")
        
        return True
        
    except Exception as e:
        logger.error(f"初始化失败: {str(e)}", exc_info=True)
        print(f"❌ 初始化失败: {str(e)}")
        return False

def create_default_configs(project_root):
    """创建默认配置文件"""
    # workflow_rules.json
    workflow_rules_path = os.path.join(project_root, ".aceflow", "config", "workflow_rules.json")
    if not os.path.exists(workflow_rules_path):
        default_workflow_rules = {
            "workflow_rules": {
                "full_workflow": ["S1", "S2", "S3", "S4", "S5", "S6", "S7", "S8"],
                "quick_workflow": ["S2", "S4", "S5", "S8"],
                "change_workflow": ["S1", "S2", "S3", "S4"],
                "emergency_workflow": ["S4", "S5", "S6", "S8"]
            },
            "memory_pool_config": {
                "storage_path": "./.aceflow/memory_pool",
                "retention_policy": "critical_forever,temporary_7d"
            },
            "ai_decision_config": {
                "trust_level": "L2",
                "success_threshold": 0.85
            }
        }
        
        with open(workflow_rules_path, 'w', encoding='utf-8') as f:
            json.dump(default_workflow_rules, f, ensure_ascii=False, indent=2)
        logger.info(f"创建默认配置: {workflow_rules_path}")

    # dynamic_thresholds.json
    thresholds_path = os.path.join(project_root, ".aceflow", "config", "dynamic_thresholds.json")
    if not os.path.exists(thresholds_path):
        default_thresholds = {
            "global": {
                "time_adjustment_range": 20,
                "memory_retention_days": 30
            },
            "stage_specific": {
                "S3": {
                    "test_case_coverage": {
                        "default": 80,
                        "payment_module": 99,
                        "ui_module": 75
                    }
                },
                "S4": {
                    "unit_test_pass_rate": {
                        "default": 90,
                        "critical_task": 95,
                        "minor_task": 85
                    }
                }
            }
        }
        
        with open(thresholds_path, 'w', encoding='utf-8') as f:
            json.dump(default_thresholds, f, ensure_ascii=False, indent=2)
        logger.info(f"创建默认配置: {thresholds_path}")

    # aceflow_agent.json (VS Code配置)
    agent_config_path = os.path.join(project_root, ".vscode", "aceflow_agent.json")
    if not os.path.exists(agent_config_path):
        default_agent_config = {
            "agent_type": "pateoas_aceflow_agent",
            "capabilities": [
                "state_awareness",
                "memory_management",
                "autonomous_navigation",
                "abnormality_handling"
            ],
            "state_templates": ".aceflow/templates/stage_templates/",
            "memory_pool_config": {
                "storage_path": ".aceflow/memory_pool/",
                "retention_policy": "critical_forever,temporary_7d"
            },
            "output_config": {
                "root_dir": "aceflow_result",
                "stage_dir_format": "S{stage_number}_{stage_name}",
                "compatibility_mode": true
            }
        }
        
        with open(agent_config_path, 'w', encoding='utf-8') as f:
            json.dump(default_agent_config, f, ensure_ascii=False, indent=2)
        logger.info(f"创建默认配置: {agent_config_path}")

    # process_index.json (流程索引)
    index_path = os.path.join(project_root, ".aceflow", "process_index.json")
    if not os.path.exists(index_path):
        default_index = {
            "process_spec": {
                "path": ".aceflow/templates/document_templates/process_spec.md",
                "sections": [
                    {"name": "核心原则", "anchor": "#_2-核心原则"},
                    {"name": "阶段定义", "anchor": "#_4-阶段定义与执行规范"},
                    {"name": "状态管理", "anchor": "#_5-状态管理规范"}
                ]
            },
            "templates": {
                "stage_templates": ".aceflow/templates/stage_templates/",
                "auxiliary_templates": ".aceflow/templates/auxiliary_templates/",
                "available_templates": [
                    "s1_user_story.md", "s2_tasks.md", "s3_testcases.md",
                    "s4_implementation.md", "s5_test_report.md", "s6_codereview.md",
                    "s7_demo_feedback.md", "s8_progress_index.md", "task-status-table.md"
                ]
            },
            "config_files": {
                "dynamic_thresholds": ".aceflow/config/dynamic_thresholds.json",
                "workflow_rules": ".aceflow/config/workflow_rules.json",
                "agent_config": ".vscode/aceflow_agent.json"
            }
        }
        
        with open(index_path, 'w', encoding='utf-8') as f:
            json.dump(default_index, f, ensure_ascii=False, indent=2)
        logger.info(f"创建流程索引: {index_path}")

def create_gitignore(project_root):
    """创建.gitignore文件，排除临时文件和敏感信息"""
    gitignore_path = os.path.join(project_root, ".gitignore")
    ignore_content = """# AceFlow-PATEOAS 临时文件
.aceflow/logs/
.aceflow/memory_pool/
.aceflow/current_state.json
.aceflow/*.log

# 产物目录
aceflow_result/

# Python 环境文件
__pycache__/
*.py[cod]
*$py.class
*.so
.Python
env/
build/
develop-eggs/
dist/
downloads/
eggs/
.eggs/
lib/
lib64/
parts/
sdist/
var/
*.egg-info/
.installed.cfg
*.egg

# VS Code 配置
.vscode/settings.json
.vscode/launch.json
!.vscode/extensions.json
!.vscode/aceflow_agent.json
"""
    
    # 如果.gitignore不存在或不包含AceFlow规则，则添加
    if not os.path.exists(gitignore_path):
        with open(gitignore_path, 'w', encoding='utf-8') as f:
            f.write(ignore_content)
        logger.info("创建.gitignore文件")
    else:
        with open(gitignore_path, 'r', encoding='utf-8') as f:
            content = f.read()
        if "# AceFlow-PATEOAS 临时文件" not in content:
            with open(gitignore_path, 'a', encoding='utf-8') as f:
                f.write("\n" + ignore_content)
            logger.info("更新.gitignore文件，添加AceFlow规则")

# 命令行入口函数
def init_project(args):
    """CLI命令：初始化项目"""
    initialize_project(reset_state=getattr(args, 'reset', False))

# 命令配置
def add_init_command(subparsers):
    """添加init命令到解析器"""
    init_parser = subparsers.add_parser('init', help='初始化AceFlow-PATEOAS项目结构')
    init_parser.add_argument('--reset', action='store_true', help='重置现有状态文件')
    init_parser.set_defaults(func=init_project)