|
import { api } from "../../../scripts/api.js";
|
|
import { app } from "../../../scripts/app.js";
|
|
|
|
app.registerExtension({
|
|
name: "KJNodes.browserstatus",
|
|
setup() {
|
|
if (!app.ui.settings.getSettingValue("KJNodes.browserStatus")) {
|
|
return;
|
|
}
|
|
api.addEventListener("status", ({ detail }) => {
|
|
let title = "ComfyUI";
|
|
let favicon = "green";
|
|
let queueRemaining = detail && detail.exec_info.queue_remaining;
|
|
|
|
if (queueRemaining) {
|
|
favicon = "red";
|
|
title = `00% - ${queueRemaining} | ${title}`;
|
|
}
|
|
let link = document.querySelector("link[rel~='icon']");
|
|
if (!link) {
|
|
link = document.createElement("link");
|
|
link.rel = "icon";
|
|
document.head.appendChild(link);
|
|
}
|
|
link.href = new URL(`../${favicon}.png`, import.meta.url);
|
|
document.title = title;
|
|
});
|
|
|
|
api.addEventListener("progress", ({ detail }) => {
|
|
const { value, max } = detail;
|
|
const progress = Math.floor((value / max) * 100);
|
|
let title = document.title;
|
|
|
|
if (!isNaN(progress) && progress >= 0 && progress <= 100) {
|
|
const paddedProgress = String(progress).padStart(2, '0');
|
|
title = `${paddedProgress}% ${title.replace(/^\d+%\s/, '')}`;
|
|
}
|
|
document.title = title;
|
|
});
|
|
},
|
|
init() {
|
|
if (!app.ui.settings.getSettingValue("KJNodes.browserStatus")) {
|
|
return;
|
|
}
|
|
const pythongossFeed = app.extensions.find(
|
|
(e) => e.name === 'pysssss.FaviconStatus',
|
|
)
|
|
if (pythongossFeed) {
|
|
console.warn("KJNodes - Overriding pysssss.FaviconStatus")
|
|
app.extensions = app.extensions.filter(item => item !== pythongossFeed);
|
|
}
|
|
},
|
|
}); |
