bizydraft 0.1.6__py3-none-any.whl → 0.1.7__py3-none-any.whl

bizydraft/static/js/handleStyle.js

@@ -31,9 +31,6 @@ const styleMenus = `
     #comfyui-body-bottom{
         display: none;
     }
-    // .p-dialog-mask.p-overlay-mask.p-overlay-mask-enter{
-    //     display: none !important;
-    // }
 `
 app.registerExtension({
     name: "comfy.BizyAir.Style",

bizydraft/static/js/postEvent.js

@@ -1,9 +1,4 @@
 import { app } from "../../scripts/app.js";
-// import { $el } from "../../scripts/ui.js";
-
-// import { WebSocketClient } from './socket.js';
-
-
 
 app.registerExtension({
     name: "comfy.BizyAir.Socket",
@@ -14,11 +9,7 @@ app.registerExtension({
 
     customSocket(callback, customUrl) {
         const url = customUrl || app.api.socket.url;
-        // const clientId = 'e07abdf5f465462f8dc43ca0812e9284';
-        // const socket = new WebSocket(url + "?clientId=" + sessionStorage.getItem("clientId"));
-        // const socket = new WebSocket(url + "?clientId=" + clientId);
         const socket = new WebSocket(url);
-
         const dispatchCustomEvent = this.dispatchCustomEvent;
 
         socket.onmessage = function (event) {
@@ -62,18 +53,22 @@ app.registerExtension({
                     }
                 } else {
                     const msg = JSON.parse(event.data);
+                    window.parent.postMessage({
+                        type: 'functionResult',
+                        method: 'progress_info_change',
+                        result: msg.progress_info
+                    }, '*');
                     switch (msg.type) {
                         case 'status':
                             if (msg.data.sid) {
                                 const clientId = msg.data.sid;
-                                window.name = clientId; // use window name so it isnt reused when duplicating tabs
-                                sessionStorage.setItem('clientId', clientId); // store in session storage so duplicate tab can load correct workflow
+                                window.name = clientId; 
+                                sessionStorage.setItem('clientId', clientId); 
                                 socket.clientId = clientId;
                             }
                             dispatchCustomEvent('status', msg.data.status ?? null);
                             break;
                         case 'executing':
-                            // msg.data.prompt_id && (msg.data.prompt_id = '');
 
                             dispatchCustomEvent(
                                 'executing',
@@ -180,46 +175,19 @@ app.registerExtension({
         const parts = value.split(`; ${name}=`);
         if (parts.length === 2) return parts.pop().split(';').shift();
     },
-    // doSendSocket() {
-    //     const socket = new WebSocketClient(`/w/v1/comfy/draft/ws?clientId=${sessionStorage.getItem("clientId")
-    //         }&userId=${this.getCookie("user_id")
-    //         }`);
-    //     const customSocket = this.customSocket.bind(this);
-    //     socket.on('message', (message) => {
-    //         let data = JSON.parse(message.data);
-    //         if (data.status == "Running") {
-    //             customSocket(() => {}, data.wsUrl);
-    //         }
-    //     });
-    // },
 
 
     async setup() {
+        
+
         const customSocket = this.customSocket.bind(this);
         const startSocket = this.startSocket.bind(this);
         const closeSocket = this.closeSocket.bind(this);
         const changeSocketUrl = this.changeSocketUrl.bind(this);
         const sendSocketMessage = this.sendSocketMessage.bind(this);
-        // let iTimer = null;
-
-        // // 直接检查一次
-        // const initialUserId = this.getCookie("user_id");
-        // if (initialUserId) {
-        //     // 初始检查到 userId
-        //     this.doSendSocket();
-        // }
-
-        // // 使用定时器监听cookie变化
-        // let lastCookieValue = this.getCookie("user_id");
-        // iTimer = setInterval(() => {
-        //     const currentCookieValue = this.getCookie("user_id");
-        //     if (currentCookieValue && currentCookieValue !== lastCookieValue) {
-        //         // 检测到 cookie user_id 变化
-        //         lastCookieValue = currentCookieValue;
-        //         this.doSendSocket();
-        //         clearInterval(iTimer);
-        //     }
-        // }, 300); // 每秒检查一次
+
+        // 设置节点事件
+        // setupNodeEvents();
 
         const methods = {
             customSocket: function (params) {
@@ -373,17 +341,11 @@ app.registerExtension({
                 };
                 console.log("-----------setCookie-----------", params)
                 setCookie(params.name, params.value, params.days);
-                // window.parent.postMessage({
-                //     type: 'functionResult',
-                //     method: 'setCookie',
-                //     result: true
-                // }, '*');
+
                 return true;
             },
             fitView: function () {
-                // window.app.canvas.ds.offset = [0, 0];
-                // window.app.canvas.ds.scale = 1;
-                // window.app.canvas.setDirty(true, true);
+
                 app.canvas.fitViewToSelectionAnimated()
 
                 window.parent.postMessage({
@@ -392,6 +354,55 @@ app.registerExtension({
                     result: true
                 }, '*');
                 return true;
+            },
+            clickAssistant: function () {
+                const assistantBtn = document.querySelector('.btn-assistant');
+                if (assistantBtn) {
+                    assistantBtn.click();
+                    window.parent.postMessage({
+                        type: 'functionResult',
+                        method: 'clickAssistant',
+                        result: true
+                    }, '*');
+                    return true;
+                } else {
+                    console.warn('Assistant button not found');
+                    window.parent.postMessage({
+                        type: 'functionResult',
+                        method: 'clickAssistant',
+                        result: false
+                    }, '*');
+                    return false;
+                }
+            },
+            clickCommunity: function () {
+                const communityBtn = document.querySelectorAll('.bizyair-menu-content .cursor-pointer.relative.px-3.rounded-sm')[1];
+                if (communityBtn) {
+                    communityBtn.click();
+                    window.parent.postMessage({
+                        type: 'functionResult',
+                        method: 'clickCommunity',
+                        result: true
+                    }, '*');
+                    return true;
+                } else {
+                    window.parent.postMessage({
+                        type: 'functionResult',
+                        method: 'clickCommunity',
+                        result: false
+                    }, '*');
+                    return false;
+                }
+            },
+            toPublish: async function () {
+                const graph = await app.graphToPrompt();
+                window.parent.postMessage({
+                    type: 'functionResult',
+                    method: 'toPublish',
+                    result: graph.workflow
+                }, '*');
+                return graph.workflow;
+
             }
         };
 
@@ -400,10 +411,17 @@ app.registerExtension({
                 const methodName = event.data.method;
                 const params = event.data.params || {};
 
+
                 if (methods[methodName]) {
                     methods[methodName](params);
                 } else {
                     console.error('方法不存在:', methodName);
+                    window.parent.postMessage({
+                        type: 'functionResult',
+                        method: methodName,
+                        error: `方法 ${methodName} 不存在`,
+                        success: false
+                    }, '*');
                 }
             }
         });

bizydraft/static/js/socket.js

@@ -76,15 +76,6 @@ export class WebSocketClient {
     this.startKeepAlive()
   }
 
-  onMessage(message) {
-    message
-    // const data = JSON.parse(message.data);
-    // if (data === 'pong') {
-    //   // Do nothing
-    // } else {
-    //   console.log('message:', data);
-    // }
-  }
 
   onError(error) {
     console.error('WebSocket Error: ', error)

{bizydraft-0.1.6.dist-info → bizydraft-0.1.7.dist-info}/METADATA

@@ -1,4 +1,4 @@
 Metadata-Version: 2.4
 Name: bizydraft
-Version: 0.1.6
+Version: 0.1.7
 Summary: bizydraft

{bizydraft-0.1.6.dist-info → bizydraft-0.1.7.dist-info}/RECORD

@@ -3,14 +3,14 @@ bizydraft/hijack_nodes.py,sha256=xPVLBBnRiJaI8ZbR80O8W4C9SlNAp7MxP7dKj_pRlYs,869
 bizydraft/hijack_routes.py,sha256=tzSIY-3wLiwoHALf7MswjAF5FqDSmXynGjpfTgRdkxE,6138
 bizydraft/resp.py,sha256=8INvKOe5Dgai3peKfqKjrhUoYeuXWXn358w30-_cY-A,369
 bizydraft/server.py,sha256=ES1fBYr_Ufv_YWsynLck1ln9bXTGjK3KXas1CK9WBqM,1358
-bizydraft/static/js/handleStyle.js,sha256=rfw5G_RfENNnwQOaGQyT8vLSMRt51JcPbsGd6itAv1Y,1272
+bizydraft/static/js/handleStyle.js,sha256=WDaMQSaZrij10pyXbQ3jinWjhb6iTt_6iQclgbnr3qc,1167
 bizydraft/static/js/hookLoadImage.js,sha256=cEaqedTnngieI-Kb0PVgLor1nVDkFQx1_7AiurwtRwI,3878
 bizydraft/static/js/main.js,sha256=cZ-7wR9T8aNLzIrTjc-g9xVZf7z6TXWl1zhP_wXSSVo,150
-bizydraft/static/js/postEvent.js,sha256=4PVl929jU1m3ooKA1lFXNWgv62mtKAFnhqbsHVBUYi8,16488
-bizydraft/static/js/socket.js,sha256=iNxbUPSQluIlx1x0MIVL7UJ35BXqLH7Z2s8vmfmbyyc,2643
+bizydraft/static/js/postEvent.js,sha256=cmwycxslbbvRAvzT0Vp_rVZh46WNrskcQydcmTrsnw8,16809
+bizydraft/static/js/socket.js,sha256=q_SWsySJdVV7ZTH-3Pdr09pkWGQj08C9xVAQES7eJaU,2440
 bizydraft/static/js/tool.js,sha256=IlDVjddodT9DtfdIGiIzSIKdxbUMztDLQ4jI0uFJCwk,206
 bizydraft/static/js/uploadFile.js,sha256=7PXMYRlgxgDWFFbsaasjf3TelsW6r9mD1TFTjQI6X0E,4957
-bizydraft-0.1.6.dist-info/METADATA,sha256=cjpKUrIR9Pt-pvxP6yb94TgtyQqhFKJtVJ2FF_UMXWc,72
-bizydraft-0.1.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bizydraft-0.1.6.dist-info/top_level.txt,sha256=XtoBq6hjZhXIM7aas4GtPDtAiKo8FdLzMABXW8qqQ8M,10
-bizydraft-0.1.6.dist-info/RECORD,,
+bizydraft-0.1.7.dist-info/METADATA,sha256=DrbYCQxUl5sv19uGmP0sJUGwOx1UVhVf_6xBAo7WhD8,72
+bizydraft-0.1.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizydraft-0.1.7.dist-info/top_level.txt,sha256=XtoBq6hjZhXIM7aas4GtPDtAiKo8FdLzMABXW8qqQ8M,10
+bizydraft-0.1.7.dist-info/RECORD,,