bizydraft 0.2.49__py3-none-any.whl → 0.2.87__py3-none-any.whl

bizydraft/static/js/hookLoad/model.js

@@ -0,0 +1,322 @@
+import { hideWidget } from "../tool.js";
+import { fetchNodeConfig } from "./configLoader.js";
+
+// 动态配置缓存
+let nodeConfigCache = null;
+let configLoadPromise = null;
+let storageClearedOnce = false;
+
+const HAZY_WHITELIST_NODES = {};
+
+export const possibleWidgetNames = [
+  "clip_name",
+  "clip_name1",
+  "clip_name2",
+  "clip_name3",
+  "clip_name4",
+  "ckpt_name",
+  "lora_name",
+  "name",
+  "lora",
+  "lora_01",
+  "lora_02",
+  "lora_03",
+  "lora_04",
+  "lora_1_name",
+  "lora_2_name",
+  "lora_3_name",
+  "lora_4_name",
+  "lora_5_name",
+  "lora_6_name",
+  "lora_7_name",
+  "lora_8_name",
+  "lora_9_name",
+  "lora_10_name",
+  "lora_11_name",
+  "lora_12_name",
+  "model_name",
+  "control_net_name",
+  "ipadapter_file",
+  "unet_name",
+  "vae_name",
+  "model",
+  "model_name",
+  "instantid_file",
+  "pulid_file",
+  "style_model_name",
+  "yolo_model",
+  "face_model",
+  "bbox_model_name",
+  "sam_model_name",
+  "model_path",
+  "upscale_model",
+  "supir_model",
+  "sdxl_model",
+  "upscale_model_1",
+  "upscale_model_2",
+  "upscale_model_3",
+  "sam_model",
+  "sam2_model",
+  "grounding_dino_model",
+];
+
+// 获取节点配置的API函数（使用共享配置加载器）
+async function fetchNodeConfigWithCache() {
+  if (nodeConfigCache) {
+    return nodeConfigCache;
+  }
+
+  if (configLoadPromise) {
+    return configLoadPromise;
+  }
+
+  configLoadPromise = (async () => {
+    const config = await fetchNodeConfig();
+    if (config) {
+      nodeConfigCache = config;
+      console.log(
+        "节点配置加载成功:",
+        Object.keys(nodeConfigCache).length,
+        "个节点"
+      );
+    }
+    return config;
+  })();
+
+  return configLoadPromise;
+}
+
+// 读取 mode_type（不再兼容旧的 modelType）
+export function getModelTypeFromInput(inputConfig) {
+  console.log("inputConfig", inputConfig);
+  if (!inputConfig) return undefined;
+  console.log("inputConfig", inputConfig);
+  return inputConfig.mode_type;
+}
+
+// 根据节点名称获取节点配置信息（名单优先，正则补充；不阻塞返回）
+export async function getNodeConfig(nodeName) {
+  if (/bizyair/i.test(nodeName)) {
+    return null;
+  }
+
+  // 1) 名单（API）优先（仅用缓存，不等待网络）
+  if (nodeConfigCache && nodeConfigCache[nodeName]) {
+    return { nodeName, config: nodeConfigCache[nodeName] };
+  }
+
+  // 若尚未发起请求，后台发起一次
+  if (!configLoadPromise) {
+    try {
+      void fetchNodeConfigWithCache();
+    } catch (e) {}
+  }
+
+  // 2) 正则补充：如 XxxLoader => Xxx（立即返回，不等待API）
+  const regex = /^(\w+).*Loader.*/i;
+  const match = nodeName.match(regex);
+  if (match) {
+    const inferredType = match[1];
+    return {
+      nodeName,
+      config: {
+        inputs: { [nodeName]: { mode_type: inferredType, required: true } },
+      },
+    };
+  }
+  return null;
+}
+
+export function createSetWidgetCallback(nodeConfig, selectedBaseModels = []) {
+  return function setWidgetCallback() {
+    if (!nodeConfig || !nodeConfig.config || !nodeConfig.config.inputs) {
+      console.warn("节点配置无效:", nodeConfig);
+      return;
+    }
+
+    const inputs = nodeConfig.config.inputs;
+    const inputKeys = Object.keys(inputs);
+
+    // 根据API配置找到对应的widget
+    const targetWidgets = [];
+    inputKeys.forEach((inputKey) => {
+      const widget = this.widgets.find((w) => w.name === inputKey);
+      if (widget) {
+        targetWidgets.push({
+          widget: widget,
+          inputKey: inputKey,
+          inputConfig: inputs[inputKey],
+        });
+      }
+    });
+
+    // 如果没有找到匹配的widget，使用原来的逻辑作为备选
+    if (targetWidgets.length === 0) {
+      const fallbackWidgets = this.widgets.filter((widget) =>
+        possibleWidgetNames.includes(widget.name)
+      );
+      fallbackWidgets.forEach((wdt, index) => {
+        const firstInput = Object.values(inputs)[0];
+        if (firstInput) {
+          targetWidgets.push({
+            widget: wdt,
+            inputKey: wdt.name,
+            inputConfig: firstInput,
+            index: index,
+          });
+        }
+      });
+    }
+
+    targetWidgets.forEach(({ widget, inputKey, inputConfig, index }) => {
+      // 检查是否禁用comfyagent
+      if (inputConfig.disable_comfyagent) {
+        console.log(`跳过禁用的widget: ${inputKey}`);
+        return;
+      }
+
+      widget.value = widget.value || "to choose";
+      widget.mouse = function (e, pos, canvas) {
+        try {
+          if (
+            e.type === "pointerdown" ||
+            e.type === "mousedown" ||
+            e.type === "click" ||
+            e.type === "pointerup"
+          ) {
+            e.preventDefault();
+            e.stopPropagation();
+            e.widgetClick = true;
+            window.parent.postMessage(
+              {
+                type: "collapsePublishWorkflowDialog",
+                method: "collapsePublishWorkflowDialog",
+                result: true,
+              },
+              "*"
+            );
+            const currentNode = this.node;
+
+            if (!currentNode || !currentNode.widgets) {
+              console.warn("Node or widgets not available");
+              return false;
+            }
+
+            if (
+              typeof bizyAirLib !== "undefined" &&
+              typeof bizyAirLib.showModelSelect === "function"
+            ) {
+              bizyAirLib.showModelSelect({
+                modelType: [getModelTypeFromInput(inputConfig)],
+                selectedBaseModels,
+                onApply: (version, model) => {
+                  if (!currentNode || !currentNode.widgets) return;
+
+                  // 更新widget值
+                  widget.value = version.file_name;
+
+                  // 找到对应的隐藏字段（固定命名：model_version_id, model_version_id2...）
+                  let modelVersionField;
+                  // 真实绑定顺序应与targetWidgets相同，因此通过当前widget在targetWidgets的索引定位
+                  const twIndex = targetWidgets.findIndex(
+                    (tw) => tw.widget === widget
+                  );
+                  const fieldName =
+                    twIndex === 0
+                      ? "model_version_id"
+                      : `model_version_id${twIndex + 1}`;
+                  modelVersionField = currentNode.widgets.find(
+                    (w) => w.name === fieldName
+                  );
+
+                  if (model && modelVersionField && version) {
+                    modelVersionField.value = version.id;
+                    currentNode.setDirtyCanvas(true);
+
+                    // 删除节点上的感叹号徽章
+                    if (
+                      currentNode &&
+                      currentNode.badges &&
+                      Array.isArray(currentNode.badges)
+                    ) {
+                      // 移除 text 为 '!' 的徽章
+                      currentNode.badges = currentNode.badges.filter(
+                        (badgeFn) => {
+                          try {
+                            const badge =
+                              typeof badgeFn === "function"
+                                ? badgeFn()
+                                : badgeFn;
+                            return badge.text !== "!";
+                          } catch (e) {
+                            return true;
+                          }
+                        }
+                      );
+                      // 同时移除 hasTips 标记
+                      if (currentNode.hasTips) {
+                        delete currentNode.hasTips;
+                      }
+                    }
+                  }
+                },
+              });
+            } else {
+              console.error("bizyAirLib not available");
+            }
+            return false;
+          }
+        } catch (error) {
+          console.error("Error handling mouse event:", error);
+        }
+      };
+
+      widget.options = widget.options || {};
+      widget.options.values = () => [];
+      widget.options.editable = false;
+      widget.clickable = true;
+      widget.processMouse = true;
+    });
+  };
+}
+
+export function setupNodeMouseBehavior(node, nodeConfig) {
+  // 固定隐藏主版本ID字段（其余编号字段为hidden类型本身不可见）
+  hideWidget(node, "model_version_id");
+
+  // 只设置必要的状态信息，不修改onMouseDown（已在上面的扩展中处理）
+  if (!node._bizyairState) {
+    node._bizyairState = {
+      lastClickTime: 0,
+      DEBOUNCE_DELAY: 300,
+      nodeConfig: nodeConfig,
+    };
+  }
+}
+
+export function addBadge(node) {
+  const customBadge = new LGraphBadge({
+    text: "!",
+    fgColor: "white",
+    bgColor: "#FF6B6B",
+    fontSize: 12,
+    padding: 8,
+    height: 20,
+    cornerRadius: 10,
+  });
+  if (!Array.isArray(node.badges)) {
+    node.badges = [];
+  }
+  if (node.hasTips) {
+    return;
+  }
+  node.badges.push(() => customBadge);
+  node.hasTips = true;
+}
+
+// 启动时后台预取（不阻塞后续逻辑）
+try {
+  void fetchNodeConfigWithCache();
+} catch (e) {
+  /* noop */
+}

