| import { app } from "../../../scripts/app.js"; | |
| const notificationSetup = () => { | |
| if (!("Notification" in window)) { | |
| console.log("This browser does not support notifications."); | |
| alert("This browser does not support notifications."); | |
| return; | |
| } | |
| if (Notification.permission === "denied") { | |
| console.log("Notifications are blocked. Please enable them in your browser settings."); | |
| alert("Notifications are blocked. Please enable them in your browser settings."); | |
| return; | |
| } | |
| if (Notification.permission !== "granted") { | |
| Notification.requestPermission(); | |
| } | |
| return true; | |
| }; | |
| app.registerExtension({ | |
| name: "pysssss.SystemNotification", | |
| async beforeRegisterNodeDef(nodeType, nodeData, app) { | |
| if (nodeData.name === "SystemNotification|pysssss") { | |
| const onExecuted = nodeType.prototype.onExecuted; | |
| nodeType.prototype.onExecuted = async function () { | |
| onExecuted?.apply(this, arguments); | |
| const mode = this.widgets.find((w) => w.name === "mode"); | |
| const message = this.widgets.find((w) => w.name === "message"); | |
| if (mode.value === "on empty queue") { | |
| if (app.ui.lastQueueSize !== 0) { | |
| await new Promise((r) => setTimeout(r, 500)); | |
| } | |
| if (app.ui.lastQueueSize !== 0) { | |
| return; | |
| } | |
| } | |
| if (!notificationSetup()) return; | |
| const notification = new Notification("ComfyUI", { body: message.value ?? "Your notification has triggered." }); | |
| }; | |
| const onNodeCreated = nodeType.prototype.onNodeCreated; | |
| nodeType.prototype.onNodeCreated = function () { | |
| onNodeCreated?.apply(this, arguments); | |
| notificationSetup(); | |
| }; | |
| } | |
| }, | |
| }); | |