11872668922

Committed 16 Nov 2024 07:25PM UTC coverage: 95.048% (-0.8%) from 95.829%

Build # 11872668922

Build Type

push

github

Committed by

lazToum

Commit Message

rewrite

Run Details

2322 of 2570 branches covered (90.35%)

Branch coverage included in aggregate %.

4728 of 5072 new or added lines in 234 files covered. (93.22%)

4 existing lines in 1 file now uncovered.

13224 of 13786 relevant lines covered (95.92%)

447.84 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

90.15

/src/waldiez/store/edges/utils.ts

import { Edge, MarkerType, Node } from '@xyflow/react';

import { nanoid } from 'nanoid';

import {
  WaldiezAgentNodeType,
  WaldiezEdgeType,
  WaldiezSourceEdge,
  WaldiezSourceEdgeData
} from '@waldiez/models';
import { AGENT_COLORS } from '@waldiez/theme';

export const edgeCommonStyle = (edgeType: WaldiezEdgeType, color: string) => ({
  markerEnd:
    edgeType !== 'nested'
      ? {
          type: MarkerType.ArrowClosed,
          color,
          width: 10,
          height: 10
        }
      : undefined,
  style: {
    stroke: color,
    strokeWidth: 3
  }
});

const getNewEdgeNodes = (allNodes: Node[], source: string, target: string) => {
  const sourceNode = allNodes.find(node => node.id === source);
  if (!sourceNode) {
    throw new Error(`Source node with id ${source} not found`);
  }
  const targetNode = allNodes.find(node => node.id === target);
  if (!targetNode) {
    throw new Error(`Target node with id ${target} not found`);
  }
  return {
    sourceNode,
    targetNode
  };
};
const getNewEdgeName = (sourceNode: Node, targetNode: Node) => {
  const sourceLabel = (sourceNode.data.label as string).slice(0, 15);
  const targetLabel = (targetNode.data.label as string).slice(0, 15);
  const edgeName = `${sourceLabel} => ${targetLabel}`;
  return edgeName;
};
const getNewChatType: (sourceNode: Node, targetNode: Node, hidden: boolean) => WaldiezEdgeType = (
  sourceNode,
  targetNode,
  hidden
) => {
  if (hidden) {
    return 'hidden' as WaldiezEdgeType;
  }
  const agentType = sourceNode.data.agentType as WaldiezAgentNodeType;
  let chatType: WaldiezEdgeType = agentType === 'manager' ? 'group' : 'chat';
  if (targetNode.data.parentId) {
    chatType = 'hidden';
  }
  return chatType;
};
export const getNewEdge = (
  source: string,
  target: string,
  hidden: boolean,
  positionGetter: (chatType: string) => number,
  nodes: Node[]
) => {
  const { sourceNode, targetNode } = getNewEdgeNodes(nodes, source, target);
  const edgeName = getNewEdgeName(sourceNode, targetNode);
  const edgeData = new WaldiezSourceEdgeData(source, target, edgeName);
  edgeData.order = -1;
  const agentType = sourceNode.data.agentType as WaldiezAgentNodeType;
  const chatType = getNewChatType(sourceNode, targetNode, hidden);
  edgeData.position = positionGetter(chatType);
  const color = AGENT_COLORS[agentType];
  const newEdge = new WaldiezSourceEdge({
    id: `we-${nanoid()}`,
    source,
    target,
    data: edgeData,
    rest: {}
  }).asEdge();
  return {
    ...newEdge,
    type: chatType,
    animated: false,
    selected: true,
    ...edgeCommonStyle(chatType, color)
  };
};

const getNewChatsOfType = (allEdges: Edge[], type: string) => {
  const edgesOfType = allEdges.filter(edge => edge.type === type);
  const edgesOfTypeBySource: { [source: string]: Edge[] } = {};
  edgesOfType.forEach(edge => {
    if (!edgesOfTypeBySource[edge.source]) {
      edgesOfTypeBySource[edge.source] = [];
    }
    edgesOfTypeBySource[edge.source].push(edge);
  });
  return edgesOfType.map((edge, index) => {
    return {
      ...edge,
      data: { ...edge.data, position: index + 1 }
    };
  });
};

export const getNewChatEdges = (allEdges: Edge[]) => {
  return getNewChatsOfType(allEdges, 'chat');
};

export const getNewNestedEdges = (allEdges: Edge[]) => {
  return getNewChatsOfType(allEdges, 'nested');
};

export const getNewGroupEdges = (allEdges: Edge[]) => {
  return getNewChatsOfType(allEdges, 'group');
};

export const getNewHiddenEdges = (allEdges: Edge[]) => {
  return getNewChatsOfType(allEdges, 'hidden');
};