bizydraft/static/js/hookLoadMedia.js

@@ -0,0 +1,196 @@
+import { app } from "../../scripts/app.js";
+import { hideWidget } from "./tool.js";
+import {
+  getMediaNodeConfig,
+  getMediaInputKeys,
+  computeIsMediaNode,
+  mediaNodeList,
+  fetchMediaConfigWithCache,
+  findMediaWidget,
+  findMediaWidgets,
+  updateWidgetsOptions,
+  handleNewUploadedFile,
+  extractUrlFromInput,
+  initMaps,
+  createImageNameWidgetCallbackForMediaWidget,
+  createMediaWidgetCallback,
+  setupVaWidgets,
+  applyWorkflowImageSettings,
+  fetchImageList,
+} from "./hookLoad/media.js";
+
+app.registerExtension({
+  name: "bizyair.image.to.oss",
+  beforeRegisterNodeDef(nodeType, nodeData) {
+    let workflowParams = null;
+    document.addEventListener("workflowLoaded", (event) => {
+      workflowParams = event.detail;
+    });
+    document.addEventListener("drop", (e) => {
+      e.preventDefault();
+      const files = e.dataTransfer.files;
+
+      Array.from(files).forEach((file) => {
+        if (file.type === "application/json" || file.name.endsWith(".json")) {
+          const reader = new FileReader();
+          reader.onload = function (event) {
+            try {
+              const jsonContent = JSON.parse(event.target.result);
+              if (jsonContent && jsonContent.nodes) {
+                window.currentWorkflowData = jsonContent;
+              }
+            } catch (error) {
+              console.error("解析JSON文件失败:", error);
+            }
+          };
+          reader.readAsText(file);
+        }
+      });
+    });
+    const originalOnNodeCreated = nodeType.prototype.onNodeCreated;
+    nodeType.prototype.onNodeCreated = async function () {
+      const result = originalOnNodeCreated?.apply(this, arguments);
+      if (!(await computeIsMediaNode(nodeData.name))) {
+        return result;
+      }
+
+      const apiHost = "https://bizyair.cn/api";
+      // const apiHost = 'http://localhost:3000/api'
+
+      // 获取 API 配置
+      const mediaNodeConfig = await getMediaNodeConfig(nodeData.name);
+      const apiInputKeys = getMediaInputKeys(mediaNodeConfig);
+
+      // 查找媒体 widget
+      const media_widget = findMediaWidget(this.widgets, apiInputKeys);
+      const va_widgets = findMediaWidgets(this.widgets, apiInputKeys);
+      let image_name_widget = this.widgets.find((w) => w.name === "image_name");
+
+      // 获取文件列表
+      const image_list = await fetchImageList(apiHost, nodeData.name);
+
+      // 初始化 Map 映射
+      const { urlToNameMap, nameToItemMap } = initMaps(image_list);
+
+      // 如果找到va_widgets，处理它们
+      if (va_widgets.length > 0) {
+        image_name_widget = setupVaWidgets(
+          this,
+          va_widgets,
+          image_list,
+          urlToNameMap,
+          nameToItemMap,
+          image_name_widget,
+          va_widgets[0]
+        );
+      }
+
+      // 如果va_widgets没有创建image_name_widget，使用原有逻辑创建
+      if (!image_name_widget && media_widget) {
+        image_name_widget = this.addWidget(
+          "combo",
+          "image_name",
+          "",
+          createImageNameWidgetCallbackForMediaWidget(
+            nameToItemMap,
+            media_widget
+          ),
+          {
+            serialize: true,
+            values: image_list.map((item) => item.name),
+          }
+        );
+      }
+
+      // 如果进入了va_widgets分支，使用va_widgets中第一个作为media_widget的替代
+      const actualMediaWidget =
+        va_widgets.length > 0 ? va_widgets[0] : media_widget;
+
+      if (image_name_widget && actualMediaWidget) {
+        const val =
+          urlToNameMap.get(actualMediaWidget.value) || actualMediaWidget.value;
+        image_name_widget.label = actualMediaWidget.label;
+        image_name_widget.value = val;
+
+        // 调整 widget 顺序
+        const currentIndex = this.widgets.indexOf(image_name_widget);
+        if (currentIndex > 1) {
+          this.widgets.splice(currentIndex, 1);
+          this.widgets.splice(1, 0, image_name_widget);
+        }
+
+        // 如果没有进入va_widgets分支，才隐藏media_widget
+        if (va_widgets.length === 0) {
+          hideWidget(this, media_widget.name);
+        }
+
+        // 更新 widget 选项列表
+        updateWidgetsOptions([actualMediaWidget], image_list);
+
+        // 对于va_widgets的情况，callback已经在上面重写过了，不需要再次重写
+        if (va_widgets.length === 0 && media_widget) {
+          const callback = media_widget.callback;
+          media_widget.callback = createMediaWidgetCallback(
+            media_widget,
+            urlToNameMap,
+            nameToItemMap,
+            image_list,
+            image_name_widget,
+            actualMediaWidget,
+            callback
+          );
+        }
+      }
+
+      // 应用工作流设置
+      const workflowData = window.currentWorkflowData || workflowParams;
+      if (workflowData) {
+        await applyWorkflowImageSettings(
+          workflowData,
+          image_list,
+          media_widget,
+          image_name_widget,
+          this.id,
+          va_widgets,
+          actualMediaWidget
+        );
+        if (window.currentWorkflowData) {
+          delete window.currentWorkflowData;
+        }
+      } else {
+        await applyWorkflowImageSettings(
+          workflowParams,
+          image_list,
+          media_widget,
+          image_name_widget,
+          this.id,
+          va_widgets,
+          actualMediaWidget
+        );
+      }
+
+      // 发送完成消息
+      window.parent.postMessage(
+        {
+          type: "functionResult",
+          method: "hookLoadImageCompleted",
+          params: {},
+        },
+        "*"
+      );
+    };
+  },
+});
+
+// app.api.addEventListener('graphChanged', (e) => {
+//     console.log('Graph 发生变化，当前 workflow JSON:', e.detail)
+//     window.parent.postMessage({
+//         type: 'functionResult',
+//         method: 'workflowChanged',
+//         result: e.detail
+//     }, '*');
+
+//     document.dispatchEvent(new CustomEvent('workflowLoaded', {
+//         detail: e.detail
+//     }));
+// })