selmer-bot-website/main.js

// const { fetch } = require('undici');
// const express = require('express');
// const fetch = require('node-fetch');
// const btoa = require('btoa');
// const { clientId, clientSecret, port } = require('./config.json');
import express from 'express'
import fetch from 'node-fetch'
import { MongoClient, ServerApiVersion } from 'mongodb'
import { v4 as uuidv4 } from 'uuid'
import { validate as uuidValidate } from 'uuid';

//Bot section (PLACE IN ENV)
import { Client, Intents } from 'discord.js';
const token = process.env.token;
const clientId = process.env.clientId;
const clientSecret = process.env.clientSecret;
const port = process.env.PORT || 3000;
const mongouri = process.env.mongouri;
const client = new MongoClient(mongouri, { useNewUrlParser: true, useUnifiedTopology: true, serverApi: ServerApiVersion.v1 });
const connection = client.connect();


const bot = new Client({ 
    intents: [
        Intents.FLAGS.GUILDS,
        Intents.FLAGS.GUILD_MESSAGES,
        Intents.FLAGS.GUILD_MESSAGE_REACTIONS,
        Intents.FLAGS.GUILD_VOICE_STATES,
        Intents.FLAGS.GUILD_EMOJIS_AND_STICKERS,
        Intents.FLAGS.GUILD_PRESENCES,
        Intents.FLAGS.GUILD_MEMBERS,
        Intents.FLAGS.DIRECT_MESSAGES,
        Intents.FLAGS.DIRECT_MESSAGE_REACTIONS,
        Intents.FLAGS.DIRECT_MESSAGE_TYPING,
    ],
    partials: [ 'CHANNEL' ]
});

bot.on('ready', async () => {console.log('BOT RUNNING'); });
bot.login(token);


async function getJSONResponse(body) {
	let fullBody = '';

	for await (const data of body) {
		fullBody += data.toString();
	}
	return JSON.parse(fullBody);
}

const app = express();
// app.use(express.json());
app.use(express.static('/assets'));


app.post('/user', async(request, response) => {
	const guilds = JSON.parse(request.headers.guilds);

	const id = request.headers.userid;

	for (let i = 0; i < guilds.length; i ++) {
		const guild = bot.guilds.cache.get(guilds[i].id);
		
		if (guild && guild.ownerId == id) {
			guilds[i].inServer = true;
		} else {
			guilds[i].inServer = false;
		}
	}

	//Add the guilds to the session data
	connection.then((client) => {
		const sessionId = uuidv4();

		const dbo = client.db('main').collection('sessions');
		dbo.insertOne({ sessionId: sessionId, userId: id, guilds: JSON.stringify(guilds) });

		response.send(sessionId);
	});
});


app.post('/logout', async (req, res) => {
	connection.then((client) => {
		const sessionId = req.headers.sessionid;
		const dbo = client.db('main').collection('sessions');
		dbo.deleteOne({ sessionId: sessionId });
	});

	res.sendStatus(200);
});


app.post('/getSessionInfo', async (req, res) => {
	const session = req.headers.session;
	if (session) {
		if (uuidValidate(session)) {
			connection.then((client) => {
				const dbo = client.db('main').collection('sessions');
				dbo.findOne({ sessionId: session }).then((doc) => {
					res.send(doc);
				})
			})
		}
	}
});


app.get('/myGuilds.html', async (req, res) => {
	return res.sendFile('myGuilds.html', { root: '.' });
});


app.post('/getServer', async (req, res) => {
	const id = req.headers.servernumber;
	
	//Get info
	connection.then((client) => {
		const dbo = client.db(id).collection('SETUP');
		dbo.find().toArray(async (err, docs) => {
	
			const m = new Map();
			m.set('Id', id);
			await Promise.all(docs.map(async (doc) => {
				m.set(doc._id, doc);
	
			})).then(() => { res.send(JSON.stringify(Object.fromEntries(m))); })
		})
	});
	
	// return res.sendFile('myGuilds.html', { root: '.' });
})


app.get('/getChannels', async (req, res) => {
	const id = req.headers.servernumber;
	const channels = bot.guilds.cache.get(id).channels.cache;
	
	const arr = {text: [], voice: []};

	channels.forEach((channel) => {
		const type = channel.type;
		if (type == 'GUILD_TEXT') {
			arr.text.push({ name: channel.name, id: channel.id });
		} else if (type == 'GUILD_VOICE') {
			arr.voice.push({ name: channel.name, id: channel.id });
		}
	});

	res.send(arr);
});


