from flask_login import UserMixin
from pathlib import Path

from database.db import get_db

class User(UserMixin):
    def __init__(self, id_, name, email, profile_pic):
        self.id = id_
        self.name = name
        self.email = email
        self.profile_pic = profile_pic

    @staticmethod
    def get(user_id):
        db = get_db()
        user = db.execute(
            "SELECT * FROM user WHERE id = ?", (user_id,)
        ).fetchone()
        if not user:
            return None

        user = User(
            id_=user[0], name=user[1], email=user[2], profile_pic=user[3]
        )
        return user

    @staticmethod
    def create(id_, name, email, profile_pic):
        db = get_db()
        db.execute(
            "INSERT INTO user (id, name, email, profile_pic) "
            "VALUES (?, ?, ?, ?)",
            (id_, name, email, profile_pic),
        )
        db.commit()

        Path(f"userinfo/{id_}").mkdir(parents=True, exist_ok=True)

class dbCalendar():
    def __init__(self, id_, name, toggle, color):
        self.usr_id = id_
        self.name = name
        self.toggle = toggle
        self.color = color

    @staticmethod
    def getCalendars(user_id):
        db = get_db()
        calendars = db.execute(
            "SELECT * FROM calendar WHERE usr_id = ?", (user_id,)
        ).fetchall()

        return calendars

    @staticmethod
    def getCalendar(user_id, calendar_id):
        db = get_db()
        calendar = db.execute(
            "SELECT * FROM calendar WHERE usr_id = ? AND calendar_id = ?", (user_id, calendar_id,)
        ).fetchone()
        if not calendar:
            return None
    
    @staticmethod
    def create(user_id, calendar_id, name, color, toggle = False):
        db = get_db()
        db.execute(
            "INSERT INTO calendar (usr_id, calendar_id, name, toggle, color) "
            "VALUES (?, ?, ?, ?, ?)",
            (user_id, calendar_id, name, toggle, color),
        )
        db.commit()