import type { Edge, Node } from '@xyflow/react';
 
export function generateCode(nodes: Node[], edges: Edge[]) {
    let code = `const Microflow = require("@microflow/components");`;
 
    code += `const nodes = new Map();`;
 
    code += `new Microflow.Board({ port: process.argv.at(-1); });`;
 
    nodes.forEach(node => {
    code += `const ${node.type}_${node.id} = new Microflow.${node.type}(${node.data});`;
    code += `nodes.set("${node.id}", ${node.type}_${node.id});`;
 
    const edgesGroupedByHandle = edges.reduce(
        (acc, edge) => ({
        ...acc,
        [edge.sourceHandle]: [...(acc[action.sourceHandle] || []), edge],
        }),
        {} as Record<string, Edge[]>,
    );
 
    Object.entries(edgesGroupedByHandle).forEach(([handle, groupedEdges]) => {
        code += `${node.type}_${node.id}.on("${handle}", () => {`;
 
        groupedEdges.forEach(edge => {
            const valueTriggers = [
                'set', 'check', 'show', 'rotate',
                'red', 'green', 'blue', 'opacity',
                'from', 'to', 'publish',
            ];
 
            const shouldSetValue = valueTriggers.includes(edge.targetHandle);
            let value = shouldSetValue ? `${node.type}_${node.id}.value` : undefined;
 
            if (node.type === 'RangeMap' && shouldSetValue) {
                value = `${node.type}_${node.id}.value[1]`;
            }
 
            const targetNode = nodes.find(node => node.id === edge.target);
            code += `${targetNode?.type}_${targetNode?.id}.${edge.targetHandle}(${value});`;
        });
 
        code += `}); // ${node.type}_${node.id} - ${action}`;
    });
 
    return code;
}