mirror of
https://github.com/NovaOSS/nova-api.git
synced 2024-11-29 10:03:56 +01:00
75 lines
1.9 KiB
Python
75 lines
1.9 KiB
Python
"""User management."""
|
|
|
|
import os
|
|
import json
|
|
import fastapi
|
|
|
|
from db import users
|
|
|
|
from dhooks import Webhook, Embed
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
router = fastapi.APIRouter(tags=['core'])
|
|
|
|
async def check_core_auth(request):
|
|
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_error = await check_core_auth(incoming_request)
|
|
|
|
if auth_error:
|
|
return auth_error
|
|
|
|
user = await users.by_discord_id(discord_id)
|
|
|
|
if not user:
|
|
return fastapi.Response(status_code=404, content='User not found.')
|
|
|
|
return user
|
|
|
|
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=user['_id'], inline=False)
|
|
embed.add_field(name='Discord', value=user['auth']['discord'])
|
|
embed.add_field(name='Github', value=user['auth']['github'])
|
|
|
|
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.')
|
|
|
|
user = await users.create(discord_id)
|
|
new_user_webhook(user)
|
|
|
|
return user
|
|
|
|
if __name__ == '__main__':
|
|
new_user_webhook({
|
|
'_id': 'JUST_A_TEST_IGNORE_ME',
|
|
'auth': {
|
|
'discord': 123,
|
|
'github': 'abc'
|
|
}
|
|
})
|