improved logging & new reject button

phish-directory · Jul 24, 2024 · 124e4df · 124e4df
1 parent 227d7de
commit 124e4df
Showing 2 changed files with 65 additions and 7 deletions.
diff --git a/src/functions/domain.ts b/src/functions/domain.ts
@@ -61,6 +61,14 @@ export async function sendNewDomainMessage(app: App, domain: String) {
           return JSON.stringify(data);
         }
 
+        async function creareRejectData(domain: String, ts: String) {
+          let data = {
+            domain: domain,
+            ts: ts,
+          };
+          return JSON.stringify(data);
+        }
+
         await app.client.chat.update({
           token: process.env.SLACK_BOT_TOKEN,
           channel: reviewChannel,
@@ -144,7 +152,7 @@ export async function sendNewDomainMessage(app: App, domain: String) {
                     text: "Reject Domain (Safe)",
                     emoji: true,
                   },
-                  value: "reject_domain",
+                  value: `${await creareRejectData(domain, ts)}`,
                   action_id: "reject_domain",
                 },
               ],

diff --git a/src/index.ts b/src/index.ts
@@ -41,14 +41,13 @@ app.event(/.*/, async ({ event, client }) => {});
 app.action(/.*?/, async (args) => {
   try {
     const { ack, respond, payload, client, body } = args;
+    let actionUser = body.user.id!;
 
     await ack();
 
     // @ts-ignore
     switch (payload.action_id) {
       case "domain_classification":
-        let actionUser = body.user.id!;
-
         // @ts-expect-error
         let rawData = payload.selected_option.value;
         let data = JSON.parse(rawData);
@@ -82,7 +81,7 @@ app.action(/.*?/, async (args) => {
         let classmsg = await client.chat.postMessage({
           token: process.env.SLACK_BOT_TOKEN,
           channel: feedChannel,
-          text: `> Domain: ${domain} has been classified as ${classification} by <@${actionUser}>`,
+          text: `> *Domain:* _${domain}_ has been *classified* as _${classification}_ by _<@${actionUser}>_`,
         });
 
         if (actionUser === "U05NX48GL3T") {
@@ -136,8 +135,6 @@ app.action(/.*?/, async (args) => {
           timestamp: classTs,
         });
 
-        console.log(rsp);
-
         if (rsp.data === "Verdict added") {
           client.chat.postMessage({
             token: process.env.SLACK_BOT_TOKEN,
@@ -171,7 +168,60 @@ app.action(/.*?/, async (args) => {
 
         break;
       case "reject_domain":
-        console.log("DOMAIN REJECTED");
+        // @ts-expect-error
+        const rawRData = payload.value;
+        console.log(rawRData);
+
+        let rData = JSON.parse(rawRData);
+        let rdomain = rData.domain;
+        let rts = rData.ts;
+
+        await client.chat
+          .delete({
+            token: process.env.SLACK_BOT_TOKEN,
+            channel: reviewChannel,
+            ts: rts,
+          })
+          .catch((error) =>
+            blog(`Error deleting chat message: ${error}`, "error")
+          );
+
+        let rclassmsg = await client.chat.postMessage({
+          token: process.env.SLACK_BOT_TOKEN,
+          channel: feedChannel,
+          text: `> *Domain:* _${rdomain}_ has been *marked* as _Safe_ by _<@${actionUser}>_`,
+        });
+
+        if (actionUser === "U05NX48GL3T") {
+          client.reactions.add({
+            token: process.env.SLACK_BOT_TOKEN,
+            name: "jasper",
+            channel: feedChannel,
+            timestamp: rclassmsg.ts,
+          });
+        } else if (actionUser === "U0616280E6P") {
+          client.reactions.add({
+            token: process.env.SLACK_BOT_TOKEN,
+            name: "aram-sq",
+            channel: feedChannel,
+            timestamp: rclassmsg.ts,
+          });
+        } else {
+          client.reactions.add({
+            token: process.env.SLACK_BOT_TOKEN,
+            name: ":bust_in_silhouette:",
+            channel: feedChannel,
+            timestamp: rclassmsg.ts,
+          });
+        }
+
+        client.reactions.add({
+          token: process.env.SLACK_BOT_TOKEN,
+          name: "large_green_circle",
+          channel: feedChannel,
+          timestamp: rclassmsg.ts,
+        });
+
         break;
       default:
         console.log("Unknown action");