Migrate to new comfyui node schema

Author: Acly

__init__.py

@@ -1,5 +1,6 @@
 import folder_paths
 import os
+from comfy_api.latest import ComfyExtension, io
 
 
 def _add_folder_path(folder_name: str, extensions_to_register: list):
@@ -21,29 +22,23 @@ def _add_folder_path(folder_name: str, extensions_to_register: list):
 
 from . import nodes
 
-NODE_CLASS_MAPPINGS = {
-    "INPAINT_LoadFooocusInpaint": nodes.LoadFooocusInpaint,
-    "INPAINT_ApplyFooocusInpaint": nodes.ApplyFooocusInpaint,
-    "INPAINT_VAEEncodeInpaintConditioning": nodes.VAEEncodeInpaintConditioning,
-    "INPAINT_MaskedFill": nodes.MaskedFill,
-    "INPAINT_MaskedBlur": nodes.MaskedBlur,
-    "INPAINT_LoadInpaintModel": nodes.LoadInpaintModel,
-    "INPAINT_InpaintWithModel": nodes.InpaintWithModel,
-    "INPAINT_ExpandMask": nodes.ExpandMask,
-    "INPAINT_ShrinkMask": nodes.ShrinkMask,
-    "INPAINT_StabilizeMask": nodes.StabilizeMask,
-    "INPAINT_DenoiseToCompositingMask": nodes.DenoiseToCompositingMask,
-}
-NODE_DISPLAY_NAME_MAPPINGS = {
-    "INPAINT_LoadFooocusInpaint": "Load Fooocus Inpaint",
-    "INPAINT_ApplyFooocusInpaint": "Apply Fooocus Inpaint",
-    "INPAINT_VAEEncodeInpaintConditioning": "VAE Encode & Inpaint Conditioning",
-    "INPAINT_MaskedFill": "Fill Masked Area",
-    "INPAINT_MaskedBlur": "Blur Masked Area",
-    "INPAINT_LoadInpaintModel": "Load Inpaint Model",
-    "INPAINT_InpaintWithModel": "Inpaint (using Model)",
-    "INPAINT_ExpandMask": "Expand Mask",
-    "INPAINT_ShrinkMask": "Shrink Mask",
-    "INPAINT_StabilizeMask": "Stabilize Mask",
-    "INPAINT_DenoiseToCompositingMask": "Denoise to Compositing Mask",
-}
+
+class InpaintNodes(ComfyExtension):
+    async def get_node_list(self) -> list[type[io.ComfyNode]]:
+        return [
+            nodes.LoadFooocusInpaint,
+            nodes.ApplyFooocusInpaint,
+            nodes.VAEEncodeInpaintConditioning,
+            nodes.MaskedFill,
+            nodes.MaskedBlur,
+            nodes.LoadInpaintModel,
+            nodes.InpaintWithModel,
+            nodes.ExpandMask,
+            nodes.ShrinkMask,
+            nodes.StabilizeMask,
+            nodes.DenoiseToCompositingMask,
+        ]
+
+
+async def comfy_entrypoint():
+    return InpaintNodes()