javaparser-core/src/main/java/com/github/javaparser/version/PostProcessors.java

package com.github.javaparser.version;

import com.github.javaparser.ParseResult;
import com.github.javaparser.ParserConfiguration;
import com.github.javaparser.ast.Node;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import static com.github.javaparser.ParseResult.PostProcessor;

/**
 * A post processor that will call a collection of post processors.
 */
public class PostProcessors implements PostProcessor {
    private final List<PostProcessor> postProcessors = new ArrayList<>();

    public PostProcessors(PostProcessor... postProcessors) {
        this.postProcessors.addAll(Arrays.asList(postProcessors));
    }

    public List<PostProcessor> getPostProcessors() {
        return postProcessors;
    }

    public PostProcessors remove(PostProcessor postProcessor) {
        if (!postProcessors.remove(postProcessor)) {
            throw new AssertionError("Trying to remove a post processor that isn't there.");
        }
        return this;
    }

    public PostProcessors replace(PostProcessor oldProcessor, PostProcessor newProcessor) {
        remove(oldProcessor);
        add(newProcessor);
        return this;
    }

    public PostProcessors add(PostProcessor newProcessor) {
        postProcessors.add(newProcessor);
        return this;
    }

    @Override
    public void process(ParseResult<? extends Node> result, ParserConfiguration configuration) {
        postProcessors.forEach(pp -> pp.process(result, configuration));
    }
}