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


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

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));
    }
}