import os
from typing import TYPE_CHECKING

from sqladmin import Admin, ModelView

from core.bot.models.chat_gpt import ChatGpt
from settings.config import settings

if TYPE_CHECKING:
    from main import Application


class ChatGptAdmin(ModelView, model=ChatGpt):
    column_list = [ChatGpt.id, ChatGpt.model, ChatGpt.priority]
    column_sortable_list = [ChatGpt.priority]
    column_default_sort = ("priority", True)
    form_widget_args = {"model": {"readonly": True}}

    can_create = False
    can_delete = False


def create_admin(application: "Application") -> Admin:
    admin = Admin(
        title="Chat GPT admin",
        app=application.fastapi_app,
        engine=application.db.async_engine,
        base_url=os.path.join(settings.URL_PREFIX, "admin"),
        authentication_backend=None,
    )
    admin.add_view(ChatGptAdmin)
    return admin