from typing import TYPE_CHECKING

from pydantic import BaseModel, BaseSettings
from sqlalchemy import create_engine
from sqlmodel import Session

from learn_sql_model.standard_config import load

if TYPE_CHECKING:
    from sqlalchemy import Engine


class ApiServer(BaseModel):
    app: str = "learn_sql_model.api.app:app"
    port: int = 5000
    reload: bool = True
    log_level: str = "info"


class Database:
    def __init__(self, config: "Config" = None) -> None:
        if config is None:

            self.config = get_config()
        else:
            self.config = config
        self.create_db_and_tables()

    @property
    def engine(self) -> "Engine":
        return create_engine(self.config.database_url)

    @property
    def session(self) -> "Session":
        return Session(self.engine)


class Config(BaseSettings):
    env: str = "dev"
    database_url: str = "sqlite:///database.db"
    api_server: ApiServer = ApiServer()

    class Config:
        extra = "ignore"
        env_nested_delimiter = "__"
        env_file = ".env", ".env.dev", ".env.qa", ".env.prod"

    @property
    def database(self) -> Database:
        return get_database(config=self)


def get_database(config: Config = None) -> Database:

    if config is None:
        config = get_config()

    return Database(config)


def get_config(overrides: dict = {}) -> Config:
    raw_config = load("learn_sql_model")
    config = Config(**raw_config, **overrides)
    return config