From 76ce1017eafc0f31e1957b88645ebda9306abe67 Mon Sep 17 00:00:00 2001
From: Michael C <michael@mchang.name>
Date: Wed, 22 Feb 2023 00:08:27 -0500
Subject: [PATCH] add warning webhook

---
 src/routes/postWarning.ts | 24 ++++++++++++++++++++++++
 src/utils/webhookUtils.ts | 28 ++++++++++++++++++++++++++++
 test.json                 |  6 ++++++
 test/mocks.ts             |  4 ++++
 4 files changed, 62 insertions(+)

diff --git a/src/routes/postWarning.ts b/src/routes/postWarning.ts
index f9eaa4c..c3c2a0c 100644
--- a/src/routes/postWarning.ts
+++ b/src/routes/postWarning.ts
@@ -5,6 +5,7 @@ import { isUserVIP } from "../utils/isUserVIP";
 import { getHashCache } from "../utils/getHashCache";
 import { HashedUserID, UserID } from "../types/user.model";
 import { config } from "../config";
+import { generateWarningDiscord, warningData, dispatchEvent } from "../utils/webhookUtils";
 
 type warningEntry = {
     userID: HashedUserID,
@@ -21,6 +22,8 @@ function checkExpiredWarning(warning: warningEntry): boolean {
     return warning.issueTime > expiry && !warning.enabled;
 }
 
+const getUsername = (userID: HashedUserID) => db.prepare("get", `SELECT "userName" FROM "userNames" WHERE "userID" = ?`, [userID], { useReplica: true });
+
 export async function postWarning(req: Request, res: Response): Promise<Response> {
     if (!req.body.userID) return res.status(400).json({ "message": "Missing parameters" });
 
@@ -62,6 +65,27 @@ export async function postWarning(req: Request, res: Response): Promise<Response
         resultStatus = "removed from";
     }
 
+    const targetUsername = await getUsername(userID) ?? null;
+    const issuerUsername = await getUsername(issuerUserID) ?? null;
+    const webhookData = {
+        target: {
+            userID,
+            username: targetUsername
+        },
+        issuer: {
+            userID: issuerUserID,
+            username: issuerUsername
+        },
+        reason
+    } as warningData;
+
+    try {
+        const warning = generateWarningDiscord(webhookData);
+        dispatchEvent("warning", warning);
+    } catch /* istanbul ignore next */ (err) {
+        Logger.error(`Error sending warning to Discord ${err}`);
+    }
+
     return res.status(200).json({
         message: `Warning ${resultStatus} user '${userID}'.`,
     });
diff --git a/src/utils/webhookUtils.ts b/src/utils/webhookUtils.ts
index 7180a98..f75fa46 100644
--- a/src/utils/webhookUtils.ts
+++ b/src/utils/webhookUtils.ts
@@ -1,6 +1,7 @@
 import { config } from "../config";
 import { Logger } from "../utils/logger";
 import axios from "axios";
+import { HashedUserID } from "../types/user.model";
 
 function getVoteAuthorRaw(submissionCount: number, isTempVIP: boolean, isVIP: boolean, isOwnSubmission: boolean): string {
     if (isOwnSubmission) {
@@ -57,8 +58,35 @@ function dispatchEvent(scope: string, data: Record<string, unknown>): void {
     }
 }
 
+interface warningData {
+    target: {
+        userID: HashedUserID
+        username: string | null
+    },
+    issuer: {
+        userID: HashedUserID,
+        username: string | null
+    },
+    reason: string
+}
+
+function generateWarningDiscord(data: warningData) {
+    return {
+        embeds: [
+            {
+                title: "Warning",
+                description: `**User:** ${data.target.username} (${data.target.userID})\n**Issuer:** ${data.issuer.username} (${data.issuer.userID})\n**Reason:** ${data.reason}`,
+                color: 0xff0000,
+                timestamp: new Date().toISOString()
+            }
+        ]
+    };
+}
+
 export {
     getVoteAuthorRaw,
     getVoteAuthor,
     dispatchEvent,
+    generateWarningDiscord,
+    warningData
 };
diff --git a/test.json b/test.json
index fef8052..0d38463 100644
--- a/test.json
+++ b/test.json
@@ -42,6 +42,12 @@
                 "vote.up",
                 "vote.down"
             ]
+        }, {
+            "url": "http://127.0.0.1:8081/WarningWebhook",
+            "key": "superSecretKey",
+            "scopes": [
+                "warning"
+            ]
         }
     ],
     "maxNumberOfActiveWarnings": 3,
diff --git a/test/mocks.ts b/test/mocks.ts
index d6673bf..f34a403 100644
--- a/test/mocks.ts
+++ b/test/mocks.ts
@@ -13,6 +13,10 @@ app.post("/webhook/FirstTimeSubmissions", (req, res) => {
     res.sendStatus(200);
 });
 
+app.post("/webhook/WarningWebhook", (req, res) => {
+    res.sendStatus(200);
+});
+
 app.post("/webhook/CompletelyIncorrectReport", (req, res) => {
     res.sendStatus(200);
 });