#!/usr/bin/env python3
"""
验证 API v3.0 重构是否完整
检查代码中是否还有旧的逻辑残留
"""

import os
import re
from pathlib import Path

def print_header(title):
    print(f"\n{'='*60}")
    print(f"  {title}")
    print('='*60)

def search_in_file(filepath, pattern, description):
    """在文件中搜索模式"""
    try:
        with open(filepath, 'r', encoding='utf-8') as f:
            content = f.read()
            matches = re.findall(pattern, content, re.MULTILINE)
            return matches
    except Exception as e:
        return None

def check_removed_code():
    """检查已移除的代码"""
    print_header("检查已移除的代码")
    
    checks = [
        {
            "file": "search/es_query_builder.py",
            "pattern": r"if field == ['\"]price_ranges['\"]",
            "description": "硬编码的 price_ranges 逻辑",
            "should_exist": False
        },
        {
            "file": "search/es_query_builder.py",
            "pattern": r"def add_dynamic_aggregations",
            "description": "add_dynamic_aggregations 方法",
            "should_exist": False
        },
        {
            "file": "api/models.py",
            "pattern": r"aggregations.*Optional\[Dict",
            "description": "aggregations 参数（在 SearchRequest 中）",
            "should_exist": False
        },
        {
            "file": "frontend/static/js/app.js",
            "pattern": r"price_ranges",
            "description": "前端硬编码的 price_ranges",
            "should_exist": False
        },
        {
            "file": "frontend/static/js/app.js",
            "pattern": r"displayAggregations",
            "description": "旧的 displayAggregations 函数",
            "should_exist": False
        }
    ]
    
    all_passed = True
    for check in checks:
        filepath = os.path.join("/home/tw/SearchEngine", check["file"])
        matches = search_in_file(filepath, check["pattern"], check["description"])
        
        if matches is None:
            print(f"  ⚠️ 无法读取：{check['file']}")
            continue
        
        if check["should_exist"]:
            if matches:
                print(f"  ✓ 存在：{check['description']}")
            else:
                print(f"  ✗ 缺失：{check['description']}")
                all_passed = False
        else:
            if matches:
                print(f"  ✗ 仍存在：{check['description']}")
                print(f"     匹配：{matches[:2]}")
                all_passed = False
            else:
                print(f"  ✓ 已移除：{check['description']}")
    
    return all_passed

def check_new_code():
    """检查新增的代码"""
    print_header("检查新增的代码")
    
    checks = [
        {
            "file": "api/models.py",
            "pattern": r"class RangeFilter",
            "description": "RangeFilter 模型",
            "should_exist": True
        },
        {
            "file": "api/models.py",
            "pattern": r"class FacetConfig",
            "description": "FacetConfig 模型",
            "should_exist": True
        },
        {
            "file": "api/models.py",
            "pattern": r"class FacetValue",
            "description": "FacetValue 模型",
            "should_exist": True
        },
        {
            "file": "api/models.py",
            "pattern": r"class FacetResult",
            "description": "FacetResult 模型",
            "should_exist": True
        },
        {
            "file": "api/models.py",
            "pattern": r"range_filters.*RangeFilter",
            "description": "range_filters 参数",
            "should_exist": True
        },
        {
            "file": "api/models.py",
            "pattern": r"facets.*FacetConfig",
            "description": "facets 参数",
            "should_exist": True
        },
        {
            "file": "search/es_query_builder.py",
            "pattern": r"def build_facets",
            "description": "build_facets 方法",
            "should_exist": True
        },
        {
            "file": "search/searcher.py",
            "pattern": r"def _standardize_facets",
            "description": "_standardize_facets 方法",
            "should_exist": True
        },
        {
            "file": "api/routes/search.py",
            "pattern": r"@router.get\(['\"]\/suggestions",
            "description": "/search/suggestions 端点",
            "should_exist": True
        },
        {
            "file": "api/routes/search.py",
            "pattern": r"@router.get\(['\"]\/instant",
            "description": "/search/instant 端点",
            "should_exist": True
        },
        {
            "file": "frontend/static/js/app.js",
            "pattern": r"function displayFacets",
            "description": "displayFacets 函数",
            "should_exist": True
        },
        {
            "file": "frontend/static/js/app.js",
            "pattern": r"rangeFilters",
            "description": "rangeFilters 状态",
            "should_exist": True
        }
    ]
    
    all_passed = True
    for check in checks:
        filepath = os.path.join("/home/tw/SearchEngine", check["file"])
        matches = search_in_file(filepath, check["pattern"], check["description"])
        
        if matches is None:
            print(f"  ⚠️ 无法读取：{check['file']}")
            continue
        
        if check["should_exist"]:
            if matches:
                print(f"  ✓ 存在：{check['description']}")
            else:
                print(f"  ✗ 缺失：{check['description']}")
                all_passed = False
        else:
            if matches:
                print(f"  ✗ 仍存在：{check['description']}")
                all_passed = False
            else:
                print(f"  ✓ 已移除：{check['description']}")
    
    return all_passed

