[Bug] Guard link fixer with try-catch (#3806)

src/composables/useWorkflowValidation.ts

@@ -7,45 +7,22 @@ import { fixBadLinks } from '@/utils/linkFixer'
 
 export interface ValidationResult {
   graphData: ComfyWorkflowJSON | null
-  linksFixes?: {
-    patched: number
-    deleted: number
-  }
 }
 
 export function useWorkflowValidation() {
   const toastStore = useToastStore()
 
-  /**
-   * Validates a workflow, including link validation and schema validation
-   */
-  async function validateWorkflow(
+  function tryFixLinks(
     graphData: ComfyWorkflowJSON,
-    options: {
-      silent?: boolean
-    } = {}
-  ): Promise<ValidationResult> {
+    options: { silent?: boolean } = {}
+  ) {
     const { silent = false } = options
 
-    let linksFixes
-    let validatedData: ComfyWorkflowJSON | null = null
-
-    // First do schema validation
-    const validatedGraphData = await validateComfyWorkflow(
-      graphData,
-      /* onError=*/ (err) => {
-        if (!silent) {
-          toastStore.addAlert(err)
-        }
-      }
-    )
-
-    if (validatedGraphData) {
     // Collect all logs in an array
     const logs: string[] = []
     // Then validate and fix links if schema validation passed
     const linkValidation = fixBadLinks(
-        validatedGraphData as unknown as ISerialisedGraph,
+      graphData as unknown as ISerialisedGraph,
       {
         fix: true,
         silent,
@@ -76,16 +53,43 @@ export function useWorkflowValidation() {
       }
     }
 
-      validatedData = linkValidation.graph as unknown as ComfyWorkflowJSON
-      linksFixes = {
-        patched: linkValidation.patched,
-        deleted: linkValidation.deleted
+    return linkValidation.graph as unknown as ComfyWorkflowJSON
+  }
+
+  /**
+   * Validates a workflow, including link validation and schema validation
+   */
+  async function validateWorkflow(
+    graphData: ComfyWorkflowJSON,
+    options: {
+      silent?: boolean
+    } = {}
+  ): Promise<ValidationResult> {
+    const { silent = false } = options
+
+    let validatedData: ComfyWorkflowJSON | null = null
+
+    // First do schema validation
+    const validatedGraphData = await validateComfyWorkflow(
+      graphData,
+      /* onError=*/ (err) => {
+        if (!silent) {
+          toastStore.addAlert(err)
+        }
+      }
+    )
+
+    if (validatedGraphData) {
+      try {
+        validatedData = tryFixLinks(validatedGraphData, { silent })
+      } catch (err) {
+        // Link fixer itself is throwing an error
+        console.error(err)
       }
     }
 
     return {
-      graphData: validatedData,
-      linksFixes
+      graphData: validatedData
     }
   }