// From the Path Demo
class Path {
    #sourceId ;
    #edgeId;
    #targetId;

    constructor(sourceId, edgeId, targetId) {
        this.#sourceId = sourceId;
        this.#edgeId = edgeId;
        this.#targetId = targetId;
    }

    get sourceId() {
        return this.#sourceId;
    }

    get edgeId() {
        return this.#edgeId;
    }

    get targetId() {
        return this.#targetId;
    }
}

class PathResolver {

    #paths;

    constructor(bpmnVisualization) {
        this.#paths = this.#buildPaths(bpmnVisualization.bpmnElementsRegistry.getElementsByKinds(Object.values(bpmnvisu.FlowKind)));
    }

    /**
     * @param {string[]} shapeIds
     */
    flatPathsBetweenShapes(shapeIds) {
        return [...shapeIds, ...this.#paths.filter(path => shapeIds.includes(path.sourceId) && shapeIds.includes(path.targetId)).map(path => path.edgeId)];
    }

    /**
     * @param {string[]} shapeIds
     */
    flatPathsWithNextEdges(shapeIds) {
        return [...shapeIds, ...this.#paths.filter(path => shapeIds.includes(path.sourceId) ).map(path => path.edgeId)];
    }

    flatAllPaths(){
        const flatPaths = new Set();

        for (const path of this.#paths) {
            flatPaths.add(path.edgeId);
            flatPaths.add(path.sourceId);
            flatPaths.add(path.targetId);
        }

        return Array.from(flatPaths);
    }
    
    #buildPaths(edges) {
        return edges.map(edge => {
            const bpmnSemantic = edge.bpmnSemantic;
            return new Path(bpmnSemantic.sourceRefId, bpmnSemantic.id , bpmnSemantic.targetRefId);
        });
    }
}