#!/usr/bin/env python3
"""
增强的零配置测试 - 支持无AI模式
"""

import sys
sys.path.insert(0, ".")

from src import WebTester
import time

def smart_test(url: str, model: str = "glm", headless: bool = False, auto_login: bool = True):
    """
    智能测试，即使AI失败也能继续
    """
    
    print("=" * 60)
    print("🤖 增强智能测试")
    print("=" * 60)
    print(f"🌐 目标URL: {url}")
    print(f"🤖 AI模型: {model}")
    print(f"🖥️  无头模式: {'是' if headless else '否'}")
    print(f"🔐 自动登录: {'是' if auto_login else '否'}")
    print("-" * 60)
    
    tester = WebTester(model=model, headless=headless)
    
    try:
        # 启动并导航
        tester.start()
        tester.goto(url)
        time.sleep(2)  # 等待页面加载
        
        # 步骤1: 处理登录
        current_url = tester.browser.page.url
        print(f"\n📍 当前URL: {current_url}")
        
        # 检查是否需要登录
        need_login = ("login" in current_url.lower() or 
                     tester.browser.page.locator("input[type='password']").count() > 0)
        
        if need_login and auto_login:
            print("\n🔐 检测到登录页面，尝试登录...")
            
            # 方法1: 尝试AI登录（如果API可用）
            login_success = False
            try:
                result = tester.test("输入用户名admin和密码password，点击登录按钮")
                time.sleep(3)
                new_url = tester.browser.page.url
                if new_url != current_url and "login" not in new_url.lower():
                    print("  ✅ AI登录成功")
                    login_success = True
            except Exception as e:
                print(f"  ⚠️ AI登录失败: {str(e)[:50]}...")
            
            # 方法2: 如果AI失败，使用DOM直接登录
            if not login_success:
                print("  🔄 尝试直接登录...")
                try:
                    # 查找用户名和密码输入框
                    page = tester.browser.page
                    
                    # 尝试多种可能的用户名输入框
                    username_selectors = [
                        "input[placeholder*='用户名']",
                        "input[placeholder*='账号']",
                        "input[name='username']",
                        "input[name='account']",
                        "input[type='text']"
                    ]
                    
                    username_input = None
                    for selector in username_selectors:
                        if page.locator(selector).count() > 0:
                            username_input = page.locator(selector).first
                            break
                    
                    # 尝试多种可能的密码输入框
                    password_selectors = [
                        "input[type='password']",
                        "input[placeholder*='密码']",
                        "input[name='password']"
                    ]
                    
                    password_input = None
                    for selector in password_selectors:
                        if page.locator(selector).count() > 0:
                            password_input = page.locator(selector).first
                            break
                    
                    # 填写并提交
                    if username_input and password_input:
                        username_input.fill("admin")
                        password_input.fill("password")
                        
                        # 查找登录按钮
                        login_selectors = [
                            "button:has-text('登录')",
                            "button:has-text('登陆')",
                            "button:has-text('确定')",
                            "input[type='submit']",
                            ".login-btn"
                        ]
                        
                        for selector in login_selectors:
                            if page.locator(selector).count() > 0:
                                page.locator(selector).first.click()
                                break
                        
                        time.sleep(3)
                        new_url = tester.browser.page.url
                        if new_url != current_url:
                            print("  ✅ 直接登录成功")
                            login_success = True
                except Exception as e:
                    print(f"  ❌ 直接登录失败: {e}")
            
            if not login_success:
                print("  ⚠️ 无法自动登录，将测试登录页面功能")
        
        # 步骤2: 开始探索
        print("\n🔍 开始智能探索...")
        
        # 根据是否登录成功调整探索策略
        if login_success:
            explore_config = {
                "max_depth": 5,
                "max_clicks": 100,
                "focus_patterns": ["管理", "查询", "新增", "详情"],
                "dangerous_patterns": ["删除", "退出", "注销"]
            }
        else:
            explore_config = {
                "max_depth": 2,
                "max_clicks": 20,
                "focus_patterns": ["登录", "用户", "密码"],
                "dangerous_patterns": []
            }
        
        # 尝试AI探索，如果失败则使用DOM探索
        try:
            explore_result = tester.explore(explore_config)
            print("  ✅ AI探索完成")
        except Exception as e:
            print(f"  ⚠️ AI探索失败: {str(e)[:50]}...")
            print("  🔄 使用基础探索...")
            
            # 基础探索：点击所有可见的按钮和链接
            page = tester.browser.page
            clickable_elements = page.locator("button, a, [role='button'], input[type='button']")
            
            clicked_count = 0
            for i in range(min(clickable_elements.count(), 10)):
                try:
                    element = clickable_elements.nth(i)
                    if element.is_visible():
                        text = element.inner_text()[:20]
                        print(f"    点击: {text or '无文本'}")
                        element.click()
                        time.sleep(1)
                        page.go_back()
                        time.sleep(1)
                        clicked_count += 1
                except:
                    continue
            
            print(f"  📊 基础探索完成，点击了 {clicked_count} 个元素")
        
        # 步骤3: 生成测试报告
        print("\n📊 测试总结:")
        print(f"  ✅ 测试完成")
        print(f"  📄 已访问页面: {tester.browser.page.url}")
        
        return True
        
    except Exception as e:
        print(f"\n❌ 测试失败: {e}")
        import traceback
        traceback.print_exc()
        return False
    finally:
        tester.stop()

def main():
    import argparse
    
    parser = argparse.ArgumentParser(description="增强智能测试工具")
    parser.add_argument("url", help="要测试的网站URL")
    parser.add_argument("--model", default="glm", 
                       choices=["claude", "openai", "glm", "mimo"],
                       help="AI模型")
    parser.add_argument("--headless", action="store_true",
                       help="无头模式")
    parser.add_argument("--no-login", action="store_true",
                       help="跳过自动登录")
    
    args = parser.parse_args()
    
    # 运行测试
    success = smart_test(args.url, args.model, args.headless, not args.no_login)
    sys.exit(0 if success else 1)

if __name__ == "__main__":
    main()