Jony
ba920f2b66
Replace the prior release tree with the current plugin, frontend, tests, and documentation. Document retired node IDs and the public Gitea update source.
143 lines
9.2 KiB
JavaScript
143 lines
9.2 KiB
JavaScript
import assert from "node:assert/strict";
|
||
import fs from "node:fs";
|
||
import path from "node:path";
|
||
import { fileURLToPath } from "node:url";
|
||
|
||
const here = path.dirname(fileURLToPath(import.meta.url));
|
||
const source = fs.readFileSync(path.join(here, "..", "web", "js", "o1keyVideoGenerator.js"), "utf8");
|
||
|
||
assert.match(source, /const GENERATOR = "O1keyVideoGenerator"/);
|
||
assert.match(source, /const RESULT = "O1keyVideoResult"/);
|
||
assert.match(source, /const SAVE_VIDEO = "SaveVideo"/);
|
||
assert.match(source, /const SAVE_IMAGE = "SaveImage"/);
|
||
assert.match(source, /import \{ openReferenceImageEditor \} from "\.\/o1keyReferenceImageEditor\.js"/);
|
||
assert.match(source, /import "\.\/o1keyImageGenerator\.js"/);
|
||
assert.doesNotMatch(source, /LiteGraph\.createNode\(RESULT\)/);
|
||
assert.match(source, /LiteGraph\.createNode\(SAVE_VIDEO\)/);
|
||
assert.match(source, /request\.return_last_frame \? LiteGraph\.createNode\(SAVE_IMAGE\) : null/);
|
||
assert.match(source, /source\.connect\(0, saveVideo, 0\)/);
|
||
assert.match(source, /source\.connect\(1, saveImage, 0\)/);
|
||
assert.match(source, /function createNativeOutputNodes\(source, request, batchId\)/);
|
||
assert.match(source, /function dispatchNativePreview\(node, descriptor, animated = false\)/);
|
||
assert.match(source, /api\.dispatchCustomEvent\("executed"/);
|
||
assert.match(source, /output\.animated = \[true\]/);
|
||
assert.match(source, /async function submitNativeJob\(source, targets, request, batchId\)/);
|
||
assert.match(source, /result_node_id: Number\(targets\.saveVideo\.id\)/);
|
||
assert.match(source, /createNativeOutputNodes\(node, request, batchId\)/);
|
||
assert.match(source, /function recoverNativeSaveVideo\(node\)/);
|
||
assert.match(source, /recoverNativeSaveVideo\(node\)/);
|
||
assert.match(source, /String\(result\.properties\?\.o1keyVideoBatchId \|\| ""\) === batchId/);
|
||
assert.match(source, /requestJson\("\/o1key\/video\/jobs"/);
|
||
assert.match(source, /api\.addEventListener\("o1key\.video_job"/);
|
||
assert.match(source, /run\.disabled = false/);
|
||
assert.doesNotMatch(source, /独立任务 · 可连续提交 · 提交即运行|o1vg-summary/);
|
||
assert.doesNotMatch(source, /最多并行|等待后台槽位/);
|
||
assert.match(source, /value: "overseas_hc", label: "海外"/);
|
||
assert.doesNotMatch(source, /label: "海外 HC"/);
|
||
assert.match(source, /const ASSET_CREATION_OPTIONS/);
|
||
assert.match(source, /value: "auto", label: "自动创建"/);
|
||
assert.match(source, /value: "manual", label: "手动"/);
|
||
assert.match(source, /makeField\("素材创建", assetCreation\)/);
|
||
assert.match(source, /assetCreationField\.style\.display = isSeedance \? "grid" : "none"/);
|
||
assert.match(source, /assetCreation\._trigger\.disabled = !assetModeAvailable/);
|
||
assert.match(source, /imageIdField\.style\.display = manualAssets \? "grid" : "none"/);
|
||
assert.match(source, /videoIdField\.style\.display = manualAssets && activeMode === "multimodal"/);
|
||
assert.match(source, /asset_creation_mode: manualAssets \? "manual" : "auto"/);
|
||
assert.match(source, /assets: manualAssets \? manualRequestAssets : \{\}/);
|
||
assert.match(source, /function requestForRetry\(node\)/);
|
||
assert.match(source, /request\.asset_creation_mode !== "auto"/);
|
||
assert.match(source, /return \{ \.\.\.request, asset_creation_mode: "manual", media: \{\}, assets \}/);
|
||
assert.match(source, /resolved_assets: detail\.resolved_assets \|\| null/);
|
||
assert.match(source, /const STAGE_LABELS = \{ starting: "正在启动"/);
|
||
assert.match(source, /STAGE_LABELS\[detail\.stage\] \|\| detail\.stage/);
|
||
assert.match(source, /const SEEDANCE_REFERENCE_LIMITS = Object\.freeze/);
|
||
assert.match(source, /minDimension: 300/);
|
||
assert.match(source, /maxDimension: 6000/);
|
||
assert.match(source, /minVideoPixels: 407696/);
|
||
assert.match(source, /maxVideoPixels: 8295044/);
|
||
assert.match(source, /async function validateReferenceImageFile\(file\)/);
|
||
assert.match(source, /async function validateReferenceVideoFile\(file\)/);
|
||
assert.match(source, /if \(kind === "image"\) await validateReferenceImageFile\(file\); else if \(kind === "video"\) await validateReferenceVideoFile\(file\)/);
|
||
assert.match(source, /宽高 300～6000px · 比例 0\.4～2\.5/);
|
||
assert.match(source, /title: "参考视频", shortTitle: "视频", hint: "40\.77万～829\.50万像素"/);
|
||
assert.doesNotMatch(source, /参考图至少需要 480p|≥ 41万像素/);
|
||
assert.match(source, /window\.o1keyCanvasImagePicker/);
|
||
assert.match(source, /canvasPick\.append\(el\("b", "", "▣"\), el\("span", "", "画布取图"\)\)/);
|
||
assert.match(source, /actions\.append\(canvasPick\)/);
|
||
assert.match(source, /const file = await picker\.readFile\(descriptor\)/);
|
||
assert.match(source, /media\[key\] = multiple \? \[\.\.\.\(media\[key\] \|\| \[\]\), uploaded\] : uploaded/);
|
||
assert.doesNotMatch(source, /真人素材 ID/);
|
||
|
||
// Keep the video node on the same visual component language as the image node.
|
||
assert.match(source, /\.o1vg-field/);
|
||
assert.match(source, /\.o1vg-select-trigger/);
|
||
assert.match(source, /\.o1vg-thumb/);
|
||
assert.match(source, /\.o1vg-generate/);
|
||
assert.match(source, /\.o1vg-prompt-write/);
|
||
assert.match(source, /\.o1vg-prompt-wrap\{[^\n]*height:auto;min-height:142px;flex:1 1 142px/);
|
||
assert.match(source, /\.o1vg-prompt\{[\s\S]*?min-height:142px;max-height:none;resize:none/);
|
||
assert.match(source, /promptWrite\.title = "AI帮写"/);
|
||
assert.match(source, /promptWrite\.append\(promptWriteIcon, el\("span", "", "AI帮写"\)\)/);
|
||
assert.match(source, /requestJson\("\/o1key\/video\/prompt-write"/);
|
||
assert.match(source, /setPromptWriteStatus\(node, "AI帮写中…", "busy"\)/);
|
||
assert.match(source, /role: "首帧"/);
|
||
assert.match(source, /role: index === 0 \? "首帧" : "尾帧"/);
|
||
assert.match(source, /reference_video_count:/);
|
||
assert.match(source, /reference_audio_count:/);
|
||
assert.match(source, /function makeVideoPreview\(item, title\)/);
|
||
assert.match(source, /video\.preload = "metadata"/);
|
||
assert.match(source, /video\.currentTime = previewTime/);
|
||
assert.match(source, /video\.src = descriptorUrl\(item\)/);
|
||
assert.match(source, /releaseVideoPreviews\(track\)/);
|
||
assert.match(source, /video\.removeAttribute\("src"\)/);
|
||
assert.match(source, /if \(kind === "video"\) tile\.append\(makeVideoPreview\(item, title\)\)/);
|
||
assert.doesNotMatch(source, /drawImage\([^)]*video|canvas\.toDataURL/);
|
||
assert.match(source, /function makeSelect/);
|
||
assert.match(source, /function mediaDropDestination/);
|
||
assert.match(source, /function moveMediaItem/);
|
||
assert.match(source, /function editMediaImage/);
|
||
assert.match(source, /openReferenceImageEditor\(\{/);
|
||
assert.match(source, /addEventListener\("dragstart"/);
|
||
assert.match(source, /addEventListener\("dragover"/);
|
||
assert.match(source, /addEventListener\("drop"/);
|
||
assert.match(source, /\.o1vg-thumb-edit/);
|
||
assert.match(source, /\.o1vg-thumb-index/);
|
||
assert.match(source, /function makeMultimodalMediaSection\(node\)/);
|
||
assert.match(source, /meta\.append\(el\("strong", "", "参考素材"\), hintElement\)/);
|
||
assert.match(source, /upload\.append\(el\("b", "", "＋"\), el\("span", "", "上传"\)\)/);
|
||
assert.match(source, /input\.accept = sources\.map\(\(source\) => source\.accept\)\.join\(","\)/);
|
||
assert.match(source, /const sourceForFile = \(file\) =>/);
|
||
assert.match(source, /addFiles\(files\)/);
|
||
const multimodalSource = source.slice(source.indexOf("function makeMultimodalMediaSection"), source.indexOf("function nativeSavePrefix"));
|
||
assert.equal((multimodalSource.match(/actions\.append\(/g) || []).length, 1);
|
||
assert.doesNotMatch(multimodalSource, /canvasPick/);
|
||
assert.doesNotMatch(multimodalSource, /o1vg-empty/);
|
||
assert.doesNotMatch(multimodalSource, /添加\$\{source\.shortTitle\}/);
|
||
assert.doesNotMatch(multimodalSource, /sourceInput|source\.input/);
|
||
assert.match(source, /\.o1vg-multimodal-assets \.o1vg-add\{min-width:58px;height:29px;gap:4px;padding:0 10px;[^}]*font-size:11px;font-weight:600/);
|
||
assert.match(source, /mediaHost\.append\(firstSection, lastSection, multimodalSection\)/);
|
||
assert.match(source, /multimodalSection\.style\.display = !manualAssets && activeMode === "multimodal" \? "flex" : "none"/);
|
||
assert.match(source, /automaticHeights = \{ text: 679, first_frame: 824, first_last_frame: 964, multimodal: 824 \}/);
|
||
assert.match(source, /widget\(node, "generation_mode"\)\?\.value \|\| "multimodal"/);
|
||
assert.match(source, /const restore = \(\) =>/);
|
||
assert.match(source, /mode\.value = value\("generation_mode", "multimodal"\)/);
|
||
assert.match(source, /firstSection\._render\?\.\(\)/);
|
||
assert.match(source, /multimodalSection\._render\?\.\(\)/);
|
||
assert.match(source, /node\._o1vgRestore = restore/);
|
||
assert.match(source, /nodeType\.prototype\.onConfigure = function \(\)/);
|
||
assert.match(source, /this\._o1vgRestore\?\.\(\)/);
|
||
assert.match(source, /loadedGraphNode\(node\).*node\._o1vgRestore\?\.\(\)/s);
|
||
assert.doesNotMatch(source, /const imagesSection = makeMediaSection/);
|
||
assert.doesNotMatch(source, /const videosSection = makeMediaSection/);
|
||
assert.doesNotMatch(source, /const audiosSection = makeMediaSection/);
|
||
assert.match(source, /function applyGeneratorDefaultSize/);
|
||
assert.match(source, /node\.setSize\?\.\(\[560, 679\]\)/);
|
||
assert.match(source, /target\.hidden = true/);
|
||
assert.match(source, /target\.options\.hidden = true/);
|
||
assert.doesNotMatch(source, /document\.createElement\("select"\)/);
|
||
|
||
// A click creates an independent O1Key job; it must never enter ComfyUI's native queue.
|
||
assert.doesNotMatch(source, /queuePrompt|app\.queuePrompt|fetchApi\("\/prompt"/);
|
||
|
||
console.log("o1key video generator frontend checks passed");
|