Add heatmap.png generator

src/db.ts

@@ -10,7 +10,7 @@ export const dbClient = await new Client().connect({
 	password: config.db.password,
 });
 
-const weekDays = ['sunday', 'monday', 'tuesday', 'wednesday', 'thursday', 'friday', 'saturday'];
+export const weekDays = ['sunday', 'monday', 'tuesday', 'wednesday', 'thursday', 'friday', 'saturday'];
 
 export const queries = {
 	insertRollLogCmd: (api: number, error: number) => `INSERT INTO roll_log(input,result,resultid,api,error) values(?,?,?,${api},${error})`,

src/intervals.ts

@@ -8,12 +8,14 @@ import {
 	// Discordeno deps
 	cache,
 	cacheHandlers,
+	// imagescript
+	is,
 	// Log4Deno deps
 	log,
 	LT,
 } from '../deps.ts';
 import { PastCommandCount } from './mod.d.ts';
-import { dbClient } from './db.ts';
+import { dbClient, weekDays } from './db.ts';
 import utils from './utils.ts';
 import config from '../config.ts';
 
@@ -41,7 +43,7 @@ const getRandomStatus = async (): Promise<string> => {
 	return status;
 };
 
-// updateListStatistics(bot ID, current guild count) returns nothing
+// updateListStatistics(bot ID, current guild count) returns nothing, posts to botlists
 // Sends the current server count to all bot list sites we are listed on
 const updateListStatistics = (botID: bigint, serverCount: number): void => {
 	config.botLists.forEach(async (e) => {
@@ -64,11 +66,11 @@ const updateListStatistics = (botID: bigint, serverCount: number): void => {
 // Keep one week of data
 const hoursToKeep = 7 * 24;
 const previousHours: Array<Array<PastCommandCount>> = [];
-// updateHourlyRates() returns nothing
+// updateHourlyRates() returns nothing, updates DB directly
 // Updates the hourlyRate for command usage
 const updateHourlyRates = async () => {
 	try {
-		const newestHour = await dbClient.query(`SELECT command, count FROM command_cnt ORDER BY command;`).catch((e) => utils.commonLoggers.dbError('intervals.ts:71', 'query', e));
+		const newestHour = await dbClient.query('SELECT command, count FROM command_cnt ORDER BY command;').catch((e) => utils.commonLoggers.dbError('intervals.ts:71', 'query', e));
 		previousHours.push(newestHour);
 		if (previousHours.length > 1) {
 			const oldestHour = previousHours[0];
@@ -99,4 +101,104 @@ const updateHourlyRates = async () => {
 	}
 };
 
-export default { getRandomStatus, updateListStatistics, updateHourlyRates };
+// getPercentOfRange(min, max, val) returns number
+// Gets a percent value of where val lies in the min-max range
+const getPercentOfRange = (minVal: number, maxVal: number, val: number): number => {
+	const localMax = maxVal - minVal;
+	const localVal = val - minVal;
+
+	return localVal / localMax;
+};
+
+// Pixel locations in heatmap-base.png, pixel locations are 0 based
+// dayPixels holds the left and right (AKA X Coord) pixel locations for each col (ex: [leftPX, rightPX])
+const dayPixels: Array<Array<number>> = [
+	[72, 159],
+	[163, 260],
+	[264, 359],
+	[363, 497],
+	[501, 608],
+	[612, 686],
+	[690, 800],
+];
+// hourPixels holds the top and bottom (AKA Y Coord) pixel locations for each row (ex: [topPX, botPX])
+const hourPixels: Array<Array<number>> = [
+	[29, 49],
+	[51, 72],
+	[74, 95],
+	[97, 118],
+	[120, 141],
+	[143, 164],
+	[166, 187],
+	[189, 209],
+	[211, 232],
+	[234, 254],
+	[256, 277],
+	[279, 299],
+	[301, 322],
+	[324, 345],
+	[347, 368],
+	[370, 391],
+	[393, 413],
+	[415, 436],
+	[438, 459],
+	[461, 482],
+	[484, 505],
+	[507, 528],
+	[530, 550],
+	[552, 572],
+];
+// updateHeatmap() returns nothing, creates new heatmap.png
+// Updates the heatmap image with latest data from the db
+const updateHeatmapPng = async () => {
+	const baseHeatmap = Deno.readFileSync('./src/endpoints/gets/heatmap-base.png');
+	const heatmap = await is.decode(baseHeatmap);
+	if (heatmap instanceof is.Image) {
+		// Get latest data from DB
+		const heatmapData = await dbClient.query('SELECT * FROM roll_time_heatmap ORDER BY hour;').catch((e) => utils.commonLoggers.dbError('intervals.ts:148', 'query', e));
+
+		let minRollCnt = Infinity;
+		let maxRollCnt = 0;
+		// determine min and max values
+		for (let hour = 0; hour < heatmapData.length; hour++) {
+			for (let day = 0; day < weekDays.length; day++) {
+				const rollCnt = heatmapData[hour][weekDays[day]];
+				log(LT.LOG, `updateHeatmapPng | finding min/max | min: ${minRollCnt} max: ${maxRollCnt} curr: ${rollCnt}`);
+				if (rollCnt > maxRollCnt) {
+					maxRollCnt = rollCnt;
+				}
+				if (rollCnt < minRollCnt) {
+					minRollCnt = rollCnt;
+				}
+			}
+		}
+
+		// Apply values to image
+		for (let hour = 0; hour < heatmapData.length; hour++) {
+			for (let day = 0; day < weekDays.length; day++) {
+				log(LT.LOG, `updateHeatmapPng | putting ${weekDays[day]} ${hour}:00 into image`);
+				const percent = getPercentOfRange(minRollCnt, maxRollCnt, heatmapData[hour][weekDays[day]]);
+				heatmap.drawBox(
+					dayPixels[day][0] + 1,
+					hourPixels[hour][0] + 1,
+					dayPixels[day][1] - dayPixels[day][0] + 1,
+					hourPixels[hour][1] - hourPixels[hour][0] + 1,
+					is.Image.rgbToColor(
+						255 * (1 - percent),
+						255 * percent,
+						0,
+					),
+				);
+			}
+		}
+
+		Deno.writeFileSync('./src/endpoints/gets/heatmap.png', await heatmap.encode());
+	}
+};
+
+export default {
+	getRandomStatus,
+	updateListStatistics,
+	updateHourlyRates,
+	updateHeatmapPng,
+};