From f6a7f5582c9ef3bedf4aca7091fd59d1aa55549b Mon Sep 17 00:00:00 2001
From: karishmas6 <carishmashukla28@gmail.com>
Date: Thu, 19 Sep 2024 18:04:32 +0530
Subject: [PATCH] feat: handle task deletion + conditional invocation

---
 .../integrations/gsheet.ts                    | 39 +++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/server/src/workflow-management/integrations/gsheet.ts b/server/src/workflow-management/integrations/gsheet.ts
index 8f0baaf4..3d4b285a 100644
--- a/server/src/workflow-management/integrations/gsheet.ts
+++ b/server/src/workflow-management/integrations/gsheet.ts
@@ -4,6 +4,16 @@ import path from 'path';
 import logger from "../../logger";
 import { readFile } from "../storage";
 
+
+interface GoogleSheetUpdateTask {
+  name: string;
+  runId: string;
+  status: 'pending' | 'completed';
+}
+
+export let googleSheetUpdateTasks: { [runId: string]: GoogleSheetUpdateTask } = {};
+
+
 // *** Temporary Path to the JSON file that will store the integration details ***
 const integrationsFilePath = path.join(__dirname, 'integrations.json');
 
@@ -80,3 +90,32 @@ export async function writeDataToSheet(spreadsheetId: string, range: string, dat
     throw error;
   }
 }
+
+const processGoogleSheetUpdates = async () => {
+  while (true) {
+    let hasPendingTasks = false; 
+    for (const runId in googleSheetUpdateTasks) {
+      const task = googleSheetUpdateTasks[runId];
+      if (task.status === 'pending') {
+        hasPendingTasks = true; 
+        try {
+          await updateGoogleSheet(task.name, task.runId);
+          console.log(`Successfully updated Google Sheets for run ${task.runId}`);
+          delete googleSheetUpdateTasks[runId];
+        } catch (error: any) {
+          console.error(`Failed to update Google Sheets for run ${task.runId}:`, error);
+        }
+      }
+    }
+    if (!hasPendingTasks) {
+      break;
+    }
+    await new Promise(resolve => setTimeout(resolve, 5000));
+  }
+};
+
+export const startProcessGoogleSheetUpdates = () => {
+  if (Object.keys(googleSheetUpdateTasks).length > 0) {
+    processGoogleSheetUpdates();
+  }
+};
\ No newline at end of file