#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
视频脚本生成引擎 (示例模板)
将此文件移动到 engines/ 目录即可启用
"""

from typing import Dict, Any
from ..base import BaseAIGCEngine, EngineResult


class VideoScriptEngine(BaseAIGCEngine):
    """
    视频脚本生成引擎
    
    根据景点信息生成短视频分镜脚本
    """
    
    engine_id = "video_script"
    engine_name = "视频脚本生成"
    version = "1.0.0"
    description = "根据景点信息生成短视频分镜脚本"
    
    def get_param_schema(self) -> Dict[str, Any]:
        return {
            "scenic_spot_id": {
                "type": "int",
                "required": True,
                "desc": "景区 ID",
            },
            "duration": {
                "type": "int",
                "required": True,
                "desc": "视频时长 (秒)",
            },
            "style": {
                "type": "str",
                "required": False,
                "default": "vlog",
                "enum": ["vlog", "documentary", "promo", "tutorial"],
                "desc": "视频风格",
            },
            "target_platform": {
                "type": "str",
                "required": False,
                "default": "xiaohongshu",
                "enum": ["xiaohongshu", "douyin", "bilibili"],
                "desc": "目标平台",
            },
        }
    
    def estimate_duration(self, params: Dict[str, Any]) -> int:
        duration = params.get('duration', 60)
        return 20 + (duration // 10)  # 基础时间 + 每10秒视频约1秒生成时间
    
    async def execute(self, params: Dict[str, Any]) -> EngineResult:
        task_id = params.get('_task_id', 'unknown')
        
        try:
            self.set_progress(task_id, 10)
            
            # 1. 获取景区信息
            scenic_spot_id = params['scenic_spot_id']
            spot = await self.db.scenic_spot.find_by_id(scenic_spot_id)
            
            if not spot:
                return EngineResult(
                    success=False,
                    error=f"景区不存在: {scenic_spot_id}",
                    error_code="SPOT_NOT_FOUND"
                )
            
            self.set_progress(task_id, 20)
            
            # 2. 构建提示词
            duration = params['duration']
            style = params.get('style', 'vlog')
            platform = params.get('target_platform', 'xiaohongshu')
            
            system_prompt, user_prompt = self.prompt.build_full(
                "video_script",
                spot_name=spot.get('name', ''),
                spot_description=spot.get('description', ''),
                duration=duration,
                style=style,
                platform=platform
            )
            
            self.set_progress(task_id, 40)
            
            # 3. 调用 LLM
            result = await self.llm.generate(
                prompt=user_prompt,
                system_prompt=system_prompt
            )
            
            self.set_progress(task_id, 80)
            
            # 4. 解析结果
            script = self.parse_json(result)
            
            if not script:
                return EngineResult(
                    success=False,
                    error="脚本解析失败",
                    error_code="PARSE_ERROR"
                )
            
            self.set_progress(task_id, 100)
            
            return EngineResult(
                success=True,
                data={
                    "script": script,
                    "scene_count": len(script.get('scenes', [])),
                    "total_duration": duration,
                },
                metadata={
                    "scenic_spot_id": scenic_spot_id,
                    "style": style,
                    "platform": platform,
                }
            )
            
        except Exception as e:
            self.log(f"视频脚本生成失败: {e}", level='error')
            return EngineResult(
                success=False,
                error=str(e),
                error_code="EXECUTION_ERROR"
            )