From 3377cd4a35822770282eb675bd8a9dd2cb851d97 Mon Sep 17 00:00:00 2001
From: "Dr.Lt.Data" <dr.lt.data@gmail.com>
Date: Mon, 31 Jul 2023 20:19:10 +0900
Subject: [PATCH] improve scanner.py update DB

---
 custom-node-list.json   | 24 +++++++++++++++++++--
 extension-node-map.json | 47 ++++++++++++++++++++++++++++++++++++++++-
 scanner.py              |  9 ++++++++
 3 files changed, 77 insertions(+), 3 deletions(-)

diff --git a/custom-node-list.json b/custom-node-list.json
index c4e20e6..0fecf03 100644
--- a/custom-node-list.json
+++ b/custom-node-list.json
@@ -838,6 +838,26 @@
             "install_type": "git-clone",
             "description": "Nodes: Remove Image Background (abg). A Anime Background Remover node for comfyui, based on this hf space, works same as AGB extention in automatic1111."
         },
+        {
+            "author": "twri",
+            "title": "SDXL Prompt Styler",
+            "reference": "https://github.com/twri/sdxl_prompt_styler",
+            "files": [
+                "https://github.com/twri/sdxl_prompt_styler"
+            ],
+            "install_type": "git-clone",
+            "description": "SDXL Prompt Styler is a node that enables you to style prompts based on predefined templates stored in a JSON file. The node specifically replaces a {prompt} placeholder in the 'prompt' field of each template with provided positive text."
+        },
+        {
+            "author": "bash-j",
+            "title": "Mikey Nodes",
+            "reference": "https://github.com/bash-j/mikey_nodes",
+            "files": [
+                "https://github.com/bash-j/mikey_nodes"
+            ],
+            "install_type": "git-clone",
+            "description": "Nodes: Prompt With Style, Prompt With SDXL, Resize Image for SDXL, Save Image With Prompt Data, HaldCLUT, Empty Latent Ratio Select/Custom SDXL"
+        },
         {
             "author": "taabata",
             "title": "Syrian Falcon Nodes",
@@ -1007,8 +1027,8 @@
             "description": "Nodes: Prompt parser. ComfyUI extra nodes. Mostly prompt parsing."
         },
         {
-            "author": "tudal",
-            "title": "SadaleNet",
+            "author": "SadaleNet",
+            "title": "ComfyUI A1111-like Prompt Custom Node Solution",
             "reference": "https://github.com/SadaleNet/CLIPTextEncodeA1111-ComfyUI",
             "files": [
                 "https://github.com/SadaleNet/CLIPTextEncodeA1111-ComfyUI/raw/master/custom_nodes/clip_text_encoder_a1111.py"
diff --git a/extension-node-map.json b/extension-node-map.json
index a912184..9fb4a1f 100644
--- a/extension-node-map.json
+++ b/extension-node-map.json
@@ -34,6 +34,15 @@
         "BSZAspectHires",
         "BSZCombinedHires"
     ],
+    "https://github.com/Bikecicle/ComfyUI-Waveform-Extensions/raw/main/EXT_AudioManipulation.py": [
+        "BatchJoinAudio",
+        "CutAudio",
+        "DuplicateAudio",
+        "JoinAudio",
+        "ResampleAudio",
+        "ReverseAudio",
+        "StretchAudio"
+    ],
     "https://github.com/BlenderNeko/ComfyUI_ADV_CLIP_emb": [
         "BNK_AddCLIPSDXLParams",
         "BNK_AddCLIPSDXLRParams",
@@ -649,6 +658,7 @@
         "Text Random Line",
         "Text Random Prompt",
         "Text String",
+        "Text String Truncate",
         "Text to Conditioning",
         "Text to Console",
         "Text to Number",
@@ -687,6 +697,20 @@
         "EditableCLIPEncode",
         "LoRAScheduler"
     ],
+    "https://github.com/bash-j/mikey_nodes": [
+        "Batch Resize Image for SDXL",
+        "Empty Latent Ratio Custom SDXL",
+        "Empty Latent Ratio Select SDXL",
+        "HaldCLUT",
+        "Prompt With SDXL",
+        "Prompt With Style",
+        "Prompt With Style V2",
+        "Prompt With Style V3",
+        "Resize Image for SDXL",
+        "Save Image With Prompt Data",
+        "Upscale Tile Calculator",
+        "VAE Decode 6GB SDXL (deprecated)"
+    ],
     "https://github.com/biegert/ComfyUI-CLIPSeg/raw/main/custom_nodes/clipseg.py": [
         "CLIPSeg",
         "CombineSegMasks"
@@ -857,6 +881,28 @@
         "YKImagePadForOutpaint",
         "YKMaskToImage"
     ],
+    "https://github.com/hnmr293/ComfyUI-nodes-hnmr": [
+        "CLIPIter",
+        "Dict2Model",
+        "GridImage",
+        "ImageBlend2",
+        "KSamplerOverrided",
+        "KSamplerSetting",
+        "KSamplerXYZ",
+        "LatentToHist",
+        "LatentToImage",
+        "ModelIter",
+        "RandomLatentImage",
+        "SaveStateDict",
+        "SaveText",
+        "StateDictLoader",
+        "StateDictMerger",
+        "StateDictMergerBlockWeighted",
+        "StateDictMergerBlockWeightedMulti",
+        "VAEDecodeBatched",
+        "VAEEncodeBatched",
+        "VAEIter"
+    ],
     "https://github.com/kwaroran/abg-comfyui": [
         "Remove Image Background (abg)"
     ],
@@ -1170,7 +1216,6 @@
         "trRouterLonger"
     ],
     "https://github.com/tudal/Hakkun-ComfyUI-nodes/raw/main/hakkun_nodes.py": [
-        "Any Converter",
         "Calculate Upscale",
         "Image size to string",
         "Multi Text Merge",
diff --git a/scanner.py b/scanner.py
index 710a21e..4d33ca5 100644
--- a/scanner.py
+++ b/scanner.py
@@ -24,6 +24,11 @@ def scan_in_file(filename):
     for key in keys:
         nodes.add(key)
 
+    pattern3 = r'NODE_CLASS_MAPPINGS\[\'(.*?)\'\]'
+    keys = re.findall(pattern3, code)
+    for key in keys:
+        nodes.add(key)
+
     matches = regex.findall(code)
     for match in matches:
         dict_text = match
@@ -32,6 +37,10 @@ def scan_in_file(filename):
         for key, value in key_value_pairs:
             class_dict[key] = value.strip()
 
+        key_value_pairs = re.findall(r"'([^']*)'\s*:\s*([^,\n]*)", dict_text)
+        for key, value in key_value_pairs:
+            class_dict[key] = value.strip()
+
         for key, value in class_dict.items():
             nodes.add(key)