import bcrypt
from bson import ObjectId
from app.db import get_db

class User:
    def __init__(self, username, email, password_hash=None, role="user", auth_type="local", microsoft_id=None):
        self.username = username
        self.email = email
        self.password_hash = password_hash
        self.role = role
        self.auth_type = auth_type
        self.microsoft_id = microsoft_id
    
    @staticmethod
    def hash_password(password):
        salt = bcrypt.gensalt()
        hashed = bcrypt.hashpw(password.encode('utf-8'), salt)
        return hashed.decode('utf-8')
    
    @staticmethod
    def check_password(password_hash, password):
        return bcrypt.checkpw(password.encode('utf-8'), password_hash.encode('utf-8'))
    
    @staticmethod
    async def find_by_username(username):
        db = await get_db()
        user_data = await db.users.find_one({"username": username})
        return user_data
    
    @staticmethod
    async def find_by_email(email):
        db = await get_db()
        user_data = await db.users.find_one({"email": email})
        return user_data
    
    @staticmethod
    async def find_by_id(user_id):
        db = await get_db()
        user_data = await db.users.find_one({"_id": ObjectId(user_id)})
        return user_data
    
    @staticmethod
    async def find_by_microsoft_id(microsoft_id):
        db = await get_db()
        user_data = await db.users.find_one({"microsoft_id": microsoft_id})
        return user_data
    
    @staticmethod
    async def update_microsoft_id(user_id, microsoft_id):
        db = await get_db()
        result = await db.users.update_one(
            {"_id": ObjectId(user_id)},
            {"$set": {"microsoft_id": microsoft_id, "auth_type": "microsoft"}}
        )
        return result.modified_count > 0
    
    def to_dict(self):
        return {
            "username": self.username,
            "email": self.email,
            "role": self.role,
            "auth_type": self.auth_type,
            "microsoft_id": self.microsoft_id
        }
    
    async def save(self):
        db = await get_db()
        user_data = {
            "username": self.username,
            "email": self.email,
            "password_hash": self.password_hash,
            "role": self.role,
            "auth_type": self.auth_type,
            "microsoft_id": self.microsoft_id
        }
        result = await db.users.insert_one(user_data)
        return result.inserted_id