| 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(); |
| }; |
| } |
| }, |
| }); |
|
|