def check_documentation():
    """检查文档"""
    print_header("检查文档")
    
    docs = [
        "API_DOCUMENTATION.md",
        "API_EXAMPLES.md",
        "MIGRATION_GUIDE_V3.md",
        "CHANGES.md"
    ]
    
    all_exist = True
    for doc in docs:
        filepath = os.path.join("/home/tw/SearchEngine", doc)
        if os.path.exists(filepath):
            size_kb = os.path.getsize(filepath) / 1024
            print(f"  ✓ 存在：{doc} ({size_kb:.1f} KB)")
        else:
            print(f"  ✗ 缺失：{doc}")
            all_exist = False
    
    return all_exist

def check_imports():
    """检查模块导入"""
    print_header("检查模块导入")
    
    import sys
    sys.path.insert(0, '/home/tw/SearchEngine')
    
    try:
        from api.models import (
            RangeFilter, FacetConfig, FacetValue, FacetResult,
            SearchRequest, SearchResponse, ImageSearchRequest,
            SearchSuggestRequest, SearchSuggestResponse
        )
        print("  ✓ API 模型导入成功")
        
        from search.es_query_builder import ESQueryBuilder
        print("  ✓ ESQueryBuilder 导入成功")
        
        from search.searcher import Searcher, SearchResult
        print("  ✓ Searcher 导入成功")
        
        # 检查方法
        qb = ESQueryBuilder('test', ['field1'])
        if hasattr(qb, 'build_facets'):
            print("  ✓ build_facets 方法存在")
        else:
            print("  ✗ build_facets 方法不存在")
            return False
        
        if hasattr(qb, 'add_dynamic_aggregations'):
            print("  ✗ add_dynamic_aggregations 方法仍存在（应该已删除）")
            return False
        else:
            print("  ✓ add_dynamic_aggregations 方法已删除")
        
        # 检查 SearchResult
        sr = SearchResult(hits=[], total=0, max_score=0, took_ms=10, facets=[])
        if hasattr(sr, 'facets'):
            print("  ✓ SearchResult.facets 属性存在")
        else:
            print("  ✗ SearchResult.facets 属性不存在")
            return False
        
        if hasattr(sr, 'aggregations'):
            print("  ✗ SearchResult.aggregations 属性仍存在（应该已删除）")
            return False
        else:
            print("  ✓ SearchResult.aggregations 属性已删除")
        
        return True
        
    except Exception as e:
        print(f"  ✗ 导入失败：{e}")
        return False

def main():
    """主函数"""
    print("\n" + "🔍 开始验证 API v3.0 重构")
    print(f"项目路径：/home/tw/SearchEngine\n")
    
    # 运行检查
    check1 = check_removed_code()
    check2 = check_new_code()
    check3 = check_documentation()
    check4 = check_imports()
    
    # 总结
    print_header("验证总结")
    
    results = {
        "已移除的代码": check1,
        "新增的代码": check2,
        "文档完整性": check3,
        "模块导入": check4
    }
    
    all_passed = all(results.values())
    
    for name, passed in results.items():
        status = "✓ 通过" if passed else "✗ 失败"
        print(f"  {status}: {name}")
    
    if all_passed:
        print(f"\n  🎉 所有检查通过！API v3.0 重构完成。")
    else:
        print(f"\n  ⚠️ 部分检查失败，请检查上述详情。")
    
    print("\n" + "="*60 + "\n")
    
    return 0 if all_passed else 1

if __name__ == "__main__":
    exit(main())