# [DEF:main:Module]
# @PURPOSE: CLI Entry point.
# @RELATION: CALLS -> translation_engine

import sys
import re
from pathlib import Path
from translation_engine import SubtitleOrchestrator
from domain_models import SubtitleLine

# [DEF:parse_srt:Function]
# @PURPOSE: Extract raw text and structure from SRT file.
def parse_srt(file_path: str) -> list[SubtitleLine]:
    with open(file_path, 'r', encoding='utf-8') as f:
        content = f.read()
    
    # Regex to capture standard SRT blocks
    pattern = re.compile(r'(\d+)\n(\d{2}:\d{2}:\d{2},\d{3} --> \d{2}:\d{2}:\d{2},\d{3})\n((?:(?!\r?\n\r?\n).)*)', re.DOTALL)
    matches = pattern.findall(content)
    
    lines = []
    for idx, times, text in matches:
        start, end = times.split(' --> ')
        # Clean text
        clean_text = text.replace('\n', ' ').strip()
        lines.append(SubtitleLine(
            index=int(idx),
            start_time=start,
            end_time=end,
            original_text=clean_text
        ))
    return lines
# [/DEF:parse_srt]

# [DEF:save_srt:Function]
def save_srt(lines: list[SubtitleLine], path: str):
    with open(path, 'w', encoding='utf-8') as f:
        for line in lines:
            text = line.translated_text if line.translated_text else line.original_text
            f.write(f"{line.index}\n{line.start_time} --> {line.end_time}\n{text}\n\n")
# [/DEF:save_srt]

if __name__ == "__main__":
    # [STATE:Entry]
    if len(sys.argv) < 2:
        print("Usage: python main.py <subs.srt>")
        sys.exit(1)
        
    input_file = sys.argv[1]
    output_file = str(Path(input_file).with_stem(Path(input_file).stem + "_ru"))
    
    engine = SubtitleOrchestrator()
    
    # 1. Load Data
    subs = parse_srt(input_file)
    full_text = " ".join([s.original_text for s in subs])
    
    # 2. Pass 1: Analyze
    print("--- PASS 1: Analyzing Context ---")
    context = engine.pass_one_analysis(full_text)
    print(f"Detected Genre: {context.genre}")
    print(f"Characters found: {len(context.characters)}")
    
    # 3. Pass 2: Translate
    print("--- PASS 2: Translating ---")
    translated_subs = engine.pass_two_translation(subs)
    
    # 4. Save
    save_srt(translated_subs, output_file)
    print(f"Saved to {output_file}")
    # [STATE:Exit]

# [/DEF:main]