# -*- coding: utf-8 -*-
"""CLI 命令构建器"""

from typing import List, Dict, Any, Optional
from ..models.task_model import TaskConfig


# CLI 支持的命令行参数（来自 cli/main.py）
CLI_SUPPORTED_ARGS = {
    # 值类型参数
    "store_id", "tasks", "pg_dsn", "pg_host", "pg_port", "pg_name", 
    "pg_user", "pg_password", "api_base", "api_token", "api_timeout",
    "api_page_size", "api_retry_max", "window_start", "window_end",
    "export_root", "log_root", "pipeline_flow", "fetch_root", 
    "ingest_source", "idle_start", "idle_end",
    # 布尔类型参数
    "dry_run", "force_window_override", "write_pretty_json", "allow_empty_advance",
}


class CLIBuilder:
    """构建 CLI 命令行参数"""
    
    def __init__(self, python_executable: str = "python"):
        self.python_executable = python_executable
    
    def build_command(self, config: TaskConfig) -> List[str]:
        """
        根据任务配置构建命令行参数列表
        
        Args:
            config: 任务配置对象
            
        Returns:
            命令行参数列表
        """
        cmd = [self.python_executable, "-m", "cli.main"]
        
        # 任务列表
        if config.tasks:
            cmd.extend(["--tasks", ",".join(config.tasks)])
        
        # Pipeline 流程
        if config.pipeline_flow:
            cmd.extend(["--pipeline-flow", config.pipeline_flow])
        
        # Dry-run 模式
        if config.dry_run:
            cmd.append("--dry-run")
        
        # 时间窗口
        if config.window_start:
            cmd.extend(["--window-start", config.window_start])
        if config.window_end:
            cmd.extend(["--window-end", config.window_end])
        
        # 数据源目录
        if config.ingest_source:
            cmd.extend(["--ingest-source", config.ingest_source])
        
        # 门店 ID
        if config.store_id is not None:
            cmd.extend(["--store-id", str(config.store_id)])
        
        # 数据库 DSN
        if config.pg_dsn:
            cmd.extend(["--pg-dsn", config.pg_dsn])
        
        # API Token
        if config.api_token:
            cmd.extend(["--api-token", config.api_token])
        
        # 额外参数（只传递 CLI 支持的参数）
        for key, value in config.extra_args.items():
            if value is not None and key in CLI_SUPPORTED_ARGS:
                arg_name = f"--{key.replace('_', '-')}"
                if isinstance(value, bool):
                    if value:
                        cmd.append(arg_name)
                else:
                    cmd.extend([arg_name, str(value)])
        
        return cmd
    
    def build_command_string(self, config: TaskConfig) -> str:
        """
        构建命令行字符串（用于显示）
        
        Args:
            config: 任务配置对象
            
        Returns:
            命令行字符串
        """
        cmd = self.build_command(config)
        # 对包含空格的参数添加引号
        quoted_cmd = []
        for arg in cmd:
            if ' ' in arg or '"' in arg:
                quoted_cmd.append(f'"{arg}"')
            else:
                quoted_cmd.append(arg)
        return " ".join(quoted_cmd)
    
    def build_from_dict(self, params: Dict[str, Any]) -> List[str]:
        """
        从字典构建命令行参数
        
        Args:
            params: 参数字典
            
        Returns:
            命令行参数列表
        """
        config = TaskConfig(
            tasks=params.get("tasks", []),
            pipeline_flow=params.get("pipeline_flow", "FULL"),
            dry_run=params.get("dry_run", False),
            window_start=params.get("window_start"),
            window_end=params.get("window_end"),
            ingest_source=params.get("ingest_source"),
            store_id=params.get("store_id"),
            pg_dsn=params.get("pg_dsn"),
            api_token=params.get("api_token"),
            extra_args=params.get("extra_args", {}),
        )
        return self.build_command(config)


# 全局实例
cli_builder = CLIBuilder()