#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
测试 B站提取器（简化版，只使用 API）
"""

import sys
import json
from pathlib import Path

# 添加 src 目录到路径
sys.path.insert(0, str(Path(__file__).parent / 'skills' / 'obsidian-kb' / 'src'))

from extractors.bilibili_extractor import BilibiliExtractor


def main():
    """测试主函数"""
    print("开始测试 B站提取器...")

    # 创建提取器（不使用配置，使用默认值）
    extractor = BilibiliExtractor(config=None)

    print(f"提取器配置: headless={extractor.headless}, timeout={extractor.timeout}")

    # 测试 BV号
    source_id = 'BV1MnFNz5EDL'

    print(f"\n正在提取: {source_id}")

    try:
        # 首先测试 source_id 解析
        bvid = extractor._parse_source_id(source_id)
        print(f"解析 BV 号: {bvid}")

        # 直接测试 API 方法
        print("\n测试 API 提取...")
        result = extractor.extract(source_id)

        print("\n✅ 提取成功！\n")

        # 打印结果
        print(json.dumps(result, indent=2, ensure_ascii=False))

    except Exception as e:
        print(f"\n❌ 提取失败: {e}")
        import traceback
        traceback.print_exc()


if __name__ == '__main__':
    main()