#!/usr/bin/env python3
"""
演示测试 - 测试基础功能（不实际发布）
"""

import asyncio
import sys
from pathlib import Path

# 简化的模型定义（基于working_solution.py）
from dataclasses import dataclass, field
from enum import Enum
from typing import List, Optional

class PlatformType(Enum):
    XIAOHONGSHU = "xiaohongshu"
    DOUYIN = "douyin"

@dataclass
class ImageNote:
    title: str
    description: str
    images: List[str] = field(default_factory=list)
    tags: List[str] = field(default_factory=list)
    visibility: str = "public"
    cover_image: Optional[str] = None

@dataclass
class VideoContent:
    title: str
    description: str
    video_path: str
    tags: List[str] = field(default_factory=list)
    visibility: str = "public"
    cover_image: Optional[str] = None

@dataclass
class AccountInfo:
    platform: PlatformType
    username: str
    cookie_file: str
    is_active: bool = True

# 测试基础功能
def test_basic_functionality():
    """测试基础功能"""
    print("🚀 开始基础功能测试")
    print("=" * 50)

    # 1. 测试内容创建
    print("📝 1. 测试内容创建")

    note = ImageNote(
        title="我的旅行分享 🏔️",
        description="今天去了很美的地方，和大家分享一些照片！",
        images=["img1.jpg", "img2.jpg", "img3.jpg"],
        tags=["旅行", "美景", "分享", "生活"]
    )
    print(f"✅ 图文笔记创建成功: {note.title}")
    print(f"   描述: {note.description[:30]}...")
    print(f"   图片数量: {len(note.images)}")
    print(f"   标签: {', '.join(note.tags)}")

    video = VideoContent(
        title="我的日常生活vlog 📸",
        description="记录一些日常生活的美好瞬间，希望大家喜欢！",
        video_path="daily_vlog.mp4",
        tags=["vlog", "日常", "生活记录", "分享"]
    )
    print(f"\n✅ 视频内容创建成功: {video.title}")
    print(f"   描述: {video.description[:30]}...")
    print(f"   视频路径: {video.video_path}")
    print(f"   标签: {', '.join(video.tags)}")

    # 2. 测试账号信息
    print(f"\n👤 2. 测试账号信息")

    xhs_account = AccountInfo(
        platform=PlatformType.XIAOHONGSHU,
        username="my_xhs_account",
        cookie_file="my_xhs_account.json"
    )
    print(f"✅ 小红书账号: {xhs_account.platform.value}/{xhs_account.username}")

    douyin_account = AccountInfo(
        platform=PlatformType.DOUYIN,
        username="my_douyin_account",
        cookie_file="my_douyin_account.json"
    )
    print(f"✅ 抖音账号: {douyin_account.platform.value}/{douyin_account.username}")

    # 3. 测试内容验证逻辑
    print(f"\n✅ 3. 测试内容验证")

    def validate_content(content, platform):
        """基础内容验证"""
        errors = []

        if not content.title.strip():
            errors.append("标题不能为空")
        elif len(content.title) > 100:
            errors.append("标题过长")

        if not content.description.strip():
            errors.append("描述不能为空")
        elif len(content.description) > 1000:
            errors.append("描述过长")

        if isinstance(content, ImageNote):
            if not content.images:
                errors.append("图文笔记必须包含图片")
            elif len(content.images) > 9:
                errors.append("图片数量不能超过9张")

        elif isinstance(content, VideoContent):
            if not content.video_path:
                errors.append("视频路径不能为空")

        return errors

    # 验证图文笔记
    note_errors = validate_content(note, PlatformType.XIAOHONGSHU)
    print(f"✅ 图文笔记验证: {'通过' if not note_errors else f'失败 - {note_errors}'}")

    # 验证视频内容
    video_errors = validate_content(video, PlatformType.DOUYIN)
    print(f"✅ 视频内容验证: {'通过' if not video_errors else f'失败 - {video_errors}'}")

    return True

async def test_playwright_basic():
    """测试Playwright基础功能"""
    print(f"\n🌐 4. 测试Playwright基础功能")

    try:
        from playwright.async_api import async_playwright

        async with async_playwright() as p:
            print("✅ Playwright启动成功")

            # 创建浏览器实例（有头模式用于演示）
            browser = await p.chromium.launch(headless=False)
            print("✅ Chromium浏览器启动成功")

            # 创建页面
            page = await browser.new_page()
            print("✅ 新页面创建成功")

            # 测试基本导航
            await page.goto("https://www.baidu.com")
            title = await page.title()
            print(f"✅ 页面导航成功: {title}")

            # 获取截图（可选）
            try:
                await page.screenshot(path="demo_screenshot.png")
                print("✅ 截图保存成功: demo_screenshot.png")
            except Exception as e:
                print(f"⚠️ 截图保存失败: {e}")

            # 关闭浏览器
            await browser.close()
            print("✅ 浏览器关闭成功")

        return True

    except ImportError:
        print("❌ Playwright未安装")
        return False
    except Exception as e:
        print(f"❌ Playwright测试失败: {e}")
        return False

def test_file_preparation():
    """测试文件准备"""
    print(f"\n📁 5. 测试文件准备")

    # 创建必要的目录
    dirs_to_create = [
        "media/images",
        "media/videos",
        "auth/cookies/xiaohongshu",
        "auth/cookies/douyin",
        "logs"
    ]

    created_dirs = []
    for dir_path in dirs_to_create:
        path = Path(dir_path)
        if not path.exists():
            path.mkdir(parents=True, exist_ok=True)
            created_dirs.append(dir_path)
            print(f"✅ 创建目录: {dir_path}")
        else:
            print(f"✅ 目录已存在: {dir_path}")

    # 创建测试文件
    test_files = [
        ("media/images/test_image.jpg", "测试图片文件"),
        ("media/videos/test_video.mp4", "测试视频文件")
    ]

    created_files = []
    for file_path, description in test_files:
        path = Path(file_path)
        if not path.exists():
            # 创建一个占位文件
            path.touch()
            created_files.append(file_path)
            print(f"✅ 创建占位文件: {file_path} ({description})")
        else:
            print(f"✅ 文件已存在: {file_path}")

    return True

async def main():
    """主测试函数"""
    print("🎭 社交媒体自动发布器 - 功能演示测试")
    print("📋 这个测试不会实际发布任何内容，只测试基础功能")
    print("=" * 60)

    success = True

    # 1. 基础功能测试
    if not test_basic_functionality():
        success = False

    # 2. 文件准备测试
    if not test_file_preparation():
        success = False

    # 3. Playwright测试
    playwright_success = await test_playwright_basic()
    if not playwright_success:
        print("⚠️ Playwright测试失败，但基础功能正常")

    print("\n" + "=" * 60)
    if success:
        print("🎉 基础功能测试完成！")
        print("\n📝 接下来你可以:")
        print("1. 安装浏览器: playwright install chromium")
        print("2. 运行实际测试（需要扫码登录）")
        print("3. 开始使用项目进行内容发布")
        print("\n⚠️  注意: 实际发布前请准备好真实的图片和视频文件")
        return True
    else:
        print("❌ 部分测试失败，请检查环境配置")
        return False

if __name__ == "__main__":
    # 运行测试
    result = asyncio.run(main())
    sys.exit(0 if result else 1)