bizydraft 0.2.31__py3-none-any.whl → 0.2.78.dev20251117024007__py3-none-any.whl

bizydraft/static/js/disableComfyWebSocket.js

@@ -0,0 +1,65 @@
+// 保存原始的 WebSocket 构造函数
+const OriginalWebSocket = window.WebSocket;
+
+// 保存原始的 fetch 函数
+const OriginalFetch = window.fetch;
+
+// 需要跳过的 URL 数组
+const skipFetchUrls = [
+  'manager/badge_mode',
+  'pysssss/autocomplete'
+];
+
+class FakeWebSocket {
+  constructor(url) {
+    this.url = url;
+    this.readyState = WebSocket.CONNECTING;  // 核心：保持 CONNECTING 状态
+    console.warn('[BizyDraft] 已阻止 WebSocket 连接:', url);
+  }
+  send() {}
+  close() {}
+  addEventListener() {}
+  removeEventListener() {}
+}
+
+window.WebSocket = function(url, protocols) {
+  //精确拦截/ws请求
+  if (typeof url === 'string' && /^wss?:\/\/[^/]+\/ws(\?.*)?$/.test(url)) {
+    return new FakeWebSocket(url);
+  }
+  // 其他连接正常创建，不影响
+  return new OriginalWebSocket(url, protocols);
+};
+
+// 保留 WebSocket 的静态属性和原型
+Object.setPrototypeOf(window.WebSocket, OriginalWebSocket);
+window.WebSocket.prototype = OriginalWebSocket.prototype;
+
+// 复制静态常量（使用 defineProperty 避免只读属性错误）
+['CONNECTING', 'OPEN', 'CLOSING', 'CLOSED'].forEach(prop => {
+  Object.defineProperty(window.WebSocket, prop, {
+    value: OriginalWebSocket[prop],
+    writable: false,
+    enumerable: true,
+    configurable: true
+  });
+});
+
+// 拦截 fetch 请求
+window.fetch = function(url) {
+  // 将 URL 转换为字符串进行比较
+  const urlString = typeof url === 'string' ? url : url.toString();
+  // 检查 URL 是否在跳过列表中
+  if (skipFetchUrls.some(skipUrl => urlString.includes(skipUrl))) {
+    console.warn('[BizyDraft] 已阻止 fetch 请求:', urlString);
+    // 返回一个模拟的 Response 对象，状态为 200
+    return Promise.resolve(new Response(null, {
+      status: 200,
+      statusText: 'OK',
+      headers: new Headers({ 'Content-Type': 'application/json' })
+    }));
+  }
+
+  // 其他请求正常发送
+  return OriginalFetch.apply(this, arguments);
+};

bizydraft/static/js/freezeModeHandler.js

@@ -39,6 +39,9 @@ function freezeWorkflow(isworkflow = false) {
 
     // 阻止拖拽上传文件
     blockDragDropEvents(true);
+
+    // 关闭并禁用最小地图按钮
+    setMiniMapButtonDisabled(true, true);
 }
 
 // 解冻工作流，移除遮罩层
@@ -70,6 +73,9 @@ function unfreezeWorkflow() {
 
     // 恢复拖拽上传文件
     blockDragDropEvents(false);
+
+    // 恢复最小地图按钮可用
+    setMiniMapButtonDisabled(false);
 }
 
 // 显示提示信息
@@ -450,6 +456,23 @@ function updateHighlightedNodes() {
         }
     }
 }
