from mcp.server.fastmcp import FastMCP
from tech_wizard.tools import TechWizardTools
import os
from dotenv import load_dotenv

# Load environment variables
load_dotenv()

# Initialize FastMCP
mcp = FastMCP("Tech-Wizard")

# Initialize Tools
tech_tools = TechWizardTools()

@mcp.tool()
def architect_system(requirements: str) -> str:
    """
    Design a technical architecture plan (Mermaid diagram, stack, patterns).
    """
    return tech_tools.architect_system(requirements)

@mcp.tool()
def design_api(features: str) -> str:
    """
    Design a comprehensive API surface (endpoints, schemas, security).
    """
    return tech_tools.design_api(features)

@mcp.tool()
def setup_webhooks(source_platform: str) -> str:
    """
    Generate secure and reliable webhook handler code.
    """
    return tech_tools.setup_webhooks(source_platform)

@mcp.tool()
def configure_dns_routing(domain: str, app_details: str) -> str:
    """
    Get a DNS configuration and SSL/TLS routing plan.
    """
    return tech_tools.configure_routing(domain, app_details)

@mcp.tool()
def engineering_consultant(query: str) -> str:
    """
    Consult the Grand Tech Wizard for high-level engineering advice.
    """
    return tech_tools.llm.chat(query)

def main():
    mcp.run()

if __name__ == "__main__":
    main()