/**
* ui_model_sources.js
* 模型来源统一中控中心 (Model Source Hub)
*
* 核心功能：
* 1. 查看/编辑当前工作流中用到的模型来源链接 (Workflow Scope)
* 2. 盘点本地全部模型库已识别/缺失信息状态 (Library Scope)
* 3. 一键在画布生成原生 ComfyUI Note 便签节点 (显式工作流节点)
* 4. 同步/保存来源元数据至工作流 JSON/PNG extra 字段 (隐式元数据)
* 5. 一键浏览器直达对应发布页 (Civitai / Hugging Face / Liblib / 网盘等)
* 6. 持久化自定义链接至本地模型 .civitai.info (全系统自动回退)
*/
import { app } from '../../../scripts/app.js';
import { translate as t } from './locales.js';
import { text, jsonResponse } from './ui_dom.js';
import { createViewScope } from './ui_lifecycle.js';
import { showWorkbenchToast } from './ui_prompt_toast.js';
import { inferModelFolderTypes, isPhysicalRenameProtectedType } from './model_policies.js';
import { foundationModelType, partitionSourceModels, shapeLibrarySourceModels, usableSourceUrl } from './model_source_data.js';
let activeSourcesModalScope = null;
export function detectPlatform(url) {
if (!url || typeof url !== 'string') return null;
const clean = url.trim();
if (!clean) return null;
if (/civitai\.(com|red)/i.test(clean)) return { name: 'Civitai', color: '#38bdf8', bg: 'rgba(56,189,248,0.15)', border: 'rgba(56,189,248,0.4)' };
if (/huggingface\.co/i.test(clean)) return { name: 'HuggingFace', color: '#fbbf24', bg: 'rgba(251,191,36,0.15)', border: 'rgba(251,191,36,0.4)' };
if (/liblib/i.test(clean)) return { name: 'LiblibAI', color: '#ec4899', bg: 'rgba(236,72,153,0.15)', border: 'rgba(236,72,153,0.4)' };
if (/modelscope\.cn/i.test(clean)) return { name: 'ModelScope', color: '#a855f7', bg: 'rgba(168,85,247,0.15)', border: 'rgba(168,85,247,0.4)' };
if (/github\.com/i.test(clean)) return { name: 'GitHub', color: '#34d399', bg: 'rgba(52,211,153,0.15)', border: 'rgba(52,211,153,0.4)' };
if (/pan\.baidu|123pan|quark|lanzou/i.test(clean)) return { name: 'CloudNet', color: '#06b6d4', bg: 'rgba(6,182,212,0.15)', border: 'rgba(6,182,212,0.4)' };
return { name: 'Web Link', color: '#94a3b8', bg: 'rgba(148,163,184,0.15)', border: 'rgba(148,163,184,0.3)' };
}
export function isModelFilename(val) {
return typeof val === 'string' && /\.(safetensors|ckpt|pt|pth|bin|sft|gguf)$/i.test(val);
}
export function normalizeUrl(url) {
if (!url) return '';
let trimmed = url.trim();
if (!trimmed) return '';
if (!/^https?:\/\//i.test(trimmed)) {
trimmed = 'https://' + trimmed;
}
return trimmed;
}
function foundationType(model) {
return foundationModelType(model);
}
function canSaveLocalSource(model) {
return !model.isMissing && (!foundationType(model) || model.nodeId == null || Boolean(model.file_path));
}
/** 提取当前画布上所有正在使用的模型 */
export function collectWorkflowModels() {
const models = [];
const seen = new Set();
const liveNodes = app.graph?._nodes || [];
const savedSources = app.graph?.extra?.anomalous_model_sources || {};
const savedHashes = app.graph?.extra?.anomalous_hashes || {};
for (const node of liveNodes) {
if (!Array.isArray(node.widgets)) continue;
for (const w of node.widgets) {
const val = w.value;
const folderTypes = inferModelFolderTypes(node, w);
// Native component options such as TAESD need not have a file extension.
const componentOption = /^(vae_name|clip_name\d*|clip_vision(?:_name)?|text_encoder(?:_name)?\d*)$/i.test(w.name || '')
&& folderTypes.some(isPhysicalRenameProtectedType)
&& typeof val === 'string' && val.trim() && val.toLowerCase() !== 'none';
if (!isModelFilename(val) && !componentOption) continue;
const dedupeKey = `${node.id}_${val}`;
if (seen.has(dedupeKey)) continue;
seen.add(dedupeKey);
const basename = val.split(/[/\\]/).pop();
const nativeValues = w.options?.values;
const isMissing = Array.isArray(nativeValues)
&& !nativeValues.some(value => typeof value === 'string' && value.replaceAll('\\', '/') === val.replaceAll('\\', '/'));
const hashObj = savedHashes[dedupeKey] || savedHashes[val] || window.anomalous_hash_cache?.[val] || window.anomalous_hash_cache?.[basename] || {};
const hash = typeof hashObj === 'string' ? hashObj : (hashObj.hash || '');
const existingSource = savedSources[val] || savedSources[basename] || savedSources[dedupeKey];
let url = usableSourceUrl(typeof existingSource === 'string' ? existingSource : existingSource?.url)
|| usableSourceUrl(hashObj?.url) || usableSourceUrl(hashObj?.civitai_url);
models.push({
key: dedupeKey,
nodeId: node.id,
nodeTitle: node.title || node.type || `Node #${node.id}`,
nodeType: node.type,
folderTypes,
filename: val,
basename,
isMissing,
hash,
url,
initialUrl: url,
platform: detectPlatform(url),
});
}
}
return models;
}
/** 异步盘点本地全部模型库 */
export async function fetchAllLibraryModels(signal = null) {
const res = await fetch('/anomalous/all_scan_models?limit=0', { signal });
const payload = await jsonResponse(res, 'load library models');
return shapeLibrarySourceModels(payload.models, detectPlatform);
}
/** 在 ComfyUI 画布生成原生 Note 便签节点 */
export function createCanvasNoteNode(models) {
if (!models.length) return false;
const creator = (typeof LiteGraph !== 'undefined' ? LiteGraph?.createNode : null) || globalThis.LiteGraph?.createNode || (typeof window !== 'undefined' ? window.LiteGraph?.createNode : null);
if (typeof creator !== 'function') return false;
const noteNode = creator('Note');
if (!noteNode) return false;
const now = new Date();
const timeStr = `${now.getFullYear()}-${String(now.getMonth() + 1).padStart(2, '0')}-${String(now.getDate()).padStart(2, '0')} ${String(now.getHours()).padStart(2, '0')}:${String(now.getMinutes()).padStart(2, '0')}`;
let textContent = `═══════════════════════════════════════════════════════════════\n`;
textContent += `📋 工作流模型来源与下载直达 (Workflow Models & Sources)\n`;
textContent += `═══════════════════════════════════════════════════════════════\n`;
textContent += `• 生成时间: ${timeStr}\n`;
textContent += `• 模型总数: ${models.length} 个\n`;
textContent += `───────────────────────────────────────────────────────────────\n\n`;
models.forEach((m, idx) => {
textContent += `[${idx + 1}] ${m.nodeTitle || m.nodeType || 'Model'}\n`;
textContent += ` 文件: ${m.filename}\n`;
if (m.url) {
textContent += ` 来源: ${m.url}\n`;
} else if (foundationType(m)) {
textContent += ` 来源: ${t('modelSourcesStatusUnfilledOptional')}\n`;
} else {
textContent += ` 来源: ⚠️ 未配置来源发布页\n`;
}
if (m.hash) {
textContent += ` 哈希: ${m.hash.slice(0, 16)}...\n`;
}
textContent += `\n`;
});
textContent += `───────────────────────────────────────────────────────────────\n`;
textContent += `由 Anomalous Model Browser 模型来源中控中心一键生成`;
noteNode.title = window.anomalous_browser_lang === 'zh' ? '📋 模型来源清单 (Sources)' : '📋 Model Sources';
if (Array.isArray(noteNode.widgets) && noteNode.widgets.length > 0) {
noteNode.widgets[0].value = textContent;
}
// 计算合适坐标（排在画布最上方或靠左侧）
let minX = Infinity;
let minY = Infinity;
for (const n of app.graph._nodes || []) {
if (n.pos) {
if (n.pos[0] < minX) minX = n.pos[0];
if (n.pos[1] < minY) minY = n.pos[1];
}
}
if (!Number.isFinite(minX)) { minX = 100; minY = 100; }
noteNode.pos = [minX, minY - 320];
noteNode.size = [480, 260];
app.graph.add(noteNode);
app.canvas?.setDirty(true, true);
return noteNode;
}
/** 同步来源至工作流元数据 (extra.anomalous_model_sources) */
export function syncWorkflowSources(models) {
if (!app.graph) return { count: 0, isUpdate: false };
app.graph.extra ||= {};
const existing = app.graph.extra.anomalous_model_sources || {};
const isUpdate = Object.keys(existing).length > 0;
const updatedMap = { ...existing };
let savedCount = 0;
for (const m of models) {
if (!m.url) continue;
const normalized = normalizeUrl(m.url);
updatedMap[m.filename] = {
name: m.filename,
url: normalized,
platform: detectPlatform(normalized)?.name || 'Custom',
nodeId: m.nodeId,
hash: m.hash || '',
updated_at: Date.now(),
};
m.url = normalized;
m.initialUrl = normalized;
m.platform = detectPlatform(normalized);
savedCount++;
}
app.graph.extra.anomalous_model_sources = updatedMap;
app.canvas?.setDirty(true, true);
return { count: savedCount, isUpdate };
}
/** 复制 Markdown 格式清单 */
export async function copySourcesSummary(models) {
if (!models.length) return;
let md = `### 📋 工作流模型下载来源清单 (Workflow Models & Sources)\n\n`;
models.forEach((m, idx) => {
const plat = m.platform ? `[${m.platform.name}]` : '';
md += `${idx + 1}. **${m.nodeTitle || m.nodeType}** \`${m.basename || m.filename}\`\n`;
if (m.url) {
md += ` - 来源链接: ${plat} ${m.url}\n`;
} else if (foundationType(m)) {
md += ` - 来源链接: ${t('modelSourcesStatusUnfilledOptional')}\n`;
} else {
md += ` - 来源链接: ⚠️ 未指定\n`;
}
});
await navigator.clipboard.writeText(md);
}
/** 保存单个模型自定义链接到本地 .civitai.info */
export async function saveSingleModelToLocalSidecar(item, url) {
if (!canSaveLocalSource(item)) throw new Error(t('modelSourcesLocalTargetUnknown'));
const cleanUrl = normalizeUrl(url);
const targetFilename = item.basename || (item.filename ? item.filename.split(/[/\\]/).pop() : '');
const body = {
filename: targetFilename,
type: item.type || 'checkpoints',
subfolder: item.subfolder || '/',
path_idx: item.path_idx || 0,
custom_source_url: cleanUrl,
};
const res = await fetch('/anomalous/update_metadata', {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify(body),
});
return jsonResponse(res, 'save custom source url');
}
/** 一键打开外部链接 */
export function openExternalUrl(rawUrl) {
const normalized = normalizeUrl(rawUrl);
if (!normalized) return;
window.open(normalized, '_blank', 'noopener,noreferrer');
}
/** 异步解析工作流中模型的本地元数据与路径详情 */
export async function resolveWorkflowModelsMetadata(models, signal = null) {
if (!Array.isArray(models) || !models.length) return false;
const components = models.filter(m => foundationType(m));
const requestedPaths = models.filter(m => !foundationType(m)).map(m => m.filename);
try {
const res = await fetch('/anomalous/resolve_paths_to_previews', {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify({ paths: requestedPaths }),
signal,
});
const data = await jsonResponse(res, 'resolve workflow models');
const modelsMap = data?.models || {};
const componentModels = {};
// The endpoint caps contextual lookups at 16; keep large workflows complete.
for (let offset = 0; offset < components.length; offset += 16) {
const response = await fetch('/anomalous/resolve_paths_to_previews', {
method: 'POST', headers: { 'Content-Type': 'application/json' }, signal,
body: JSON.stringify({ paths: [], context_requests: components.slice(offset, offset + 16).map(m => ({
key: m.key, path: m.relPath || m.filename, folder_types: m.folderTypes || [m.type], exact_only: true,
})) }),
});
const batch = await jsonResponse(response, 'resolve component sources');
Object.assign(componentModels, batch.context_models || {});
}
let hasChanges = false;
for (const m of models) {
const info = foundationType(m) ? componentModels[m.key] : modelsMap[m.filename] || modelsMap[m.basename];
if (info) {
m.type = info.type;
m.subfolder = info.subfolder;
m.path_idx = info.path_idx;
m.file_path = info.file_path;
hasChanges = true;
if (foundationType(m)) m.isMissing = false;
if (!m.hash && info.metadata?.hash) {
m.hash = info.metadata.hash;
hasChanges = true;
}
const resolvedUrl = usableSourceUrl(info.metadata?.source_url)
|| usableSourceUrl(info.metadata?.civitai_url);
if (!m.url && resolvedUrl) {
m.url = resolvedUrl;
m.initialUrl = resolvedUrl;
m.platform = detectPlatform(resolvedUrl);
hasChanges = true;
}
if (typeof window !== 'undefined' && window.anomalous_hash_cache) {
if (m.filename && info.metadata?.hash) {
window.anomalous_hash_cache[m.filename] = {
hash: info.metadata.hash,
url: resolvedUrl,
};
}
}
}
}
return hasChanges;
} catch (e) {
if (!signal?.aborted) {
console.warn('[Model Source Hub] Failed to resolve workflow model info', e);
}
return false;
}
}
// -----------------------------------------------------------------------------
// UI RENDERERS (Sub-divided into concise blocks <= 50 lines)
// -----------------------------------------------------------------------------
function renderModalHeader(headerEl, scope, state, onScopeChange, onClose) {
headerEl.className = 'anomalous-sources-header';
const topRow = text(headerEl, 'div', '', 'anomalous-sources-top-row');
const titleWrap = text(topRow, 'div', '', 'anomalous-sources-title-wrap');
text(titleWrap, 'h3', t('modelSourcesModalTitle'));
const closeBtn = text(topRow, 'button', '✕', 'anomalous-sources-close-btn');
closeBtn.title = t('close') || '关闭';
closeBtn.onclick = onClose;
// Scope Tabs (Active Workflow vs All Local Models)
const tabsRow = text(headerEl, 'div', '', 'anomalous-sources-scope-tabs');
const wfTab = text(tabsRow, 'button', '', `anomalous-sources-scope-tab${state.scope === 'workflow' ? ' is-active' : ''}`);
wfTab.innerHTML = `🎛️ ${t('modelSourcesScopeWorkflow')} ${state.workflowModels.length}`;
wfTab.onclick = () => onScopeChange('workflow');
const libTab = text(tabsRow, 'button', '', `anomalous-sources-scope-tab${state.scope === 'library' ? ' is-active' : ''}`);
const libraryCount = state.libraryStatus === 'ready' ? state.libraryModels.length : '...';
libTab.innerHTML = `📚 ${t('modelSourcesScopeLibrary')} ${libraryCount}`;
libTab.onclick = () => onScopeChange('library');
}
function renderFilterAndSearch(filterBarEl, state, onFilterChange, onSearch) {
filterBarEl.className = 'anomalous-sources-filter-bar';
const pillsWrap = text(filterBarEl, 'div', '', 'anomalous-sources-pills');
const filters = [
{ id: 'all', label: t('modelSourcesFilterAll') },
{ id: 'resolved', label: t('modelSourcesFilterResolved') },
{ id: 'unresolved', label: t('modelSourcesFilterUnresolved') },
];
filters.forEach(f => {
const pill = text(pillsWrap, 'button', f.label, `anomalous-sources-pill${state.filter === f.id ? ' is-active' : ''}`);
pill.onclick = () => onFilterChange(f.id);
});
const searchInput = text(filterBarEl, 'input', '', 'anomalous-sources-search-input');
searchInput.placeholder = t('modelSourcesSearchPlaceholder');
searchInput.value = state.searchKeyword;
searchInput.oninput = () => onSearch(searchInput.value.trim().toLowerCase());
}
function renderModelTitle(infoCol, m) {
const titleRow = text(infoCol, 'div', '', 'anomalous-source-row-title');
text(titleRow, 'span', m.nodeTitle || m.type || 'Model', 'anomalous-source-node-tag');
text(titleRow, 'strong', m.basename || m.filename, 'anomalous-source-model-name');
const component = foundationType(m);
if (component) {
const label = { clip: 'modelSourcesTypeTextEncoder', text_encoders: 'modelSourcesTypeTextEncoder',
vae: 'modelSourcesTypeVae', vae_approx: 'modelSourcesTypeVaeApprox', clip_vision: 'modelSourcesTypeVisionEncoder' }[component];
text(titleRow, 'span', t(label), 'anomalous-source-badge-neutral');
}
const sourceBadge = text(titleRow, 'span', t(component ? 'modelSourcesStatusUnfilledOptional' : 'modelSourcesStatusUnfilled'), 'anomalous-source-badge-neutral');
sourceBadge.hidden = Boolean(m.url?.trim());
if (m.isMissing) {
const missingBadge = text(titleRow, 'span', t('modelSourcesStatusMissingOnDisk'), 'anomalous-source-badge-danger');
missingBadge.title = t('modelSourcesStatusMissingOnDisk');
}
if (m.isEditing) {
const editBadge = text(titleRow, 'span', window.anomalous_browser_lang === 'zh' ? '✏️ 正在编辑' : '✏️ Editing', 'anomalous-source-edit-badge');
editBadge.title = window.anomalous_browser_lang === 'zh' ? '当前处于来源链接修改状态' : 'In editing mode';
}
if (m.hash) {
text(titleRow, 'code', `SHA: ${m.hash.slice(0, 10)}...`, 'anomalous-source-hash-chip');
}
return sourceBadge;
}
function renderModelUrlInput(inputWrap, m, onRefresh, getActions) {
const isLocked = Boolean(m.url && !m.isEditing);
const inputClass = `anomalous-source-url-input${isLocked ? ' is-locked' : ''}${m.isEditing ? ' is-editing' : ''}`;
const urlInput = text(inputWrap, 'input', '', inputClass);
urlInput.placeholder = t('modelSourcesUrlPlaceholder');
urlInput.value = m.url || '';
urlInput.readOnly = isLocked;
const platBadge = text(inputWrap, 'span', m.platform?.name || '', 'anomalous-source-plat-badge');
if (m.platform) {
platBadge.style.color = m.platform.color;
platBadge.style.borderColor = m.platform.border;
platBadge.style.backgroundColor = m.platform.bg;
} else {
platBadge.style.display = 'none';
}
if (isLocked) {
urlInput.title = window.anomalous_browser_lang === 'zh'
? '当前为展示锁定态。双击或点击右侧【✏️ 修改】可解锁编辑'
: 'Protected link. Double-click or click [✏️ Edit] to modify';
urlInput.ondblclick = () => {
m.isEditing = true;
onRefresh();
};
} else {
urlInput.oninput = () => {
m.url = urlInput.value.trim();
m.platform = detectPlatform(m.url);
if (m.platform) {
platBadge.style.display = '';
platBadge.textContent = m.platform.name;
platBadge.style.color = m.platform.color;
platBadge.style.borderColor = m.platform.border;
platBadge.style.backgroundColor = m.platform.bg;
} else {
platBadge.style.display = 'none';
}
const { localBtn, jumpBtn, sourceBadge } = getActions();
if (sourceBadge) sourceBadge.hidden = Boolean(m.url);
const isDirty = Boolean(m.url && m.url.trim() !== (m.initialUrl || '').trim());
if (localBtn) localBtn.style.display = (isDirty && canSaveLocalSource(m)) ? '' : 'none';
if (jumpBtn) jumpBtn.disabled = !m.url;
};
urlInput.onkeydown = (e) => {
if (e.key === 'Escape') {
m.url = m.initialUrl;
m.platform = detectPlatform(m.url);
m.isEditing = false;
onRefresh();
} else if (e.key === 'Enter') {
m.isEditing = false;
onRefresh();
}
};
if (m.isEditing) {
setTimeout(() => {
urlInput.focus();
urlInput.select();
}, 0);
}
}
return urlInput;
}
function renderModelRowActions(actionsRow, m, state, urlInput, onRefresh, onAutoDetect, onSaveLocal) {
const isDirty = Boolean(m.url && m.url.trim() !== (m.initialUrl || '').trim());
const jumpBtn = text(actionsRow, 'button', t('modelSourcesJump'), 'anomalous-btn-primary anomalous-btn-sm anomalous-btn-jump');
jumpBtn.disabled = !m.url;
jumpBtn.title = window.anomalous_browser_lang === 'zh' ? '在浏览器新标签页中打开对应网址' : 'Open website in new tab';
jumpBtn.onclick = () => openExternalUrl(m.url);
const detectBtn = text(actionsRow, 'button', t('modelSourcesAutoDetect'), 'anomalous-btn-ghost anomalous-btn-sm');
detectBtn.title = window.anomalous_browser_lang === 'zh' ? '从本地元数据或 SHA256 自动解析来源' : 'Auto detect from hash or metadata';
detectBtn.onclick = () => onAutoDetect(m, urlInput);
let editBtn = null;
let cancelBtn = null;
if (m.isEditing) {
cancelBtn = text(actionsRow, 'button', window.anomalous_browser_lang === 'zh' ? '✕ 取消' : '✕ Cancel', 'anomalous-btn-ghost anomalous-btn-sm');
cancelBtn.title = window.anomalous_browser_lang === 'zh' ? '放弃本次修改并还原' : 'Cancel edits and revert';
cancelBtn.onclick = () => {
m.url = m.initialUrl;
m.platform = detectPlatform(m.url);
m.isEditing = false;
onRefresh();
};
} else if (m.url) {
editBtn = text(actionsRow, 'button', window.anomalous_browser_lang === 'zh' ? '✏️ 修改' : '✏️ Edit', 'anomalous-btn-ghost anomalous-btn-sm');
editBtn.title = window.anomalous_browser_lang === 'zh' ? '解锁输入框以修改来源网址' : 'Unlock to edit URL';
editBtn.onclick = () => {
m.isEditing = true;
onRefresh();
};
}
let localBtn = null;
if (canSaveLocalSource(m)) {
localBtn = text(actionsRow, 'button', t('modelSourcesSaveLocal'), 'anomalous-btn-primary anomalous-btn-sm anomalous-btn-save-local');
localBtn.title = window.anomalous_browser_lang === 'zh' ? '检测到链接已修改，点击记入本地模型的 .civitai.info' : 'Modified link detected. Save to local .civitai.info';
localBtn.style.display = isDirty ? '' : 'none';
localBtn.onclick = () => onSaveLocal(m);
}
return { jumpBtn, detectBtn, editBtn, cancelBtn, localBtn };
}
function renderModelRow(listEl, m, state, onRefresh, onAutoDetect, onSaveLocal) {
const card = text(listEl, 'div', '', `anomalous-source-row-item${m.url ? ' has-url' : ' is-missing-url'}`);
const infoCol = text(card, 'div', '', 'anomalous-source-col-info');
const sourceBadge = renderModelTitle(infoCol, m);
const inputRow = text(card, 'div', '', 'anomalous-source-col-input-row');
const inputWrap = text(inputRow, 'div', '', 'anomalous-source-input-wrap');
const actionsRow = text(inputRow, 'div', '', 'anomalous-source-row-actions');
let actionRefs = {};
const urlInput = renderModelUrlInput(inputWrap, m, onRefresh, () => ({ ...actionRefs, sourceBadge }));
actionRefs = renderModelRowActions(actionsRow, m, state, urlInput, onRefresh, onAutoDetect, onSaveLocal);
}
function renderComponentDisclosure(listEl, state, group, renderRow) {
const button = text(listEl, 'button', '', 'anomalous-sources-components-toggle');
button.setAttribute('aria-expanded', String(state.componentsExpanded));
const marker = state.componentsExpanded ? '▾' : '▸';
const label = text(button, 'span', '', 'anomalous-sources-components-label');
text(label, 'span', `${marker} ${t('modelSourcesComponentsAdvanced', { count: group.componentModels.length })}`);
text(label, 'small', t('modelSourcesComponentsOptionalHint'));
if (group.componentMissingCount) {
text(button, 'span', t('modelSourcesComponentsMissing', { count: group.componentMissingCount }), 'anomalous-source-badge-danger');
}
button.onclick = () => {
state.componentsExpanded = !state.componentsExpanded;
state.refresh();
};
if (!state.componentsExpanded) return;
const rows = text(listEl, 'div', '', 'anomalous-sources-components-list');
group.componentModels.forEach(model => renderRow(rows, model));
if (!group.componentModels.length) {
text(rows, 'div', t('modelSourcesComponentsNoSearchMatch'), 'anomalous-sources-components-empty');
}
}
function renderFooterBar(footerEl, state, activeMainModels, onSaveWorkflow, onGenerateNote, onCopySummary) {
footerEl.className = 'anomalous-sources-footer';
footerEl.replaceChildren();
const statsEl = text(footerEl, 'div', '', 'anomalous-sources-footer-stats');
const withUrlCount = activeMainModels.filter(m => Boolean(m.url?.trim())).length;
statsEl.innerHTML = `${t('modelSourcesMainStats')}: ${withUrlCount} / ${activeMainModels.length}`;
const btnsWrap = text(footerEl, 'div', '', 'anomalous-sources-footer-actions');
const copyBtn = text(btnsWrap, 'button', t('modelSourcesCopySummary'), 'anomalous-btn-ghost anomalous-btn-sm');
copyBtn.onclick = onCopySummary;
if (state.scope === 'workflow') {
const noteBtn = text(btnsWrap, 'button', t('modelSourcesGenerateNoteNode'), 'anomalous-btn-ghost anomalous-btn-sm');
noteBtn.title = window.anomalous_browser_lang === 'zh' ? '在 ComfyUI 画布生成一个原生 Note 便签节点，任何人打开都能直接看到下载链接' : 'Generate native Note node with source links on canvas';
noteBtn.onclick = onGenerateNote;
// Dynamic Save vs Sync text
const hasExisting = Boolean(Object.keys(app.graph?.extra?.anomalous_model_sources || {}).length);
const saveLabel = hasExisting ? t('modelSourcesSyncToWorkflow') : t('modelSourcesSaveToWorkflow');
const saveBtn = text(btnsWrap, 'button', saveLabel, 'anomalous-btn-primary anomalous-btn-sm');
saveBtn.onclick = onSaveWorkflow;
}
}
// -----------------------------------------------------------------------------
// MAIN MODAL OPENER
// -----------------------------------------------------------------------------
export function openModelSourcesModal(initialScope = 'workflow') {
activeSourcesModalScope?.dispose();
const scope = createViewScope();
activeSourcesModalScope = scope;
const overlay = document.createElement('div');
overlay.className = 'anomalous-model-sources-overlay';
scope.onDispose(() => {
overlay.remove();
if (activeSourcesModalScope === scope) activeSourcesModalScope = null;
});
const modal = document.createElement('div');
modal.className = 'anomalous-model-sources-modal';
overlay.appendChild(modal);
const state = {
scope: initialScope,
filter: 'all',
searchKeyword: '',
workflowModels: collectWorkflowModels(),
libraryModels: [],
libraryStatus: 'idle',
componentsExpanded: false,
libraryRequestId: 0,
};
const headerEl = text(modal, 'header', '', '');
const filterBarEl = text(modal, 'div', '', '');
const bodyEl = text(modal, 'div', '', 'anomalous-sources-body');
const footerEl = text(modal, 'footer', '', '');
const ensureLibraryLoaded = async (force = false) => {
if (state.libraryStatus === 'loading') return;
if (!force && state.libraryStatus === 'ready') {
refreshUi();
return;
}
const requestId = ++state.libraryRequestId;
state.libraryStatus = 'loading';
refreshUi();
try {
const items = await fetchAllLibraryModels(scope.signal);
if (scope.signal.aborted || requestId !== state.libraryRequestId) return;
state.libraryModels = items;
state.libraryStatus = 'ready';
} catch (error) {
if (scope.signal.aborted || requestId !== state.libraryRequestId) return;
console.error('[Model Source Hub] Failed to fetch library models', error);
state.libraryStatus = 'error';
}
if (!scope.signal.aborted && requestId === state.libraryRequestId) refreshUi();
};
const refreshUi = () => {
headerEl.replaceChildren();
filterBarEl.replaceChildren();
bodyEl.replaceChildren();
renderModalHeader(headerEl, scope, state, (newScope) => {
if (state.scope === newScope) return;
state.scope = newScope;
if (newScope === 'library') {
ensureLibraryLoaded();
return;
}
refreshUi();
}, () => scope.dispose());
renderFilterAndSearch(filterBarEl, state, (newFilter) => {
state.filter = newFilter;
refreshUi();
}, (keyword) => {
state.searchKeyword = keyword;
refreshUi();
});
const rawList = state.scope === 'workflow' ? state.workflowModels : state.libraryModels;
const group = partitionSourceModels(rawList, state.filter, state.searchKeyword);
const renderRow = (parent, model) => renderModelRow(parent, model, state,
() => refreshUi(),
async (item, inputEl) => {
inputEl.disabled = true;
try {
const detected = await autoDetectModelSource(item);
if (detected) {
item.url = detected;
item.initialUrl = item.initialUrl || detected;
item.platform = detectPlatform(detected);
inputEl.value = detected;
if (state.filter === 'unresolved') state.filter = 'resolved';
refreshUi();
showWorkbenchToast(window.anomalous_browser_lang === 'zh' ? '✓ 已识别模型来源！' : '✓ Model source detected!');
} else {
showWorkbenchToast(window.anomalous_browser_lang === 'zh' ? '未能在本地或云端匹配到官方页面' : 'No online source matched');
}
} finally {
inputEl.disabled = false;
}
},
async (item) => {
try {
await saveSingleModelToLocalSidecar(item, item.url);
item.initialUrl = item.url;
item.isEditing = false;
refreshUi();
showWorkbenchToast(t('modelSourcesSavedLocal'));
} catch (e) {
showWorkbenchToast(window.anomalous_browser_lang === 'zh' ? '保存至本地失败' : 'Failed to save local');
}
});
if (state.scope === 'library' && state.libraryStatus === 'loading') {
const loadingBox = text(bodyEl, 'div', '', 'anomalous-sources-empty');
text(loadingBox, 'div', `⏳ ${t('modelSourcesLibraryLoading')}`);
} else if (state.scope === 'library' && state.libraryStatus === 'error') {
const errorBox = text(bodyEl, 'div', '', 'anomalous-sources-empty');
text(errorBox, 'div', t('modelSourcesLibraryLoadFailed'));
const retry = text(errorBox, 'button', t('modelSourcesRetry'), 'anomalous-btn-ghost anomalous-btn-sm');
retry.onclick = () => {
state.libraryStatus = 'idle';
ensureLibraryLoaded();
};
} else {
if (state.scope === 'library') {
text(bodyEl, 'div', t('modelSourcesLibraryRangeHint'), 'anomalous-sources-library-hint');
}
group.mainModels.forEach(model => renderRow(bodyEl, model));
if (group.allComponentCount) {
state.refresh = refreshUi;
renderComponentDisclosure(bodyEl, state, group, renderRow);
}
}
const libraryUnavailable = state.scope === 'library'
&& (state.libraryStatus === 'loading' || state.libraryStatus === 'error');
if (!libraryUnavailable
&& !group.mainModels.length && !group.allComponentCount) {
const emptyBox = text(bodyEl, 'div', '', 'anomalous-sources-empty');
emptyBox.innerHTML = `
🔍
${t('modelSourcesNoModelsFound')}
`;
}
const visibleModels = [...group.mainModels, ...(state.componentsExpanded ? group.componentModels : [])];
renderFooterBar(footerEl, state, group.mainModels,
() => {
const res = syncWorkflowSources(state.workflowModels);
showWorkbenchToast(res.isUpdate ? t('modelSourcesSyncedToWorkflow') : t('modelSourcesSavedToWorkflow'));
refreshUi();
},
() => {
const ok = createCanvasNoteNode(state.workflowModels);
if (ok) showWorkbenchToast(t('modelSourcesNoteCreated'));
},
async () => {
await copySourcesSummary(visibleModels);
showWorkbenchToast(t('modelSourcesCopied'));
}
);
};
refreshUi();
resolveWorkflowModelsMetadata(state.workflowModels, scope.signal).then(hasChanges => {
if (hasChanges && !scope.signal.aborted) {
refreshUi();
}
});
overlay.onclick = (e) => {
if (e.target === overlay) scope.dispose();
};
scope.listen(window, 'keydown', (e) => {
if (e.key === 'Escape') scope.dispose();
});
document.body.appendChild(overlay);
if (initialScope === 'library') ensureLibraryLoaded();
return () => scope.dispose();
}
/** 自动识别单条模型来源 */
export async function autoDetectModelSource(item) {
if (item.url) return item.url;
if (foundationType(item)) {
await resolveWorkflowModelsMetadata([item]);
return item.url || '';
}
// 1. Priority: check local sidecar & metadata via backend resolver
try {
const queryPath = item.filename || item.basename;
const res = await fetch('/anomalous/resolve_paths_to_previews', {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify({ paths: [queryPath, item.basename].filter(Boolean) }),
});
const data = await jsonResponse(res, 'detect model source local');
const resolved = data?.models?.[queryPath] || data?.models?.[item.basename];
if (resolved) {
item.type = resolved.type;
item.subfolder = resolved.subfolder;
item.path_idx = resolved.path_idx;
if (resolved.metadata?.hash) item.hash = resolved.metadata.hash;
const foundUrl = usableSourceUrl(resolved.metadata?.source_url)
|| usableSourceUrl(resolved.metadata?.civitai_url);
if (foundUrl) return foundUrl;
}
} catch (e) {
// Backend lookup failed, proceed to cloud/search fallback
}
// 2. Cloud lookup if hash exists
if (item.hash) {
try {
const res = await fetch(`https://civitai.com/api/v1/model-versions/by-hash/${item.hash}`);
if (res.ok) {
const data = await res.json();
if (data.modelId) {
const domain = (data.model?.nsfw || data.nsfwLevel > 1) ? 'civitai.red' : 'civitai.com';
return `https://${domain}/models/${data.modelId}${data.id ? '?modelVersionId=' + data.id : ''}`;
}
}
} catch (e) {
// Network fallback
}
}
// A search-results page is not a model source. Keep the item unresolved
// unless local metadata or an exact hash lookup identifies a release.
return '';
}