aboutsummaryrefslogtreecommitdiff
path: root/src/main/java/org/zwobble/mammoth/internal/docx/ReadResult.java
blob: 9efd74751d3b04ab4eaddecb92a43fd0b3d4a452 (plain)
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
package org.zwobble.mammoth.internal.docx;

import org.zwobble.mammoth.internal.documents.DocumentElement;
import org.zwobble.mammoth.internal.results.InternalResult;

import java.util.List;
import java.util.function.BiFunction;
import java.util.function.Function;

import static org.zwobble.mammoth.internal.util.Iterables.lazyConcat;
import static org.zwobble.mammoth.internal.util.Iterables.lazyFlatMap;
import static org.zwobble.mammoth.internal.util.Lists.*;

public class ReadResult {
    public static final ReadResult EMPTY_SUCCESS = success(list());

    public static <T> ReadResult flatMap(Iterable<T> iterable, Function<T, ReadResult> function) {
        List<ReadResult> results = eagerMap(iterable, function);
        return new ReadResult(
            eagerFlatMap(results, result -> result.elements),
            eagerFlatMap(results, result -> result.extra),
            lazyFlatMap(results, result -> result.warnings));
    }

    public static <T> ReadResult map(
        InternalResult<T> first,
        ReadResult second,
        BiFunction<T, List<DocumentElement>, DocumentElement> function)
    {
        return new ReadResult(
            list(function.apply(first.getValue(), second.elements)),
            second.extra,
            lazyConcat(first.getWarnings(), second.warnings));
    }

    public static ReadResult success(DocumentElement element) {
        return success(list(element));
    }

    public static ReadResult success(List<DocumentElement> elements) {
        return new ReadResult(elements, list(), list());
    }

    public static ReadResult emptyWithWarning(String warning) {
        return withWarning(list(), warning);
    }

    public static ReadResult withWarning(DocumentElement element, String warning) {
        return withWarning(list(element), warning);
    }

    public static ReadResult withWarning(List<DocumentElement> elements, String warning) {
        return new ReadResult(elements, list(), list(warning));
    }

    private final List<DocumentElement> elements;
    private final List<DocumentElement> extra;
    private final Iterable<String> warnings;

    public ReadResult(List<DocumentElement> elements, List<DocumentElement> extra, Iterable<String> warnings) {
        this.elements = elements;
        this.extra = extra;
        this.warnings = warnings;
    }

    public ReadResult map(Function<List<DocumentElement>, List<DocumentElement>> function) {
        return new ReadResult(
            function.apply(elements),
            extra,
            warnings);
    }

    public ReadResult flatMap(Function<List<DocumentElement>, ReadResult> function) {
        ReadResult result = function.apply(elements);
        return new ReadResult(
            result.elements,
            eagerConcat(extra, result.extra),
            lazyConcat(warnings, result.warnings));
    }

    public ReadResult toExtra() {
        return new ReadResult(list(), eagerConcat(extra, elements), warnings);
    }

    public ReadResult appendExtra() {
        return new ReadResult(eagerConcat(elements, extra), list(), warnings);
    }

    public InternalResult<List<DocumentElement>> toResult() {
        return new InternalResult<>(elements, warnings);
    }
}