from sqlmodel import Field, Relationship, Session, SQLModel
from typing import List

class Base(SQLModel):
    pass

#### User
class UserBase(Base):
    name: str = Field(index=True)
    email: str | None = None
    is_admin: bool = False

class UserResponse(UserBase):
    id: int

class UserCreate(UserBase):
    password: str

class UserDB(UserBase, table=True):
    id: int | None = Field(default=None, primary_key=True)
    password: str

#### Special
class AaGroupLink(Base, table=True):
    group_id: int | None = Field(default=None, foreign_key="group.id", primary_key=True)
    accessauth_id: int | None = Field(default=None, foreign_key="accessauthorization.id", primary_key=True)

#### Group
class GroupBase(Base):
    name: str = Field(index=True)
    

class GroupDB(GroupBase, table=True):
    id: int | None = Field(default=None, primary_key=True)
    cards: List["Card"] = Relationship(back_populates="group")
    accessauths: List["AccessAuthorizationBase"] = Relationship(back_populates="groups", link_model=AaGroupLink)

class GroupResponse(GroupBase):
    id: int
    cards: List["Card"]
    accessauths: List["AccessAuthorizationBase"]

#### AccessAuthorization
class AccessAuthorizationBase(Base):
    name: str = Field(index=True)
    is_active: bool
    groups: List["GroupDB"] = Relationship(back_populates="accessauths", link_model=AaGroupLink)
    timetables: List["Timetable"] = Relationship(back_populates="accessauth")

class AccessAuthorizationDB(Base, table=True):
    id: int | None = Field(default=None, primary_key=True)

class AccessAuthorizationCreate(AccessAuthorizationBase):
    pass

class AccessAuthorizationResponse(AccessAuthorizationBase):
    id: int

#### Card
class Card(Base, table=True):
    id: int | None = Field(default=None, primary_key=True)
    uuid: str
    group_id: int | None = Field(default=None, foreign_key="groupdb.id")
    group: GroupDB | None = Relationship(back_populates="cards")

class Timetable(Base, table=True):
    id: int | None = Field(default=None, primary_key=True)
    weekday: int
    starttime: str
    duration: int
    accessauth_id: int | None = Field(default=None, foreign_key="accessauthorizationdb.id")
    accessauth: AccessAuthorizationDB | None = Relationship(back_populates="timetables")