"""
PM Agent - 产品经理智能体
负责需求消歧与规格生成
"""
from autogen import AssistantAgent
from typing import Dict, Any, Optional
import os
from pathlib import Path

from config.llm_config import get_agent_llm_config, PM_PROMPT


class ProductManagerAgent:
    """产品经理 Agent，负责生成软件需求规格说明书"""
    
    def __init__(self, llm_config: Optional[Dict] = None):
        """
        初始化 PM Agent
        
        Args:
            llm_config: LLM 配置，为 None 时使用默认配置
        """
        self.llm_config = llm_config or get_agent_llm_config("PM_Agent")
        
        # 创建 AutoGen AssistantAgent
        self.agent = AssistantAgent(
            name="PM_Agent",
            system_message=PM_PROMPT,
            llm_config=self.llm_config,
            description="资深软件产品经理，专注于汽车嵌入式系统领域",
            human_input_mode="NEVER"  # 全自动模式
        )
        
        self.workspace_dir = Path("workspace")
        self.workspace_dir.mkdir(exist_ok=True)
    
    def generate_srs(self, user_requirement: str) -> str:
        """
        根据用户需求生成 SRS 文档
        
        Args:
            user_requirement: 用户输入的原始需求
        
        Returns:
            生成的 SRS 文档内容
        """
        prompt = f"""
请根据以下用户需求生成完整的《软件需求规格说明书 (SRS)》：

用户需求：{user_requirement}

请确保输出包含：
1. 文档标题和版本信息
2. 功能性需求列表（FR-001, FR-002...）
3. 非功能性需求（NFR-001, NFR-002...）
4. 验收标准（AC-001, AC-002...）
5. 潜在风险与边缘情况

请以 Markdown 格式输出完整文档。
"""
        
        # 调用 Agent 生成 SRS
        response = self.agent.generate_reply(
            messages=[{"role": "user", "content": prompt}]
        )
        
        srs_content = response if isinstance(response, str) else str(response)
        
        # 保存到文件
        srs_file = self.workspace_dir / "SRS.md"
        with open(srs_file, 'w', encoding='utf-8') as f:
            f.write(srs_content)
        
        print(f"✅ SRS 文档已生成：{srs_file}")
        return srs_content
    
    def refine_requirements(
        self,
        original_srs: str,
        feedback: str
    ) -> str:
        """
        根据反馈优化需求
        
        Args:
            original_srs: 原始 SRS 文档
            feedback: 反馈意见
        
        Returns:
            优化后的 SRS 文档
        """
        prompt = f"""
请根据以下反馈优化现有的 SRS 文档：

原始 SRS:
{original_srs[:2000]}...  # 限制长度避免超出上下文

反馈意见：
{feedback}

请输出优化后的完整 SRS 文档。
"""
        
        response = self.agent.generate_reply(
            messages=[{"role": "user", "content": prompt}]
        )
        
        refined_srs = response if isinstance(response, str) else str(response)
        
        # 更新文件
        srs_file = self.workspace_dir / "SRS.md"
        with open(srs_file, 'w', encoding='utf-8') as f:
            f.write(refined_srs)
        
        print(f"✅ SRS 文档已更新：{srs_file}")
        return refined_srs


def create_pm_agent(llm_config: Optional[Dict] = None) -> AssistantAgent:
    """
    创建 PM Agent（AutoGen 原生格式）
    
    Args:
        llm_config: LLM 配置
    
    Returns:
        AutoGen AssistantAgent 实例
    """
    config = llm_config or get_agent_llm_config("PM_Agent")
    
    agent = AssistantAgent(
        name="PM_Agent",
        system_message=PM_PROMPT,
        llm_config=config,
        description="资深软件产品经理",
        human_input_mode="NEVER"
    )
    
    return agent