""" Defines the table for storing users """

# Import our mapped SQL types from SQLAlchemy
from sqlalchemy import Integer, String, DateTime, ARRAY


# Import mapping capabilities from the SQLAlchemy ORM
from sqlalchemy.orm import Mapped, mapped_column


# Import the EntityBase that we are extending
from .entity_base import EntityBase


# Import datetime for created_at type
from datetime import datetime


# Import enums for Role and Program
from .program_enum import ProgramEnum
from .user_enum import RoleEnum

#Import models for User methods
from ..models.user_model import User

from typing import Self


class UserEntity(EntityBase):
    """Serves as the database model for User table"""

    # set table name to user in the database
    __tablename__ = "user"

    # set fields or 'columns' for the user table
    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.now)
    username: Mapped[str] = mapped_column(String(32), nullable=False, default="", unique=True )
    role: Mapped[RoleEnum] = mapped_column(String, nullable=False)
    email: Mapped[str] = mapped_column(String(50), nullable=False, unique=True)
    program: Mapped[list[str]] = mapped_column(ARRAY(String), nullable=False)
    experience: Mapped[int] = mapped_column(Integer, nullable=False)
    group: Mapped[str] = mapped_column(String(50))

    
    @classmethod
    def from_model(cls, model: User) -> Self:
        """
        Create a user entity from model

        Args: model (User): the model to create the entity from

        Returns:
            self: The entity
        """

        return cls(
            id=model.id,
            created_at=model.created_at,
            username=model.username,
            role=model.usertype,
            email=model.email,
            program=model.programtype,
            experience=model.experience,
            group=model.group,         
        )

    def to_model(self) -> User:
        """
        
        Create a user model from entity

        Returns:
            User: A User model for API usage

        """    
        

        return User(
            id=self.id,
            username=self.username,
            email= self.email ,
            experience= self.experience,
            group= self.group,
            programtype= self.program,
            usertype= self.role,
            created_at= self.created_at,
        )