from datetime import datetime
from typing import Optional
from uuid import UUID
from pydantic import BaseModel, EmailStr, Field


class UserBase(BaseModel):
    email: EmailStr
    name: str = Field(..., min_length=1, max_length=255)


class UserCreate(UserBase):
    password: str = Field(..., min_length=10, description="Password must be at least 10 characters")
    role: str = Field(default="engineer", description="User role: admin, engineer, or viewer")
    invite_code: Optional[str] = Field(None, description="Invite code for registration (required when invite system is enabled)")


class UserUpdate(BaseModel):
    name: Optional[str] = Field(None, min_length=1, max_length=255)
    email: Optional[EmailStr] = None


class UserLogin(BaseModel):
    email: EmailStr
    password: str


class UserResponse(UserBase):
    id: UUID
    role: str
    is_team_admin: bool = False
    team_id: Optional[UUID] = None
    created_at: datetime
    last_login: Optional[datetime] = None

    class Config:
        from_attributes = True