start adding metrics

package.json

@@ -8,6 +8,7 @@
     "aws-sdk": "^2.1565.0",
     "discord.js": "^14.13.0",
     "express": "^4.18.3",
+    "prom-client": "^15.1.2",
     "sqids": "^0.3.0",
     "zod": "^3.22.3"
   },

src/index.ts

@@ -12,6 +12,7 @@ const bot = new Powercord(env.DSC_TOKEN, {
 });
 export default bot;
 registerSentryButtons();
+import "./modules/metrics";
 
 import { finishPackOrder } from "./modules/pack"; // this file imports the bot, so it must be imported after the bot is created
 

src/modules/api.ts

@@ -3,6 +3,7 @@ import {
   handleIncomingApplication,
   submittedApplicationSchema,
 } from "./applications";
+import { register } from "prom-client";
 
 const api = express();
 api.use(express.json());
@@ -17,6 +18,12 @@ api.post("/api/applications/submit", (req, res) => {
   }
 });
 
+api.get("/metrics", async (req, res) => {
+  res
+    .header("Content-Type", register.contentType)
+    .send(await register.metrics());
+});
+
 api.get("/", (req, res) => {
   res.send(200);
 });

src/modules/deliver.ts

@@ -16,6 +16,7 @@ import {
   clearKitchenMessages,
   sendKitchenMessage,
 } from "../utils/kitchenChannels";
+import { updateProcessingOrders } from "./metrics";
 
 const sendDeliveryContent = async (
   o: order,
@@ -122,6 +123,7 @@ bot.registerButton(/order:(\d+):deliver/, async (interaction) => {
       invite: invite.url,
     },
   });
+  updateProcessingOrders(orderStatus.DELIVERING, order.id);
 
   await clearKitchenMessages(order.id);
   const deliveringActionRow =
@@ -191,6 +193,7 @@ bot.registerButton(/order:(\d+):complete/, async (interaction) => {
         status: orderStatus.DELIVERED,
       },
     });
+    updateProcessingOrders(orderStatus.DELIVERED, order.id);
   } catch (e) {
     return { success: false, message: "Failed to mark delivery finished" };
   }

src/modules/fill.ts

@@ -15,6 +15,7 @@ import {
   clearKitchenMessages,
   sendKitchenMessage,
 } from "../utils/kitchenChannels";
+import { updateProcessingOrders } from "./metrics";
 
 bot.registerButton("order:(\\d+):fill", async (interaction) => {
   const orderId = interaction.customId.split(":")[1];
@@ -42,6 +43,7 @@ bot.registerButton("order:(\\d+):fill", async (interaction) => {
         chefUsername: interaction.user.username,
       },
     });
+    updateProcessingOrders(orderStatus.FILLING, order.id);
   } catch (e) {
     return interaction.reply({
       content: "Failed to claim order",

src/modules/metrics.ts

@@ -0,0 +1,58 @@
+import { OrderStatus } from "aws-sdk/clients/outposts";
+import { Gauge, Histogram, collectDefaultMetrics } from "prom-client";
+import { prisma } from "..";
+import { orderStatus } from "@prisma/client";
+
+collectDefaultMetrics();
+
+const processingOrders = new Gauge({
+  name: "processing_orders",
+  help: "Gauge of processing orders",
+  labelNames: ["status"],
+});
+const ordersCache = new Map<number, OrderStatus>();
+const initOrdersCache = async () => {
+  const orders = await prisma.order.findMany({
+    where: {
+      status: {
+        notIn: [
+          orderStatus.DELIVERED,
+          orderStatus.CANCELLED,
+          orderStatus.REJECTED,
+        ],
+      },
+    },
+  });
+  orders.forEach((order) => {
+    ordersCache.set(order.id, order.status);
+  });
+  updateCounts();
+};
+initOrdersCache();
+
+const updateCounts = () => {
+  for (const status of Object.values(orderStatus)) {
+    if (status === orderStatus.DELIVERED) continue;
+    if (status === orderStatus.CANCELLED) continue;
+    if (status === orderStatus.REJECTED) continue;
+    const count = [...ordersCache.entries()].filter(
+      ([, s]) => s === status
+    ).length;
+    processingOrders.set({ status }, count);
+  }
+};
+export const updateProcessingOrders = (
+  status: OrderStatus,
+  orderNumber: number
+) => {
+  if (
+    status === orderStatus.DELIVERED ||
+    status === orderStatus.CANCELLED ||
+    status === orderStatus.REJECTED
+  ) {
+    ordersCache.delete(orderNumber);
+  } else {
+    ordersCache.set(orderNumber, status);
+  }
+  updateCounts();
+};

src/modules/order.ts

@@ -11,6 +11,8 @@ import { emojiInline } from "../utils/emoji";
 import env from "../utils/env";
 import { KitchenChannel, sendKitchenMessage } from "../utils/kitchenChannels";
 import { closed, closedReason } from "./closed";
+import { updateProcessingOrders } from "./metrics";
+import { orderStatus } from "@prisma/client";
 
 bot.addGlobalCommand(
   new SlashCommandBuilder()
@@ -20,7 +22,9 @@ bot.addGlobalCommand(
     .addStringOption((option) =>
       option
         .setName("order")
-        .setDescription("What would you like to order? - 1 reasonable item per order :)")
+        .setDescription(
+          "What would you like to order? - 1 reasonable item per order :)"
+        )
         .setRequired(true)
         .setMaxLength(241)
     ) as SlashCommandBuilder,
@@ -98,6 +102,7 @@ bot.addGlobalCommand(
         statusMessageId: message.id,
       },
     });
+    updateProcessingOrders(orderStatus.ORDERED, record.id);
 
     const kitchenActionRow =
       new ActionRowBuilder<ButtonBuilder>().addComponents([

src/modules/pack.ts

@@ -21,6 +21,7 @@ import {
 import s3 from "./s3";
 import handleError from "./sentry";
 import Sqids from "sqids";
+import { updateProcessingOrders } from "./metrics";
 
 // this took way too long to get copilot to spit out it had better work
 const URL_REGEX =
@@ -108,6 +109,7 @@ bot.client.on("messageCreate", async (message) => {
           fileUrl: c,
         },
       });
+      updateProcessingOrders(orderStatus.PACKING, order.id);
       setTimeout(() => finishPackOrder(order.id), 1000 * 60 * 5);
 
       clearKitchenMessages(order.id);
@@ -177,6 +179,7 @@ export const finishPackOrder = async (orderId: number) => {
         status: orderStatus.PACKED,
       },
     });
+    updateProcessingOrders(orderStatus.PACKED, order.id);
   } catch (e) {
     sendKitchenMessage(KitchenChannel.logs, {
       content: `:x: Failed to finish packing order ${orderId}!! <@!${

src/modules/reject.ts

@@ -14,6 +14,7 @@ import {
   clearKitchenMessages,
   sendKitchenMessage,
 } from "../utils/kitchenChannels";
+import { updateProcessingOrders } from "./metrics";
 
 bot.registerButton(/order:(\d+):reject/, async (interaction) => {
   const orderId = interaction.customId.split(":")[1];
@@ -48,6 +49,7 @@ bot.registerModal(/order:(\d+):reject:modal/, async (interaction) => {
         rejectorId: interaction.user.id,
       },
     });
+    updateProcessingOrders(orderStatus.REJECTED, order.id);
   } catch (e) {
     return interaction.reply({
       content: "Failed to reject order",