wip: tracking

Author: GalvinPython

api/src/db/queries/tracking.ts

@@ -0,0 +1,33 @@
+import { CronJob } from "cron";
+
+let usersToUpdate: Record<string, string[]> = {};
+
+export async function addUserToTrackingData(userId: string, guildId: string) : Promise<void> {
+	console.log("Adding user to tracking data:", userId, guildId);
+	if (!usersToUpdate[guildId]) {
+		usersToUpdate[guildId] = [];
+	}
+	if (!usersToUpdate[guildId].includes(userId)) {
+		usersToUpdate[guildId].push(userId);
+	}
+	return;
+}
+
+async function doTrackingJob() {
+	const usersToUpdateTemp = { ...usersToUpdate };
+	usersToUpdate = {};
+	if (!Object.keys(usersToUpdateTemp).length) {
+		console.log("No users to update!");
+		return;
+	}
+	const guildIds = Object.keys(usersToUpdateTemp);
+	for (const guildId of guildIds) {
+		const userIds = usersToUpdateTemp[guildId];
+		for (const userId of userIds) {
+			console.log(userId, guildId);
+		}
+	}
+}
+
+const trackingJob = new CronJob("*/5 * * * * *", doTrackingJob);
+trackingJob.start();

api/src/index.ts

@@ -1,6 +1,6 @@
 import express, { type NextFunction, type Request, type Response } from "express";
 import cors from "cors";
-import { getBotInfo, getGuild, getUser, getUsers, initTables, pool, updateGuild, enableUpdates, disableUpdates, setCooldown, setUpdatesChannel, setXP, setLevel, removeGuild, removeUser, getAllServersWithUpdatesEnabled } from "./db";
+import { getBotInfo, getGuild, getUser, getUsers, initTables, pool, updateGuild, enableUpdates, disableUpdates, setCooldown, setUpdatesChannel, setXP, setLevel, removeGuild, removeUser, getAllServersWithUpdatesEnabled, addUserToTrackingData } from "./db";
 
 const app = express();
 const PORT = 18103;
@@ -478,6 +478,10 @@ app.post("/admin/:action/:guild/:target", authMiddleware, async (req, res) => {
 					return res.status(500).json({ message: "Internal server error" });
 			}
 		}
+		case "tracking": {
+			await addUserToTrackingData(target, guild);
+			break;
+		}
 		default:
 			return res.status(400).json({ message: "Illegal request" });
 	}
@@ -739,3 +743,7 @@ async function syncFromLurkr(guild: string) {
 	}
 }
 //#endregion
+
+//#region Tracking
+
+//#endregion

bot/src/events/messageCreate.ts

@@ -1,6 +1,6 @@
 import { Message } from 'discord.js';
 import client from '../index';
-import { getCooldown, makePOSTRequest, updateGuildInfo } from '../utils/requestAPI';
+import { addUserToTrackingData, getCooldown, makePOSTRequest, updateGuildInfo } from '../utils/requestAPI';
 
 const cooldowns = new Map<string, number>();
 
@@ -24,4 +24,6 @@ client.on('messageCreate', async (message: Message) => {
 	const guildIcon = message.guild?.iconURL() ?? 'https://cdn.discordapp.com/embed/avatars/0.png';
 	const guildMembers = message.guild?.memberCount;
 	await updateGuildInfo(message.guildId as string, guildName as string, guildIcon as string, guildMembers as number);
+
+	await addUserToTrackingData(message.author.id, message.guildId as string);
 });

bot/src/utils/requestAPI.ts

@@ -110,6 +110,16 @@ export async function getDBSize() {
 	return response.json();
 }
 
+export async function addUserToTrackingData(userId: string, guildId: string) {
+	await fetch(`http://localhost:18103/admin/tracking/${guildId}/${userId}`, {
+		"headers": {
+			'Content-Type': 'application/json',
+			'Authorization': process.env.AUTH as string,
+		},
+		"method": "POST"
+	});
+}
+
 //#region Roles
 export async function getRoles(guild: string) {
 	const response = await fetch(`http://localhost:18103/admin/roles/${guild}/get`, {

package.json

@@ -15,7 +15,7 @@
   },
   "devDependencies": {
     "@eslint/js": "^9.7.0",
-    "@types/bun": "latest",
+    "@types/bun": "1.1.6",
     "@types/eslint__js": "^8.42.3",
     "dotenv-cli": "^7.4.2",
 	"eslint": "^8.56.0",