1	import { Edge, MarkerType, Node } from '@xyflow/react';	1✔
2
3	import { nanoid } from 'nanoid';
4
5	import {
6	WaldiezAgentNodeType,
7	WaldiezEdgeType,
8	WaldiezSourceEdge,
9	WaldiezSourceEdgeData
10	} from '@waldiez/models';
11	import { AGENT_COLORS } from '@waldiez/theme';
12
13	export const edgeCommonStyle = (edgeType: WaldiezEdgeType, color: string) => ({	1✔
14	markerEnd:	26✔
15	edgeType !== 'nested'	26✔
16	? {	19✔
17	type: MarkerType.ArrowClosed,	19✔
18	color,	19✔
19	width: 10,	19✔
20	height: 10	19✔
21	}	19✔
22	: undefined,	7✔
23	style: {	26✔
24	stroke: color,	26✔
25	strokeWidth: 3	26✔
26	}	26✔
27	});	26✔
28
29	const getNewEdgeNodes = (allNodes: Node[], source: string, target: string) => {	1✔
30	const sourceNode = allNodes.find(node => node.id === source);	1✔
31	if (!sourceNode) {	1!
32	throw new Error(`Source node with id ${source} not found`);	×
33	}	×
34	const targetNode = allNodes.find(node => node.id === target);	1✔
35	if (!targetNode) {	1!
36	throw new Error(`Target node with id ${target} not found`);	×
37	}	×
38	return {	1✔
39	sourceNode,	1✔
40	targetNode	1✔
41	};	1✔
42	};	1✔
43	const getNewEdgeName = (sourceNode: Node, targetNode: Node) => {	1✔
44	const sourceLabel = (sourceNode.data.label as string).slice(0, 15);	1✔
45	const targetLabel = (targetNode.data.label as string).slice(0, 15);	1✔
46	const edgeName = `${sourceLabel} => ${targetLabel}`;	1✔
47	return edgeName;	1✔
48	};	1✔
49	const getNewChatType: (sourceNode: Node, targetNode: Node, hidden: boolean) => WaldiezEdgeType = (	1✔
50	sourceNode,	1✔
51	targetNode,	1✔
52	hidden	1✔
53	) => {	1✔
54	if (hidden) {	1!
NEW 55	return 'hidden' as WaldiezEdgeType;	×
NEW 56	}	×
57	const agentType = sourceNode.data.agentType as WaldiezAgentNodeType;	1✔
58	let chatType: WaldiezEdgeType = agentType === 'manager' ? 'group' : 'chat';	1!
59	if (targetNode.data.parentId) {	1!
60	chatType = 'hidden';	×
61	}	×
62	return chatType;	1✔
63	};	1✔
64	export const getNewEdge = (	1✔
65	source: string,	1✔
66	target: string,	1✔
67	hidden: boolean,	1✔
68	positionGetter: (chatType: string) => number,	1✔
69	nodes: Node[]	1✔
70	) => {	1✔
71	const { sourceNode, targetNode } = getNewEdgeNodes(nodes, source, target);	1✔
72	const edgeName = getNewEdgeName(sourceNode, targetNode);	1✔
73	const edgeData = new WaldiezSourceEdgeData(source, target, edgeName);	1✔
74	edgeData.order = -1;	1✔
75	const agentType = sourceNode.data.agentType as WaldiezAgentNodeType;	1✔
76	const chatType = getNewChatType(sourceNode, targetNode, hidden);	1✔
77	edgeData.position = positionGetter(chatType);	1✔
78	const color = AGENT_COLORS[agentType];	1✔
79	const newEdge = new WaldiezSourceEdge({	1✔
80	id: `we-${nanoid()}`,	1✔
81	source,	1✔
82	target,	1✔
83	data: edgeData,	1✔
84	rest: {}	1✔
85	}).asEdge();	1✔
86	return {	1✔
87	...newEdge,	1✔
88	type: chatType,	1✔
89	animated: false,	1✔
90	selected: true,	1✔
91	...edgeCommonStyle(chatType, color)	1✔
92	};	1✔
93	};	1✔
94
95	const getNewChatsOfType = (allEdges: Edge[], type: string) => {	1✔
96	const edgesOfType = allEdges.filter(edge => edge.type === type);	404✔
97	const edgesOfTypeBySource: { [source: string]: Edge[] } = {};	404✔
98	edgesOfType.forEach(edge => {	404✔
99	if (!edgesOfTypeBySource[edge.source]) {	104✔
100	edgesOfTypeBySource[edge.source] = [];	76✔
101	}	76✔
102	edgesOfTypeBySource[edge.source].push(edge);	104✔
103	});	404✔
104	return edgesOfType.map((edge, index) => {	404✔
105	return {	104✔
106	...edge,	104✔
107	data: { ...edge.data, position: index + 1 }	104✔
108	};	104✔
109	});	404✔
110	};	404✔
111
112	export const getNewChatEdges = (allEdges: Edge[]) => {	1✔
113	return getNewChatsOfType(allEdges, 'chat');	101✔
114	};	101✔
115
116	export const getNewNestedEdges = (allEdges: Edge[]) => {	1✔
117	return getNewChatsOfType(allEdges, 'nested');	101✔
118	};	101✔
119
120	export const getNewGroupEdges = (allEdges: Edge[]) => {	1✔
121	return getNewChatsOfType(allEdges, 'group');	101✔
122	};	101✔
123
124	export const getNewHiddenEdges = (allEdges: Edge[]) => {	1✔
125	return getNewChatsOfType(allEdges, 'hidden');	101✔
126	};	101✔

waldiez / react / 11872668922

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous