Add new ComfyUI support for parsing and executing workflows

Author: mythz

AiServer.ServiceInterface/AppData.cs

@@ -86,7 +86,22 @@ T[] LoadModels<T>(string name) where T : IHasId<string>
         }
         return models.ToArray();
     }
-    
+
+    public void LoadComfyWorkflow()
+    {
+        var objectInfoJson = ReadTextFile("wwwroot/lib/data/object_info.json")!;
+        var nodeDefs = ComfyMetadata.Instance.LoadObjectInfo(objectInfoJson);
+        var overrideObjectInfoJson = ReadTextFile("App_Data/overrides/object_info.json");
+        if (overrideObjectInfoJson != null)
+        {
+            var overrideNodeDefs = ComfyMetadata.ParseNodeDefinitions(overrideObjectInfoJson);
+            foreach (var entry in overrideNodeDefs)
+            {
+                nodeDefs[entry.Key] = entry.Value;
+            }
+        }
+    }
+
     public void Reload(IDbConnection db)
     {
         Prompts = PocoDataSource.Create(LoadModels<Prompt>("prompts.json")
@@ -101,6 +116,7 @@ public void Reload(IDbConnection db)
 
         LoadModelDefaults();
         LogWorkerInfo(AiProviders, "API");
+        LoadComfyWorkflow();
     }
 
     public void ResetAiProviders(IDbConnection db)

AiServer.ServiceInterface/AppExtensions.cs

@@ -136,4 +136,30 @@ public static string ComputeSha256(this Stream stream)
                 return element.GetRawText();
         }
     }
+    
+    /// <summary>
+    /// Validates that a relative path is safe and doesn't traverse outside the intended directory.
+    /// </summary>
+    /// <param name="basePath">The base directory path.</param>
+    /// <param name="relativePath">The relative path to validate.</param>
+    /// <returns>True if the path is safe, false otherwise.</returns>
+    public static bool IsPathSafe(this string relativePath, string basePath)
+    {
+        try
+        {
+            // Normalize paths to handle different directory separators
+            basePath = Path.GetFullPath(basePath);
+            
+            // Combine the base path with the relative path
+            string fullPath = Path.GetFullPath(Path.Combine(basePath, relativePath));
+            
+            // Check if the resulting path starts with the base path
+            return fullPath.StartsWith(basePath, StringComparison.OrdinalIgnoreCase);
+        }
+        catch (Exception ex) when (ex is ArgumentException or NotSupportedException or PathTooLongException)
+        {
+            // Path contains invalid characters or is in an invalid format
+            return false;
+        }
+    }
 }