🚸 Auto-update typebot if it was migrated while getting it

Author: baptisteArno

apps/builder/src/features/typebot/api/getTypebot.ts

@@ -1,9 +1,14 @@
 import { TRPCError } from "@trpc/server";
 import prisma from "@typebot.io/prisma";
+import { DbNull } from "@typebot.io/prisma/enum";
 import { getTypebotAccessRight } from "@typebot.io/typebot/helpers/getTypebotAccessRight";
 import { isReadTypebotForbidden } from "@typebot.io/typebot/helpers/isReadTypebotForbidden";
 import { migrateTypebot } from "@typebot.io/typebot/migrations/migrateTypebot";
-import { typebotSchema } from "@typebot.io/typebot/schemas/typebot";
+import {
+  type Typebot,
+  type TypebotV6,
+  typebotSchema,
+} from "@typebot.io/typebot/schemas/typebot";
 import { z } from "@typebot.io/zod";
 import { publicProcedure } from "@/helpers/server/trpc";
 
@@ -70,12 +75,47 @@ export const getTypebot = publicProcedure
         });
 
       try {
-        const parsedTypebot = migrateToLatestVersion
-          ? await migrateTypebot(typebotSchema.parse(existingTypebot))
-          : typebotSchema.parse(existingTypebot);
+        const { typebot, wasMigrated } = (
+          migrateToLatestVersion
+            ? await migrateTypebot(typebotSchema.parse(existingTypebot))
+            : {
+                typebot: typebotSchema.parse(existingTypebot),
+                wasMigrated: false,
+              }
+        ) as
+          | { typebot: TypebotV6; wasMigrated: true }
+          | { typebot: Typebot; wasMigrated: false };
+
+        if (wasMigrated)
+          await prisma.typebot.update({
+            where: {
+              id: existingTypebot.id,
+            },
+            data: {
+              version: typebot.version,
+              name: typebot.name,
+              icon: typebot.icon,
+              selectedThemeTemplateId: typebot.selectedThemeTemplateId,
+              events: typebot.events,
+              groups: typebot.groups,
+              theme: typebot.theme ? typebot.theme : undefined,
+              settings: typebot.settings ? typebot.settings : undefined,
+              folderId: typebot.folderId,
+              variables: typebot.variables,
+              edges: typebot.edges,
+              resultsTablePreferences:
+                typebot.resultsTablePreferences === null
+                  ? DbNull
+                  : typebot.resultsTablePreferences,
+              publicId: typebot.publicId === null ? null : typebot.publicId,
+              customDomain: typebot.customDomain,
+              isClosed: typebot.isClosed,
+              whatsAppCredentialsId: typebot.whatsAppCredentialsId ?? undefined,
+            },
+          });
 
         return {
-          typebot: parsedTypebot,
+          typebot,
           currentUserMode: getTypebotAccessRight(user, existingTypebot),
         };
       } catch (err) {

apps/builder/src/features/typebot/api/importTypebot.ts

@@ -69,7 +69,7 @@ const importingTypebotSchema = z.preprocess(
 
 type ImportingTypebot = z.infer<typeof importingTypebotSchema>;
 
-const migrateImportingTypebot = (
+const migrateImportingTypebot = async (
   typebot: ImportingTypebot,
 ): Promise<TypebotV6> => {
   const fullTypebot = {
@@ -87,7 +87,7 @@ const migrateImportingTypebot = (
     publicId: null,
     riskLevel: null,
   } satisfies Typebot;
-  return migrateTypebot(fullTypebot);
+  return (await migrateTypebot(fullTypebot)).typebot;
 };
 
 export const importTypebot = authenticatedProcedure

apps/builder/src/features/typebot/api/updateTypebot.ts

@@ -216,7 +216,7 @@ export const updateTypebot = authenticatedProcedure
         },
       });
 
-      const migratedTypebot = await migrateTypebot(
+      const { typebot: migratedTypebot } = await migrateTypebot(
         typebotSchema.parse(newTypebot),
       );
 

packages/scripts/src/inspectTypebot.ts

@@ -28,6 +28,7 @@ const inspectTypebot = async () => {
   const typebot = await prisma.typebot.findFirst({
     where,
     select: {
+      version: true,
       id: true,
       name: true,
       riskLevel: true,

packages/typebot/src/migrations/migrateTypebot.ts

@@ -5,14 +5,16 @@ import type { Typebot, TypebotV6 } from "../schemas/typebot";
 import { migrateTypebotFromV3ToV4 } from "./migrateTypebotFromV3ToV4";
 import { migrateTypebotFromV5ToV6 } from "./migrateTypebotFromV5ToV6";
 
-export const migrateTypebot = async (typebot: Typebot): Promise<TypebotV6> => {
-  if (isTypebotAtLeastV6(typebot)) return typebot;
+export const migrateTypebot = async (
+  typebot: Typebot,
+): Promise<{ typebot: TypebotV6; wasMigrated: boolean }> => {
+  if (isTypebotAtLeastV6(typebot)) return { typebot, wasMigrated: false };
   let migratedTypebot: any = typebot;
   if (migratedTypebot.version === "3")
     migratedTypebot = await migrateTypebotFromV3ToV4(typebot);
   if (migratedTypebot.version === "4" || migratedTypebot.version === "5")
-    migratedTypebot = migrateTypebotFromV5ToV6(migratedTypebot);
-  return migratedTypebot;
+    migratedTypebot = await migrateTypebotFromV5ToV6(migratedTypebot);
+  return { typebot: migratedTypebot, wasMigrated: true };
 };
 
 export const migratePublicTypebot = async (