+//关闭并禁用小地图按钮
+function setMiniMapButtonDisabled(disabled, clickIfActive = false) {
+    const miniMapBtn = document.querySelector('button[data-testid="toggle-minimap-button"]');
+    if (!miniMapBtn) return;
+
+    if (disabled) {
+        const isActive = miniMapBtn.classList.contains('minimap-active') ||
+            miniMapBtn.hasAttribute('minimap-active') ||
+            miniMapBtn.getAttribute('aria-pressed') === 'true';
+        if (clickIfActive && isActive) {
+            miniMapBtn.click();
+        }
+        miniMapBtn.disabled = true;
+    } else {
+        miniMapBtn.disabled = false;
+    }
+}
 
 // 处理遮罩层上的鼠标按下事件
 function handleOverlayMouseDown(e) {

bizydraft/static/js/handleStyle.js

@@ -44,6 +44,22 @@ const styleMenus = `
     .p-dialog-mask.p-overlay-mask.p-overlay-mask-enter[data-pc-name="dialog"]{
         display: none !important;
     }
+    .p-button.p-component.p-button-icon-only.p-button-text.templates-tab-button.side-bar-button.p-button-secondary{
+        display: none;
+    }
+    .p-button.p-component.p-button-icon-only.p-button-text.queue-tab-button.side-bar-button.p-button-secondary{
+        display: none;
+    }
+    .w-full.flex.content-end{
+        display: none;
+    }
+    .side-tool-bar-container.small-sidebar .side-bar-button-label{
+        display: none;
+    }
+    body .comfy-img-preview video{
+        width: 100%;
+        height: 100%;
+    }
 `;
 app.registerExtension({
   name: "comfy.BizyAir.Style",
@@ -52,5 +68,62 @@ app.registerExtension({
       textContent: styleMenus,
       parent: document.head,
     });
+    const getCloseBtn = () => {
+        // let temp = null
+        // document.querySelectorAll('h2').forEach(e => {
+        //     if (e.innerHTML == "<span>模板</span>") {
+        //         const dialogContent = e.closest('.p-dialog-content')
+        //         if (dialogContent) {
+        //             temp = dialogContent.querySelector('i.pi.pi-times.text-sm')
+        //         }
+        //     }
+        // })
+        return document.querySelector('i.pi.pi-times.text-sm')
+    }
+    const getFengrossmentBtn = () => {
+        let temp = null
+        document.querySelectorAll('button').forEach(e => {
+            if (e.getAttribute('aria-label') == "专注模式 (F)") {
+                temp = e
+            }
+        })
+        return temp
+    }
+    let indexCloseLayout = 0;
+    let indexAddSmlBar = 0;
+    let indexFengrossment = 0;
+    let iTimer = setInterval(() => {
+        indexCloseLayout++
+        if (indexCloseLayout > 10) {
+            clearInterval(iTimer)
+            return
+        }
+        if (getCloseBtn()) {
+            getCloseBtn().click()
+            clearInterval(iTimer)
+        }
+    }, 300)
+    let iTimerSmlBar = setInterval(() => {
+        indexAddSmlBar++
+        if (indexAddSmlBar > 10) {
+            clearInterval(iTimerSmlBar)
+            return
+        }
+        if (document.querySelector('.side-tool-bar-container')) {
+            document.querySelector('.side-tool-bar-container').classList.add('small-sidebar')
+            clearInterval(iTimerSmlBar)
+        }
+    }, 300)
+    let iTimerFengrossment = setInterval(() => {
+        indexFengrossment++
+        if (indexFengrossment > 10) {
+            clearInterval(iTimerFengrossment)
+            return
+        }
+        if (getFengrossmentBtn()) {
+            getFengrossmentBtn().style.display = 'none'
+            clearInterval(iTimerFengrossment)
+        }
+    }, 300)
   }
 });

bizydraft/static/js/hookLoad/configLoader.js

@@ -0,0 +1,68 @@
+// 统一的配置加载器 - 合并 model.js 和 media.js 的配置请求
+// 只发送一次 API 请求，同时获取 weight_load_nodes 和 media_load_nodes
+
+// 统一的配置缓存
+let fullConfigCache = null;
+let configLoadPromise = null;
+
+// API配置（统一使用一个域名，根据实际需要调整）
+const CONFIG_API_URL = 'https://bizyair.cn/api/special/comfyagent_node_config?t=' + Math.floor(Date.now() / 60000);
+
+// 获取完整配置的API函数（返回包含 weight_load_nodes 和 media_load_nodes 的完整对象）
+export async function fetchFullConfig() {
+    if (fullConfigCache) {
+        return fullConfigCache;
+    }
+
+    if (configLoadPromise) {
+        return configLoadPromise;
+    }
+
+    configLoadPromise = (async () => {
+        try {
+            console.log('正在从API获取完整节点配置...');
+            const response = await fetch(CONFIG_API_URL, { credentials: 'include' });
+
+            if (!response.ok) {
+                throw new Error(`HTTP error! status: ${response.status}`);
+            }
+
+            const result = await response.json();
+
+            if (result.code === 20000 && result.data) {
+                fullConfigCache = {
+                    weight_load_nodes: result.data.weight_load_nodes || {},
+                    media_load_nodes: result.data.media_load_nodes || {}
+                };
+                console.log('完整配置加载成功:', {
+                    weight_nodes: Object.keys(fullConfigCache.weight_load_nodes).length,
+                    media_nodes: Object.keys(fullConfigCache.media_load_nodes).length
+                });
+                return fullConfigCache;
+            } else {
+                throw new Error('API返回数据格式不正确');
+            }
+        } catch (error) {
+            console.error('获取完整配置失败:', error);
+            fullConfigCache = null;
+            return null;
+        }
+    })();
+
+    return configLoadPromise;
+}
+
+// 获取模型节点配置（从完整配置中提取 weight_load_nodes）
+export async function fetchNodeConfig() {
+    const fullConfig = await fetchFullConfig();
+    return fullConfig ? fullConfig.weight_load_nodes : null;
+}
+
+// 获取媒体节点配置（从完整配置中提取 media_load_nodes）
+export async function fetchMediaConfig() {
+    const fullConfig = await fetchFullConfig();
+    return fullConfig ? fullConfig.media_load_nodes : null;
+}
+
+// 启动时后台预取（不阻塞后续逻辑）
+try { void fetchFullConfig(); } catch (e) { /* noop */ }