(async function() { if (typeof gradioApp != 'undefined') return; const { app } = await import("../../../scripts/app.js") const { api } = await import("../../../scripts/api.js") const { $el } = await import('../../../scripts/ui.js') await import('./socket.io.js') const socketio = globalThis.io let layerStrs = []; let boundsStrs = []; let setLayerStrs = []; let socket = socketio(location.origin, { transports: ["websocket"], path: '/sd-ppp/', query: { api_level: 2, type: 'comfyui' } }); api.addEventListener("progress", ({ detail }) => { if (!detail || isNaN(detail.value / detail.max)) return; socket.emit('c_progress', { progress: Math.round(detail.value / detail.max * 100) }); }); console.log("[sd-ppp]", "Loading js extension"); const id = "Comfy.SD-PPP" app.ui.settings.addSetting({ id, name: "Download SD-PPP plugin", type: (name, setter, value) => { return $el("tr", [ $el("td", [ $el("label", { for: id.replaceAll(".", "-"), textContent: "Download SD-PPP plugin", }), ]), $el("td", [ $el("input", { type: "button", value: "Photoshop .ccx", onclick: async () => { location.href = "/extensions/sd-ppp/plugins/sd-ppp_PS.ccx" }, }) ]) ]) } }) app.registerExtension({ name: id, init() { }, async setup() { socket.on('connect', () => { socket.emit('c_reset_changes'); socket.emit('c_reset_instance_name', { name: document.title }) }); socket.on('disconnect', () => { }); socket.on('connect_error', () => { console.error('sdppp socket connect_error') }); socket.on('s_run', () => { document.getElementById('queue-button').click() }); socket.on('s_trigger_graph_change', () => { api.dispatchEvent(new CustomEvent("graphChanged")); }); setInterval(() => { checkHistoryChanges(); socket.emit('c_get_layers', (data) => { layerStrs = data.layer_strs; boundsStrs = data.bound_strs; setLayerStrs = data.set_layer_strs; }) socket.emit('c_reset_instance_name', { name: document.title }) }, 2000); }, async beforeRegisterNodeDef(nodeType, nodeData, app) { if (nodeType.comfyClass === 'Get Image From Photoshop Layer') { const onSelected = nodeType.prototype.onSelected; const onMouseEnter = nodeType.prototype.onMouseEnter; let this_handler = function() { this.widgets[0].options.values = layerStrs; this.widgets[1].options.values = boundsStrs; } nodeType.prototype.onSelected = async function(...args) { if(onSelected) await onSelected.call(this, ...args); this_handler.call(this); } nodeType.prototype.onMouseEnter = async function(...args) { if(onMouseEnter) await onMouseEnter.call(this, ...args); this_handler.call(this); } } else if (nodeType.comfyClass === 'Send Images To Photoshop') { const onSelected = nodeType.prototype.onSelected; const onMouseEnter = nodeType.prototype.onMouseEnter; let this_handler = function() { this.widgets[0].options.values = setLayerStrs; } nodeType.prototype.onSelected = async function(...args) { if(onSelected) await onSelected.call(this, ...args); this_handler.call(this); } nodeType.prototype.onMouseEnter = async function(...args) { if(onMouseEnter) await onMouseEnter.call(this, ...args); this_handler.call(this); } nodeType.prototype.onDrawBackground = async function(...args) { console.log(args); } } } }); const SDPPPNodes = [ 'Get Image From Photoshop Layer', 'Send Images To Photoshop', ] async function checkHistoryChanges() { try { const currentState = app.graph.serialize(); const mode0NodeTypes = currentState.nodes.filter(node => node.mode == 0).map(node => node.type); const containsSDPPPNodes = mode0NodeTypes.some(nodeType => SDPPPNodes.includes(nodeType)); if (!containsSDPPPNodes) return; socket.emit('c_check_changes') } catch (e) { console.error("[sd-ppp]", "Failed to check changes", e); } } })();