fix: enable status updates

Klimatbyran · Feb 28, 2024 · 44063cd · 44063cd
1 parent 928bb09
commit 44063cd
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/src/elastic.ts b/src/elastic.ts
@@ -119,10 +119,11 @@ class Elastic {
     }
   }
 
-  /*async updateDocumentState(url: string, newState: string) {
+  async updateDocumentState(documentId: string, newState: string) {
     try {
       await this.client.update({
         index: this.indexName,
+        id: documentId,
         body: {
           doc: {
             state: newState
@@ -133,7 +134,7 @@ class Elastic {
     } catch (error) {
       console.error(`Error updating document state for Document ID ${documentId}:`, error);
     }
-  }*/
+  }
 }
 
 export default new Elastic(config)
diff --git a/src/workers/discordReview.ts b/src/workers/discordReview.ts
@@ -72,13 +72,16 @@ const worker = new Worker(
     job.updateProgress(40)
 
     discord.client.on('interactionCreate', async (interaction) => {
+      let reportState = ''
       if (interaction.isButton() && interaction.customId === 'approve') {
+        reportState = 'approved'
         interaction.update({
           content: 'Approved!',
           embeds: [],
           components: [],
         })
       } else if (interaction.isButton() && interaction.customId === 'edit') {
+        reportState = 'edited'
         const input = new TextInputBuilder()
           .setCustomId('editInput')
           .setLabel(`Granska utsläppsdata`)
@@ -115,14 +118,22 @@ const worker = new Worker(
         }
       } else if (interaction.isButton() && interaction.customId === 'reject') {
         // todo diskutera vad vill vill händer. ska man ens få rejecta?
+        reportState = 'rejected'
         interaction.update({
           content: 'Rejected!',
           embeds: [],
           components: [],
         })
       }
-    })
 
+      if (reportState !== '') {
+        try {
+          await elastic.updateDocumentState(job.data.documentId, reportState)
+        } catch (error) {
+          job.log(`Error updating document state: ${error.message}`)
+        }
+      }
+    })
     job.updateProgress(100)
   },
   {