from pydantic import BaseModel
from typing import List

# --- Token Schemas ---

class Token(BaseModel):
    access_token: str
    token_type: str

class TokenData(BaseModel):
    username: str | None = None

# --- Server Schemas ---

class ServerBase(BaseModel):
    name: str

class ServerCreate(ServerBase):
    pass

class Server(ServerBase):
    id: int
    owner_id: int

    class Config:
        orm_mode = True

# --- User Schemas ---

class UserBase(BaseModel):
    username: str

class UserCreate(UserBase):
    password: str

class User(UserBase):
    id: int
    owned_servers: List[Server] = []

    class Config:
        orm_mode = True