from mcp.server.fastmcp import FastMCP

# Create an MCP server
mcp = FastMCP("Demo")


# Add an addition tool
@mcp.tool()
def add(a: int, b: int) -> int:
    """Add two numbers"""
    return a + b


# Add a dynamic greeting resource
@mcp.resource("greeting://{name}")
def get_greeting(name: str) -> str:
    """Get a personalized greeting"""
    return f"Hello, {name}!"

import MySQLdb
from mcp.server.fastmcp import FastMCP

mcp = FastMCP("MySQL Explorer")

conn = MySQLdb.connect(
    host="127.0.0.1",
    port=3306,
    user="root",
    password="root",
)

@mcp.resource("schema://main")
def get_schema() -> str:
    """Provide the database schema as a resource"""
    cursor = conn.cursor()
    cursor.execute("SHOW TABLES")  # Get a list of all tables

    tables = cursor.fetchall()
    schema = []

    for table in tables:
        table_name = table[0]
        cursor.execute(f"SHOW CREATE TABLE `{table_name}`")  # Get the create statement for each table
        create_table_sql = cursor.fetchone()[1]
        schema.append(create_table_sql)
    return "\n".join(schema)



@mcp.tool()
def query_data(sql: str) -> str:
    """Execute SQL queries safely"""
    try:
        cursor = conn.cursor()
        cursor.execute(sql)
        result = cursor.fetchall()
        return "\n".join(str(row) for row in result)
    except Exception as e:
        return f"Error: {str(e)}"