247 lines
9.9 KiB
JavaScript
247 lines
9.9 KiB
JavaScript
/* global window, document, requestAnimationFrame */
|
||
// AgentUI serialization helpers extracted from editor.html
|
||
// Exposes window.AgentUISer.{toPipelineJSON, fromPipelineJSON}
|
||
// Depends on globals defined by editor.html: editor, NODE_IO, addNode, applyNodeDefaults, status
|
||
(function (w) {
|
||
'use strict';
|
||
|
||
function ensureDeps() {
|
||
if (!w || !w.editor) throw new Error('AgentUISer: global editor is not available');
|
||
if (!w.NODE_IO) throw new Error('AgentUISer: global NODE_IO is not available');
|
||
}
|
||
|
||
// Drawflow -> pipeline JSON
|
||
function toPipelineJSON() {
|
||
ensureDeps();
|
||
const editor = w.editor;
|
||
const NODE_IO = w.NODE_IO;
|
||
|
||
const data = editor.export();
|
||
const nodes = [];
|
||
const idMap = {}; // drawflow id -> generated id like n1, n2
|
||
|
||
const dfNodes = (data && data.drawflow && data.drawflow.Home && data.drawflow.Home.data) ? data.drawflow.Home.data : {};
|
||
|
||
// 1) Собираем ноды
|
||
let idx = 1;
|
||
for (const id in dfNodes) {
|
||
const df = dfNodes[id];
|
||
const genId = `n${idx++}`;
|
||
idMap[id] = genId;
|
||
const el = document.querySelector(`#node-${id}`);
|
||
// Берём источник правды из DOM.__data (куда жмём «Сохранить параметры») или из drawflow.data
|
||
const datacopySrc = el && el.__data ? el.__data : (df.data || {});
|
||
const datacopy = typeof w.applyNodeDefaults === 'function'
|
||
? w.applyNodeDefaults(df.name, JSON.parse(JSON.stringify(datacopySrc)))
|
||
: (JSON.parse(JSON.stringify(datacopySrc)));
|
||
nodes.push({
|
||
id: genId,
|
||
type: df.name,
|
||
pos_x: df.pos_x,
|
||
pos_y: df.pos_y,
|
||
config: datacopy,
|
||
in: {}
|
||
});
|
||
}
|
||
|
||
// 2) Восстанавливаем связи по входам (inputs)
|
||
// В Drawflow v0.0.55 inputs/outputs — это объекты вида input_1/output_1
|
||
for (const id in dfNodes) {
|
||
const df = dfNodes[id];
|
||
const targetNode = nodes.find(n => n.id === idMap[id]);
|
||
if (!targetNode) continue;
|
||
const io = NODE_IO[targetNode.type] || { inputs: [], outputs: [] };
|
||
|
||
for (let i = 0; i < io.inputs.length; i++) {
|
||
const inputKey = `input_${i + 1}`;
|
||
const input = df.inputs && df.inputs[inputKey];
|
||
if (!input || !Array.isArray(input.connections) || input.connections.length === 0) continue;
|
||
|
||
// Собираем все связи этого входа и сохраняем строку либо массив строк (для depends поддерживаем мульти-коннекты)
|
||
const refs = [];
|
||
for (const conn of (input.connections || [])) {
|
||
if (!conn) continue;
|
||
const sourceDfId = String(conn.node);
|
||
const outKey = String(conn.output ?? '');
|
||
|
||
// conn.output может быть "output_1", "1" (строкой), либо числом 1
|
||
let sourceOutIdx = -1;
|
||
let m = outKey.match(/output_(\d+)/);
|
||
if (m) {
|
||
sourceOutIdx = parseInt(m[1], 10) - 1;
|
||
} else if (/^\d+$/.test(outKey)) {
|
||
sourceOutIdx = parseInt(outKey, 10) - 1;
|
||
} else if (typeof conn.output === 'number') {
|
||
sourceOutIdx = conn.output - 1;
|
||
}
|
||
if (!(sourceOutIdx >= 0)) sourceOutIdx = 0; // safety
|
||
|
||
const sourceNode = nodes.find(n => n.id === idMap[sourceDfId]);
|
||
if (!sourceNode) continue;
|
||
const sourceIo = NODE_IO[sourceNode.type] || { outputs: [] };
|
||
const sourceOutName = (sourceIo.outputs && sourceIo.outputs[sourceOutIdx] != null)
|
||
? sourceIo.outputs[sourceOutIdx]
|
||
: `out${sourceOutIdx}`;
|
||
refs.push(`${sourceNode.id}.${sourceOutName}`);
|
||
}
|
||
|
||
// Каноничное имя входа: по NODE_IO, иначе in{0-based}
|
||
const targetInName = (io.inputs && io.inputs[i] != null)
|
||
? io.inputs[i]
|
||
: `in${i}`;
|
||
|
||
if (!targetNode.in) targetNode.in = {};
|
||
targetNode.in[targetInName] = (refs.length <= 1 ? refs[0] : refs);
|
||
}
|
||
}
|
||
|
||
return { id: 'pipeline_editor', name: 'Edited Pipeline', nodes };
|
||
}
|
||
|
||
// pipeline JSON -> Drawflow
|
||
async function fromPipelineJSON(p) {
|
||
ensureDeps();
|
||
const editor = w.editor;
|
||
const NODE_IO = w.NODE_IO;
|
||
|
||
editor.clear();
|
||
let x = 100; let y = 120; // Fallback
|
||
const idMap = {}; // pipeline id -> drawflow id
|
||
const logs = [];
|
||
const $ = (sel) => document.querySelector(sel);
|
||
|
||
const resolveOutIdx = (type, outName) => {
|
||
const outs = ((NODE_IO[type] && NODE_IO[type].outputs) || []);
|
||
let idx = outs.indexOf(outName);
|
||
if (idx < 0 && typeof outName === 'string') {
|
||
// поддержка: out-1, out_1, output_1, out1, out0
|
||
const s = String(outName);
|
||
let m = s.match(/^out(?:put)?[_-]?(\d+)$/);
|
||
if (m) {
|
||
const n = parseInt(m[1], 10);
|
||
idx = n > 0 ? n - 1 : 0;
|
||
} else {
|
||
m = s.match(/^out(\d+)$/); // совместимость со старым out0
|
||
if (m) idx = parseInt(m[1], 10) | 0;
|
||
}
|
||
}
|
||
return idx;
|
||
};
|
||
const resolveInIdx = (type, inName) => {
|
||
const ins = ((NODE_IO[type] && NODE_IO[type].inputs) || []);
|
||
let idx = ins.indexOf(inName);
|
||
if (idx < 0 && typeof inName === 'string') {
|
||
// поддержка: in-1, in_1, in1, in0
|
||
const s = String(inName);
|
||
let m = s.match(/^in[_-]?(\d+)$/);
|
||
if (m) {
|
||
const n = parseInt(m[1], 10);
|
||
idx = n > 0 ? n - 1 : 0;
|
||
} else {
|
||
m = s.match(/^in(\d+)$/); // совместимость со старым in0
|
||
if (m) idx = parseInt(m[1], 10) | 0;
|
||
}
|
||
}
|
||
return idx;
|
||
};
|
||
|
||
// Ожидание появления порта в DOM (устранение гонки рендера)
|
||
async function waitForPort(dfid, kind, idx, tries = 60, delay = 16) {
|
||
// Drawflow создаёт DOM-узел с id="node-${dfid}"
|
||
const sel = `#node-${dfid} .${kind}_${idx}`;
|
||
for (let i = 0; i < tries; i++) {
|
||
if ($(sel)) return true;
|
||
await new Promise(r => setTimeout(r, delay));
|
||
}
|
||
logs.push(`port missing: #${dfid} ${kind}_${idx}`);
|
||
return false;
|
||
}
|
||
// Повторные попытки соединить порты, пока DOM не готов
|
||
async function connectWithRetry(srcDfId, tgtDfId, outNum, inNum, tries = 120, delay = 25) {
|
||
const outClass = `output_${outNum}`;
|
||
const inClass = `input_${inNum}`;
|
||
for (let i = 0; i < tries; i++) {
|
||
const okOut = await waitForPort(srcDfId, 'output', outNum, 1, delay);
|
||
const okIn = await waitForPort(tgtDfId, 'input', inNum, 1, delay);
|
||
if (okOut && okIn) {
|
||
try {
|
||
editor.addConnection(srcDfId, tgtDfId, outClass, inClass);
|
||
return true;
|
||
} catch (e) {
|
||
// retry on next loop
|
||
}
|
||
}
|
||
await new Promise(r => setTimeout(r, delay));
|
||
}
|
||
return false;
|
||
}
|
||
|
||
// 1) Создаём ноды
|
||
for (const n of p.nodes) {
|
||
const pos = { x: n.pos_x || x, y: n.pos_y || y };
|
||
const data = { ...(n.config || {}), _origId: n.id };
|
||
const dfid = typeof w.addNode === 'function'
|
||
? w.addNode(n.type, pos, data)
|
||
: (function () { throw new Error('AgentUISer: addNode() is not defined'); })();
|
||
idMap[n.id] = dfid;
|
||
if (!n.pos_x) x += 260; // раскладываем по горизонтали, если нет сохраненной позиции
|
||
}
|
||
|
||
// 2) Дождёмся полного рендера DOM
|
||
await new Promise(r => setTimeout(r, 0));
|
||
if (typeof requestAnimationFrame === 'function') {
|
||
await new Promise(r => requestAnimationFrame(() => r()));
|
||
await new Promise(r => requestAnimationFrame(() => r())); // двойной rAF для надежности
|
||
} else {
|
||
await new Promise(r => setTimeout(r, 32));
|
||
}
|
||
// 3) Проставляем связи из in (поддержка строк и массивов ссылок)
|
||
for (const n of p.nodes) {
|
||
if (!n.in) continue;
|
||
const targetDfId = idMap[n.id];
|
||
const targetIo = NODE_IO[n.type] || { inputs: [] };
|
||
for (const [inName, ref] of Object.entries(n.in)) {
|
||
const refs = Array.isArray(ref) ? ref : [ref];
|
||
for (const oneRef of refs) {
|
||
if (!oneRef || typeof oneRef !== 'string' || !oneRef.includes('.')) continue;
|
||
const [srcId, outName] = oneRef.split('.');
|
||
const sourceDfId = idMap[srcId];
|
||
if (!sourceDfId) { logs.push(`skip: src ${srcId} not found`); continue; }
|
||
const srcType = p.nodes.find(nn => nn.id === srcId)?.type;
|
||
|
||
let outIdx = resolveOutIdx(srcType, outName);
|
||
let inIdx = resolveInIdx(n.type, inName);
|
||
|
||
// Fallback на первый порт, если неизвестные имена, но порт существует
|
||
if (outIdx < 0) outIdx = 0;
|
||
if (inIdx < 0) inIdx = 0;
|
||
|
||
const ok = await connectWithRetry(sourceDfId, targetDfId, outIdx + 1, inIdx + 1, 200, 25);
|
||
if (ok) {
|
||
logs.push(`connect: ${srcId}.${outName} (#${sourceDfId}.output_${outIdx + 1}) -> ${n.id}.${inName} (#${targetDfId}.input_${inIdx + 1})`);
|
||
} else {
|
||
logs.push(`skip connect (ports not ready after retries): ${srcId}.${outName} -> ${n.id}.${inName}`);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
// 4) Обновим линии и выведем лог
|
||
try {
|
||
Object.values(idMap).forEach((dfid) => {
|
||
if (editor.updateConnectionNodes) {
|
||
editor.updateConnectionNodes(`node-${dfid}`);
|
||
}
|
||
});
|
||
} catch {}
|
||
if (logs.length) {
|
||
try { (typeof w.status === 'function') && w.status('Загружено (links):\n' + logs.join('\n')); } catch {}
|
||
try { console.debug('[fromPipelineJSON]', logs); } catch {}
|
||
}
|
||
}
|
||
|
||
w.AgentUISer = {
|
||
toPipelineJSON,
|
||
fromPipelineJSON,
|
||
};
|
||
})(window); |