From fa6278108e90580c51d95ccbfa6f75bbcadaae36 Mon Sep 17 00:00:00 2001
From: karishmas6 <carishmashukla28@gmail.com>
Date: Tue, 29 Oct 2024 04:22:19 +0530
Subject: [PATCH] feat: get count of items

---
 server/src/routes/storage.ts | 21 +++++++++++++++------
 1 file changed, 15 insertions(+), 6 deletions(-)

diff --git a/server/src/routes/storage.ts b/server/src/routes/storage.ts
index db65eeb1..759e8dec 100644
--- a/server/src/routes/storage.ts
+++ b/server/src/routes/storage.ts
@@ -256,14 +256,22 @@ router.post('/runs/run/:id', requireSignIn, async (req: AuthenticatedRequest, re
       let extractedScreenshotsCount = 0;
       let extractedItemsCount = 0;
 
-      if (run.dataValues.binaryOutput) {
-        extractedScreenshotsCount = run.dataValues.binaryOutput['item-0'].length;
-      }
-      if (run.dataValues.serializableOutput) {
-        extractedItemsCount = run.dataValues.serializableOutput['item-0'].length;
+      if (run.dataValues.binaryOutput && run.dataValues.binaryOutput["item-0"]) {
+        extractedScreenshotsCount = 1;
       }
 
-      console.log(`${extractedItemsCount} ${extractedScreenshotsCount}`)
+      if (run.dataValues.serializableOutput && run.dataValues.serializableOutput["item-0"]) {
+        const itemsArray = run.dataValues.serializableOutput["item-0"];
+        extractedItemsCount = itemsArray.length;
+
+        totalRowsExtracted = itemsArray.reduce((total, item) => {
+          return total + Object.keys(item).length;
+        }, 0);
+      }
+
+      console.log(`Extracted Items Count: ${extractedItemsCount}`);
+      console.log(`Extracted Screenshots Count: ${extractedScreenshotsCount}`);
+      console.log(`Total Rows Extracted: ${totalRowsExtracted}`);
 
       capture(
         'maxun-oss-run-created-manual',
@@ -272,6 +280,7 @@ router.post('/runs/run/:id', requireSignIn, async (req: AuthenticatedRequest, re
           user_id: req.user?.id,
           created_at: new Date().toISOString(),
           status: 'success',
+          totalRowsExtracted,
           extractedItemsCount,
           extractedScreenshotsCount,
         }