Refactor modules and add some image nodes

debug.py

@@ -0,0 +1,96 @@
+import sys
+import os
+import textwrap
+import code
+
+from . import register_node
+
+
+@register_node
+class Stringify:
+    """Convert any input to str/repr."""
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "x": ("*", ),
+            },
+        }
+
+    RETURN_TYPES = ("STRING", "STRING")
+    RETURN_NAMES = ("str", "repr")
+    FUNCTION = "go"
+    CATEGORY = "ali1234/debug"
+
+    def go(self, x):
+        return (str(x), repr(x))
+
+
+class RestoreStdStreams(object):
+    # ComfyUI-Manager patches sys.stdout and sys.stder
+    # which breaks GNU Readline support and makes the
+    # REPL annoying to use. This context manager temporarily
+    # puts back the originals.
+
+    def __enter__(self):
+        self._stdout, self._stderr = sys.stdout, sys.stderr
+        sys.stdout, sys.stderr = sys.__stdout__, sys.__stderr__
+        self._stdout.flush(); self._stderr.flush()
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        _stdout, _stderr = sys.stdout, sys.stderr
+        sys.stdout, sys.stderr = self._stdout, self._stderr
+        _stdout.flush(); _stderr.flush()
+
+
+EOF = 'Ctrl-Z plus Return' if os.sep == '\\' else 'Ctrl-D (i.e. EOF)'
+MESSAGE = f"Press {EOF} to continue processing."
+
+
+class Quitter:
+    # Replace exit() and quit() to make sure the user exits with EOF
+    # otherwise the whole server gets terminated.
+
+    def __repr__(self):
+        return MESSAGE
+
+    def __call__(self):
+        print(MESSAGE)
+
+
+@register_node
+class Interact:
+    """Opens an interactive REPL whenever the node is evaluated."""
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {},
+            "optional": {x: "*" for x in ('a', 'b', 'c', 'd')},
+        }
+
+    RETURN_TYPES = ()
+    RETURN_NAMES = ()
+    FUNCTION = "interact"
+    CATEGORY = "ali1234/debug"
+    OUTPUT_NODE = True
+
+    def interact(self, **kwargs):
+        if sys.__stdout__.isatty():
+            with RestoreStdStreams():
+                code.interact(
+                    banner=textwrap.dedent(f"""
+                        Interactive debugging started.
+                        Try `print(a)`.
+                        {MESSAGE}
+                    """),
+                    exitmsg="Resuming workflow...",
+                    local={
+                        **kwargs,
+                        'quit': Quitter(),
+                        'exit': Quitter()
+                    }
+                )
+        else:
+            # Don't block the server if there is no tty.
+            print("Skipping interactive prompt because there is no tty.")
+        return ()