Added special cases handling and a setting class for it to be defined externally

ComfySharp/ComfyClient.cs

@@ -23,7 +23,7 @@ public class ComfyClient {
         nodes= new();
 
         try {
-            dbGenerator = new(ConversionSettings.FromFile("conv_config.json"));
+            dbGenerator = new(ConversionSettings.FromFile(Path.Combine(Environment.CurrentDirectory, "conv_config.json")));
         }
         catch (Exception e) {
             Console.WriteLine(e);
@@ -68,7 +68,7 @@ public class ComfyClient {
 
         if (req is { IsSuccessStatusCode: true, Content: not null }) {
             var doc = await req.Content.ReadFromJsonAsync<JsonDocument>();
-            
+            dbGenerator.GenerateClasses(doc);
             //ObjectInfoParser.Parse(doc, out nodes);
         }
 

ComfySharp/ComfySharp.csproj

@@ -8,7 +8,7 @@
 
     <ItemGroup>
       <None Update="conv_config.json">
-        <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
+        <CopyToOutputDirectory>Always</CopyToOutputDirectory>
       </None>
     </ItemGroup>
 

ComfySharp/ConversionSettings.cs

@@ -1,11 +1,13 @@
-using System.Text.Json;
+using System.Runtime.Serialization;
+using System.Text.Json;
+using System.Text.Json.Serialization;
 
 namespace ComfySharp;
 
-[Serializable]
+[DataContract, JsonSerializable(typeof(ConversionSettings))]
 public class ConversionSettings {
-    public List<string> EnumConvertAsString = new();
+    [DataMember(IsRequired = true)] public List<string> EnumConvertAsString = new();
-    public List<string> EnumConvertAsBool = new();
+    [DataMember(IsRequired = true)] public List<string> EnumConvertAsBool = new();
 
     public static ConversionSettings FromFile(string path) {
         if (!File.Exists(path)) throw new FileNotFoundException("Could not find settings file", path);
@@ -24,6 +26,6 @@ public class ConversionSettings {
 
     public void Save(string path) => File.WriteAllText(path, ToJson());
 
-    public string ToJson() => JsonSerializer.Serialize(this);
+    public string ToJson() => JsonSerializer.Serialize(this, new JsonSerializerOptions { WriteIndented = true, IncludeFields = true });
 
 }

ComfySharp/Types/NodeDBGenerator.cs

@@ -18,6 +18,9 @@ public class NodeDBGenerator {
     private Dictionary<string, List<string>> knownEnums = new();
     public Dictionary<string, List<string>> GetKnownEnums() => knownEnums;
     
+    private Dictionary<string, List<string>> knownStringLists = new();
+    public Dictionary<string, List<string>> GetKnownStringLists() => knownStringLists;
+    
     public int Count => nodes.Count;
     private int typeFields = 0;
     private int enumFields = 0;
@@ -62,6 +65,26 @@ public class NodeDBGenerator {
         enumFields++;
     }
     
+    /// <summary>
+    /// Adds or updates a known string list, avoids duplicates.
+    /// </summary>
+    /// <param name="type">stringList typename</param>
+    /// <param name="values">Collection of string values for this StringList</param>
+    private void AddKnownStringList(string type, ICollection<string> values) {
+        if (!knownStringLists.ContainsKey(type)) {
+            knownStringLists.Add(type, values.ToList());
+            Console.WriteLine("Added new known stringList: {0}", type);
+        }
+        else {
+            values.ToList().ForEach(value => {
+                if (!knownStringLists[type].Contains(value)) {
+                    knownStringLists[type].Add(value);
+                    Console.WriteLine("Added new value to already known stringList: {0}", value);
+                }
+            });
+        }
+    }
+    
     public void GenerateClasses(JsonDocument document) {
         foreach (var node in document.RootElement.EnumerateObject()) ScanNode(node);
 
@@ -120,15 +143,18 @@ public class NodeDBGenerator {
                 Console.WriteLine("Encountered a special case: {0}", inputProperty.Name);
                 return;
             }
-            // these are all lists of strings and not enums
-            if (inputProperty.Name == "image" || inputProperty.Name == "ckpt_name" || inputProperty.Name == "lora_name" || inputProperty.Name == "control_net_name"){
             
-            }
+            List<string> enumValues = new(); //holds all the values of the enum, valid for both following cases
-            List<string> enumValues = new();
             inputProperty.Value[0].EnumerateArray().ToList().ForEach(value => enumValues.Add(value.ToString()));
+            
+            // these are all lists of strings and not enums
+            if (settings.EnumConvertAsString.Contains(inputProperty.Name)) {
+                AddKnownStringList(inputProperty.Name, enumValues);
+            } else {
                 AddKnownEnum(inputProperty.Name, enumValues);
             }
         }
+    }
 
     private void ScanOutputs(JsonProperty output) { }
     

ComfySharp/conv_config.json

@@ -0,0 +1,33 @@
+{
+  "EnumConvertAsString": [
+    "ckpt_name",
+    "image",
+    "lora_name",
+    "control_net_name",
+    "config_name",
+    "clip_name",
+    "clip_name1",
+    "clip_name2",
+    "unet_name",
+    "model_path",
+    "hypernetwork_name",
+    "model_name",
+    "Select to add LoRA",
+    "Select to add Wildcard",
+    "preset",
+    "preset_repo_id",
+    "checkpoint",
+    "video",
+    "model",
+    "bbox_detector",
+    "pose_estimator",
+    "preprocessor",
+    "cascade_xml",
+    "style",
+    "style1",
+    "style2",
+    "style3",
+    "style4"
+  ],
+  "EnumConvertAsBool": []
+}