uinodes.com doc
This commit is contained in:
code羊
@@ -0,0 +1,222 @@
|
||||
import { app } from "../../../scripts/app.js";
|
||||
|
||||
|
||||
function normalize_node_name(node_name) {
|
||||
if (!node_name) return '';
|
||||
|
||||
let text = String(node_name).normalize('NFKC');
|
||||
|
||||
// 替换加减乘除为英文
|
||||
const symbolMap = {
|
||||
'+': 'plus',
|
||||
'*': 'times',
|
||||
'×': 'times',
|
||||
'/': 'div',
|
||||
'÷': 'div',
|
||||
};
|
||||
|
||||
for (const [symbol, replacement] of Object.entries(symbolMap)) {
|
||||
const escapedSymbol = symbol.replace(/[-/\\^$*+?.()|[\]{}]/g, '\\$&');
|
||||
const regex = new RegExp(escapedSymbol, 'g');
|
||||
text = text.replace(regex, replacement);
|
||||
}
|
||||
|
||||
// 将所有非字母数字（包括空格、标点等）替换为 `-`
|
||||
text = text.replace(/[^a-zA-Z0-9]+/g, '-');
|
||||
|
||||
// 去除首尾和重复的连字符
|
||||
text = text.replace(/-+/g, '-').replace(/^-|-$/g, '');
|
||||
|
||||
return text;
|
||||
}
|
||||
|
||||
/** 根据 python_module 和节点名构建 URL */
|
||||
function buildNodeUrl(python_module, nodeName) {
|
||||
if (!python_module || !nodeName) return null;
|
||||
|
||||
let category = '';
|
||||
|
||||
if (python_module === "nodes") {
|
||||
category = "nodes";
|
||||
} else if (python_module.startsWith("comfy_extras.")) {
|
||||
category = "extra_nodes";
|
||||
} else if (python_module.startsWith("custom_nodes.")) {
|
||||
const parts = python_module.split(".");
|
||||
if (parts.length >= 2) {
|
||||
category = parts[1]; // 取第二个部分作为仓库名
|
||||
} else {
|
||||
category = "custom_nodes";
|
||||
}
|
||||
} else if (python_module.startsWith("comfy_api_nodes.")) {
|
||||
category = "nodes_api";
|
||||
} else {
|
||||
return null;
|
||||
}
|
||||
|
||||
const slug = normalize_node_name(nodeName);
|
||||
|
||||
return `https://uinodes.com/plugins/${category}/${slug}`;
|
||||
}
|
||||
|
||||
|
||||
// 部分代码参考kjnodes https://github.com/kijai/ComfyUI-KJNodes
|
||||
app.registerExtension({
|
||||
name: "fsy.Link2uinodesCOM",
|
||||
async beforeRegisterNodeDef(nodeType, nodeData) {
|
||||
|
||||
try {
|
||||
addLinkIcon(nodeData, nodeType);
|
||||
} catch (error) {
|
||||
console.error("Error in registering fsy.Link2uinodesCOM", error);
|
||||
}
|
||||
},
|
||||
|
||||
});
|
||||
|
||||
/** 获取节点的名字 */
|
||||
function getNodeName(nodeData, nodeInstance) {
|
||||
// 1. 从节点实例获取标题
|
||||
if (nodeInstance && nodeInstance.title) {
|
||||
return nodeInstance.title;
|
||||
}
|
||||
|
||||
// 2. 从节点数据获取显示名称
|
||||
if (nodeData.display_name) {
|
||||
return nodeData.display_name;
|
||||
}
|
||||
|
||||
// 3. 从节点数据获取名称
|
||||
if (nodeData.name) {
|
||||
return nodeData.name;
|
||||
}
|
||||
|
||||
// 4. 从节点类型获取
|
||||
if (nodeInstance && nodeInstance.type) {
|
||||
return nodeInstance.type;
|
||||
}
|
||||
|
||||
return "Unknown Node";
|
||||
}
|
||||
|
||||
/** 计算图标位置 */
|
||||
function getIconPosition(nodeSize, iconSize, iconMargin, position) {
|
||||
const positions = {
|
||||
'left_of_help': {
|
||||
x: nodeSize[0] - iconSize - iconMargin - 18 - 5,
|
||||
y: iconSize - 34
|
||||
},
|
||||
'top_right': {
|
||||
x: nodeSize[0] - iconSize - iconMargin,
|
||||
y: iconSize - 34 - 20
|
||||
},
|
||||
'top_left': {
|
||||
x: iconMargin,
|
||||
y: iconSize - 34
|
||||
}
|
||||
};
|
||||
|
||||
return positions[position] || positions['left_of_help'];
|
||||
}
|
||||
|
||||
/** 为节点添加链接图标 */
|
||||
export const addLinkIcon = (
|
||||
nodeData,
|
||||
nodeType,
|
||||
opts = { icon_size: 14, icon_margin: 4 }
|
||||
) => {
|
||||
opts = opts || {}
|
||||
const iconSize = opts.icon_size ? opts.icon_size : 14
|
||||
const iconMargin = opts.icon_margin ? opts.icon_margin : 4
|
||||
|
||||
const drawFg = nodeType.prototype.onDrawForeground
|
||||
nodeType.prototype.onDrawForeground = function (ctx) {
|
||||
const r = drawFg ? drawFg.apply(this, arguments) : undefined
|
||||
if (this.flags.collapsed) return r
|
||||
|
||||
// 获取图标位置
|
||||
const position = app.ui.settings.getSettingValue("fsy.Link2uinodesCOM") || 'left_of_help';
|
||||
const iconPos = getIconPosition(this.size, iconSize, iconMargin, position);
|
||||
|
||||
// 绘制链接图标
|
||||
ctx.save()
|
||||
ctx.translate(iconPos.x - 2, iconPos.y)
|
||||
ctx.scale(iconSize / 32, iconSize / 32)
|
||||
ctx.strokeStyle = 'rgba(255,255,255,0.3)'
|
||||
ctx.lineCap = 'round'
|
||||
ctx.lineJoin = 'round'
|
||||
ctx.lineWidth = 2.4
|
||||
ctx.font = 'bold 36px monospace'
|
||||
ctx.fillStyle = '#4A9EFF'; // 蓝色链接图标
|
||||
ctx.fillText('🔗', 0, 24) // 使用链接emoji
|
||||
ctx.restore()
|
||||
|
||||
return r
|
||||
}
|
||||
|
||||
// 处理点击事件
|
||||
const mouseDown = nodeType.prototype.onMouseDown
|
||||
nodeType.prototype.onMouseDown = function (e, localPos, canvas) {
|
||||
const r = mouseDown ? mouseDown.apply(this, arguments) : undefined
|
||||
|
||||
// 获取图标位置
|
||||
const position = app.ui.settings.getSettingValue("fsy.Link2uinodesCOM") || 'left_of_help';
|
||||
const iconPos = getIconPosition(this.size, iconSize, iconMargin, position);
|
||||
|
||||
// 检查是否点击了链接图标
|
||||
if (
|
||||
localPos[0] > iconPos.x &&
|
||||
localPos[0] < iconPos.x + iconSize &&
|
||||
localPos[1] > iconPos.y &&
|
||||
localPos[1] < iconPos.y + iconSize
|
||||
) {
|
||||
|
||||
// 构建 URL 链接
|
||||
const url = buildNodeUrl(nodeData.python_module, nodeData.name);
|
||||
// console.log("构建的URL:", url);
|
||||
|
||||
// 打开链接前确认
|
||||
if (url) {
|
||||
const confirmed = confirm(`是否要打开以下链接？\n\n${url}\n\n点击"确定"将在新标签页中打开此链接。`);
|
||||
if (confirmed) {
|
||||
window.open(url, '_blank');
|
||||
}
|
||||
}
|
||||
|
||||
return true; // 阻止事件继续传播
|
||||
}
|
||||
|
||||
return r;
|
||||
}
|
||||
|
||||
// 添加悬停提示
|
||||
const onMouseMove = nodeType.prototype.onMouseMove
|
||||
nodeType.prototype.onMouseMove = function (e, localPos, canvas) {
|
||||
const r = onMouseMove ? onMouseMove.apply(this, arguments) : undefined;
|
||||
|
||||
const position = app.ui.settings.getSettingValue("fsy.Link2uinodesCOM") || 'left_of_help';
|
||||
const iconPos = getIconPosition(this.size, iconSize, iconMargin, position);
|
||||
|
||||
// 检查鼠标是否悬停在链接图标上
|
||||
if (
|
||||
localPos[0] > iconPos.x &&
|
||||
localPos[0] < iconPos.x + iconSize &&
|
||||
localPos[1] > iconPos.y &&
|
||||
localPos[1] < iconPos.y + iconSize
|
||||
) {
|
||||
// 设置鼠标样式为指针
|
||||
canvas.canvas.style.cursor = 'pointer';
|
||||
|
||||
// 构建URL并设置悬停提示
|
||||
const url = buildNodeUrl(nodeData.python_module, nodeData.name);
|
||||
if (url) {
|
||||
canvas.canvas.title = `即将打开 ${url}`;
|
||||
}
|
||||
|
||||
} else {
|
||||
// 移除悬停提示
|
||||
canvas.canvas.title = '';
|
||||
}
|
||||
|
||||
return r;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user