#!/usr/bin/env python3
"""
全面检查所有页面的按钮显示（忽略移动端菜单）
检查多个页面的 RippleButton 是否正常显示
"""

from playwright.sync_api import sync_playwright
import sys

def check_page_buttons(page, url, page_name):
    """检查指定页面的按钮"""
    print(f"\n{'='*60}")
    print(f"🔍 检查页面: {page_name}")
    print(f"📍 URL: {url}")
    print(f"{'='*60}")
    
    try:
        page.goto(url, timeout=30000)
        page.wait_for_load_state('networkidle')
        
        # 截图保存
        screenshot_name = page_name.replace('/', '-').replace(' ', '_')
        screenshot_path = f'test-results/{screenshot_name}.png'
        page.screenshot(path=screenshot_path, full_page=True)
        print(f"📸 截图已保存: {screenshot_path}")
        
        # 查找所有可能的按钮（包括 a 标签和 button 标签）
        all_buttons = page.locator('a, button').all()
        
        # 过滤出包含文本的按钮，并排除移动端菜单按钮
        buttons_with_text = []
        mobile_menu_buttons = ['首页', '服务', '产品', '新闻', '联系']  # 移动端菜单按钮
        
        for button in all_buttons:
            try:
                text = button.inner_text().strip()
                # 只关注短文本按钮，并排除移动端菜单
                if text and len(text) < 50 and text not in mobile_menu_buttons:
                    buttons_with_text.append({
                        'element': button,
                        'text': text
                    })
            except:
                pass
        
        print(f"\n📊 找到 {len(buttons_with_text)} 个按钮/链接（已排除移动端菜单）")
        
        # 检查每个按钮
        issues = []
        for btn_info in buttons_with_text:
            button = btn_info['element']
            text = btn_info['text']
            
            try:
                is_visible = button.is_visible()
                text_color = button.evaluate('el => window.getComputedStyle(el).color')
                bg_color = button.evaluate('el => window.getComputedStyle(el).backgroundColor')
                opacity = button.evaluate('el => window.getComputedStyle(el).opacity')
                
                # 检查文字是否可见（文字颜色不应与背景色相同）
                if 'rgb(196, 30, 58)' in text_color and 'rgb(196, 30, 58)' in bg_color:
                    issue = f"❌ 按钮 '{text}': 红色文字 + 红色背景 (可能不可见)"
                    issues.append(issue)
                    print(f"  {issue}")
                elif float(opacity) < 0.1:
                    issue = f"❌ 按钮 '{text}': 透明度过低 ({opacity})"
                    issues.append(issue)
                    print(f"  {issue}")
                elif not is_visible:
                    issue = f"⚠️  按钮 '{text}': 不可见"
                    issues.append(issue)
                    print(f"  {issue}")
                else:
                    print(f"  ✅ 按钮 '{text}': 正常")
                    
            except Exception as e:
                print(f"  ⚠️  按钮 '{text}': 检查失败 - {e}")
        
        if issues:
            print(f"\n⚠️  发现 {len(issues)} 个问题:")
            for issue in issues:
                print(f"  {issue}")
            return False
        else:
            print(f"\n✅ 所有按钮正常")
            return True
            
    except Exception as e:
        print(f"\n❌ 页面检查失败: {e}")
        return False

def main():
    pages_to_check = [
        {"url": "http://localhost:3000/", "name": "首页"},
        {"url": "http://localhost:3000/services/software", "name": "软件开发服务"},
        {"url": "http://localhost:3000/services/data", "name": "数据分析服务"},
        {"url": "http://localhost:3000/products/erp", "name": "ERP产品"},
        {"url": "http://localhost:3000/products/crm", "name": "CRM产品"},
        {"url": "http://localhost:3000/solutions/manufacturing", "name": "制造业解决方案"},
        {"url": "http://localhost:3000/contact", "name": "联系我们"},
    ]
    
    with sync_playwright() as p:
        browser = p.chromium.launch(headless=True)
        page = browser.new_page()
        
        results = {}
        for page_info in pages_to_check:
            result = check_page_buttons(page, page_info['url'], page_info['name'])
            results[page_info['name']] = result
        
        browser.close()
        
        # 总结
        print(f"\n{'='*60}")
        print("📋 检查总结")
        print(f"{'='*60}")
        
        all_passed = True
        for page_name, passed in results.items():
            status = "✅ 通过" if passed else "❌ 失败"
            print(f"{status} - {page_name}")
            if not passed:
                all_passed = False
        
        if all_passed:
            print(f"\n🎉 所有页面检查通过！")
            return 0
        else:
            print(f"\n⚠️  部分页面存在问题，请检查截图和日志。")
            return 1

if __name__ == '__main__':
    sys.exit(main())