-
-
+
-
尚未扫描
`;
+
等待扫描当前工作流
+
+
尚未扫描
+
+
+
+
+
+
外部模型库指定本机外部模型目录，只精确匹配当前工作流缺失的模型文件名。
+
+
+
+
+
+
`;
document.body.appendChild(panel);
- panel.querySelector('[data-action="close"]').onclick = () => panel.classList.remove("open");
- panel.querySelector('[data-action="scan"]').onclick = () => scan(false);
+ const savedWidth = Number(localStorage.getItem("findmodels.panelWidth"));
+ if (Number.isFinite(savedWidth) && savedWidth >= 420) panel.style.width = `${savedWidth}px`;
+ enablePanelResize(panel);
+ panel.querySelector('[data-tab-panel="settings"]')?.insertAdjacentHTML("beforeend", `
+
夸克链接查找下载来源时后台精确搜索以下模型库；公开分享大文件受限时会优先回退到可验证的快速直链。
+
+
+
+
+
+
+
正在检查夸克登录态...
+
`);
+ get("/findmodels/quark-auth").then((result) => {
+ const status = panel.querySelector("[data-quark-auth-status]");
+ if (status) status.textContent = result.configured ? "已保存夸克登录态，可重试直链下载。" : "未保存夸克登录态，公开分享大文件可能被夸克限制。";
+ }).catch(() => {});
+ panel.querySelectorAll("[data-tab]").forEach((button) => {
+ button.onclick = () => setActiveTab(panel, button.dataset.tab);
+ });
+ setActiveTab(panel, activeTab);
+ panel.querySelector('[data-action="close"]').onclick = () => {
+ panelUserOpened = false;
+ closeDockedPanel(panel);
+ };
+ panel.querySelectorAll('[data-action="scan"]').forEach((button) => {
+ button.onclick = () => scheduleScan(0, true);
+ });
+ panel.querySelector('[data-action="refresh-downloads"]').onclick = () => refreshDownloadJobs();
+ panel.querySelector('[data-action="install-missing-nodes"]').onclick = async () => {
+ const button = panel.querySelector('[data-action="install-missing-nodes"]');
+ button.disabled = true;
+ let installed = 0;
+ const failures = [];
+ const packages = lastResult?.missing_node_packages || [];
+ for (const nodePackage of packages) {
+ const nodeType = nodePackage.node_types?.[0] || "";
+ const packageId = nodePackage.known ? nodePackage.id : "";
+ const activityId = packageId || nodeType;
+ nodeActivities.set(activityId, { title: nodePackage.title, status: "downloading", message: "正在查找可信插件来源…" });
+ await refreshDownloadJobs();
+ button.textContent = `正在查找 ${nodePackage.title}…`;
+ try {
+ const found = await post("/findnodes/candidates", { node_type: nodeType, package_id: packageId });
+ const candidate = (found.candidates || []).find((item) => [
+ "workflow_package_id",
+ "workflow_package_github",
+ "official_node_mapping",
+ "comfy_manager_node_mapping",
+ "comfy_manager_github",
+ ].includes(item.reason));
+ if (!candidate) throw new Error("没有找到由工作流 aux_id 或 ComfyUI-Manager 官方映射确认的 GitHub 仓库");
+ button.textContent = `正在安装 ${nodePackage.title}…`;
+ nodeActivities.set(activityId, { title: nodePackage.title, status: "downloading", message: "正在检查依赖并安装…" });
+ await refreshDownloadJobs();
+ await post("/findnodes/install", {
+ node_type: nodeType,
+ package_id: packageId,
+ plugin_id: candidate.id,
+ install_dependencies: installDependenciesEnabled(panel),
+ });
+ resolvedNodePackages.add(activityId);
+ nodeActivities.set(activityId, { title: nodePackage.title, status: "completed", message: "安装完成，重启 ComfyUI 后生效" });
+ installed += 1;
+ } catch (error) {
+ nodeActivities.set(activityId, { title: nodePackage.title, status: "failed", message: `安装失败：${error.message}` });
+ failures.push(`${nodePackage.title}: ${error.message}`);
+ }
+ await refreshDownloadJobs();
+ }
+ button.disabled = false;
+ button.textContent = "安装缺失节点";
+ panel.querySelector(".fm-summary").textContent =
+ `已安装或更新 ${installed} 个精确匹配插件。${failures.length ? `失败 ${failures.length} 个：${failures.join("；")}` : "请重启 ComfyUI。"}`
+ if (installed) {
+ render(lastResult, true);
+ scheduleScan(0, true);
+ }
+ };
+ panel.querySelector('[data-action="select-external-folder"]').onclick = async () => {
+ const button = panel.querySelector('[data-action="select-external-folder"]');
+ const input = panel.querySelector("[data-external-folder]");
+ button.disabled = true;
+ button.textContent = "等待选择…";
+ try {
+ const result = await post("/findmodels/external-folder/select", {});
+ if (!result.cancelled) {
+ input.value = result.path;
+ scheduleScan(0, true);
+ scheduleEnrichedScan();
+ }
+ } catch (error) {
+ panel.querySelector(".fm-summary").textContent = `选择外部模型库失败：${error.message}`;
+ } finally {
+ button.disabled = false;
+ button.textContent = "选择文件夹";
+ }
+ };
+ panel.querySelector('[data-action="install-custom-github"]').onclick = async () => {
+ const button = panel.querySelector('[data-action="install-custom-github"]');
+ const input = panel.querySelector("[data-custom-github]");
+ const nodePackage = lastResult?.missing_node_packages?.[0];
+ const nodeType = nodePackage?.node_types?.[0] || "CustomNode";
+ if (!input.value.trim()) {
+ panel.querySelector(".fm-summary").textContent = "请输入完整的 GitHub 插件仓库链接。";
+ return;
+ }
+ button.disabled = true;
+ button.textContent = "正在安装…";
+ try {
+ const result = await post("/findnodes/install", {
+ node_type: nodeType,
+ package_id: nodePackage?.known ? nodePackage.id : "custom-github",
+ repo_url: input.value.trim(),
+ install_dependencies: installDependenciesEnabled(panel),
+ });
+ input.value = "";
+ panel.querySelector(".fm-summary").textContent = `${result.title} 已安装，请重启 ComfyUI。`;
+ await refreshDownloadJobs();
+ } catch (error) {
+ panel.querySelector(".fm-summary").textContent = `GitHub 插件安装失败：${error.message}`;
+ } finally {
+ button.disabled = false;
+ button.textContent = "链接安装插件";
+ }
+ };
+ panel.querySelector('[data-action="save-quark-cookie"]').onclick = async () => {
+ const input = panel.querySelector("[data-quark-cookie]");
+ const status = panel.querySelector("[data-quark-auth-status]");
+ try {
+ const result = await post("/findmodels/quark-auth", { cookie: input.value });
+ input.value = "";
+ status.textContent = result.configured ? "已保存夸克登录态，可重试直链下载。" : "Cookie 为空，未保存登录态。";
+ } catch (error) {
+ status.textContent = `保存夸克登录态失败：${error.message}`;
+ }
+ };
+ panel.querySelector('[data-action="clear-quark-cookie"]').onclick = async () => {
+ const input = panel.querySelector("[data-quark-cookie]");
+ const status = panel.querySelector("[data-quark-auth-status]");
+ try {
+ await post("/findmodels/quark-auth", { cookie: "" });
+ input.value = "";
+ status.textContent = "已清除夸克登录态。";
+ } catch (error) {
+ status.textContent = `清除夸克登录态失败：${error.message}`;
+ }
+ };
panel.querySelector('[data-action="adapt"]').onclick = async () => {
const button = panel.querySelector('[data-action="adapt"]');
button.disabled = true;
@@ -228,15 +794,114 @@ function ensurePanel() {
let failed = 0;
for (const model of lastResult?.models || []) {
if (!model.match?.auto_apply) continue;
- if (await applyMatch(model)) count += 1;
+ if (await applyMatchEverywhere(model)) count += 1;
else failed += 1;
}
panel.querySelector(".fm-summary").textContent =
`已加载 ${count} 个模型${failed ? `，${failed} 个加载失败` : ""}。请检查节点后保存工作流。`;
button.disabled = false;
- if (count) window.setTimeout(() => scan(false), 100);
+ if (count) {
+ for (const model of [...(lastResult?.models || [])]) {
+ if (model.match?.auto_apply) removeResolvedModel(model.name);
+ }
+ scheduleScan(0, true);
+ }
};
panel.addEventListener("click", async (event) => {
+ const copyButton = event.target.closest("[data-copy-model-name]");
+ if (copyButton) {
+ try {
+ await copyText(copyButton.dataset.copyModelName);
+ const original = copyButton.textContent;
+ copyButton.textContent = "已复制";
+ window.setTimeout(() => { copyButton.textContent = original; }, 1200);
+ } catch (error) {
+ panel.querySelector(".fm-summary").textContent = `复制模型名称失败：${error.message}`;
+ }
+ return;
+ }
+
+ const locateButton = event.target.closest("[data-locate-node]");
+ if (locateButton) {
+ if (!locateNode(locateButton.dataset.locateNode)) {
+ panel.querySelector(".fm-summary").textContent = "无法定位节点，节点可能已被删除。";
+ }
+ return;
+ }
+
+ const controlButton = event.target.closest("[data-download-control]");
+ if (controlButton) {
+ controlButton.disabled = true;
+ try {
+ await post(`/findmodels/download/${controlButton.dataset.downloadControl}`, {
+ job_id: controlButton.dataset.jobId,
+ });
+ await refreshDownloadJobs();
+ } catch (error) {
+ panel.querySelector(".fm-summary").textContent = `下载任务操作失败：${error.message}`;
+ controlButton.disabled = false;
+ }
+ return;
+ }
+
+ const nodeRefreshButton = event.target.closest("[data-node-refresh]");
+ if (nodeRefreshButton) {
+ nodeRefreshButton.disabled = true;
+ nodeRefreshButton.textContent = "正在查找…";
+ const target = nodeRefreshButton.closest(".fm-item").querySelector(".fm-node-candidates");
+ try {
+ const data = await post("/findnodes/candidates", {
+ node_type: nodeRefreshButton.dataset.nodeRefresh,
+ package_id: nodeRefreshButton.dataset.packageId,
+ });
+ target.innerHTML = nodeCandidatesHtml(
+ data.node_type,
+ data.candidates || [],
+ data.github_search_url,
+ data.package_id,
+ );
+ } catch (error) {
+ nodeRefreshButton.disabled = false;
+ nodeRefreshButton.textContent = "重新查找插件";
+ panel.querySelector(".fm-summary").textContent = `查找缺失节点插件失败：${error.message}`;
+ }
+ return;
+ }
+
+ const moveButton = event.target.closest("[data-external-move]");
+ if (moveButton) {
+ moveButton.disabled = true;
+ moveButton.textContent = "正在剪切…";
+ try {
+ const moved = await post("/findmodels/external-move", {
+ source: moveButton.dataset.externalMove,
+ name: moveButton.dataset.name,
+ category: moveButton.dataset.category,
+ });
+ const model = (lastResult?.models || []).find((item) => item.name === moveButton.dataset.name);
+ const references = model?.referencing_nodes?.length
+ ? model.referencing_nodes
+ : [{ node_id: moveButton.dataset.nodeId, widget: moveButton.dataset.widget }];
+ for (const reference of references) {
+ await applyMatch({
+ ...reference,
+ name: moveButton.dataset.name,
+ match: { name: moved.relative_name },
+ });
+ }
+ removeResolvedModel(moveButton.dataset.name);
+ panel.querySelector(".fm-summary").textContent =
+ `已剪切 ${moved.relative_name} 到 ${moved.category} 模型目录。`;
+ scheduleScan(0, true);
+ scheduleEnrichedScan();
+ } catch (error) {
+ moveButton.disabled = false;
+ moveButton.textContent = "剪切失败，重试";
+ panel.querySelector(".fm-summary").textContent = `剪切模型失败：${error.message}`;
+ }
+ return;
+ }
+
const installButton = event.target.closest("[data-node-install]");
if (installButton) {
installButton.disabled = true;
@@ -244,13 +909,25 @@ function ensurePanel() {
try {
const result = await post("/findnodes/install", {
node_type: installButton.dataset.nodeType,
+ package_id: installButton.dataset.packageId,
plugin_id: installButton.dataset.nodeInstall,
+ install_dependencies: installDependenciesEnabled(panel),
+ });
+ const activityId = installButton.dataset.packageId || installButton.dataset.nodeType;
+ resolvedNodePackages.add(activityId);
+ nodeActivities.set(activityId, {
+ title: result.title,
+ status: "completed",
+ message: "安装完成，重启 ComfyUI 后生效",
});
installButton.textContent = result.action === "updated" ? "更新完成，需重启" : "安装完成，需重启";
panel.querySelector(".fm-summary").textContent =
`${result.title} 已${result.action === "updated" ? "更新" : "安装"}；`
+ `${result.new_conflicts?.length ? `发现 ${result.new_conflicts.length} 个新增依赖冲突，请查看终端。` : "未发现新增依赖冲突。"}`
+ "请重启 ComfyUI。";
+ await refreshDownloadJobs();
+ render(lastResult, true);
+ scheduleScan(0, true);
} catch (error) {
installButton.disabled = false;
installButton.textContent = "安装失败，重试";
@@ -270,13 +947,23 @@ function ensurePanel() {
name: sourceButton.dataset.source,
category: model.category,
});
+ if (data.size) {
+ model.size = data.size;
+ const sizeLabel = sourceButton.closest(".fm-item").querySelector("[data-model-size]");
+ if (sizeLabel) sizeLabel.textContent = `大小：${formatSize(data.size)}`;
+ }
const links = data.candidates.map((item) => sourceHtml(item, model)).join("");
- target.innerHTML = links || "
未找到文件名完全一致的可下载模型";
+ const searches = data.search_urls
+ ? `
`
+ : "";
+ target.innerHTML = links
+ ? `${data.exact_count ? "" : '
未找到完全同名文件，以下为高置信相似候选，需人工核对。'}${links}${searches}`
+ : `
未找到完全同名或高置信候选。${searches}`;
} catch (error) {
target.textContent = `获取下载项失败：${error.message}`;
} finally {
sourceButton.disabled = false;
- sourceButton.textContent = "下载缺失模型";
+ sourceButton.textContent = "查找下载来源";
}
return;
}
@@ -310,43 +997,82 @@ function ensurePanel() {
function render(result, quiet) {
const panel = ensurePanel();
+ result.models = (result.models || []).filter((model) => !resolvedModels.has(normalizedModelValue(model.name)));
+ result.summary = { ...result.summary, unresolved: result.models.length };
lastResult = result;
const summary = result.summary;
const missingNodes = result.missing_nodes || [];
+ const missingNodePackages = (result.missing_node_packages || []).filter((nodePackage) => {
+ const nodeType = nodePackage.node_types?.[0] || "";
+ return !resolvedNodePackages.has(nodePackage.id) && !resolvedNodePackages.has(nodeType);
+ });
const missingNodeCandidates = result.missing_node_candidates || {};
- updateToolbarButton(summary.unresolved + missingNodes.length);
- panel.querySelector(".fm-summary").textContent =
- `缺失节点：${missingNodes.length}；未加载模型：${summary.unresolved}`;
- const nodeHtml = missingNodes.map((nodeType) => `
+ updateToolbarButton(summary.unresolved + missingNodePackages.length);
+ panel.querySelector('[data-tab-count="models"]').textContent = summary.unresolved;
+ panel.querySelector('[data-tab-count="nodes"]').textContent = missingNodePackages.length;
+ panel.querySelector('[data-section-count="models"]').textContent = summary.unresolved;
+ panel.querySelector('[data-section-count="nodes"]').textContent = missingNodePackages.length;
+ const scanStatus = panel.querySelector("[data-scan-status]");
+ if (scanStatus) {
+ scanStatus.textContent = result.quick
+ ? "已快速识别，正在补充外部模型库候选…"
+ : "当前工作流扫描完成";
+ }
+ const externalInput = panel.querySelector("[data-external-folder]");
+ if (externalInput && document.activeElement !== externalInput) {
+ externalInput.value = result.external_folder || "";
+ }
+ panel.querySelector(".fm-summary").innerHTML =
+ `
✓当前页面仅显示尚未解决的依赖项`;
+ const nodeHtml = missingNodePackages.map((nodePackage) => {
+ const nodeType = nodePackage.node_types?.[0] || "";
+ const nodeIds = nodePackage.node_ids || [];
+ return `
- ${escapeHtml(nodeType)}缺失节点
- ${nodeCandidatesHtml(nodeType, missingNodeCandidates[nodeType] || [])}
- `).join("");
- panel.querySelector(".fm-list").innerHTML = nodeHtml + result.models.map((model) => `
-
- ${escapeHtml(model.name)}${escapeHtml(model.category)} · ${escapeHtml(model.official_missing ? "官方判定缺失" : statusText(model.status))}
- ${model.match ? `本地候选：${escapeHtml(model.match.name)} (${Math.round(model.match.confidence * 100)}%) ${model.match.auto_apply ? `` : ""}
` : ""}
- ${model.status === "missing" ? `` : ""}
- `).join("") || "
当前工作流中未发现缺失节点或模型。
";
+
!${escapeHtml(nodePackage.title)}${nodePackage.known ? "缺失节点包" : "未知包"}
+
${nodePackage.count} 个节点${nodePackage.version ? `${escapeHtml(nodePackage.version)}` : ""}
+
${(nodePackage.node_types || []).map((item) => `${escapeHtml(item)}`).join("")}
+ ${nodeIds[0] ? `
` : ""}
+
${nodeCandidatesHtml(nodeType, missingNodeCandidates[nodeType] || [], "", nodePackage.known ? nodePackage.id : "")}
+ `;
+ }).join("");
+ const modelsHtml = (result.models || []).map(modelCardHtml).join("");
+ panel.querySelector(".fm-model-list").innerHTML = modelsHtml
+ ? `
文件名类型 / 引用 / 本地候选操作
${modelsHtml}`
+ : '
未发现缺失模型当前工作流模型已就绪
';
+ panel.querySelector(".fm-node-list").innerHTML = nodeHtml
+ ? `
包名（节点集）引用 / 来源操作
${nodeHtml}`
+ : '
未发现缺失节点当前工作流节点均已注册
';
panel.querySelectorAll("[data-apply]").forEach((button) => {
button.onclick = async () => {
const model = result.models.find((item) => `${item.node_id}:${item.widget}` === button.dataset.apply);
- if (model && await applyMatch(model)) window.setTimeout(() => scan(false), 100);
+ if (model && await applyMatchEverywhere(model)) {
+ removeResolvedModel(model.name);
+ scheduleScan(0, true);
+ }
};
});
- if (!quiet) panel.classList.add("open");
}
-async function scan(quiet = false) {
+async function scan(quiet = false, { quick = false } = {}) {
const panel = ensurePanel();
+ const snapshot = workflowSnapshot();
+ const signature = workflowSignature(snapshot);
+ const requestId = ++scanRequestId;
updateToolbarButton(null, "扫描中");
- panel.querySelector(".fm-summary").textContent = "正在扫描…";
+ const scanStatus = panel.querySelector("[data-scan-status]");
+ if (scanStatus) scanStatus.textContent = quick ? "正在快速识别当前工作流…" : "正在补充外部模型库候选…";
+ if (!quiet && !lastResult) panel.querySelector(".fm-summary").textContent = "正在扫描…";
try {
- render(await post("/findmodels/scan", workflowSnapshot()), quiet);
+ const result = await post("/findmodels/scan", { ...snapshot, quick });
+ if (requestId !== scanRequestId || signature !== workflowSignature(workflowSnapshot())) return;
+ lastWorkflowSignature = signature;
+ render(result, quiet);
+ if (quick) scheduleEnrichedScan();
} catch (error) {
+ if (requestId !== scanRequestId) return;
updateToolbarButton(null, "扫描失败");
- if (!quiet) panel.classList.add("open");
- panel.querySelector(".fm-summary").textContent = `扫描失败：${error.message}`;
+ if (scanStatus) scanStatus.textContent = `扫描失败：${error.message}`;
}
}
@@ -371,27 +1097,33 @@ function findRunButton() {
return selectors.map((selector) => document.querySelector(selector)).find(Boolean);
}
-function findImageStreamButton() {
+function findPropertyPanelButton() {
const selectors = [
- "[data-testid='image-feed-button']",
- "[data-testid='image-stream-button']",
- "button[aria-label='显示图像流']",
- "button[title='显示图像流']",
- "button[aria-label*='图像流']",
- "button[title*='图像流']",
- "button[aria-label*='image feed']",
- "button[title*='image feed']",
+ "button[aria-label='开关属性面板']",
+ "button[title='开关属性面板']",
+ "button[aria-label*='属性面板']",
+ "button[title*='属性面板']",
+ "button[aria-label*='Toggle property panel']",
+ "button[title*='Toggle property panel']",
];
const direct = selectors.map((selector) => document.querySelector(selector)).find(Boolean);
if (direct) return direct;
return [...document.querySelectorAll("button")].find((button) =>
- ["显示图像流", "Show image feed", "Show image stream"].includes(button.textContent?.trim()),
+ ["开关属性面板", "Toggle property panel"].includes(button.textContent?.trim()),
+ );
+}
+
+function findActiveTasksButton() {
+ return [...document.querySelectorAll("button")].find((button) =>
+ /(?:\d+\s*个活动任务|\d+\s*active tasks?)/i.test(button.textContent?.trim() || ""),
);
}
function findTopToolbar() {
- const imageStreamButton = findImageStreamButton();
- if (imageStreamButton?.parentElement) return imageStreamButton.parentElement;
+ const propertyButton = findPropertyPanelButton();
+ if (propertyButton?.parentElement) return propertyButton.parentElement;
+ const activeTasksButton = findActiveTasksButton();
+ if (activeTasksButton?.parentElement) return activeTasksButton.parentElement;
const runButton = findRunButton();
if (runButton?.parentElement) return runButton.parentElement;
return document.querySelector(
@@ -401,7 +1133,8 @@ function findTopToolbar() {
function mountToolbarButton() {
let existing = document.getElementById("find-models-launcher");
- const imageStreamButton = findImageStreamButton();
+ const propertyButton = findPropertyPanelButton();
+ const activeTasksButton = findActiveTasksButton();
const runButton = findRunButton();
const toolbar = findTopToolbar();
@@ -411,18 +1144,30 @@ function mountToolbarButton() {
existing.type = "button";
existing.textContent = "查找模型";
existing.title = "扫描当前工作流中的缺失模型";
- existing.onclick = () => scan(false);
document.body.appendChild(existing);
}
+ existing.onclick = async () => {
+ panelUserOpened = true;
+ const panel = ensurePanel();
+ if (panel.classList.contains("open")) {
+ scan(true, { quick: true });
+ return;
+ }
+ await openDockedPanel(panel);
+ scan(true, { quick: true });
+ };
if (toolbar) {
existing.classList.remove("toolbar-fallback");
if (existing.parentElement !== toolbar) {
- if (imageStreamButton) imageStreamButton.insertAdjacentElement("beforebegin", existing);
+ if (propertyButton) propertyButton.insertAdjacentElement("beforebegin", existing);
+ else if (activeTasksButton) activeTasksButton.insertAdjacentElement("afterend", existing);
else if (runButton) runButton.insertAdjacentElement("afterend", existing);
else toolbar.appendChild(existing);
- } else if (imageStreamButton && existing.nextElementSibling !== imageStreamButton) {
- imageStreamButton.insertAdjacentElement("beforebegin", existing);
+ } else if (propertyButton && existing.nextElementSibling !== propertyButton) {
+ propertyButton.insertAdjacentElement("beforebegin", existing);
+ } else if (!propertyButton && activeTasksButton && activeTasksButton.nextElementSibling !== existing) {
+ activeTasksButton.insertAdjacentElement("afterend", existing);
}
return true;
}
@@ -447,12 +1192,18 @@ app.registerExtension({
style.href = new URL("./find_models.css", import.meta.url).href;
document.head.appendChild(style);
ensurePanel();
+ panelUserOpened = false;
+ ensurePanel().classList.remove("open");
watchTopToolbar();
startDownloadMonitor();
- window.setTimeout(() => scan(true), 1200);
+ startWorkflowMonitor();
+ scheduleScan(0, true);
},
async afterConfigureGraph() {
- window.clearTimeout(window.__findModelsTimer);
- window.__findModelsTimer = window.setTimeout(() => scan(true), 700);
+ scanRequestId += 1;
+ lastWorkflowSignature = "";
+ resolvedModels.clear();
+ resolvedNodePackages.clear();
+ scheduleScan(0, true);
},
});