"""
通讯录子图构建器
Contact Subgraph Builder
"""

from langgraph.graph import StateGraph, START, END

from .state import ContactState
from .nodes import (
    parse_intent,
    list_contacts,
    add_contact,
    list_emails,
    generate_email_draft,
    human_review,
    send_email,
    sniff_contacts,
    format_result,
    should_continue
)


def build_contact_subgraph() -> StateGraph:
    """
    构建通讯录子图
    
    Returns:
        配置好的 StateGraph
    """
    # 创建图
    graph = StateGraph(ContactState)
    
    # 添加节点
    graph.add_node("parse_intent", parse_intent)
    graph.add_node("list_contacts", list_contacts)
    graph.add_node("add_contact", add_contact)
    graph.add_node("list_emails", list_emails)
    graph.add_node("generate_email_draft", generate_email_draft)
    graph.add_node("human_review", human_review)
    graph.add_node("send_email", send_email)
    graph.add_node("sniff_contacts", sniff_contacts)
    graph.add_node("format_result", format_result)
    
    # 添加边
    # 从START开始
    graph.add_edge(START, "parse_intent")
    
    # 从parse_intent根据条件路由
    graph.add_conditional_edges(
        "parse_intent",
        should_continue,
        {
            "list_contacts": "list_contacts",
            "add_contact": "add_contact",
            "list_emails": "list_emails",
            "generate_email_draft": "generate_email_draft",
            "sniff_contacts": "sniff_contacts",
        }
    )
    
    # 从各个操作节点到format_result
    graph.add_edge("list_contacts", "format_result")
    graph.add_edge("add_contact", "format_result")
    graph.add_edge("list_emails", "format_result")
    graph.add_edge("sniff_contacts", "format_result")
    
    # 邮件发送的特殊流程
    graph.add_edge("generate_email_draft", "human_review")
    
    # 从human_review根据条件路由
    graph.add_conditional_edges(
        "human_review",
        should_continue,
        {
            "send_email": "send_email",
            "format_result": "format_result",
        }
    )
    
    # 发送邮件后到格式化
    graph.add_edge("send_email", "format_result")
    
    # 最终到END
    graph.add_edge("format_result", END)
    
    return graph