# 公共方法
import json
import requests  # 新增
import pysrt
from loguru import logger
from typing import List, Dict


def load_srt(file_path: str) -> List[Dict]:
    """加载并解析SRT文件（使用 pysrt 库，支持多种编码和格式）

    Args:
        file_path: SRT文件路径

    Returns:
        字幕内容列表，格式：
        [
            {
                'number': int,           # 字幕序号
                'timestamp': str,        # "00:00:01,000 --> 00:00:03,000"
                'text': str,             # 字幕文本
                'start_time': str,       # "00:00:01,000"
                'end_time': str          # "00:00:03,000"
            },
            ...
        ]

    Raises:
        FileNotFoundError: 文件不存在
        ValueError: 文件编码不支持或格式错误
    """
    # 编码自动检测：依次尝试常见编码
    encodings = ['utf-8', 'utf-8-sig', 'gbk', 'gb2312']
    subs = None
    detected_encoding = None

    for encoding in encodings:
        try:
            subs = pysrt.open(file_path, encoding=encoding)
            detected_encoding = encoding
            logger.info(f"成功加载字幕文件 {file_path}，编码：{encoding}，共 {len(subs)} 条")
            break
        except UnicodeDecodeError:
            continue
        except Exception as e:
            logger.warning(f"使用编码 {encoding} 加载失败: {e}")
            continue

    if subs is None:
        # 所有编码都失败
        raise ValueError(
            f"无法读取字幕文件 {file_path}，"
            f"请检查文件编码（支持 UTF-8、GBK、GB2312）"
        )

    # 检查是否为空
    if not subs:
        logger.warning(f"字幕文件 {file_path} 解析后无有效内容")
        return []

    # 转换为原格式（向后兼容）
    subtitles = []
    for sub in subs:
        # 合并多行文本为单行（某些 SRT 文件会有换行）
        text = sub.text.replace('\n', ' ').strip()

        # 跳过空字幕
        if not text:
            continue

        subtitles.append({
            'number': sub.index,
            'timestamp': f"{sub.start} --> {sub.end}",
            'text': text,
            'start_time': str(sub.start),
            'end_time': str(sub.end)
        })

    logger.info(f"成功解析 {len(subtitles)} 条有效字幕")
    return subtitles


def load_srt_from_content(srt_content: str) -> List[Dict]:
    """从字符串内容解析SRT（用于直接传入字幕内容，无需依赖文件路径）

    Args:
        srt_content: SRT格式的字幕文本内容

    Returns:
        字幕内容列表，格式同 load_srt 函数

    Raises:
        ValueError: 字幕内容为空或格式错误
    """
    if srt_content is None or not str(srt_content).strip():
        raise ValueError("字幕内容为空")

    try:
        subs = pysrt.from_string(str(srt_content))
    except Exception as e:
        logger.error(f"无法解析字幕内容: {e}")
        raise ValueError("无法解析字幕内容，请确保为标准 SRT 格式") from e

    if not subs:
        logger.warning("字幕内容解析后无有效内容")
        return []

    subtitles = []
    for sub in subs:
        text = sub.text.replace('\n', ' ').strip()
        if not text:
            continue

        subtitles.append({
            'number': sub.index,
            'timestamp': f"{sub.start} --> {sub.end}",
            'text': text,
            'start_time': str(sub.start),
            'end_time': str(sub.end)
        })

    logger.info(f"成功从内容解析 {len(subtitles)} 条有效字幕")
    return subtitles