from pydantic import BaseModel, Field
from typing import Optional, List
from uuid import UUID


# TG Models
class RefResponse(BaseModel):
    ref: str
    description: str
    promocode: str

class RefAddRequest(BaseModel):
    description: str

class TokenRequest(BaseModel):
    tg_id: int

class RegisterRequest(BaseModel):
    tg_id: int
    chat_id: Optional[int] = None
    phone: Optional[str] = None
    name: Optional[str] = None
    login: Optional[str] = None
    company_key: str

# New Response Models for TG APIs
class RefAddResponse(BaseModel):
    ref: str
    promocode: str
    description: str

class RefStatItem(BaseModel):
    description: str
    sales: int
    income: float

class RefStatResponse(BaseModel):
    refData: List[RefStatItem]

class StatResponse(BaseModel):
    totalSales: int
    totalIncome: float
    availableWithdrawal: float