#!/bin/bash

# 递归查找所有 MKV 文件
find . -type f -name "*.mkv" | while read -r file; do
    # 获取不带扩展名的文件名
    base_name="${file%.*}"
    
    # 提取中文字幕 (简体/繁体)
    ffmpeg -i "$file" -map 0:s:m:language:chi? -c:s srt "${base_name}.chi.srt" 2>/dev/null
    ffmpeg -i "$file" -map 0:s:m:language:zho? -c:s srt "${base_name}.zho.srt" 2>/dev/null
    ffmpeg -i "$file" -map 0:s:m:language:zh? -c:s srt "${base_name}.zh.srt" 2>/dev/null
    
    # 提取英文字幕
    ffmpeg -i "$file" -map 0:s:m:language:eng? -c:s srt "${base_name}.eng.srt" 2>/dev/null
    ffmpeg -i "$file" -map 0:s:m:language:en? -c:s srt "${base_name}.en.srt" 2>/dev/null
    
    echo "处理完成: $file"
done

echo "所有 MKV 文件字幕提取完成！"