from typing import Any, Dict, List
from db.mysql_client import MySQLClient


# Initialize client
mysql_client = MySQLClient()


def query_mysql(sql: str, params: List = None) -> Dict[str, Any]:
    """
    Execute a SQL query against MySQL.

    Args:
        sql: The SQL query to execute
        params: Optional query parameters

    Returns:
        Query results or error message
    """
    try:
        results = mysql_client.query(sql, tuple(params) if params else None)
        return {
            "status": "success",
            "count": len(results),
            "results": results
        }
    except Exception as e:
        return {"status": "error", "message": str(e)}


def get_mysql_table_schema(table: str) -> Dict[str, Any]:
    """
    Get schema information for a MySQL table.

    Args:
        table: Table name

    Returns:
        Table schema information
    """
    try:
        schema = mysql_client.get_table_schema(table)
        return {
            "status": "success",
            "table": table,
            "schema": schema
        }
    except Exception as e:
        return {"status": "error", "message": str(e)}