nova-api/api/core.py

"""User management."""

import os
import json
import fastapi

from db.users import UserManager

from dhooks import Webhook, Embed
from dotenv import load_dotenv

load_dotenv()
router = fastapi.APIRouter(tags=['core'])

async def check_core_auth(request):
    """
    
    ### Checks the request's auth
    Auth is taken from environment variable `CORE_API_KEY`

    """
    received_auth = request.headers.get('Authorization')

    if received_auth != os.getenv('CORE_API_KEY'):
        return fastapi.Response(status_code=403, content='Invalid or no API key given.')

@router.get('/users')
async def get_users(discord_id: int, incoming_request: fastapi.Request):
    auth = await check_core_auth(incoming_request)
    if auth:
        return auth

    # Get user by discord ID
    manager = UserManager()
    user = await manager.user_by_discord_id(discord_id)
    if not user:
        return fastapi.Response(status_code=404, content='User not found.')

    return user

async def new_user_webhook(user: dict) -> None:    
    dhook = Webhook(os.getenv('DISCORD_WEBHOOK__USER_CREATED'))

    embed = Embed(
        description='New User',
        color=0x90ee90,
    )

    embed.add_field(name='ID', value=str(user['_id']), inline=False)
    embed.add_field(name='Discord', value=user['auth']['discord'] or '-')
    embed.add_field(name='Github', value=user['auth']['github'] or '-')

    dhook.send(embed=embed)

@router.post('/users')
async def create_user(incoming_request: fastapi.Request):
    auth_error = await check_core_auth(incoming_request)

    if auth_error:
        return auth_error

    try:
        payload = await incoming_request.json()
        discord_id = payload.get('discord_id')
    except (json.decoder.JSONDecodeError, AttributeError):
        return fastapi.Response(status_code=400, content='Invalid or no payload received.')
    
    # Create the user 
    manager = UserManager()
    user = await manager.create(discord_id)
    await new_user_webhook(user)

    return user

@router.put('/users')
async def update_user(incoming_request: fastapi.Request):
    auth_error = await check_core_auth(incoming_request)

    if auth_error:
        return auth_error

    try:
        payload = await incoming_request.json()
        discord_id = payload.get('discord_id')
        updates = payload.get('updates')
    except (json.decoder.JSONDecodeError, AttributeError):
        return fastapi.Response(status_code=400, content='Invalid or no payload received.')
    
    # Update the user 
    manager = UserManager()
    user = await manager.update_by_discord_id(discord_id, updates)

    return user