SequentialDependencyGraphBuilder.java

package blue.contract.packager.graphbuilder;

import blue.contract.packager.model.DependencyGraph;
import blue.contract.packager.model.DirectoryNode;
import blue.language.model.Node;

import java.io.IOException;
import java.util.List;
import java.util.Map;

public class SequentialDependencyGraphBuilder implements DependencyGraphBuilder {
    private final List<DependencyGraphBuilder> builders;

    public SequentialDependencyGraphBuilder(List<DependencyGraphBuilder> builders) {
        this.builders = builders;
    }

    @Override
    public DependencyGraph buildDependencyGraph(String rootDir) throws IOException {
        DependencyGraph combinedGraph = new DependencyGraph();

        for (DependencyGraphBuilder builder : builders) {
            DependencyGraph graph = builder.buildDependencyGraph(rootDir);
            mergeGraphs(combinedGraph, graph);
        }

        return combinedGraph;
    }

    private void mergeGraphs(DependencyGraph target, DependencyGraph source) {
        for (Map.Entry<String, Map<String, DirectoryNode>> typeEntry : source.getDirectories().entrySet()) {
            String typeName = typeEntry.getKey();
            Map<String, DirectoryNode> sourceVersions = typeEntry.getValue();

            for (Map.Entry<String, DirectoryNode> versionEntry : sourceVersions.entrySet()) {
                String version = versionEntry.getKey();
                DirectoryNode sourceDir = versionEntry.getValue();

                // Check if this type+version combination already exists in the target
                Map<String, DirectoryNode> targetVersions = target.getDirectories().get(typeName);
                if (targetVersions != null && targetVersions.containsKey(version)) {
                    throw new IllegalStateException(
                            String.format("Directory collision detected: %s version %s", typeName, version)
                    );
                }

                // Add the directory with its dependency
                target.addDirectory(typeName, version, sourceDir.getDependency());

                // Add all nodes from the source directory
                for (Node node : sourceDir.getNodes().values()) {
                    target.addNode(typeName, version, node);
                }
            }
        }
    }
}