Publish current ComfyUI O1Key code baseline

Replace the prior release tree with the current plugin, frontend, tests, and documentation. Document retired node IDs and the public Gitea update source.
2026-09-24 19:56:48 +08:00
parent 3e337722ab
commit ba920f2b66
183 changed files with 49496 additions and 9934 deletions
@@ -0,0 +1,89 @@
+import assert from "node:assert/strict";
+import { readFileSync } from "node:fs";
+import { runInNewContext } from "node:vm";
+
+const source = readFileSync(new URL("../web/js/omniFlashVideo.js", import.meta.url), "utf8")
+    .replace(/^import .*;\s*/m, "");
+const migrationSource = readFileSync(new URL("../web/js/migrateWorkflow.js", import.meta.url), "utf8")
+    .replace(/^import .*;\s*/gm, "");
+let extension;
+const queued = [];
+const app = {
+    registerExtension(value) { extension = value; },
+    async queuePrompt(...args) { queued.push(args); },
+};
+runInNewContext(source, { app });
+const node = {
+    comfyClass: "O1keyOmniFlashVideo", id: 42, widgets: [],
+    addWidget(_type, name, _value, callback, options) {
+        const widget = { name, callback, options };
+        this.widgets.push(widget);
+        return widget;
+    },
+};
+extension.nodeCreated(node);
+extension.loadedGraphNode(node);
+assert.equal(node.widgets.length, 1);
+assert.equal(node.widgets[0].name, "开始生成");
+assert.equal(node.widgets[0].options.serialize, false);
+await node.widgets[0].callback();
+assert.deepEqual(JSON.parse(JSON.stringify(queued)), [[0, 1, [42]]]);
+
+const mode = { name: "生成模式", value: "文生视频", callback() {} };
+const autogrow = { names: ["参考图片1", "参考图片2"], min: 0 };
+const removedLinks = [];
+const videoNode = {
+    comfyClass: "O1keyOmniFlashVideo", id: 43, widgets: [mode],
+    comfyDynamic: { autogrow: { 参考图片: autogrow } },
+    inputs: [
+        { name: "参考图片.参考图片1", type: "IMAGE", link: null },
+        { name: "首帧图片", type: "IMAGE", link: null },
+        { name: "尾帧图片", type: "IMAGE", link: null },
+        { name: "源视频", type: "VIDEO", link: null },
+        { name: "提示词", type: "STRING", link: 99 },
+    ],
+    addWidget: node.addWidget,
+    addInput(name, type, options) { this.inputs.push({ name, type, ...options, link: null }); },
+    removeInput(index) {
+        if (this.inputs[index].link != null) removedLinks.push(this.inputs[index].link);
+        this.onConnectionsChange?.(1, index, false);
+        this.inputs.splice(index, 1);
+    },
+};
+const names = () => videoNode.inputs.map((input) => input.name);
+extension.nodeCreated(videoNode);
+assert.deepEqual(names(), ["提示词"]);
+mode.value = "参考图视频";
+mode.callback(mode.value);
+assert.deepEqual(names(), ["提示词", "参考图片.参考图片1"]);
+videoNode.inputs.at(-1).link = 12;
+mode.value = "视频编辑";
+mode.callback(mode.value);
+assert.deepEqual(names(), ["提示词", "参考图片.参考图片1", "源视频"]);
+assert.equal(videoNode.inputs.find((input) => input.name === "参考图片.参考图片1").link, 12);
+mode.value = "首尾帧";
+mode.callback(mode.value);
+assert.deepEqual(names(), ["提示词", "首帧图片", "尾帧图片"]);
+assert.deepEqual(removedLinks, [12]);
+assert.equal(videoNode.comfyDynamic.autogrow.参考图片, undefined);
+extension.loadedGraphNode(videoNode);
+assert.deepEqual(names(), ["提示词", "首帧图片", "尾帧图片"]);
+mode.value = "文生视频";
+mode.callback(mode.value);
+assert.deepEqual(names(), ["提示词"]);
+mode.value = "参考图视频";
+mode.callback(mode.value);
+assert.equal(videoNode.comfyDynamic.autogrow.参考图片, autogrow);
+assert.deepEqual(names(), ["提示词", "参考图片.参考图片1"]);
+
+let migrationExtension;
+runInNewContext(migrationSource, {
+    app: { registerExtension(extension) { migrationExtension = extension; } },
+    console: { log() {}, warn() {} },
+});
+const legacyValues = ["提示词", "文生视频", "omni_flash_8s", "1080p", "9:16"];
+const graph = { nodes: [{ type: "O1keyOmniFlashVideo", widgets_values: legacyValues }] };
+migrationExtension.beforeConfigureGraph(graph);
+assert.deepEqual(legacyValues, ["提示词", "文生视频", "1080p", "9:16"]);
+migrationExtension.beforeConfigureGraph(graph);
+assert.deepEqual(legacyValues, ["提示词", "文生视频", "1080p", "9:16"]);