#!/usr/bin/env python3
"""
根据抓取的 API 自动生成 CLI 代码
"""
import json
import sys
from pathlib import Path


def generate_cli_from_capture(capture_file: str):
    """从抓取的 API 生成 CLI 代码"""
    
    with open(capture_file, 'r') as f:
        apis = json.load(f)
    
    # 分析并去重
    endpoints = {}
    for api in apis:
        url = api['url']
        method = api['method']
        
        # 提取路径
        if 'fumamx.com' in url:
            path = url.split('fumamx.com')[-1]
        else:
            path = url
        
        key = f"{method} {path.split('?')[0]}"
        if key not in endpoints:
            endpoints[key] = {
                'method': method,
                'path': path.split('?')[0],
                'params': [],
                'examples': []
            }
        
        # 提取查询参数
        if '?' in path:
            query = path.split('?')[1]
            params = [p.split('=')[0] for p in query.split('&') if p]
            endpoints[key]['params'].extend(params)
            endpoints[key]['params'] = list(set(endpoints[key]['params']))
        
        endpoints[key]['examples'].append(api)
    
    # 生成分类
    categories = {
        'user': [],
        'customer': [],
        'email': [],
        'order': [],
        'workbench': [],
        'other': []
    }
    
    for key, info in endpoints.items():
        path = info['path'].lower()
        if any(x in path for x in ['user', 'account', 'auth', 'login']):
            categories['user'].append(info)
        elif any(x in path for x in ['customer', 'crm', 'contact']):
            categories['customer'].append(info)
        elif any(x in path for x in ['email', 'mail', 'message']):
            categories['email'].append(info)
        elif any(x in path for x in ['order', 'trade', 'deal']):
            categories['order'].append(info)
        elif any(x in path for x in ['workbench', 'dashboard', 'home', 'main']):
            categories['workbench'].append(info)
        else:
            categories['other'].append(info)
    
    # 生成代码
    print("=" * 70)
    print("生成的 API 端点:")
    print("=" * 70)
    
    for cat_name, cat_apis in categories.items():
        if cat_apis:
            print(f"\n## {cat_name.upper()} ({len(cat_apis)} 个端点)")
            for api in cat_apis:
                params_str = ", ".join(api['params']) if api['params'] else "无"
                print(f"  {api['method']:6} {api['path']}")
                print(f"         参数: {params_str}")
    
    # 生成 client.py 代码
    print("\n" + "=" * 70)
    print("生成的 client.py 方法:")
    print("=" * 70)
    
    for cat_name, cat_apis in categories.items():
        if cat_apis:
            for api in cat_apis:
                path = api['path']
                method = api['method'].lower()
                
                # 生成方法名
                parts = path.strip('/').split('/')
                func_name = '_'.join(parts[-2:]) if len(parts) >= 2 else parts[-1]
                func_name = func_name.replace('-', '_').replace('.', '_')
                
                if method == 'get':
                    print(f"""
    def {func_name}(self{', ' + ', '.join([f'{p}=None' for p in api['params']]) if api['params'] else ''}):
        \"\"\"{path}\"\"\"
        params = {{{', '.join([f"'{p}': {p}" for p in api['params']])}}}
        return self.get('{path}')""")
                else:
                    print(f"""
    def {func_name}(self, data=None):
        \"\"\"{path}\"\"\"
        return self.post('{path}', data=data)""")
    
    # 保存结果
    output = Path('/home/ubuntu/nanocode/fumamxCLI/discovered_apis.json')
    with open(output, 'w') as f:
        json.dump({
            'endpoints': list(endpoints.values()),
            'categories': {k: v for k, v in categories.items() if v}
        }, f, ensure_ascii=False, indent=2)
    
    print(f"\n✅ 结果已保存到: {output}")


if __name__ == '__main__':
    if len(sys.argv) < 2:
        print("Usage: python setup_apis.py <fumamx_apis.json>")
        print("")
        print("1. 在浏览器控制台运行 capture_api.js")
        print("2. 操作页面后运行 exportFumamxAPIs()")
        print("3. 下载 fumamx_apis.json 到本目录")
        print("4. 运行: python setup_apis.py fumamx_apis.json")
        sys.exit(1)
    
    generate_cli_from_capture(sys.argv[1])