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

import os
import argparse

# 定义要查找和替换的注释内容
# 使用原始字符串数组，支持多个旧内容，避免Unicode转义问题
OLD_COMMENTS = [
    r"""
<?php
/**
 
 
 


 
 */
 """.strip(), 
    r"""
<?php

/**
 
 
 


 
 */
 """.strip(),    
    # 格式1：标准注释格式
    r"""
<?php
/**
 
 
 



 
 */
 """.strip(),
    r"""
<?php

/**
 * Niushop商城系统 - 团队十年电商经验汇集巨献!
 * =========================================================
 * Copy right 2019-2029 杭州牛之云科技有限公司, 保留所有权利。
 * ----------------------------------------------
 * 官方网址: https://www.niushop.com
 * =========================================================
 */
 """.strip(),    
    # 格式1：标准注释格式
    r"""
<?php
/**
 * Niushop商城系统 - 团队十年电商经验汇集巨献!
 * =========================================================
 * Copy right 2019-2029 杭州牛之云科技有限公司, 保留所有权利。
 * ----------------------------------------------
 * 官方网址: https://www.niushop.com
 * =========================================================
 */""".strip(), 
    # 格式2：带有额外空行的注释格式
    r"""
<?php
/**
 * Niushop商城系统 - 团队十年电商经验汇集巨献!
 * =========================================================
 * Copy right 2019-2029 杭州牛之云科技有限公司, 保留所有权利。
 * ----------------------------------------------
 * 官方网址: https://www.niushop.com

 * =========================================================
 */""".strip(),
]

NEW_COMMENT = r"""
<?php
""".strip()

# 定义要处理的文件类型
FILE_TYPES = [".php", ".js", ".css", ".html", ".vue", ".ts", ".tsx", ".jsx", ".scss", ".less"]

def replace_comments(file_path):
    """
    替换文件中的注释
    """
    try:
        # 读取文件内容
        with open(file_path, 'r', encoding='utf-8') as f:
            content = f.read()
        
        # 检查文件是否包含任何旧注释
        has_old_comment = False
        for old_comment in OLD_COMMENTS:
            if old_comment in content:
                has_old_comment = True
                break
        
        if has_old_comment:
            # 替换所有旧注释
            new_content = content
            for old_comment in OLD_COMMENTS:
                if old_comment in new_content:
                    new_content = new_content.replace(old_comment, NEW_COMMENT)
            
            # 写回文件
            with open(file_path, 'w', encoding='utf-8') as f:
                f.write(new_content)
            
            print(f"✓ 已处理: {file_path}")
            return True
        return False
        
    except UnicodeDecodeError:
        # 跳过二进制文件
        print(f"✗ 跳过二进制文件: {file_path}")
        return False
    except PermissionError:
        # 跳过没有权限的文件
        print(f"✗ 权限不足: {file_path}")
        return False
    except Exception as e:
        # 处理其他异常
        print(f"✗ 处理失败 {file_path}: {str(e)}")
        return False

def main():
    """
    主函数
    """
    parser = argparse.ArgumentParser(description="替换Niushop商城系统的注释")
    parser.add_argument("--path", type=str, default=".", help="要遍历的目录路径")
    args = parser.parse_args()
    
    root_path = args.path
    total_files = 0
    processed_files = 0
    
    print(f"开始遍历目录: {root_path}")
    print(f"将处理的文件类型: {', '.join(FILE_TYPES)}")
    print("=" * 60)
    
    # 遍历所有文件
    for root, dirs, files in os.walk(root_path):
        # 跳过某些目录（如.git、vendor、node_modules等）
        dirs[:] = [d for d in dirs if d not in ['.git', 'vendor', 'node_modules', 'runtime', 'upload', 'public', 'static']]
        
        for file in files:
            # 检查文件类型
            if any(file.endswith(ext) for ext in FILE_TYPES):
                total_files += 1
                file_path = os.path.join(root, file)
                if replace_comments(file_path):
                    processed_files += 1
    
    print("=" * 60)
    print(f"处理完成！")
    print(f"总文件数: {total_files}")
    print(f"已处理文件数: {processed_files}")
    print(f"替换率: {processed_files / total_files * 100:.2f}%" if total_files > 0 else "未找到匹配的文件")

if __name__ == "__main__":
    main()