| import { api } from "../../../scripts/api.js";
|
| import { app } from "../../../scripts/app.js";
|
|
|
|
|
|
|
|
|
| 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 }) => {
|
|
|
| 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();
|
| },
|
| });
|
|
|