554 lines
21 KiB
JavaScript
554 lines
21 KiB
JavaScript
import { app } from "/scripts/app.js";
|
|
|
|
const STYLE_ID = "comfy-sidebar-classic-layout-override";
|
|
|
|
const CLASSIC_LAYOUT_CSS_MEDIA = `
|
|
/* Style and align the custom plugins bar (Manager, LoRA, etc.) */
|
|
[class*="actionbar"]:not(.actionbar),
|
|
[class*="actionbar-buttons"],
|
|
.actionbar-buttons {
|
|
position: fixed !important;
|
|
top: -5px !important;
|
|
right: 4px !important;
|
|
left: auto !important;
|
|
transform: none !important;
|
|
z-index: 1010 !important;
|
|
|
|
background: transparent !important;
|
|
border: none !important;
|
|
box-shadow: none !important;
|
|
padding: 0 !important;
|
|
margin: 0 !important;
|
|
display: flex !important;
|
|
align-items: center !important;
|
|
}
|
|
|
|
/* Style overrides for the new native Extensions button container when embedded in top actionbar */
|
|
.comfy-sidebar-extensions-override {
|
|
background: transparent !important;
|
|
border: none !important;
|
|
box-shadow: none !important;
|
|
padding: 0 !important;
|
|
margin: 0 8px 0 0 !important;
|
|
height: auto !important;
|
|
position: static !important;
|
|
order: -10 !important;
|
|
display: inline-flex !important;
|
|
align-items: center !important;
|
|
}
|
|
|
|
/* Scale down performance indicators (Crystools) uniformly */
|
|
#crysmonitor-monitors-root,
|
|
#crystools-monitors-root,
|
|
.crysmonitor-monitors-container {
|
|
zoom: 0.8 !important;
|
|
display: inline-flex !important;
|
|
align-items: center !important;
|
|
}
|
|
|
|
/* Hide user account profile button and PrimeVue components */
|
|
img[alt*="User Avatar"],
|
|
img[alt*="user avatar"],
|
|
button:has(img[alt*="User Avatar"]),
|
|
[role="button"]:has(img[alt*="User Avatar"]),
|
|
button:has(img[alt*="user avatar"]),
|
|
[role="button"]:has(img[alt*="user avatar"]),
|
|
.p-avatar,
|
|
[data-pc-name="avatar"],
|
|
button:has(.p-avatar),
|
|
[role="button"]:has(.p-avatar),
|
|
button:has([data-pc-name="avatar"]),
|
|
[role="button"]:has([data-pc-name="avatar"]) {
|
|
display: none !important;
|
|
}
|
|
|
|
/* Base rules: Large/Wide desktop screens (>= 1600px width) */
|
|
.p-tabview-nav-content,
|
|
[class*="tabview"] {
|
|
padding-left: 150px !important;
|
|
padding-right: 580px !important;
|
|
}
|
|
|
|
/* Medium desktop screens (between 1200px and 1599px width) */
|
|
@media (max-width: 1599px) {
|
|
.p-tabview-nav-content,
|
|
[class*="tabview"] {
|
|
padding-left: 100px !important;
|
|
padding-right: 420px !important;
|
|
}
|
|
}
|
|
|
|
/* Standard laptops / small monitors (between 900px and 1199px width) */
|
|
@media (max-width: 1199px) {
|
|
.p-tabview-nav-content,
|
|
[class*="tabview"] {
|
|
padding-left: 40px !important;
|
|
padding-right: 280px !important;
|
|
}
|
|
}
|
|
|
|
/* Very narrow or heavily zoomed displays (< 900px width) */
|
|
@media (max-width: 899px) {
|
|
.p-tabview-nav-content,
|
|
[class*="tabview"] {
|
|
padding-left: 10px !important;
|
|
padding-right: 180px !important;
|
|
}
|
|
}
|
|
`;
|
|
|
|
export function syncStockHistoryAndProgressSettings(enable) {
|
|
if (!app.ui?.settings) return;
|
|
|
|
const dockedVal = !!enable;
|
|
const progressVal = !enable;
|
|
const QPOV2_ID = "Comfy.Queue.QPOV2";
|
|
|
|
try {
|
|
app.ui.settings.setSettingValue(QPOV2_ID, dockedVal);
|
|
localStorage.setItem(QPOV2_ID, JSON.stringify(dockedVal));
|
|
} catch (e) {}
|
|
|
|
try {
|
|
const qpovContainer = document.querySelector('[data-setting-id="Comfy.Queue.QPOV2"]') ||
|
|
document.getElementById(QPOV2_ID)?.closest('.setting-item, tr, div');
|
|
if (qpovContainer) {
|
|
const toggle = qpovContainer.querySelector('input[type="checkbox"], [role="switch"], .p-toggleswitch');
|
|
if (toggle) {
|
|
const isChecked = toggle.checked ||
|
|
toggle.getAttribute('data-p-checked') === 'true' ||
|
|
toggle.classList.contains('p-toggleswitch-checked');
|
|
if (isChecked !== dockedVal) toggle.click();
|
|
}
|
|
}
|
|
} catch (e) {}
|
|
|
|
const progressKey = "Comfy.Queue.ShowRunProgressBar";
|
|
try {
|
|
app.ui.settings.setSettingValue(progressKey, progressVal);
|
|
localStorage.setItem(progressKey, JSON.stringify(progressVal));
|
|
} catch (e) {}
|
|
|
|
try {
|
|
const progressBarBtn = document.querySelector('[data-testid="show-run-progress-bar-action"]');
|
|
if (progressBarBtn) {
|
|
const isChecked = !!progressBarBtn.querySelector('[class*="lucide--check"], .icon-\\[lucide--check\\]');
|
|
if (isChecked !== progressVal) progressBarBtn.click();
|
|
}
|
|
} catch (e) {}
|
|
|
|
try { window.dispatchEvent(new Event("storage")); } catch (e) {}
|
|
}
|
|
|
|
export function applyClassicLayout(enable, updateSetting = false) {
|
|
let styleEl = document.getElementById(STYLE_ID);
|
|
|
|
if (enable) {
|
|
if (updateSetting && app.ui?.settings) {
|
|
app.ui.settings.setSettingValue("Comfy.Workflow.WorkflowTabsPosition", "Topbar");
|
|
}
|
|
if (!styleEl) {
|
|
styleEl = document.createElement("style");
|
|
styleEl.id = STYLE_ID;
|
|
document.head.appendChild(styleEl);
|
|
}
|
|
styleEl.textContent = CLASSIC_LAYOUT_CSS_MEDIA;
|
|
} else {
|
|
if (styleEl) styleEl.remove();
|
|
if (updateSetting && app.ui?.settings) {
|
|
app.ui.settings.setSettingValue("Comfy.Workflow.WorkflowTabsPosition", "Sidebar");
|
|
}
|
|
}
|
|
}
|
|
|
|
let savedButtonData = null;
|
|
let domObserver = null;
|
|
let syncScheduled = false;
|
|
|
|
function findOriginalButton() {
|
|
const icons = document.querySelectorAll('[class*="lucide--panel-right"], [class*="lucide--panel-left"], [class*="lucide--panel-bottom"]');
|
|
for (const icon of icons) {
|
|
const btn = icon.closest('button, [role="button"], .comfyui-menu-item, .p-button');
|
|
if (btn && !btn.classList.contains("comfy-sidebar-custom-properties-toggle")) {
|
|
return btn;
|
|
}
|
|
}
|
|
|
|
const buttons = document.querySelectorAll('button, [role="button"], .p-button');
|
|
for (const btn of buttons) {
|
|
if (btn.classList.contains("comfy-sidebar-custom-properties-toggle")) continue;
|
|
const title = (btn.getAttribute("title") || btn.getAttribute("aria-label") || "").toLowerCase();
|
|
if (title.includes("workflow overview") || title.includes("properties") || title.includes("toggle panel")) {
|
|
return btn;
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
function findTopbarContainer() {
|
|
const container = document.querySelector('[data-testid="action-bar-buttons"], [class*="actionbar-buttons"], .actionbar-buttons, [class*="actionbar"]');
|
|
if (container) return container;
|
|
|
|
const orig = findOriginalButton();
|
|
if (orig?.parentNode) return orig.parentNode;
|
|
|
|
const buttons = document.querySelectorAll("button, .p-button");
|
|
for (const btn of buttons) {
|
|
if (btn.classList.contains("comfy-sidebar-custom-properties-toggle")) continue;
|
|
const title = (btn.getAttribute("title") || btn.getAttribute("aria-label") || "").toLowerCase();
|
|
if (title.includes("run") || title.includes("manager")) {
|
|
if (btn.parentNode) return btn.parentNode;
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
function isPropertiesPanelOpen() {
|
|
const panel = document.querySelector('[data-testid="properties-panel"]');
|
|
if (panel && panel.offsetWidth > 0 && panel.offsetHeight > 0) {
|
|
return true;
|
|
}
|
|
|
|
const headings = document.querySelectorAll('h1, h2, h3, h4');
|
|
for (const h of headings) {
|
|
if (h.textContent.trim() === "Workflow Overview" && h.offsetWidth > 0 && h.offsetHeight > 0) {
|
|
return true;
|
|
}
|
|
}
|
|
|
|
const origBtn = findOriginalButton();
|
|
if (origBtn) {
|
|
if (origBtn.classList.contains("p-button-active") ||
|
|
origBtn.classList.contains("active") ||
|
|
origBtn.getAttribute("aria-expanded") === "true" ||
|
|
origBtn.getAttribute("aria-pressed") === "true") {
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
function findActiveQueueIndicator() {
|
|
const container = findTopbarContainer();
|
|
if (!container) return null;
|
|
|
|
const buttons = container.querySelectorAll("button, .p-button, .comfyui-menu-item");
|
|
for (const btn of buttons) {
|
|
if (btn.classList.contains("comfy-sidebar-custom-properties-toggle")) continue;
|
|
const text = (btn.textContent || "").toLowerCase();
|
|
if (text.includes("active") || text.includes("queued") || text.includes("running") || text.includes("pending")) {
|
|
return btn;
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
function findNativeExtensionsPanel() {
|
|
return Array.from(document.querySelectorAll('.shadow-interface:not(.actionbar-container)'))
|
|
.find(el => el.textContent.toLowerCase().includes('extensions') || el.querySelector('button')?.textContent.toLowerCase().includes('extensions'));
|
|
}
|
|
|
|
function updateSidebarTabsVisibility() {
|
|
const sidebar = document.querySelector('.comfyui-sidebar, .comfy-sidebar, .sidebar, [class*="sidebar-nav"], [class*="sidebar"]');
|
|
if (!sidebar) return;
|
|
|
|
const tabIconSelectors = {
|
|
"Assets": '[class*="comfy--image-ai-edit"]',
|
|
"Nodes": '[class*="comfy--node"]',
|
|
"Models": '[class*="comfy--ai-model"]',
|
|
"Workflows": '[class*="comfy--workflow"]',
|
|
"Apps": '[class*="lucide--panels-top-left"]',
|
|
"Templates": '[class*="comfy--template"]'
|
|
};
|
|
|
|
Object.entries(tabIconSelectors).forEach(([tab, selector]) => {
|
|
const shouldHide = app.ui?.settings?.getSettingValue(`Comfy Sidebar.Hide Junk.${tab}`) ?? false;
|
|
const icon = sidebar.querySelector(selector);
|
|
if (icon) {
|
|
const tabBtn = icon.closest('.comfyui-sidebar-tab, button, [role="tab"], li, a, .comfyui-sidebar-item') || icon;
|
|
if (shouldHide) {
|
|
if (tabBtn.style.display !== "none") tabBtn.style.setProperty("display", "none", "important");
|
|
} else {
|
|
if (tabBtn.style.display === "none") tabBtn.style.removeProperty("display");
|
|
}
|
|
}
|
|
});
|
|
|
|
const hideStockHistory = app.ui?.settings?.getSettingValue("Comfy Sidebar.Hide Junk.Override Stock Job History Tab") ?? false;
|
|
const historyIcon = sidebar.querySelector('[class*="lucide--history"]');
|
|
if (historyIcon) {
|
|
const historyBtn = historyIcon.closest('.comfyui-sidebar-tab, button, [role="tab"], li, a, .comfyui-sidebar-item') || historyIcon;
|
|
if (hideStockHistory) {
|
|
if (historyBtn.style.display !== "none") historyBtn.style.setProperty("display", "none", "important");
|
|
} else {
|
|
if (historyBtn.style.display === "none") historyBtn.style.removeProperty("display");
|
|
}
|
|
}
|
|
}
|
|
|
|
function findGraphButton() {
|
|
const elements = document.querySelectorAll('.bg-secondary-background.rounded-lg.items-center.inline-flex.pointer-events-auto');
|
|
for (const el of elements) {
|
|
if (el.closest('.p-dialog, .comfy-modal, [role="dialog"], .p-sidebar, .comfy-settings')) continue;
|
|
return el;
|
|
}
|
|
|
|
const fallbacks = document.querySelectorAll('button, .bg-secondary-background');
|
|
for (const el of fallbacks) {
|
|
if (el.closest('.p-dialog, .comfy-modal, [role="dialog"], .p-sidebar, .comfy-settings')) continue;
|
|
const text = (el.textContent || el.getAttribute("title") || "").toLowerCase();
|
|
if (text.includes("graph") || text.includes("workflow") || el.querySelector('[class*="sitemap"], [class*="workflow"]')) {
|
|
return el;
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
function syncErrorBadge(originalBtn, customBtn) {
|
|
if (!customBtn) return;
|
|
|
|
const stockDot = document.querySelector('.comfy-sidebar-hide-original-properties-btn ~ span, .comfy-sidebar-hide-original-properties-btn + span');
|
|
let customDot = customBtn.querySelector('.comfy-sidebar-error-dot');
|
|
|
|
if (stockDot) {
|
|
if (!customDot) {
|
|
customDot = document.createElement("span");
|
|
customDot.className = "comfy-sidebar-error-dot inline-flex items-center justify-center rounded-full bg-destructive-background text-white size-2 absolute -top-1 -right-1";
|
|
customBtn.appendChild(customDot);
|
|
}
|
|
} else if (customDot) {
|
|
customDot.remove();
|
|
}
|
|
}
|
|
|
|
function cleanHTML(html) {
|
|
if (!html) return "";
|
|
const tempDiv = document.createElement("div");
|
|
tempDiv.innerHTML = html;
|
|
const tempDot = tempDiv.querySelector('[class*="destructive"]');
|
|
if (tempDot) tempDot.remove();
|
|
return tempDiv.innerHTML;
|
|
}
|
|
|
|
export function syncClassicLayout() {
|
|
if (domObserver) domObserver.disconnect();
|
|
|
|
try {
|
|
const isClassicLayoutEnabled = app.ui?.settings?.getSettingValue("Comfy Sidebar.Comfy Layout") ?? false;
|
|
|
|
const hideGraphBtn = app.ui?.settings?.getSettingValue("Comfy Sidebar.Hide Junk.Graph Button") ?? false;
|
|
const graphBtn = findGraphButton();
|
|
if (graphBtn) {
|
|
if (hideGraphBtn) {
|
|
if (graphBtn.style.display !== "none") graphBtn.style.setProperty("display", "none", "important");
|
|
} else {
|
|
if (graphBtn.style.display === "none") graphBtn.style.removeProperty("display");
|
|
}
|
|
}
|
|
|
|
const extensionsPanel = findNativeExtensionsPanel();
|
|
if (extensionsPanel) {
|
|
if (isClassicLayoutEnabled) {
|
|
const container = findTopbarContainer();
|
|
if (container && extensionsPanel.parentNode !== container) {
|
|
if (!extensionsPanel._originalParent) {
|
|
extensionsPanel._originalParent = extensionsPanel.parentNode;
|
|
extensionsPanel._originalNextSibling = extensionsPanel.nextSibling;
|
|
}
|
|
container.appendChild(extensionsPanel);
|
|
extensionsPanel.classList.add("comfy-sidebar-extensions-override");
|
|
}
|
|
} else if (extensionsPanel._originalParent && extensionsPanel.parentNode !== extensionsPanel._originalParent) {
|
|
extensionsPanel._originalParent.insertBefore(extensionsPanel, extensionsPanel._originalNextSibling || null);
|
|
extensionsPanel.classList.remove("comfy-sidebar-extensions-override");
|
|
}
|
|
}
|
|
|
|
if (!isClassicLayoutEnabled) {
|
|
const originalBtn = findOriginalButton();
|
|
if (originalBtn?.classList.contains("comfy-sidebar-hide-original-properties-btn")) {
|
|
originalBtn.classList.remove("comfy-sidebar-hide-original-properties-btn");
|
|
}
|
|
|
|
const customBtn = document.querySelector(".comfy-sidebar-custom-properties-toggle");
|
|
if (customBtn) customBtn.remove();
|
|
|
|
const indicator = findActiveQueueIndicator();
|
|
if (indicator && indicator.style.display === "none") indicator.style.removeProperty("display");
|
|
|
|
updateSidebarTabsVisibility();
|
|
return;
|
|
}
|
|
|
|
const originalBtn = findOriginalButton();
|
|
const container = findTopbarContainer();
|
|
const openState = isPropertiesPanelOpen();
|
|
|
|
if (originalBtn) {
|
|
if (!originalBtn.classList.contains("comfy-sidebar-hide-original-properties-btn")) {
|
|
originalBtn.classList.add("comfy-sidebar-hide-original-properties-btn");
|
|
}
|
|
|
|
savedButtonData = {
|
|
className: originalBtn.className.replace("comfy-sidebar-hide-original-properties-btn", "").trim(),
|
|
innerHTML: originalBtn.innerHTML,
|
|
tagName: originalBtn.tagName,
|
|
attributes: Array.from(originalBtn.attributes).map(attr => ({
|
|
name: attr.name,
|
|
value: attr.value
|
|
}))
|
|
};
|
|
}
|
|
|
|
let customBtn = document.querySelector(".comfy-sidebar-custom-properties-toggle");
|
|
|
|
if (!customBtn && savedButtonData && container) {
|
|
customBtn = document.createElement(savedButtonData.tagName);
|
|
customBtn.className = savedButtonData.className + " comfy-sidebar-custom-properties-toggle";
|
|
customBtn.innerHTML = cleanHTML(savedButtonData.innerHTML);
|
|
|
|
for (const attr of savedButtonData.attributes) {
|
|
if (attr.name !== "class" && attr.name !== "id" && attr.name !== "style") {
|
|
customBtn.setAttribute(attr.name, attr.value);
|
|
}
|
|
}
|
|
|
|
customBtn.addEventListener("click", (e) => {
|
|
e.preventDefault();
|
|
e.stopPropagation();
|
|
|
|
const nativeBtn = findOriginalButton();
|
|
if (nativeBtn) {
|
|
nativeBtn.click();
|
|
requestAnimationFrame(() => syncClassicLayout());
|
|
}
|
|
});
|
|
|
|
container.appendChild(customBtn);
|
|
}
|
|
|
|
if (customBtn) {
|
|
if (originalBtn) {
|
|
const cleanedHTML = cleanHTML(originalBtn.innerHTML);
|
|
if (cleanedHTML && customBtn.innerHTML !== cleanedHTML) {
|
|
const existingDot = customBtn.querySelector('.comfy-sidebar-error-dot');
|
|
customBtn.innerHTML = cleanedHTML;
|
|
if (existingDot) customBtn.appendChild(existingDot);
|
|
}
|
|
|
|
const titleVal = originalBtn.getAttribute("title") || originalBtn.getAttribute("aria-label") || "Toggle properties panel";
|
|
customBtn.setAttribute("title", titleVal);
|
|
customBtn.setAttribute("aria-label", titleVal);
|
|
|
|
for (const cls of originalBtn.classList) {
|
|
if (cls !== "comfy-sidebar-hide-original-properties-btn" && !customBtn.classList.contains(cls)) {
|
|
customBtn.classList.add(cls);
|
|
}
|
|
}
|
|
|
|
syncErrorBadge(originalBtn, customBtn);
|
|
} else if (!customBtn.getAttribute("title")) {
|
|
customBtn.setAttribute("title", "Toggle properties panel");
|
|
customBtn.setAttribute("aria-label", "Toggle properties panel");
|
|
}
|
|
customBtn.classList.toggle("comfy-panel-open", openState);
|
|
customBtn.style.display = "inline-flex";
|
|
}
|
|
|
|
const hideQueueIndicator = app.ui?.settings?.getSettingValue("Comfy Sidebar.Hide Junk.Override Stock Job History Tab") ?? false;
|
|
const indicator = findActiveQueueIndicator();
|
|
if (indicator) {
|
|
if (hideQueueIndicator) {
|
|
if (indicator.style.display !== "none") indicator.style.setProperty("display", "none", "important");
|
|
} else if (indicator.style.display === "none") {
|
|
indicator.style.removeProperty("display");
|
|
}
|
|
}
|
|
|
|
updateSidebarTabsVisibility();
|
|
|
|
} catch (err) {
|
|
console.error("Comfy Sidebar: Error inside layout sync routine:", err);
|
|
} finally {
|
|
if (domObserver) {
|
|
domObserver.observe(document.body, {
|
|
childList: true,
|
|
subtree: true
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
export function setupPropertiesPanelToggleFix() {
|
|
if (!document.getElementById("comfy-sidebar-layout-fix-styles")) {
|
|
const style = document.createElement("style");
|
|
style.id = "comfy-sidebar-layout-fix-styles";
|
|
style.textContent = `
|
|
.comfy-sidebar-hide-original-properties-btn {
|
|
display: none !important;
|
|
}
|
|
.comfy-sidebar-hide-original-properties-btn ~ span,
|
|
.comfy-sidebar-hide-original-properties-btn + span {
|
|
display: none !important;
|
|
}
|
|
.comfy-sidebar-custom-properties-toggle {
|
|
order: 99999 !important;
|
|
position: relative !important;
|
|
overflow: visible !important;
|
|
margin-left: -8px !important;
|
|
margin-right: 0px !important;
|
|
display: inline-flex !important;
|
|
align-items: center !important;
|
|
justify-content: center !important;
|
|
padding: 0 !important;
|
|
}
|
|
.comfy-sidebar-custom-properties-toggle.comfy-panel-open {
|
|
margin-left: 0px !important;
|
|
}
|
|
.comfy-sidebar-custom-properties-toggle > span.comfy-sidebar-error-dot {
|
|
display: inline-flex !important;
|
|
}
|
|
`;
|
|
document.head.appendChild(style);
|
|
}
|
|
|
|
if (domObserver) domObserver.disconnect();
|
|
|
|
domObserver = new MutationObserver((mutations) => {
|
|
let shouldSync = false;
|
|
|
|
const isInternalSidebarMutation = (node) => {
|
|
if (!node || node.nodeType !== 1) return false;
|
|
return node.closest?.('.comfyui-sidebar, .comfy-sidebar, .comfy-sidebar-comparison-overlay') ||
|
|
node.classList?.contains('comfy-sidebar-card');
|
|
};
|
|
|
|
for (const mutation of mutations) {
|
|
if (mutation.addedNodes.length > 0 || mutation.removedNodes.length > 0) {
|
|
const hasExternalElementMutation =
|
|
Array.from(mutation.addedNodes).some(node => node.nodeType === 1 && !isInternalSidebarMutation(node)) ||
|
|
Array.from(mutation.removedNodes).some(node => node.nodeType === 1 && !isInternalSidebarMutation(node));
|
|
|
|
if (hasExternalElementMutation) {
|
|
shouldSync = true;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
if (shouldSync && !syncScheduled) {
|
|
syncScheduled = true;
|
|
requestAnimationFrame(() => {
|
|
syncClassicLayout();
|
|
syncScheduled = false;
|
|
});
|
|
}
|
|
});
|
|
|
|
domObserver.observe(document.body, {
|
|
childList: true,
|
|
subtree: true
|
|
});
|
|
|
|
syncClassicLayout();
|
|
} |