feat(*): 插件中心与模型协议 - 完善声明式协议运行时并增加功能开放控制
This commit is contained in:
ddCat committed
1 parent
7cc6021f6e
commit
919037e5d9
106 files changed
+8816
-499
No files matched your search
@@ -0,0 +1,273 @@
|
||||
import crypto from "node:crypto";
|
||||
|
||||
import type { CanvasConnection, CanvasNode, CanvasSnapshot } from "./types.js";
|
||||
|
||||
export type CanvasResourceSummary = {
|
||||
nodeId: string;
|
||||
nodeTitle: string;
|
||||
nodeType: CanvasNode["type"];
|
||||
status?: string;
|
||||
resourceId?: string;
|
||||
storageKey?: string;
|
||||
assetId?: string;
|
||||
assetCategory?: string;
|
||||
mimeType?: string;
|
||||
bytes?: number;
|
||||
width?: number;
|
||||
height?: number;
|
||||
durationMs?: number;
|
||||
isReady: boolean;
|
||||
};
|
||||
|
||||
export type CanvasContext = {
|
||||
schemaVersion: 1;
|
||||
stateHash: string;
|
||||
canvas: {
|
||||
projectId?: string;
|
||||
domainProjectId?: string;
|
||||
title?: string;
|
||||
viewport?: CanvasSnapshot["viewport"];
|
||||
nodeCount: number;
|
||||
connectionCount: number;
|
||||
selectedNodeCount: number;
|
||||
nodeTypeCounts: Record<string, number>;
|
||||
};
|
||||
selection: ReturnType<typeof compactContextNode>[];
|
||||
nodes: ReturnType<typeof compactContextNode>[];
|
||||
connections: Array<{
|
||||
id: string;
|
||||
fromNodeId: string;
|
||||
fromTitle: string;
|
||||
toNodeId: string;
|
||||
toTitle: string;
|
||||
fromHandleId?: string;
|
||||
toHandleId?: string;
|
||||
}>;
|
||||
resources: CanvasResourceSummary[];
|
||||
warnings: string[];
|
||||
};
|
||||
|
||||
export function buildCanvasContext(state: CanvasSnapshot | null): CanvasContext {
|
||||
if (!state) throw new Error("当前没有已连接画布");
|
||||
const nodes = state.nodes || [];
|
||||
const nodeById = new Map(nodes.map((node) => [node.id, node]));
|
||||
const selectedIds = new Set(state.selectedNodeIds || []);
|
||||
const nodeTypeCounts = nodes.reduce<Record<string, number>>((counts, node) => {
|
||||
counts[node.type] = (counts[node.type] || 0) + 1;
|
||||
return counts;
|
||||
}, {});
|
||||
const resources = nodes.flatMap((node) => {
|
||||
const resource = resourceSummary(node);
|
||||
return resource ? [resource] : [];
|
||||
});
|
||||
const warnings: string[] = [];
|
||||
if (!nodes.length) warnings.push("画布为空；创建前先确认用户要放置的区域或使用默认网格布局。");
|
||||
if (nodes.some((node) => node.metadata?.status === "error")) warnings.push("画布中存在生成失败节点；重试前应读取节点的 errorDetails 或 generationErrorCode。");
|
||||
if (resources.some((resource) => !resource.isReady)) warnings.push("存在尚未就绪或缺少持久化资源引用的媒体节点；不要把占位节点当作可用参考素材。");
|
||||
|
||||
return {
|
||||
schemaVersion: 1,
|
||||
stateHash: hashState(state),
|
||||
canvas: {
|
||||
projectId: state.projectId,
|
||||
domainProjectId: state.domainProjectId,
|
||||
title: state.title,
|
||||
viewport: state.viewport,
|
||||
nodeCount: nodes.length,
|
||||
connectionCount: (state.connections || []).length,
|
||||
selectedNodeCount: selectedIds.size,
|
||||
nodeTypeCounts,
|
||||
},
|
||||
selection: nodes.filter((node) => selectedIds.has(node.id)).map(compactContextNode),
|
||||
nodes: nodes.map(compactContextNode),
|
||||
connections: (state.connections || []).map((connection) => connectionSummary(connection, nodeById)),
|
||||
resources,
|
||||
warnings,
|
||||
};
|
||||
}
|
||||
|
||||
export function findCanvasNodes(state: CanvasSnapshot | null, input: { query?: string; ids?: string[]; types?: string[]; statuses?: string[]; resourceOnly?: boolean; limit?: number }) {
|
||||
if (!state) throw new Error("当前没有已连接画布");
|
||||
const query = input.query?.trim().toLocaleLowerCase();
|
||||
const ids = input.ids?.length ? new Set(input.ids) : null;
|
||||
const types = input.types?.length ? new Set(input.types) : null;
|
||||
const statuses = input.statuses?.length ? new Set(input.statuses) : null;
|
||||
const limit = Math.min(Math.max(input.limit || 50, 1), 200);
|
||||
const nodes = (state.nodes || []).filter((node) => {
|
||||
const metadata = node.metadata || {};
|
||||
if (ids && !ids.has(node.id)) return false;
|
||||
if (types && !types.has(node.type)) return false;
|
||||
if (statuses && !statuses.has(String(metadata.status || "idle"))) return false;
|
||||
if (input.resourceOnly && !resourceSummary(node)) return false;
|
||||
if (!query) return true;
|
||||
return [node.id, node.title, metadata.content, metadata.prompt, metadata.composerContent, metadata.assetId, Array.isArray(metadata.assetTags) ? metadata.assetTags.join(" ") : "", metadata.workflowKind, metadata.workflowTitle, metadata.characterName]
|
||||
.some((value) => String(value || "").toLocaleLowerCase().includes(query));
|
||||
});
|
||||
return { query: input.query || "", total: nodes.length, truncated: nodes.length > limit, nodes: nodes.slice(0, limit).map(compactContextNode) };
|
||||
}
|
||||
|
||||
export function getCanvasResources(state: CanvasSnapshot | null, input: { nodeIds?: string[]; status?: string; limit?: number }) {
|
||||
if (!state) throw new Error("当前没有已连接画布");
|
||||
const nodeIds = input.nodeIds?.length ? new Set(input.nodeIds) : null;
|
||||
const limit = Math.min(Math.max(input.limit || 100, 1), 300);
|
||||
const resources = (state.nodes || []).flatMap((node) => {
|
||||
if (nodeIds && !nodeIds.has(node.id)) return [];
|
||||
const resource = resourceSummary(node);
|
||||
if (!resource || (input.status && resource.status !== input.status)) return [];
|
||||
return [resource];
|
||||
});
|
||||
return { total: resources.length, truncated: resources.length > limit, resources: resources.slice(0, limit) };
|
||||
}
|
||||
|
||||
export function validateCanvasOps(state: CanvasSnapshot | null, ops: unknown[]) {
|
||||
if (!state) throw new Error("当前没有已连接画布");
|
||||
const nodes = state.nodes || [];
|
||||
const nodeIds = new Set(nodes.map((node) => node.id));
|
||||
const issues: Array<{ index: number; severity: "error" | "warning"; message: string }> = [];
|
||||
const addedIds = new Set<string>();
|
||||
const requireNode = (index: number, id: unknown, label: string) => {
|
||||
if (typeof id !== "string" || !id) issues.push({ index, severity: "error", message: `${label} 缺少节点 id` });
|
||||
else if (!nodeIds.has(id) && !addedIds.has(id)) issues.push({ index, severity: "error", message: `${label}「${id}」不在当前画布状态中；先重新读取 canvas_get_context 或 canvas_find_nodes` });
|
||||
};
|
||||
ops.forEach((raw, index) => {
|
||||
if (!raw || typeof raw !== "object" || Array.isArray(raw)) {
|
||||
issues.push({ index, severity: "error", message: "操作必须是对象" });
|
||||
return;
|
||||
}
|
||||
const op = raw as Record<string, unknown>;
|
||||
switch (op.type) {
|
||||
case "add_node":
|
||||
if (typeof op.id === "string") {
|
||||
if (nodeIds.has(op.id) || addedIds.has(op.id)) issues.push({ index, severity: "error", message: `新增节点 id「${op.id}」重复` });
|
||||
addedIds.add(op.id);
|
||||
}
|
||||
break;
|
||||
case "update_node":
|
||||
case "run_generation":
|
||||
requireNode(index, op.nodeId ?? op.id, op.type === "run_generation" ? "生成目标" : "更新目标");
|
||||
break;
|
||||
case "delete_node":
|
||||
if (Array.isArray(op.ids)) op.ids.forEach((id) => requireNode(index, id, "删除目标"));
|
||||
else requireNode(index, op.id, "删除目标");
|
||||
break;
|
||||
case "connect_nodes":
|
||||
requireNode(index, op.fromNodeId, "连接起点");
|
||||
requireNode(index, op.toNodeId, "连接终点");
|
||||
if (op.fromNodeId === op.toNodeId) issues.push({ index, severity: "error", message: "不能连接节点自身" });
|
||||
break;
|
||||
case "select_nodes":
|
||||
if (Array.isArray(op.ids)) op.ids.forEach((id) => requireNode(index, id, "选区节点"));
|
||||
break;
|
||||
case "delete_connections":
|
||||
if (!op.all && !op.id && !(Array.isArray(op.ids) && op.ids.length)) issues.push({ index, severity: "error", message: "删除连线必须提供 id、ids 或 all=true" });
|
||||
break;
|
||||
case "set_viewport":
|
||||
if (!op.viewport || typeof op.viewport !== "object") issues.push({ index, severity: "error", message: "视口参数无效" });
|
||||
break;
|
||||
default:
|
||||
issues.push({ index, severity: "error", message: `不支持的操作类型「${String(op.type)}」` });
|
||||
}
|
||||
});
|
||||
const errors = issues.filter((item) => item.severity === "error");
|
||||
return { ok: errors.length === 0, issues, operationCount: ops.length, currentStateHash: hashState(state) };
|
||||
}
|
||||
|
||||
export function hashState(state: CanvasSnapshot) {
|
||||
return crypto.createHash("sha256").update(stableStringify({
|
||||
projectId: state.projectId,
|
||||
domainProjectId: state.domainProjectId,
|
||||
title: state.title,
|
||||
nodes: state.nodes || [],
|
||||
connections: state.connections || [],
|
||||
selectedNodeIds: state.selectedNodeIds || [],
|
||||
viewport: state.viewport,
|
||||
})).digest("hex").slice(0, 16);
|
||||
}
|
||||
|
||||
function compactContextNode(node: CanvasNode) {
|
||||
const metadata = node.metadata || {};
|
||||
const resource = resourceSummary(node);
|
||||
return {
|
||||
id: node.id,
|
||||
type: node.type,
|
||||
title: node.title || "未命名节点",
|
||||
position: node.position,
|
||||
size: { width: node.width, height: node.height },
|
||||
parentId: node.parentId,
|
||||
status: String(metadata.status || "idle"),
|
||||
content: preview(metadata.content, 240),
|
||||
prompt: preview(metadata.prompt || metadata.composerContent, 300),
|
||||
generation: metadata.generationMode || metadata.workflowKind ? {
|
||||
mode: metadata.generationMode,
|
||||
model: metadata.model,
|
||||
workflowKind: metadata.workflowKind,
|
||||
workflowTitle: metadata.workflowTitle,
|
||||
} : undefined,
|
||||
asset: metadata.assetId || metadata.characterAssetId ? {
|
||||
assetId: metadata.assetId || metadata.characterAssetId,
|
||||
versionId: metadata.characterVersionId,
|
||||
category: metadata.assetCategory,
|
||||
tags: metadata.assetTags,
|
||||
characterName: metadata.characterName,
|
||||
} : undefined,
|
||||
resource: resource ? { resourceId: resource.resourceId, storageKey: resource.storageKey, mimeType: resource.mimeType, bytes: resource.bytes, width: resource.width, height: resource.height, durationMs: resource.durationMs, ready: resource.isReady } : undefined,
|
||||
};
|
||||
}
|
||||
|
||||
function resourceSummary(node: CanvasNode): CanvasResourceSummary | null {
|
||||
const metadata = node.metadata || {};
|
||||
const storageKey = stringValue(metadata.storageKey);
|
||||
const resourceId = storageKey?.startsWith("resource:") ? storageKey.slice("resource:".length) : undefined;
|
||||
const hasResourceSignal = Boolean(storageKey || metadata.resourceId || metadata.assetId || metadata.primaryImageId || metadata.mimeType || ["image", "video", "audio"].includes(node.type));
|
||||
if (!hasResourceSignal) return null;
|
||||
const status = stringValue(metadata.status) || "idle";
|
||||
return {
|
||||
nodeId: node.id,
|
||||
nodeTitle: node.title || "未命名节点",
|
||||
nodeType: node.type,
|
||||
status,
|
||||
resourceId: resourceId || stringValue(metadata.resourceId),
|
||||
storageKey,
|
||||
assetId: stringValue(metadata.assetId || metadata.characterAssetId),
|
||||
assetCategory: stringValue(metadata.assetCategory),
|
||||
mimeType: stringValue(metadata.mimeType),
|
||||
bytes: numberValue(metadata.bytes),
|
||||
width: numberValue(metadata.naturalWidth),
|
||||
height: numberValue(metadata.naturalHeight),
|
||||
durationMs: numberValue(metadata.durationMs),
|
||||
isReady: status === "success" && Boolean(storageKey || metadata.resourceId || metadata.primaryImageId),
|
||||
};
|
||||
}
|
||||
|
||||
function connectionSummary(connection: CanvasConnection, nodeById: Map<string, CanvasNode>) {
|
||||
return {
|
||||
id: connection.id,
|
||||
fromNodeId: connection.fromNodeId,
|
||||
fromTitle: nodeById.get(connection.fromNodeId)?.title || "未知节点",
|
||||
toNodeId: connection.toNodeId,
|
||||
toTitle: nodeById.get(connection.toNodeId)?.title || "未知节点",
|
||||
fromHandleId: connection.fromHandleId,
|
||||
toHandleId: connection.toHandleId,
|
||||
};
|
||||
}
|
||||
|
||||
function preview(value: unknown, limit: number) {
|
||||
if (typeof value !== "string") return undefined;
|
||||
const text = value.trim();
|
||||
if (!text) return undefined;
|
||||
return text.length > limit ? `${text.slice(0, limit)}…` : text;
|
||||
}
|
||||
|
||||
function stringValue(value: unknown) {
|
||||
return typeof value === "string" && value.trim() ? value.trim() : undefined;
|
||||
}
|
||||
|
||||
function numberValue(value: unknown) {
|
||||
return typeof value === "number" && Number.isFinite(value) ? value : undefined;
|
||||
}
|
||||
|
||||
function stableStringify(value: unknown): string {
|
||||
if (Array.isArray(value)) return `[${value.map(stableStringify).join(",")}]`;
|
||||
if (value && typeof value === "object") return `{${Object.entries(value as Record<string, unknown>).sort(([a], [b]) => a.localeCompare(b)).map(([key, item]) => `${JSON.stringify(key)}:${stableStringify(item)}`).join(",")}}`;
|
||||
return JSON.stringify(value);
|
||||
}
|
||||
Reference in new issue
Block a user