import json
from typing import TypeVar

from api import models, utils

T = TypeVar("T")


async def get_setting(scheme: T, name: str | None = None) -> T:
    name = name or scheme.__name__.lower()
    item = await utils.database.get_object(
        models.Setting, custom_query=models.Setting.query.where(models.Setting.name == name), raise_exception=False
    )
    data = scheme() if not item else scheme(**json.loads(item.value))
    if hasattr(data, "async_init"):
        await data.async_init()
    return data


async def set_setting(scheme, name: str | None = None):
    name = name or scheme.__class__.__name__.lower()
    json_data = scheme.model_dump(exclude_unset=True)
    data = {"name": name, "value": json_data}
    model = await utils.database.get_object(
        models.Setting, custom_query=models.Setting.query.where(models.Setting.name == name), raise_exception=False
    )
    if model:
        value = json.loads(model.value)
        for key in json_data:
            value[key] = json_data[key]
        data["value"] = json.dumps(value)
        await utils.database.modify_object(model, data)
    else:
        data["value"] = json.dumps(json_data)
        await utils.database.create_object(models.Setting, data)
    data = scheme.__class__(**json.loads(data["value"]))
    if hasattr(data, "async_init"):
        await data.async_init()
    return data