popa_minecraft_launcher_api/auth/app/auth.py

import base64
import json
from fastapi import HTTPException
from .models import UserLogin, UserInDB, Session, UserCreate
from .utils import (
    verify_password,
    get_password_hash,
    create_access_token,
    decode_token,
)
from .database import users_collection, sessions_collection
import uuid
from datetime import datetime, timedelta

class AuthService:
    async def register(self, user: UserCreate):
        # Проверяем, существует ли пользователь
        if await users_collection.find_one({"username": user.username}):
            raise HTTPException(status_code=400, detail="Username already taken")

        # Хешируем пароль
        hashed_password = get_password_hash(user.password)
        
        # Создаём UUID для Minecraft
        user_uuid = str(uuid.uuid4())

        # Сохраняем в MongoDB
        new_user = UserInDB(
            username=user.username,
            email=user.email,
            hashed_password=hashed_password,
            uuid=user_uuid,
        )
        await users_collection.insert_one(new_user.dict())
        return {"status": "success", "uuid": user_uuid}

    async def login(self, credentials: UserLogin):
        # Ищем пользователя
        user = await users_collection.find_one({"username": credentials.username})
        if not user or not verify_password(credentials.password, user["hashed_password"]):
            raise HTTPException(status_code=401, detail="Invalid credentials")

        # Генерируем токены
        access_token = create_access_token({"sub": user["username"], "uuid": user["uuid"]})
        client_token = str(uuid.uuid4())

        # Сохраняем сессию
        session = Session(
            access_token=access_token,
            client_token=client_token,
            user_uuid=user["uuid"],
            expires_at=datetime.utcnow() + timedelta(minutes=1440),
        )
        await sessions_collection.insert_one(session.dict())

        return {
            "accessToken": access_token,
            "clientToken": client_token,
            "selectedProfile": {
                "id": user["uuid"],
                "name": user["username"],
            },
        }

    async def validate(self, access_token: str, client_token: str):
        print(f"Searching for access_toke and client_token: '{access_token}', '{client_token}")
        session = await sessions_collection.find_one({
            "access_token": access_token,
            "client_token": client_token,
        })
        print("Session from DB:", session)
        if not session or datetime.utcnow() > session["expires_at"]:
            return False
        return True

    async def refresh(self, access_token: str, client_token: str):
        if not await self.validate(access_token, client_token):
            return None

        # Обновляем токен
        new_access_token = create_access_token({"sub": "user", "uuid": "user_uuid"})
        await sessions_collection.update_one(
            {"access_token": access_token},
            {"$set": {"access_token": new_access_token}},
        )
        return {"accessToken": new_access_token, "clientToken": client_token}
    
    async def get_minecraft_profile(self, uuid: str):
        formatted_uuid = f"{uuid[:8]}-{uuid[8:12]}-{uuid[12:16]}-{uuid[16:20]}-{uuid[20:]}"
        user = await users_collection.find_one({"uuid": formatted_uuid})
        if not user:
            raise HTTPException(
                status_code=404,
                detail=f"User not found (searched for UUID: {formatted_uuid})"
            )
        
        textures = {
            "timestamp": int(datetime.now().timestamp()),
            "profileId": formatted_uuid,
            "profileName": user["username"],
            "textures": {
                "SKIN": {"url": user.get("skin_url", "")},
                "CAPE": {"url": user.get("cloak_url", "")}
            } if user.get("skin_url") or user.get("cloak_url") else {}
        }
        
        textures_json = json.dumps(textures).encode()
        base64_textures = base64.b64encode(textures_json).decode()

        return {
            "id": formatted_uuid,
            "name": user["username"],
            "properties": [
                {
                    "name": "textures",
                    "value": base64_textures
                }
            ]
        }

    async def join_server(self, request_data: dict):
        access_token = request_data.get("accessToken")
        selected_profile = request_data.get("selectedProfile")
        server_id = request_data.get("serverId")

        if not all([access_token, selected_profile, server_id]):
            raise HTTPException(status_code=400, detail="Missing required parameters")

        decoded_token = decode_token(access_token)
        if not decoded_token:
            raise HTTPException(status_code=401, detail="Invalid access token")

        token_uuid = decoded_token.get("uuid", "").replace("-", "")
        if token_uuid != selected_profile:
            raise HTTPException(status_code=403, detail="Token doesn't match selected profile")

        return True

    async def has_joined(self, username: str, server_id: str):
        user = await users_collection.find_one({"username": username})
        if not user:
            raise HTTPException(status_code=404, detail="User not found")
        
        response_uuid = user["uuid"].replace("-", "")
        
        textures = {}
        if user.get("skin_url"):
            textures["SKIN"] = {"url": user["skin_url"]}
        if user.get("cloak_url"):
            textures["CAPE"] = {"url": user["cloak_url"]}
        
        textures_value = base64.b64encode(json.dumps({
            "timestamp": int(datetime.now().timestamp()),
            "profileId": response_uuid,
            "profileName": username,
            "textures": textures
        }).encode()).decode()

        return {
            "id": response_uuid,
            "name": username,
            "properties": [{
                "name": "textures",
                "value": textures_value
            }] if textures else []
        }