# coding=utf-8
"""
@Time   :2026/6/24
@Author :tao.chen
"""
from common.logging import logger
from spark_executor.core.connection_store import ConnectionStore, store
from spark_executor.models import Connection


def save_connection(
    *,
    name: str,
    master: str,
    deploy_mode: str = "cluster",
    yarn_rm_url: str | None = None,
    spark_conf: dict[str, str] | None = None,
) -> dict[str, str]:
    conn = Connection(
        name=name,
        master=master,
        deploy_mode=deploy_mode,
        yarn_rm_url=yarn_rm_url,
        spark_conf=spark_conf or {},
    )
    store.save(conn)
    logger.info(f"save_connection name={name} master={master}")
    return {"name": name, "status": "SAVED"}


def list_connections() -> list[dict[str, object]]:
    return [c.model_dump() for c in store.list_all()]


def get_connection(name: str) -> dict[str, object]:
    conn = store.get(name)
    if conn is None:
        raise KeyError(f"Unknown connection: {name}")
    return conn.model_dump()


def delete_connection(name: str) -> dict[str, str]:
    removed = store.delete(name)
    if not removed:
        raise KeyError(f"Unknown connection: {name}")
    logger.info(f"delete_connection name={name}")
    return {"name": name, "status": "DELETED"}