from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
from sqlalchemy.orm import DeclarativeBase, sessionmaker
from sqlalchemy import Column, Integer, String, Boolean, select,func, DateTime
import asyncio
from datetime import datetime,timezone

from pathlib import Path
db_folder = Path(__file__).parent / "DB"
db_folder.mkdir(parents=True, exist_ok=True)
db_path = db_folder / "guests.db"

async_engine = create_async_engine(f"sqlite+aiosqlite:///{db_path}", echo=True)

class Base(DeclarativeBase):
    pass

AsyncSessionLocal = sessionmaker(async_engine,class_=AsyncSession, expire_on_commit=False)

class User(Base):
    __tablename__ = "users"

    id = Column(Integer, primary_key=True, index=True)
    code = Column(String, unique=True, nullable=True)
    
    name = Column(String, nullable=True)
    surname = Column(String, nullable=True)
    text_field = Column(String, nullable=True)
    food = Column(Boolean)
    alco = Column(Boolean)
    types_of_alco = Column(String, default="Nothing")

    activated = Column(Boolean)
    created_at = Column(DateTime(timezone=True), server_default=func.now())
    updated_at = Column(DateTime(timezone=True), onupdate=func.now())
    last_login = Column(DateTime(timezone=True))

    admin = Column(Boolean, default=False)

async def create_user(user_info):
    async with AsyncSessionLocal() as session:
        user_data = user_info.dict(exclude_unset=True)
        new_user = User(**user_data)
        session.add(new_user)
        await session.commit()
        await session.refresh(new_user)

async def update_user(user_info):
    async with AsyncSessionLocal() as session:
        result = await session.execute(select(User).where(User.code==user_info.code))
        user = result.scalar_one_or_none()
        if user:
            update_data = user_info.dict(exclude_unset=True)
            for key, value in update_data.items():
                if hasattr(user, key):
                    setattr(user, key, value)
            await session.commit()
        return user
        
async def list_users():
    async with AsyncSessionLocal() as session:
        result = await session.execute(select(User))
        users = result.scalars().all() 
    if users:
        return users
    else:
        return None

async def login_user(code):
    async with AsyncSessionLocal() as session:
        result = await session.execute(select(User).where(User.code == code.code))
        user = result.scalar_one_or_none()
        if user:
            user.last_login=datetime.now(timezone.utc)
            await session.commit()
            return user
        else:
            return None