#!/usr/bin/env python3
"""手动添加B站视频到obsidian-kb收件箱"""

import sys
import os
import urllib.request
from urllib.request import Request, urlopen

# 添加路径以便导入模块
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
sys.path.insert(0, os.path.join(os.path.dirname(os.path.abspath(__file__)), 'src'))

from sources.bilibili import BilibiliAdapter
from models import Content, ContentStatus
from storage import Storage

def resolve_short_url(url: str) -> str:
    """
    解析短链接（如 b23.tv）到完整URL，或接受BV号

    Args:
        url: 可能是短链接的URL、完整URL或BV号

    Returns:
        str: 完整URL或BV号
    """
    # 如果已经是完整URL，直接返回
    if 'bilibili.com/video/' in url:
        return url

    # 如果是BV号，直接返回（bilibili.py 的 extract_content 会处理）
    if url.upper().startswith('BV'):
        return url

    # 如果是短链接，尝试解析
    if 'b23.tv' in url:
        print(f"  ⚠️  检测到短链接 {url}")
        print(f"  💡 B站对短链接有反爬虫保护，请提供:")
        print(f"     1. 完整URL: https://www.bilibili.com/video/BVxxxxx")
        print(f"     2. 或BV号: BVxxxxx")
        print(f"\n  提示: 可以在浏览器打开短链接，从地址栏复制完整URL")
        return None

    # 默认返回原URL
    return url

def add_video_to_inbox(url: str) -> str:
    """
    添加B站视频到收件箱

    Args:
        url: B站视频URL

    Returns:
        str: Content ID
    """
    print(f"📥 正在处理视频: {url}")

    # 解析短链接
    resolved_url = resolve_short_url(url)
    if resolved_url is None:
        return None

    # 创建适配器
    adapter = BilibiliAdapter()

    # 提取视频元数据
    print("  → 获取视频信息...")
    content = adapter.extract_content(resolved_url)
    if not content:
        print("  ❌ 无法提取视频信息")
        return None

    print(f"  → 标题: {content.title}")
    print(f"  → UP主: {content.author}")
    print(f"  → 时长: {content.duration}秒")

    # 获取字幕
    print("  → 获取字幕...")
    transcript = adapter.get_transcript(content)
    if transcript:
        content.transcript = transcript
        print(f"  ✓ 字幕已获取 ({len(transcript)} 字符)")
    else:
        print("  ⚠️  无法获取字幕")

    # 设置状态为已确认（准备处理）
    content.status = ContentStatus.CONFIRMED

    # 保存到收件箱
    print("  → 保存到收件箱...")
    storage = Storage()
    storage.save_content(content)

    print(f"\n✅ 视频已添加到收件箱!")
    print(f"   Content ID: {content.id}")
    print(f"\n💡 使用以下命令处理:")
    print(f"   python3 src/cli.py process {content.id}")
    print(f"   或者:")
    print(f"   python3 src/cli.py process all")

    return content.id

if __name__ == '__main__':
    if len(sys.argv) < 2:
        print("用法: python3 add_video.py <B站视频URL>")
        print("示例: python3 add_video.py https://b23.tv/oLBZZfK")
        sys.exit(1)

    url = sys.argv[1]
    add_video_to_inbox(url)