app.get('/getCal', async (req, res) => {
	const userId = req.headers.userid || false;
	const guildId = req.headers.guildid || false;

	connection.then((client) => {
		var times;

		const dbo = client.db('main').collection('reminderKeys');
		dbo.findOne({$or: [ {userId: userId}, {userId: guildId} ]}).then((doc) => {
			if (!doc) { return res.send([[], []]); }
			times = doc.times;
			let tbo = client.db('main').collection('reminders');

			// { $where: function() { return (times.indexOf(this.time) != -1 && this.userId == userId); }}
			// tbo.find({time: {$in: times}}).toArray((err, docs) => { console.log("A", docs); throw 1; });
			
			tbo.find({time: {$in: times}}).toArray((err, docs) => {

				//There's gotta be a better way
				let newdoc = [];
				for (let i = 0; i < docs.length; i ++) {
					newdoc.push({});
					for (let j in docs[i]) {
						if (!isNaN(j) && (docs[i][j].userId == userId || docs[i][j].guildId == guildId)) {
							// console.log(`${docs[i][j].userId} == ${userId}`, `${docs[i][j].guildId} == ${guildId}`);
							newdoc[i][j] = docs[i][j];
						}
					}

					newdoc[i]._id = docs[i]._id;
					newdoc[i].time = docs[i].time;
					newdoc[i].amt = docs[i].amt;

					//If there's nothing on that date, skip
					// if (newdoc[i].amt == 0) { console.log(newdoc[i]); }
				}
				res.send(JSON.stringify([times, newdoc]));
			});
		});
	})
});

app.get('/dashboard.html', async (req, res) => {
	return res.sendFile('dashboard.html', { root: '.' });
});


app.get('/premium.html', async (req, res) => {
	return res.sendFile('premium.html', { root: '.' });
});

app.get('/index.html', async (req, res) => {
	return res.sendFile('index.html', { root: '.' });
});

app.get('/calEvent.html', async (req, res) => {
	return res.sendFile('calEvent.html', { root: '.' });
});

app.get('/newCalEvent.html', async (req, res) => {
	return res.sendFile('newCalEvent.html', { root: '.' });
});

app.get('/calendar.html', async (req, res) => {
	return res.sendFile('calendar.html', { root: '.' });
})


//Stripe stuff
app.get('/.well-known/apple-developer-merchantid-domain-association', async (req, res) => {
	return res.sendFile('apple-developer-merchantid-domain-association', { root: '.' });
});

app.post('/verifypremium', async (req, res) => {
	if (req.headers.userid) {
		const id = req.headers.userid;

		connection.then((client) => {
			const dbo = client.db('main').collection('authorized');
			dbo.findOne({ discordID: id }).then((doc) => {
				if (doc != undefined) {
					res.sendStatus(200);
				} else {
					res.sendStatus(401);
				}
			}).catch((err) => {
				console.error(err); return res.sendStatus(500);
			})
		});
	}
});

app.post('/sendData', async (req, res) => {
	try {
		if (req.headers.reminders) {
			connection.then((client) => {

				const delObjKeys = JSON.parse(req.headers.delobjkeys);
				// const userId = req.headers.userid;
				const dbo = client.db('main').collection('reminders');
				const kbo = client.db('main').collection('reminderKeys');

				//Update the Time object
				for (var i in delObjKeys) {
					var obj = delObjKeys[i];

					dbo.findOne({ time: obj.time }).then((doc) => {
						if (doc) {
							// kbo.findOne({ 'userId': doc[obj.eventInd].userId }).then((doc) => {
							// 	console.log(doc);
							// }); return console.log(obj.time);
							doc.amt --;

							// console.log(doc[obj.eventInd]); return;
							var searchId;
							if (doc[obj.eventInd].userId != null) {
								searchId = doc[obj.eventInd].userId;
							} else if (doc[obj.eventInd].guildId != null) {
								searchId = doc[obj.eventInd].guildId;
							}

							kbo.findOne({ 'userId': searchId }).then((kdoc) => {
								if ((kdoc.times.length - 1) > 0) {
									kbo.updateOne({ 'userId': searchId }, {$pull: { times: obj.time }});
								} else {
									kbo.deleteOne({ 'userId': searchId });
								}
							});

							//If there's nothing left in the list, delete it
							if (doc.amt > 0) {
								delete doc[obj.eventInd];
								// dbo.findOneAndUpdate({ time: obj.time }, newdoc);
								dbo.findOneAndReplace({ time: obj.time }, doc);
							} else {
								dbo.deleteOne({ time: obj.time });
							}
						} // else { console.log("NONE", obj.time ); }

					}).catch((err) =>  {
						console.log("ERR");
						console.error(err)
						res.sendStatus(500);
					});

					// kbo.findOneAndUpdate({ userId: userId }, { $pull: {}})
				}
			}).then(() => {
				res.sendStatus(200);
			});
		} else {
			const pref = JSON.parse(req.headers.serversettings);

			connection.then(async (client) => {
				const dbo = client.db(pref.Id).collection('SETUP');

				await dbo.updateOne({ _id: 'WELCOME' }, {$set: { welcomechannel: pref.WELCOME.welcomechannel, welcomemessage: pref.WELCOME.welcomemessage }});
				await dbo.updateOne({ _id: 'LOG' }, {$set: { keepLogs: pref.LOG.keepLogs, logchannel: pref.LOG.logchannel, severity: pref.LOG.severity }});
			}).then(() => { res.send("DONE"); })
			.catch((err) => {
				console.error(err);
				res.send("FAILED");
			});
		}
	} catch (err) {
		console.error(err);
		res.send("FAILED");
	}
});


