From b0e7304bffed246ad159599cb5e6e0f28c83a4d3 Mon Sep 17 00:00:00 2001
From: Maidul Islam <maidul98@gmail.com>
Date: Mon, 5 Jun 2023 20:19:15 -0700
Subject: [PATCH] Patch backfill data

---
 backend/src/utils/setup/backfillData.ts | 30 +++++++++++++++++++------
 1 file changed, 23 insertions(+), 7 deletions(-)

diff --git a/backend/src/utils/setup/backfillData.ts b/backend/src/utils/setup/backfillData.ts
index ba56e266b..f78ec08fc 100644
--- a/backend/src/utils/setup/backfillData.ts
+++ b/backend/src/utils/setup/backfillData.ts
@@ -335,6 +335,19 @@ export const backfillSecretFolders = async () => {
     }
   );
 
+  await SecretVersion.updateMany(
+    {
+      folder: {
+        $exists: false,
+      },
+    },
+    {
+      $set: {
+        folder: "root",
+      },
+    }
+  );
+
   let secretSnapshots = await SecretSnapshot.find({
     environment: {
       $exists: false,
@@ -352,12 +365,15 @@ export const backfillSecretFolders = async () => {
         groupSnapByEnv[secVer.environment].push(secVer);
       });
 
-      const newSnapshots = Object.keys(groupSnapByEnv).map((snapEnv) => ({
-        ...secSnapshot.toObject({ virtuals: false }),
-        _id: new Types.ObjectId(),
-        environment: snapEnv,
-        secretVersions: groupSnapByEnv[snapEnv],
-      }));
+      const newSnapshots = Object.keys(groupSnapByEnv).map((snapEnv) => {
+        const secretIdsOfEnvGroup = groupSnapByEnv[snapEnv] ? groupSnapByEnv[snapEnv].map(secretVersion => secretVersion._id) : []
+        return {
+          ...secSnapshot.toObject({ virtuals: false }),
+          _id: new Types.ObjectId(),
+          environment: snapEnv,
+          secretVersions: secretIdsOfEnvGroup,
+        }
+      });
 
       await SecretSnapshot.insertMany(newSnapshots);
       await secSnapshot.delete();
@@ -369,7 +385,7 @@ export const backfillSecretFolders = async () => {
       },
     })
       .populate<{ secretVersions: ISecretVersion[] }>("secretVersions")
-      .limit(50);
+      .limit(5000);
   }
 
   console.log("Migration: Folder migration v1 complete")