79 lines
6.6 KiB
TypeScript
79 lines
6.6 KiB
TypeScript
export type NodeKind = 'text' | 'image' | 'prompt' | 'script';
|
|
export type TextStyle = {bold?:boolean;italic?:boolean;color?:string;size?:number};
|
|
export type CanvasNode = {
|
|
id:string;kind:NodeKind;title:string;text:string;x:number;y:number;
|
|
image?:string;width?:number;height?:number;style?:TextStyle;
|
|
instruction?:string;
|
|
operation?:'generate'|'upscale'|'script';
|
|
};
|
|
export type CanvasDocument = {version:1;cards:CanvasNode[];links:[string,string][]};
|
|
const kinds:NodeKind[]=['text','image','prompt','script'];
|
|
const object=(v:unknown):v is Record<string,unknown>=>!!v&&typeof v==='object'&&!Array.isArray(v);
|
|
const bounded=(v:unknown,min:number,max:number)=>typeof v==='number'&&Number.isFinite(v)&&v>=min&&v<=max;
|
|
const text=(v:unknown,max:number)=>typeof v==='string'&&v.length<=max;
|
|
const image=(v:unknown)=>typeof v==='string'&&v.length<3000000&&/^data:image\/(png|jpeg|webp);base64,[A-Za-z0-9+/=]+$/.test(v);
|
|
|
|
export function parseCanvasDocument(value:unknown):CanvasDocument {
|
|
const d=typeof value==='string'?JSON.parse(value):value;
|
|
if(!object(d)||d.version!==1||!Array.isArray(d.cards)||d.cards.length>80||!Array.isArray(d.links)||d.links.length>200)throw Error('画布格式不正确');
|
|
const ids=new Set<string>();
|
|
const cards:CanvasNode[]=d.cards.map((n:unknown)=>{
|
|
if(!object(n)||!text(n.id,100)||!n.id||ids.has(n.id as string)||!kinds.includes(n.kind as NodeKind)||!text(n.title,200)||!text(n.text,10000)||!bounded(n.x,-100000,100000)||!bounded(n.y,-100000,100000))throw Error('节点格式不正确');
|
|
if(n.width!==undefined&&!bounded(n.width,180,1200)||n.height!==undefined&&!bounded(n.height,160,1200)||n.image!==undefined&&!image(n.image))throw Error('节点尺寸或图片不正确');
|
|
if(n.instruction!==undefined&&!text(n.instruction,10000))throw Error('创作要求格式不正确');
|
|
if(n.operation!==undefined&&!['generate','upscale','script'].includes(n.operation as string))throw Error('生成任务类型不正确');
|
|
let style:TextStyle|undefined;
|
|
if(n.style!==undefined){
|
|
if(!object(n.style)||n.style.bold!==undefined&&typeof n.style.bold!=='boolean'||n.style.italic!==undefined&&typeof n.style.italic!=='boolean'||n.style.color!==undefined&&(typeof n.style.color!=='string'||!/^#[a-f\d]{6}$/i.test(n.style.color))||n.style.size!==undefined&&!bounded(n.style.size,10,72))throw Error('文字样式不正确');
|
|
style={...(n.style.bold===undefined?{}:{bold:n.style.bold as boolean}),...(n.style.italic===undefined?{}:{italic:n.style.italic as boolean}),...(n.style.color===undefined?{}:{color:n.style.color as string}),...(n.style.size===undefined?{}:{size:n.style.size as number})};
|
|
}
|
|
ids.add(n.id as string);
|
|
return {id:n.id as string,kind:n.kind as NodeKind,title:n.title as string,text:n.text as string,x:n.x as number,y:n.y as number,...(n.width===undefined?{}:{width:n.width as number}),...(n.height===undefined?{}:{height:n.height as number}),...(n.image===undefined?{}:{image:n.image as string}),...(style?{style}:{}),...(n.instruction===undefined?{}:{instruction:n.instruction as string}),...(n.operation===undefined?{}:{operation:n.operation as CanvasNode['operation']})};
|
|
});
|
|
const pairs=new Set<string>();
|
|
const links:[string,string][]=d.links.map((l:unknown)=>{
|
|
if(!Array.isArray(l)||l.length!==2||l[0]===l[1]||!l.every(id=>typeof id==='string'&&ids.has(id)))throw Error('引用节点不存在');
|
|
const key=JSON.stringify(l);if(pairs.has(key))throw Error('引用重复');pairs.add(key);return [l[0],l[1]];
|
|
});
|
|
const outgoing=new Map<string,string[]>(),indegree=new Map(cards.map(n=>[n.id,0]));
|
|
for(const [from,to] of links){outgoing.set(from,[...(outgoing.get(from)||[]),to]);indegree.set(to,indegree.get(to)!+1)}
|
|
const roots=cards.filter(n=>indegree.get(n.id)===0).map(n=>n.id);let visited=0;
|
|
while(roots.length){const id=roots.pop()!;visited++;for(const to of outgoing.get(id)||[]){const count=indegree.get(to)!-1;indegree.set(to,count);if(!count)roots.push(to)}}
|
|
if(visited!==cards.length)throw Error('引用不能形成循环');
|
|
return {version:1,cards,links};
|
|
}
|
|
|
|
export function referencedNodes(doc:CanvasDocument,target:string):CanvasNode[]{
|
|
return doc.links.filter(([,to])=>to===target).map(([from])=>doc.cards.find(n=>n.id===from)!).filter(Boolean);
|
|
}
|
|
export function connectNodes(doc:CanvasDocument,from:string,to:string):CanvasDocument {
|
|
if(from===to||![from,to].every(id=>doc.cards.some(n=>n.id===id)))throw Error('请选择不同的有效节点');
|
|
if(doc.links.some(([a,b])=>a===from&&b===to))return doc;
|
|
if(doc.links.length>=200)throw Error('最多 200 条引用');
|
|
const seen=new Set<string>();const pending=[to];
|
|
while(pending.length){const id=pending.pop()!;if(id===from)throw Error('引用不能形成循环');if(seen.has(id))continue;seen.add(id);pending.push(...doc.links.filter(([a])=>a===id).map(([,b])=>b));}
|
|
return {...doc,links:[...doc.links,[from,to]]};
|
|
}
|
|
export function removeCanvasNode(doc:CanvasDocument,id:string):CanvasDocument {
|
|
return {...doc,cards:doc.cards.filter(n=>n.id!==id),links:doc.links.filter(l=>!l.includes(id))};
|
|
}
|
|
export function deriveNode(doc:CanvasDocument,source:string,node:CanvasNode):CanvasDocument {
|
|
if(doc.cards.length>=80||doc.cards.some(n=>n.id===node.id))throw Error('节点数量或标识不正确');
|
|
const target=parseCanvasDocument({version:1,cards:[node],links:[]}).cards[0];
|
|
for(let attempt=0;attempt<=doc.cards.length+1;attempt++){
|
|
const blockers=doc.cards.filter(n=>target.x<n.x+(n.width||320)+64&&target.x+(target.width||320)+64>n.x&&target.y<n.y+(n.height||350)+64&&target.y+(target.height||350)+64>n.y);
|
|
if(!blockers.length)return parseCanvasDocument(connectNodes({...doc,cards:[...doc.cards,target]},source,target.id));
|
|
const nextY=Math.max(...blockers.map(n=>n.y+(n.height||350)+64));
|
|
target.y=nextY>100000?-100000:nextY;
|
|
}
|
|
throw Error('请移动现有节点后再创建引用节点');
|
|
}
|
|
export function agentTask(doc:CanvasDocument,id:string){
|
|
const node=doc.cards.find(n=>n.id===id);if(!node)throw Error('目标节点不存在');
|
|
const instruction=node.instruction??node.text;
|
|
if(!instruction.trim()&&node.operation!=='upscale')throw Error('请先填写创作要求');
|
|
const references=referencedNodes(doc,id);
|
|
if(node.operation==='upscale'&&!references.some(n=>n.kind==='image'&&n.image))throw Error('请先引用一张图片');
|
|
return {schema:'studio.canvas-task/1',target:{id:node.id,kind:node.kind,title:node.title},operation:node.operation||'generate',instruction,...(node.image?{sourceImage:node.image}:{}),references:references.map(n=>({id:n.id,kind:n.kind,title:n.title,text:n.text,...(n.image?{image:n.image}:{}),...(n.style?{style:n.style}:{})})),result:{targetId:node.id,allowedFields:node.kind==='image'?['image','text']:['text']}};
|
|
}
|