//Reminder format = { time: 1212122, event: { guildId: "930148608400035860", userId: "12", name: "Some Generic Name", description: "Some description", offset: "15", link: "https://www.example.com" } }
app.post('/newCalEvent', async (req, res) => {
	if (req.headers.newcalevent) {
		try {
			const obj = JSON.parse(req.headers.newcalevent);
			// console.log(obj.time, typeof obj.time); return;
			var searchId;
			if (obj.event.userId != null) {
				searchId = obj.event.userId;
			} else if (obj.event.guildId != null) {
				searchId = obj.event.guildId;
			} else { return res.sendStatus(400); }

			connection.then((client) => {
				// Update the Key object first to check if the time is already there
				const kbo = client.db('main').collection('reminderKeys');
				kbo.findOne({ 'userId': searchId }).then((doc) => {

					if (doc) {
						if (doc.times.indexOf(obj.time) == -1) {
							kbo.updateOne({ 'userId': searchId }, { $push: { times: obj.time } })
							.catch((err) => { console.error(err); res.sendStatus(500); });
						} else {
							return res.sendStatus(409);
						}
					} else {
						doc = { 'userId': searchId, times: [obj.time] }
						kbo.insertOne(doc);
					}


					//Update the Time object
					const dbo = client.db('main').collection('reminders');
					dbo.findOne({ time: obj.time }).then((doc) => {
						let n = 0;
						if (doc) {
							n = doc.amt;
							doc.amt ++;

							doc[`${n}`] = obj.event;
							dbo.findOneAndReplace({ time: obj.time }, doc);
						} else {
							const d = new Date(Number(obj.time));
							doc = { "0": obj.event, "time": obj.time, "month": d.getMonth(), "amt": 1 }; //Month used for clearing when the calendar month begins (maybe modify the garbage collection with an `else if (day == 1? clear last month)` )
							dbo.insertOne(doc);
						}

						res.sendStatus(200);
					}).catch((err) =>  {
						console.log("ERR");
						console.error(err);
						res.sendStatus(500);
					});
				}).catch((err) =>  {
					console.log("ERR");
					console.error(err);
					res.sendStatus(500);
				});

				// //Make sure there hasn't been an error (header not yet sent);
				// if (res.headersSent) { return console.log("ERROR"); }
			});
		} catch (err) {
			console.error(err);
			return res.sendStatus(500);
		}
	}
});



app.get('/', async ({ query }, response) => {
	const { code } = query;

	if (code) {
		try {
			await fetch('https://discord.com/api/oauth2/token', {
				method: 'POST',
				body: new URLSearchParams({
					client_id: clientId,
					client_secret: clientSecret,
					code,
					grant_type: 'authorization_code',
					redirect_uri: `http://www.selmerbot.com/`,
					scope: 'identify',
				}),
				headers: {
					'Content-Type': 'application/x-www-form-urlencoded',
				},
			}).then(async response => {
				const oauthData = await getJSONResponse(response.body);
				// console.log(oauthData);
			}).catch(err => { console.error(err); });
		} catch (error) {
			// NOTE: An unauthorized token will not throw an error
			// tokenResponseData.statusCode will be 401
			console.error(error);
		}
	}

	return response.sendFile('index.html', { root: '.' });
});


app.get("*",(req,res) => {
    res.sendFile("404.html", {root: '.'});
})

app.listen(port, () => console.log(`App listening on port ${port}`));