# -*- coding: utf-8 -*-
"""
从 数据.jsonl 中读取成功记录，将 MP4 和 SRT 文件下载到本地 ./成功/ 目录。
用法：python download_srt.py [数据.jsonl路径] [本地输出目录]
默认：python download_srt.py 数据.jsonl ./成功
"""
import json
import os
import subprocess
import sys
from config.config_utils import config

JSONL_PATH = sys.argv[1] if len(sys.argv) > 1 else "数据.jsonl"
OUTPUT_DIR = sys.argv[2] if len(sys.argv) > 2 else "./成功"


def download(coscli_path, cos_config_path, cos_src, local_path):
    """下载单个文件，返回是否成功"""
    if os.path.exists(local_path):
        return "skip"
    cmd = [coscli_path, "-c", cos_config_path, "cp", cos_src, local_path]
    result = subprocess.run(cmd, capture_output=True, text=True, check=False)
    return "ok" if result.returncode == 0 else result.stderr.strip()


def main():
    if not os.path.exists(JSONL_PATH):
        print(f"找不到文件: {JSONL_PATH}")
        return

    cfg             = config.get_splitframe_config()
    coscli_path     = cfg.get('coscli_path', './coscli-linux-amd64')
    cos_config_path = cfg.get('cosconfig_path', '.cos.yaml')

    # 读取所有成功记录
    records = []
    with open(JSONL_PATH, 'r', encoding='utf-8') as f:
        for line in f:
            line = line.strip()
            if not line:
                continue
            try:
                rec = json.loads(line)
                if rec.get('status') == 'success' and rec.get('srt_cos_path'):
                    records.append(rec)
            except json.JSONDecodeError:
                pass
    records = records[:50]
    total = len(records)
    print(f"共找到 {total} 条成功记录，开始下载...")
    os.makedirs(OUTPUT_DIR, exist_ok=True)

    success = skip = failed = 0

    for i, rec in enumerate(records, 1):
        video_cos = rec.get('video_cos_path').replace(".mp4", ".mp3")
        srt_cos   = rec.get('srt_cos_path')

        video_name = os.path.splitext(os.path.basename(video_cos))[0] if video_cos else None

        print(f"[{i}/{total}] {video_name}")

        # 下载 MP4
        if video_cos:
            local_mp4 = os.path.join(OUTPUT_DIR, os.path.basename(video_cos))
            ret = download(coscli_path, cos_config_path, video_cos, local_mp4)
            if ret == "ok":
                success += 1
                print(f"  ✓ 视频: {os.path.basename(video_cos)}")
            elif ret == "skip":
                skip += 1
                print(f"  - 视频已存在，跳过")
            else:
                failed += 1
                print(f"  ✗ 视频下载失败: {ret}")

        # 下载 SRT
        if srt_cos:
            local_srt = os.path.join(OUTPUT_DIR, os.path.basename(srt_cos))
            ret = download(coscli_path, cos_config_path, srt_cos, local_srt)
            if ret == "ok":
                success += 1
                print(f"  ✓ 字幕: {os.path.basename(srt_cos)}")
            elif ret == "skip":
                skip += 1
                print(f"  - 字幕已存在，跳过")
            else:
                failed += 1
                print(f"  ✗ 字幕下载失败: {ret}")

    print(f"\n完成 — 下载: {success}  失败: {failed}  跳过(已存在): {skip}")
    print(f"文件保存在: {os.path.abspath(OUTPUT_DIR)}")


if __name__ == '__main__':
    main()