AI-Copyright-Application-Generator/scripts/init/init_project_safe.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
AI驱动的企业级软件开发工作流程 - 项目初始化脚本 (安全版本)
版本: 1.1
描述: 自动创建新项目的目录结构和固定文档，修复编码问题
"""

import os
import sys
import json
import shutil
import argparse
import locale
from datetime import datetime
from pathlib import Path

# 设置系统编码环境
def setup_encoding():
    """设置系统编码环境"""
    try:
        # 设置locale为UTF-8
        locale.setlocale(locale.LC_ALL, 'en_US.UTF-8')
    except locale.Error:
        try:
            locale.setlocale(locale.LC_ALL, 'C.UTF-8')
        except locale.Error:
            pass  # 忽略locale设置错误
    
    # 确保环境变量支持UTF-8
    os.environ['LANG'] = 'en_US.UTF-8'
    os.environ['LC_ALL'] = 'en_US.UTF-8'
    os.environ['PYTHONIOENCODING'] = 'utf-8'
    
    # Windows特殊处理
    if sys.platform.startswith('win'):
        import codecs
        if hasattr(sys.stdout, 'buffer'):
            sys.stdout = codecs.getwriter('utf-8')(sys.stdout.buffer)
        if hasattr(sys.stderr, 'buffer'):
            sys.stderr = codecs.getwriter('utf-8')(sys.stderr.buffer)

# 初始化编码设置
setup_encoding()

class Colors:
    """终端颜色定义"""
    RED = '\033[91m'
    GREEN = '\033[92m'
    YELLOW = '\033[93m'
    BLUE = '\033[94m'
    MAGENTA = '\033[95m'
    CYAN = '\033[96m'
    WHITE = '\033[97m'
    BOLD = '\033[1m'
    UNDERLINE = '\033[4m'
    RESET = '\033[0m'

def print_message(color, message):
    """打印带颜色的消息"""
    print(f"{color}{message}{Colors.RESET}")

def print_success(message):
    print_message(Colors.GREEN, f"✓ {message}")

def print_info(message):
    print_message(Colors.BLUE, f"ℹ {message}")

def print_warning(message):
    print_message(Colors.YELLOW, f"⚠ {message}")

def print_error(message):
    print_message(Colors.RED, f"✗ {message}")

def safe_input(prompt, default=""):
    """安全的用户输入函数，处理编码问题"""
    try:
        if default:
            full_prompt = f"{prompt} (默认: {default}): "
        else:
            full_prompt = f"{prompt}: "
        
        # 使用bytes处理输入以避免编码问题
        print(full_prompt, end='', flush=True)
        
        # 直接从stdin读取bytes并解码
        if hasattr(sys.stdin, 'buffer'):
            line = sys.stdin.buffer.readline()
            user_input = line.decode('utf-8', errors='replace').strip()
        else:
            user_input = input().strip()
        
        return user_input if user_input else default
        
    except (UnicodeDecodeError, UnicodeError) as e:
        print_error(f"输入编码错误: {e}")
        print_warning("请确保终端支持UTF-8编码")
        return default
    except KeyboardInterrupt:
        print_error("\n用户中断操作")
        sys.exit(1)
    except EOFError:
        print_error("\n输入结束")
        return default
    except Exception as e:
        print_error(f"输入错误: {e}")
        return default

def safe_yes_no_input(prompt, default_no=True):
    """安全的是/否输入函数"""
    try:
        suffix = "(y/N)" if default_no else "(Y/n)"
        response = safe_input(f"{prompt} {suffix}", "").lower()
        
        if default_no:
            return response in ['y', 'yes', '是', '1', 'true']
        else:
            return response not in ['n', 'no', '否', '0', 'false']
    except Exception as e:
        print_error(f"输入处理错误: {e}")
        return not default_no

def get_ui_design_style():
    """获取UI设计风格选择"""
    print_info("请选择UI设计风格 (12种专业设计风格):")
    print_info("专业商务类型:")
    print("  1. 企业商务风格 (Corporate) - 默认推荐")
    print("  2. 包豪斯风格 (Bauhaus) - 功能主义设计")  
    print("  3. 艺术装饰风格 (ArtDeco) - 奢华几何美学")
    print_info("现代科技类型:")
    print("  4. 暗黑科技风格 (Cyberpunk) - 科技感强烈")
    print("  5. 未来科技风格 (Futuristic) - 数字未来美学")
    print("  6. 大胆现代风格 (Bold) - 视觉冲击力强")
    print_info("极简清新类型:")
    print("  7. 极简主义风格 (Minimal) - 简洁专注")
    print("  8. 日式极简风格 (Japanese) - 禅意侘寂")
    print("  9. 斯堪的纳维亚风格 (Scandinavian) - 北欧简约")
    print_info("创意艺术类型:")
    print("  10. 孟菲斯风格 (Memphis) - 后现代叛逆")
    print("  11. 波普艺术风格 (PopArt) - 大众文化美学")
    print("  12. 优雅复古风格 (Elegant) - 经典印刷美学")
    
    styles = [
        ("1", "corporate", "企业商务风格"),
        ("2", "bauhaus", "包豪斯风格"),
        ("3", "artdeco", "艺术装饰风格"),
        ("4", "cyberpunk", "暗黑科技风格"),
        ("5", "futuristic", "未来科技风格"),
        ("6", "bold", "大胆现代风格"),
        ("7", "minimal", "极简主义风格"),
        ("8", "japanese", "日式极简风格"),
        ("9", "scandinavian", "斯堪的纳维亚风格"),
        ("10", "memphis", "孟菲斯风格"),
        ("11", "popart", "波普艺术风格"),
        ("12", "elegant", "优雅复古风格")
    ]
    
    while True:
        choice = safe_input("请输入选择 (1-12)", "1")
        try:
            choice_num = int(choice)
            if 1 <= choice_num <= 12:
                return styles[choice_num - 1][1]  # 返回key
            else:
                print_warning("请输入1-12之间的数字")
        except ValueError:
            print_warning("请输入有效的数字")

def get_generation_mode():
    """获取生成模式"""
    print_info("请选择生成模式:")
    print("  1. fast - 快速验证模式 (5个核心页面)")
    print("  2. full - 完整生产模式 (10个完整页面)")
    
    while True:
        choice = safe_input("请输入选择 (1-2)", "1")
        if choice in ["1", "fast"]:
            return "fast"
        elif choice in ["2", "full"]:
            return "full"
        else:
            print_warning("请输入1或2")

def create_config_file(title, short_title, ui_style, mode, front_tech, backend_tech):
    """创建配置文件"""
    config = {
        "_comment_init": "=== 项目初始化配置（用户设置） ===",
        "front": front_tech,
        "backend": backend_tech,
        "title": title,
        "short_title": short_title,
        "requirements_description": "requires_docs/需求文档.md",
        "dev_tech_stack": "requires_docs/技术栈说明文档.md",
        "ui_design_style": ui_style,
        
        "_comment_fixed": "=== 固定配置（不变） ===",
        "system_prompt_dir": "system_prompts",
        "ui_design_spec_default": f"specs_docs/ui_design_specs/01-UI设计规范_默认_Corporate.md",
        "ui_design_spec": "requires_docs/UI设计规范.md",
        
        "_comment_generation": "=== 生成配置（可调整） ===",
        "page_count_fast": 5,
        "page_count_full": 10,
        "api_count_min": 8,
        "api_count_max": 35,
        "generation_mode": mode,
        
        "_comment_generated": "=== 流程生成配置（自动生成） ===",
        "framework_design": "process_docs/框架设计文档.md",
        "page_list": "process_docs/页面清单.md",
        "database_schema": "output_sourcecode/db/database_schema.sql",
        "copyright_application": "output_docs/软件著作权登记信息表.md"
    }
    
    try:
        with open("ai-copyright-config.json", "w", encoding="utf-8") as f:
            json.dump(config, f, ensure_ascii=False, indent=2)
        print_success("配置文件创建完成: ai-copyright-config.json")
        return True
    except Exception as e:
        print_error(f"创建配置文件失败: {e}")
        return False

def create_directories():
    """创建项目目录结构"""
    dirs = [
        "requires_docs",
        "process_docs", 
        "output_docs",
        "output_sourcecode/front",
        "output_sourcecode/backend",
        "output_sourcecode/db"
    ]
    
    for dir_path in dirs:
        try:
            os.makedirs(dir_path, exist_ok=True)
            print_success(f"创建目录: {dir_path}")
        except Exception as e:
            print_error(f"创建目录失败 {dir_path}: {e}")

def main():
    """主函数"""
    print_message(Colors.BOLD + Colors.CYAN, "🚀 AI驱动的软件著作权申请材料生成系统 - 项目初始化")
    print_message(Colors.CYAN, "=" * 70)
    
    try:
        # 获取项目信息
        print_info("请输入项目配置信息:")
        
        system_title = safe_input("系统完整名称", "我的软件系统")
        if not system_title:
            system_title = "我的软件系统"
            
        short_title = safe_input("系统简称", system_title[:10] if len(system_title) > 10 else system_title)
        if not short_title:
            short_title = system_title[:10] if len(system_title) > 10 else system_title
        
        # 获取技术选择
        print_info("请选择技术栈:")
        front_tech = safe_input("前端技术 (vue/react/angular)", "vue")
        if not front_tech:
            front_tech = "vue"
            
        backend_tech = safe_input("后端技术 (java/nodejs/python)", "java")
        if not backend_tech:
            backend_tech = "java"
        
        # 获取UI风格
        ui_style = get_ui_design_style()
        
        # 获取生成模式
        mode = get_generation_mode()
        
        # 确认信息
        print_info("项目配置信息确认:")
        print(f"  系统名称: {system_title}")
        print(f"  系统简称: {short_title}")
        print(f"  前端技术: {front_tech}")
        print(f"  后端技术: {backend_tech}")
        print(f"  UI风格: {ui_style}")
        print(f"  生成模式: {mode}")
        
        if not safe_yes_no_input("确认以上信息是否正确?", default_no=False):
            print_warning("用户取消操作")
            return
        
        # 创建目录结构
        print_info("创建项目目录结构...")
        create_directories()
        
        # 创建配置文件
        print_info("创建项目配置文件...")
        if create_config_file(system_title, short_title, ui_style, mode, front_tech, backend_tech):
            print_success("项目初始化完成!")
            print_info("下一步请:")
            print("  1. 编辑 requires_docs/需求文档.md 文件")
            print("  2. 运行质量检查: python3 scripts/validators/check_project.py --quick")
        else:
            print_error("项目初始化失败")
            
    except Exception as e:
        print_error(f"初始化过程中发生错误: {e}")
        import traceback
        traceback.print_exc()

if __name__ == "__main__":
    main()