Spaces:
Running
on
Zero
Running
on
Zero
| import { api } from "../../../scripts/api.js"; | |
| import { app } from "../../../scripts/app.js"; | |
| // Simple script that adds the current queue size to the window title | |
| // Adds a favicon that changes color while active | |
| app.registerExtension({ | |
| name: "pysssss.FaviconStatus", | |
| async setup() { | |
| let link = document.querySelector("link[rel~='icon']"); | |
| if (!link) { | |
| link = document.createElement("link"); | |
| link.rel = "icon"; | |
| document.head.appendChild(link); | |
| } | |
| const getUrl = (active, user) => new URL(`assets/favicon${active ? "-active" : ""}${user ? ".user" : ""}.ico`, import.meta.url); | |
| const testUrl = async (active) => { | |
| const url = getUrl(active, true); | |
| const r = await fetch(url, { | |
| method: "HEAD", | |
| }); | |
| if (r.status === 200) { | |
| return url; | |
| } | |
| return getUrl(active, false); | |
| }; | |
| const activeUrl = await testUrl(true); | |
| const idleUrl = await testUrl(false); | |
| let executing = false; | |
| const update = () => (link.href = executing ? activeUrl : idleUrl); | |
| for (const e of ["execution_start", "progress"]) { | |
| api.addEventListener(e, () => { | |
| executing = true; | |
| update(); | |
| }); | |
| } | |
| api.addEventListener("executing", ({ detail }) => { | |
| // null will be sent when it's finished | |
| executing = !!detail; | |
| update(); | |
| }); | |
| api.addEventListener("status", ({ detail }) => { | |
| let title = "ComfyUI"; | |
| if (detail && detail.exec_info.queue_remaining) { | |
| title = `(${detail.exec_info.queue_remaining}) ${title}`; | |
| } | |
| document.title = title; | |
| update(); | |
| executing = false; | |
| }); | |
| update(); | |
| }, | |
| }); | |