path: root/authentication/utils.py

import time
import dotenv
import os
import requests
import secrets
from authentication.models import User
from user_profile.models import UserPreferences
from pypresence import Client

dotenv.load_dotenv()

user_clients = {}

def get_or_create_client(user):
    if user.id not in user_clients:
        client_id = os.environ.get("DISCORD_CLIENT_ID")
        RPC = Client(client_id)
        RPC.start()
        RPC.authenticate(user.discord_access_token)
        user_clients[user.id] = RPC
    return user_clients[user.id]

def generate_mal_code_challenge():
    # 128 bytes PKCE code challenge
    token = secrets.token_urlsafe(128)
    return token[:128]

def get_mal_redirect_uri():
    mal_client_id = os.environ.get("MAL_CLIENT_ID")
    # mal_client_secret = os.environ.get("MAL_CLIENT_SECRET")
    mal_redirect_uri = os.environ.get("MAL_REDIRECT_URI")

    code_challenge = generate_mal_code_challenge()
    code_challenge_method = "plain"

    redirect_uri = f"https://myanimelist.net/v1/oauth2/authorize?response_type=code&client_id={mal_client_id}&code_challenge={code_challenge}&code_challenge_method={code_challenge_method}&redirect_uri={mal_redirect_uri}&state={code_challenge}"

    return redirect_uri, code_challenge

def get_redirect_uri():
    # Only Authenticated Users who are in our Discord Server can access the website
    discord_client_id = os.environ.get("DISCORD_CLIENT_ID")
    discord_redirect_uri = os.environ.get("DISCORD_REDIRECT_URI")
    discord_scope = "identify guilds guilds.members.read rpc"
    redirect_uri = f"https://discord.com/oauth2/authorize?client_id={discord_client_id}&response_type=code&redirect_uri={discord_redirect_uri}&scope={discord_scope}"
    return redirect_uri

def update_discord_rpc(user, anime_title, episode, time_watched):
    if not user.discord_access_token:
            return  # Skip if user hasn't connected Discord

    try:
        RPC = get_or_create_client(user)
        RPC.set_activity(
            state=episode,
            activity_type=3,
            details=anime_title,
            start=int(time.time()) - time_watched,
            large_image="yugen",
            large_text="Yugen: The Streaming Service for the Elite",
        )
        
        print(f"Successfully updated Discord RPC for user {user.username}: Watching {anime_title} {episode} for {time_watched} seconds")
    except Exception as e:
        print(f"Failed to update Discord RPC for user {user.username}: {str(e)}")

def clear_discord_rpc(user):
    if not user.discord_access_token:
        return  # Skip if user hasn't connected Discord

    if user.id in user_clients:
        try:
            RPC = user_clients[user.id]
            RPC.clear_activity()
            RPC.close()
            del user_clients[user.id]
            print(f"Successfully cleared Discord RPC for user {user.username}")
        except Exception as e:
            print(f"Failed to clear Discord RPC for user {user.username}: {str(e)}")

def exchange_mal_code(code, code_verifier):
    mal_client_id = os.environ.get("MAL_CLIENT_ID")
    mal_client_secret = os.environ.get("MAL_CLIENT_SECRET")
    mal_redirect_uri = os.environ.get("MAL_REDIRECT_URI")

    data = {
        "client_id": mal_client_id,
        "client_secret": mal_client_secret,
        "grant_type": "authorization_code",
        "code": code,
        "code_verifier": code_verifier,
        "grant_type": "authorization_code",
        "redirect_uri": mal_redirect_uri,
    }

    headers = {"Content-Type": "application/x-www-form-urlencoded"}

    response = requests.post(
        "https://myanimelist.net/v1/oauth2/token", data=data, headers=headers
    )
    return response.json()

def exchange_code(code):
    discord_client_id = os.environ.get("DISCORD_CLIENT_ID")
    discord_client_secret = os.environ.get("DISCORD_CLIENT_SECRET")
    discord_redirect_uri = os.environ.get("DISCORD_REDIRECT_URI")

    data = {
        "client_id": discord_client_id,
        "client_secret": discord_client_secret,
        "grant_type": "authorization_code",
        "code": code,
        "redirect_uri": discord_redirect_uri,
        "scope": "identify guilds guilds.members.read rpc",
    }

    headers = {"Content-Type": "application/x-www-form-urlencoded"}

    response = requests.post(
        "https://discord.com/api/oauth2/token", data=data, headers=headers
    )
    return response.json()

def get_user_mal_list(access_token, limit=10, offset=0, filter=""):
    if not access_token or access_token == "":
        return [], None, None

    base_url = f"https://api.myanimelist.net/v2/users/@me/animelist?limit={limit}&offset={offset}&fields=my_list_status,alternative_titles,anime&sort=list_updated_at"

    if filter:
        base_url += f"&status={filter}"

    headers = {"Authorization": f"Bearer {access_token}"}
    response = requests.get(base_url, headers=headers)

    if response.status_code == 200:
        data = response.json()
        user_anime_list = data["data"]

        if data["data"] and "paging" in data:
            if "next" in data["paging"]:
                page_next = data["paging"]["next"]
            else:
                page_next = None
            
            if "previous" in data["paging"]:
                page_previous = data["paging"]["previous"]
            else:
                page_previous = None
        else:
            page_next = None
            page_previous = None
    else:
        user_anime_list = []
        page_next = None
        page_previous = None
            
    return user_anime_list, page_previous, page_next

def get_single_anime_mal(access_token, mal_id):
    if not access_token or access_token == "":
        return None

    base_url = f"https://api.myanimelist.net/v2/anime/{mal_id}?fields=alternative_titles,average_episode_duration,broadcast,created_at,end_date,genres,id,main_picture,media_type,nsfw,num_episodes,num_favorites,num_list_users,num_scoring_users,popularity,rank,start_date,start_season,status,synopsis,source,studios,title,updated_at,my_list_status,background,related_anime,rating,mean"

    headers = {"Authorization": f"Bearer {access_token}"}
    response = requests.get(base_url, headers=headers)

    if response.status_code == 200:
        data = response.json()
        return data
    else:
        print(response.json(), "MAL API Error")
        return None

def get_discord_user(access_token, token_type):
    guilds = requests.get(
        "https://discord.com/api/users/@me/guilds",
        headers={"Authorization": f"{token_type} {access_token}"},
    ).json()

    authorized_guilds = os.environ.get("DISCORD_AUTHORIZED_GUILDS").split(",")
    preferred_guild = os.environ.get("DISCORD_PREFERRED_GUILD")
    user = {}
    user["is_authorized"] = False
    if isinstance(guilds, list):
        for guild in guilds:
            if guild["id"] in authorized_guilds:
                member = requests.get(
                    f"https://discord.com/api/users/@me/guilds/{guild['id']}/member",
                    headers={"Authorization": f"{token_type} {access_token}"},
                ).json()
                
                user = member["user"]
                user["is_authorized"] = True
                user["rate_limited"] = False
                user["guild_name"] = member["nick"] if member["nick"] is not None else ""

                if preferred_guild == guild["id"]:
                    print("Preferred Guild Found. Updating User Nickname to", member["nick"])
                    user["guild_name"] = member["nick"] if member["nick"] is not None else user["username"]
                    break
                
    else:
        # maybe user is rate limited
        # {'message': 'You are being rate limited.', 'retry_after': 0.3, 'global': False}
        if guilds.get("message") == "You are being rate limited.":
            user["is_authorized"] = True # Temporarily authorized
            user["rate_limited"] = True
        else:
            user["is_authorized"] = False
            user["rate_limited"] = False

    user["banner"] = user["banner"] if "banner" in user else ""
    return user

def update_user_discord_info(user_data):
    user = User.objects.get(discord_id=user_data["id"])
    print("Updating User Info for", user.username)
    user.discord_username = user_data["username"]
    user.discord_avatar = user_data["avatar"]
    user.discord_banner = user_data["banner"]
    user.discord_global_name = user_data["global_name"]
    user.discord_guild_name = user_data["guild_name"]
    user.save()

def authenticate_user(exchange_response):
    access_token = exchange_response.get("access_token")
    token_type = exchange_response.get("token_type")
    refresh_token = exchange_response.get("refresh_token")

    discord_user = get_discord_user(access_token, token_type)

    if discord_user["is_authorized"]:
        user, created = User.objects.update_or_create(
            discord_id=discord_user["id"],
            defaults={
                "username": discord_user["username"],
                "discord_id": discord_user["id"],
                "discord_access_token": access_token,
                "discord_refresh_token": refresh_token,
                "discord_token_type": token_type,
                "discord_username": discord_user["username"],
                "discord_avatar": discord_user["avatar"] if discord_user["avatar"] is not None else "",
                "discord_banner": discord_user["banner"] if discord_user["banner"] is not None else "",
                "discord_global_name": discord_user["global_name"] if discord_user["global_name"] is not None else "",
                "discord_guild_name": discord_user["guild_name"] if discord_user["guild_name"] is not None else "",
            },
        )

        if not created:
            user.username = discord_user["username"]
            user.discord_access_token = access_token
            user.discord_refresh_token = refresh_token
            user.discord_token_type = token_type
            user.discord_username = discord_user["username"]
            user.discord_avatar = discord_user["avatar"]
            user.discord_banner = (
                discord_user["banner"] if discord_user["banner"] is not None else ""
            )
            user.discord_global_name = discord_user["global_name"]
            user.discord_guild_name = discord_user["guild_name"]
            user.save()

        UserPreferences.objects.get_or_create(user=user)

        return user

    return None