#!/usr/bin/env python3
"""
Video Validation Tool
视频验证工具

验证切分后的视频文件是否有效
"""

import sys
import subprocess
from pathlib import Path


def validate_video_file(video_path: Path) -> dict:
    """验证单个视频文件"""
    try:
        # 使用ffprobe获取视频信息
        cmd = [
            'ffprobe',
            '-v', 'quiet',
            '-print_format', 'json',
            '-show_format',
            '-show_streams',
            str(video_path)
        ]
        
        result = subprocess.run(
            cmd,
            capture_output=True,
            text=True,
            timeout=30
        )
        
        if result.returncode == 0:
            import json
            data = json.loads(result.stdout)
            
            # 提取基本信息
            format_info = data.get('format', {})
            streams = data.get('streams', [])
            
            video_stream = None
            audio_stream = None
            
            for stream in streams:
                if stream.get('codec_type') == 'video':
                    video_stream = stream
                elif stream.get('codec_type') == 'audio':
                    audio_stream = stream
            
            return {
                'valid': True,
                'file_size': int(format_info.get('size', 0)),
                'duration': float(format_info.get('duration', 0)),
                'bitrate': int(format_info.get('bit_rate', 0)),
                'video_codec': video_stream.get('codec_name') if video_stream else None,
                'video_resolution': f"{video_stream.get('width')}x{video_stream.get('height')}" if video_stream else None,
                'video_fps': eval(video_stream.get('r_frame_rate', '0/1')) if video_stream else 0,
                'audio_codec': audio_stream.get('codec_name') if audio_stream else None,
                'audio_sample_rate': int(audio_stream.get('sample_rate', 0)) if audio_stream else 0,
                'error': None
            }
        else:
            return {
                'valid': False,
                'error': f"ffprobe failed: {result.stderr}"
            }
            
    except Exception as e:
        return {
            'valid': False,
            'error': str(e)
        }


def validate_split_videos(scenes_dir: Path):
    """验证切分目录中的所有视频"""
    print(f"🔍 验证目录: {scenes_dir}")
    
    if not scenes_dir.exists():
        print(f"❌ 目录不存在: {scenes_dir}")
        return False
    
    # 查找所有mp4文件
    video_files = list(scenes_dir.glob("*.mp4"))
    
    if not video_files:
        print(f"❌ 目录中没有找到mp4文件")
        return False
    
    print(f"📹 找到 {len(video_files)} 个视频文件")
    
    valid_count = 0
    total_size = 0
    total_duration = 0
    
    for video_file in sorted(video_files):
        print(f"\n🎬 验证: {video_file.name}")
        
        validation = validate_video_file(video_file)
        
        if validation['valid']:
            valid_count += 1
            file_size = validation['file_size']
            duration = validation['duration']
            total_size += file_size
            total_duration += duration
            
            print(f"   ✅ 有效")
            print(f"   📏 大小: {file_size:,} bytes ({file_size/1024/1024:.1f} MB)")
            print(f"   ⏱️ 时长: {duration:.2f}秒")
            print(f"   🎥 视频: {validation['video_codec']} {validation['video_resolution']} {validation['video_fps']:.1f}fps")
            if validation['audio_codec']:
                print(f"   🔊 音频: {validation['audio_codec']} {validation['audio_sample_rate']}Hz")
            print(f"   📊 码率: {validation['bitrate']:,} bps")
        else:
            print(f"   ❌ 无效: {validation['error']}")
    
    print(f"\n📊 验证摘要:")
    print(f"   有效文件: {valid_count}/{len(video_files)}")
    print(f"   成功率: {valid_count/len(video_files)*100:.1f}%")
    print(f"   总大小: {total_size:,} bytes ({total_size/1024/1024:.1f} MB)")
    print(f"   总时长: {total_duration:.2f}秒")
    
    return valid_count == len(video_files)


def main():
    """主函数"""
    print("🚀 视频验证工具")
    print("=" * 50)
    
    # 验证最新的切分结果
    test_scenes_dir = Path("/tmp/test_batch_output/1752032011698/scenes")
    
    if test_scenes_dir.exists():
        print("🎯 验证最新的批量切分结果")
        success = validate_split_videos(test_scenes_dir)
        
        if success:
            print("\n🎉 所有视频文件都有效！切分成功！")
            return 0
        else:
            print("\n⚠️ 部分视频文件无效")
            return 1
    else:
        print("❌ 没有找到测试切分结果")
        print("💡 请先运行批量切分命令:")
        print("   python3 -m python_core.cli scene batch /tmp/test_batch_input /tmp/test_batch_output --verbose --no-ai")
        return 1


if __name__ == "__main__":
    sys.exit(main())