/* * Vencord, a Discord client mod * Copyright (c) 2023 Vendicated and contributors * SPDX-License-Identifier: GPL-3.0-or-later */ import { DataStore } from "@api/index"; import { Logger } from "@utils/Logger"; import { openModal } from "@utils/modal"; import { findByPropsLazy } from "@webpack"; import { showToast, Toasts, UserStore } from "@webpack/common"; import { ReviewDBAuth } from "./entities"; const DATA_STORE_KEY = "rdb-auth"; const { OAuth2AuthorizeModal } = findByPropsLazy("OAuth2AuthorizeModal"); export let Auth: ReviewDBAuth = {}; export async function initAuth() { Auth = await getAuth() ?? {}; } export async function getAuth(): Promise { const auth = await DataStore.get(DATA_STORE_KEY); return auth?.[UserStore.getCurrentUser()?.id]; } export async function getToken() { const auth = await getAuth(); return auth?.token; } export async function updateAuth(newAuth: ReviewDBAuth) { return DataStore.update(DATA_STORE_KEY, auth => { auth ??= {}; Auth = auth[UserStore.getCurrentUser().id] ??= {}; if (newAuth.token) Auth.token = newAuth.token; if (newAuth.user) Auth.user = newAuth.user; return auth; }); } export function authorize(callback?: any) { openModal(props => { try { const url = new URL(response.location); url.searchParams.append("clientMod", "vencord"); const res = await fetch(url, { headers: new Headers({ Accept: "application/json" }) }); const { token, success } = await res.json(); if (success) { updateAuth({ token }); showToast("Successfully logged in!", Toasts.Type.SUCCESS); callback?.(); } else if (res.status === 1) { showToast("An Error occurred while logging in.", Toasts.Type.FAILURE); } } catch (e) { new Logger("ReviewDB").error("Failed to authorize", e); } }} /> ); }