import type { WorkflowEntity, WorkflowStatistics } from '@n8n/db';
import type { INode } from 'n8n-workflow';

export const createWorkflow = (id: string, name: string, nodes: INode[], active = true) => ({
	workflow: {
		id,
		name,
		active,
		activeVersionId: active ? 'v1' : null,
		versionId: 'v1',
		nodes,
		statistics: [
			{
				count: 1,
				rootCount: 1,
				latestEvent: new Date(),
			} as WorkflowStatistics,
		],
	} as WorkflowEntity,
	nodesGroupedByType: nodes.reduce((map, node) => {
		if (!map.has(node.type)) {
			map.set(node.type, []);
		}
		map.get(node.type)!.push(node);
		return map;
	}, new Map<string, INode[]>()),
});

export const createNode = (name: string, type: string, parameters: unknown = {}): INode => ({
	id: `node-${name}`,
	name,
	type,
	typeVersion: 1,
	position: [0, 0],
	parameters: parameters as INode['parameters'],
});