discord-easy-dashboard/routes/auth.js

💻 Complete framework to facilitate the creation of dashboard using discord.js and express

const { Router } = require('express');
const CheckAuth = (req, res, next) => req.session.user ? next() : res.status(401).redirect('/auth/login');
const btoa = require('btoa');
const delay = (ms) => new Promise((resolve) => setTimeout(resolve, ms));

const Auth = Router()
	.get('/login', async (req, res) => {
		if (req.query.code) {
			/* Obtain token - used to fetch user guilds and user informations */
			const params = new URLSearchParams();
			params.set('grant_type', 'authorization_code');
			params.set('code', req.query.code);
			params.set(
				'redirect_uri',
				`${req.dashboardConfig.baseUrl}${req.dashboardConfig.noPortIncallbackUrl ? '' : ':' + req.dashboardConfig.port}/auth/login`,
			);

			let response = await fetch('https://discord.com/api/v10/oauth2/token', {
				method: 'POST',
				body: params.toString(),
				headers: {
					Authorization: `Basic ${btoa(
						`${req.client.user.id}:${req.dashboardConfig.secret}`,
					)}`,
					'Content-Type': 'application/x-www-form-urlencoded',
				},
			});

			// Fetch tokens (used to fetch user informations)
			const tokens = await response.json();
			// If the code isn't valid

			if (tokens.error || !tokens.access_token) return res.redirect('/auth/login');
			const userData = {
				infos: null,
				guilds: null,
			};
			while (!userData.infos || !userData.guilds) {
				/* User infos */
				if (!userData.infos) {
					response = await fetch('https://discord.com/api/v10/users/@me', {
						method: 'GET',
						headers: { Authorization: `Bearer ${tokens.access_token}` },
					});
					const json = await response.json();
					if (json.retry_after) await delay(json.retry_after);
					else userData.infos = json;
				}
				/* User guilds */
				if (!userData.guilds) {
					response = await fetch('https://discord.com/api/v10/users/@me/guilds', {
						method: 'GET',
						headers: { Authorization: `Bearer ${tokens.access_token}` },
					});
					const json = await response.json();
					if (json.retry_after) await delay(json.retry_after);
					else userData.guilds = json;
				}
			}

			// Update session
			req.session.user = Object.assign(userData.infos, {
				guilds: Object.values(userData.guilds),
			});

			req.dashboardEmit('newUser', req.session.user);
			res.status(200).redirect('/');
		}
		else {
			res.redirect(`https://discord.com/api/v10/oauth2/authorize?client_id=${req.client?.user?.id}&scope=identify%20guilds&response_type=code&redirect_uri=${encodeURIComponent(`${req.dashboardConfig.baseUrl}${req.dashboardConfig.noPortIncallbackUrl ? '' : ':' + req.dashboardConfig.port}/auth/login`)}`);
		}
	})
	.get('/logout', [CheckAuth], function(req, res) {
		req.session.destroy();
		res.status(200).redirect('/');
	});
module.exports.Router = Auth;

module.exports.name = '/auth';