from typing import Optional

from pydantic import field_serializer
from sqlmodel import BigInteger, Field, Text, SQLModel

from common.core.models import SnowflakeBase
from common.core.schemas import BaseCreatorDTO


class AiModelBase:
    supplier: int = Field(nullable=False)
    name: str = Field(max_length=255, nullable=False)
    model_type: int = Field(nullable=False)
    base_model: str = Field(max_length=255, nullable=False)
    default_model: bool = Field(default=False, nullable=False)


class AiModelDetail(SnowflakeBase, AiModelBase, table=True):
    __tablename__ = "ai_model"
    api_key: str | None = Field(default=None, nullable=True, sa_type=Text())
    api_domain: str = Field(nullable=False, sa_type=Text())
    protocol: int = Field(nullable=False, default=1)
    config: str = Field(sa_type=Text())
    status: int = Field(nullable=False, default=1)
    create_time: int = Field(default=0, sa_type=BigInteger())


class AiModelWorkspaceMapping(SnowflakeBase, table=True):
    __tablename__ = "ai_model_workspace_mapping"
    ai_model_id: int = Field(default=None, nullable=True, sa_type=BigInteger())
    workspace_id: int = Field(default=None, nullable=True, sa_type=BigInteger())


class AiModelBrief(SQLModel):
    id: int
    name: str
    default_model: bool
    supplier: int

    @field_serializer("id")
    def id_to_str(self, v: int) -> str:
        return str(v)


class WorkspaceBase(SQLModel):
    name: str = Field(max_length=255, nullable=False)


class WorkspaceEditor(WorkspaceBase, BaseCreatorDTO):
    pass


class WorkspaceModel(SnowflakeBase, WorkspaceBase, table=True):
    __tablename__ = "sys_workspace"
    create_time: int = Field(default=0, sa_type=BigInteger())


class UserWsBaseModel(SQLModel):
    uid: int = Field(nullable=False, sa_type=BigInteger())
    oid: int = Field(nullable=False, sa_type=BigInteger())
    weight: int = Field(default=0, nullable=False)


class UserWsModel(SnowflakeBase, UserWsBaseModel, table=True):
    __tablename__ = "sys_user_ws"


class AssistantBaseModel(SQLModel):
    name: str = Field(max_length=255, nullable=False)
    type: int = Field(nullable=False, default=0)
    domain: str = Field(max_length=255, nullable=False)
    description: Optional[str] = Field(sa_type=Text(), nullable=True)
    configuration: Optional[str] = Field(sa_type=Text(), nullable=True)
    create_time: int = Field(default=0, sa_type=BigInteger())
    app_id: Optional[str] = Field(default=None, max_length=255, nullable=True)
    app_secret: Optional[str] = Field(default=None, max_length=255, nullable=True)
    oid: Optional[int] = Field(nullable=True, sa_type=BigInteger(), default=1)
    enable_custom_model: Optional[bool] = Field(default=False, nullable=True)
    custom_model: Optional[str] = Field(default=None, max_length=255, nullable=True)


class AssistantModel(SnowflakeBase, AssistantBaseModel, table=True):
    __tablename__ = "sys_assistant"


class AuthenticationBaseModel(SQLModel):
    name: str = Field(max_length=255, nullable=False)
    type: int = Field(nullable=False, default=0)
    config: Optional[str] = Field(sa_type=Text(), nullable=True)


class AuthenticationModel(SnowflakeBase, AuthenticationBaseModel, table=True):
    __tablename__ = "sys_authentication"
    create_time: Optional[int] = Field(default=0, sa_type=BigInteger())
    enable: bool = Field(default=False, nullable=False)
    valid: bool = Field(default=False, nullable=False)


class ApiKeyBaseModel(SQLModel):
    access_key: str = Field(max_length=255, nullable=False)
    secret_key: str = Field(max_length=255, nullable=False)
    create_time: int = Field(default=0, sa_type=BigInteger())
    uid: int = Field(default=0, nullable=False, sa_type=BigInteger())
    status: bool = Field(default=True, nullable=False)


class ApiKeyModel(SnowflakeBase, ApiKeyBaseModel, table=True):
    __tablename__ = "sys_apikey"