apply block migrations in missed cases

Author: icecrasher321

apps/sim/app/api/workflows/[id]/deployments/[version]/revert/route.ts

@@ -6,7 +6,10 @@ import { AuditAction, AuditResourceType, recordAudit } from '@/lib/audit/log'
 import { env } from '@/lib/core/config/env'
 import { generateRequestId } from '@/lib/core/utils/request'
 import { syncMcpToolsForWorkflow } from '@/lib/mcp/workflow-mcp-sync'
-import { saveWorkflowToNormalizedTables } from '@/lib/workflows/persistence/utils'
+import {
+  applyBlockMigrations,
+  saveWorkflowToNormalizedTables,
+} from '@/lib/workflows/persistence/utils'
 import { validateWorkflowPermissions } from '@/lib/workflows/utils'
 import { createErrorResponse, createSuccessResponse } from '@/app/api/workflows/utils'
 
@@ -73,8 +76,13 @@ export async function POST(
       return createErrorResponse('Invalid deployed state structure', 500)
     }
 
+    const { blocks: migratedBlocks } = await applyBlockMigrations(
+      deployedState.blocks,
+      workflowRecord?.workspaceId ?? undefined
+    )
+
     const saveResult = await saveWorkflowToNormalizedTables(id, {
-      blocks: deployedState.blocks,
+      blocks: migratedBlocks,
       edges: deployedState.edges,
       loops: deployedState.loops || {},
       parallels: deployedState.parallels || {},

apps/sim/app/api/workflows/[id]/deployments/[version]/route.ts

@@ -7,7 +7,7 @@ import { AuditAction, AuditResourceType, recordAudit } from '@/lib/audit/log'
 import { generateRequestId } from '@/lib/core/utils/request'
 import { syncMcpToolsForWorkflow } from '@/lib/mcp/workflow-mcp-sync'
 import { restorePreviousVersionWebhooks, saveTriggerWebhooksForDeploy } from '@/lib/webhooks/deploy'
-import { activateWorkflowVersion } from '@/lib/workflows/persistence/utils'
+import { activateWorkflowVersion, applyBlockMigrations } from '@/lib/workflows/persistence/utils'
 import {
   cleanupDeploymentVersion,
   createSchedulesForDeploy,
@@ -78,7 +78,13 @@ export async function GET(
       return createErrorResponse('Deployment version not found', 404)
     }
 
-    return createSuccessResponse({ deployedState: row.state })
+    const state = row.state as { blocks?: Record<string, BlockState> }
+    if (state.blocks) {
+      const { blocks } = await applyBlockMigrations(state.blocks)
+      state.blocks = blocks
+    }
+
+    return createSuccessResponse({ deployedState: state })
   } catch (error: any) {
     logger.error(
       `[${requestId}] Error fetching deployment version ${version} for workflow ${id}`,

apps/sim/lib/workflows/persistence/utils.ts

@@ -158,7 +158,7 @@ function createMigrationPipeline(migrations: BlockMigration[]) {
   }
 }
 
-const applyBlockMigrations = createMigrationPipeline([
+export const applyBlockMigrations = createMigrationPipeline([
   (ctx) => {
     const { blocks } = sanitizeAgentToolsInBlocks(ctx.blocks)
     return { ...ctx, blocks }