feat: terminal log node

__init__.py

@@ -20,7 +20,7 @@ import nodes
 import torch
 
 
-version = "V1.11.5"
+version = "V1.12"
 print(f"### Loading: ComfyUI-Manager ({version})")
 
 
@@ -1517,6 +1517,28 @@ async def install_model(request):
     return web.Response(status=400)
 
 
+class ManagerTerminalHook:
+    def write_stderr(self, msg):
+        server.PromptServer.instance.send_sync("manager-terminal-feedback", {"data": msg})
+
+    def write_stdout(self, msg):
+        server.PromptServer.instance.send_sync("manager-terminal-feedback", {"data": msg})
+
+
+manager_terminal_hook = ManagerTerminalHook()
+
+
+@server.PromptServer.instance.routes.get("/manager/terminal")
+async def terminal_mode(request):
+    if "mode" in request.rel_url.query:
+        if request.rel_url.query['mode'] == 'true':
+            sys.__comfyui_manager_terminal_hook.add_hook('cm', manager_terminal_hook)
+        else:
+            sys.__comfyui_manager_terminal_hook.remove_hook('cm')
+
+    return web.Response(status=200)
+
+
 @server.PromptServer.instance.routes.get("/manager/preview_method")
 async def preview_method(request):
     if "value" in request.rel_url.query:

js/terminal.js

@@ -0,0 +1,80 @@
+import {app} from "../../scripts/app.js";
+import {ComfyWidgets} from "../../scripts/widgets.js";
+// Node that add notes to your project
+
+let terminal_node;
+
+app.registerExtension({
+	name: "Comfy.Manager.Terminal",
+
+	registerCustomNodes() {
+		class TerminalNode {
+			color = "#222222";
+			bgcolor = "#000000";
+			groupcolor = LGraphCanvas.node_colors.black.groupcolor;
+			constructor() {
+				if(terminal_node) {
+					try {
+						terminal_node.widgets[0].value = 'The output of this node is disabled because another terminal node has appeared.';
+						node.widgets[1].value = terminal_node.widgets[1].value;
+					}
+					catch {}
+				}
+
+				terminal_node = this;
+				this.logs = [];
+
+				if (!this.properties) {
+					this.properties = {};
+					this.properties.text="";
+				}
+
+				ComfyWidgets.STRING(this, "", ["", {default:this.properties.text, multiline: true}], app)
+				ComfyWidgets.BOOLEAN(this, "mode", ["", {default:true, label_on:'Logging', label_off:'Stop'}], app)
+				ComfyWidgets.INT(this, "lines", ["", {default:500, min:10, max:10000, steps:1}], app)
+
+				Object.defineProperty(this.widgets[1], 'value', {
+					set: (v) => {
+						api.fetchApi(`/manager/terminal?mode=${v}`, {});
+						this._value = v;
+					},
+					get: () => {
+						return this._value;
+					}
+				});
+
+				this.serialize_widgets = false;
+				this.isVirtualNode = true;
+			}
+		}
+
+		// Load default visibility
+		LiteGraph.registerNodeType(
+			"Terminal Log //CM",
+			Object.assign(TerminalNode, {
+				title_mode: LiteGraph.NORMAL_TITLE,
+				title: "Terminal Log (Manager)",
+				collapsable: true,
+			})
+		);
+
+		TerminalNode.category = "utils";
+	},
+});
+
+
+import { api } from "../../scripts/api.js";
+
+function terminalFeedback(event) {
+	if(terminal_node) {
+		terminal_node.logs.push(event.detail.data);
+		if(terminal_node.logs.length > terminal_node.widgets[2].value) {
+			terminal_node.logs.shift();
+			if(terminal_node.logs[0] == '' || terminal_node.logs[0] == '\n')
+				terminal_node.logs.shift();
+		}
+		terminal_node.widgets[0].value = [...terminal_node.logs].reverse().join('').trim();
+	}
+}
+
+api.addEventListener("manager-terminal-feedback", terminalFeedback);

prestartup_script.py

@@ -33,6 +33,36 @@ restore_snapshot_path = os.path.join(startup_script_path, "restore-snapshot.json
 git_script_path = os.path.join(comfyui_manager_path, "git_helper.py")
 
 
+class TerminalHook:
+    def __init__(self):
+        self.hooks = {}
+
+    def add_hook(self, k, v):
+        self.hooks[k] = v
+
+    def remove_hook(self, k):
+        if k in self.hooks:
+            del self.hooks[k]
+
+    def write_stderr(self, msg):
+        for v in self.hooks.values():
+            try:
+                v.write_stderr(msg)
+            except Exception:
+                pass
+
+    def write_stdout(self, msg):
+        for v in self.hooks.values():
+            try:
+                v.write_stdout(msg)
+            except Exception:
+                pass
+
+
+terminal_hook = TerminalHook()
+sys.__comfyui_manager_terminal_hook = terminal_hook
+
+
 def handle_stream(stream, prefix):
     stream.reconfigure(encoding=locale.getpreferredencoding(), errors='replace')
     for msg in stream:
@@ -157,9 +187,11 @@ try:
             if self.is_stdout:
                 original_stdout.write(message)
                 original_stdout.flush()
+                terminal_hook.write_stderr(message)
             else:
                 original_stderr.write(message)
                 original_stderr.flush()
+                terminal_hook.write_stdout(message)
 
         def flush(self):
             log_file.flush()