From 3e2786a004eddf968a5fa7d5a92dff5b1b5f7734 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Wed, 24 Feb 2016 08:16:16 +0000 Subject: Create internal package --- .../zwobble/mammoth/internal/docx/ReadResult.java | 91 ++++++++++++++++++++++ 1 file changed, 91 insertions(+) create mode 100644 src/main/java/org/zwobble/mammoth/internal/docx/ReadResult.java (limited to 'src/main/java/org/zwobble/mammoth/internal/docx/ReadResult.java') diff --git a/src/main/java/org/zwobble/mammoth/internal/docx/ReadResult.java b/src/main/java/org/zwobble/mammoth/internal/docx/ReadResult.java new file mode 100644 index 0000000..366fd12 --- /dev/null +++ b/src/main/java/org/zwobble/mammoth/internal/docx/ReadResult.java @@ -0,0 +1,91 @@ +package org.zwobble.mammoth.internal.docx; + +import com.google.common.collect.ImmutableList; +import org.zwobble.mammoth.internal.documents.DocumentElement; +import org.zwobble.mammoth.results.Result; +import org.zwobble.mammoth.results.Warning; + +import java.util.List; +import java.util.function.BiFunction; +import java.util.function.Function; + +import static org.zwobble.mammoth.internal.util.MammothLists.list; + +public class ReadResult { + public static final ReadResult EMPTY_SUCCESS = success(list()); + + public static ReadResult concat(Iterable results) { + ImmutableList.Builder elements = ImmutableList.builder(); + ImmutableList.Builder extra = ImmutableList.builder(); + ImmutableList.Builder warnings = ImmutableList.builder(); + for (ReadResult result : results) { + elements.addAll(result.elements); + extra.addAll(result.extra); + warnings.addAll(result.warnings); + } + return new ReadResult(elements.build(), extra.build(), warnings.build()); + } + + public static ReadResult map( + Result first, + ReadResult second, + BiFunction, DocumentElement> function) + { + return new ReadResult( + list(function.apply(first.getValue(), second.elements)), + second.extra, + concat(first.getWarnings(), second.warnings)); + } + + public static ReadResult success(DocumentElement element) { + return success(list(element)); + } + + public static ReadResult success(List elements) { + return new ReadResult(elements, list(), list()); + } + + public static ReadResult emptyWithWarning(Warning warning) { + return new ReadResult(list(), list(), list(warning)); + } + + public static ReadResult withWarning(DocumentElement element, Warning warning) { + return new ReadResult(list(element), list(), list(warning)); + } + + private final List elements; + private final List extra; + private final List warnings; + + public ReadResult(List elements, List extra, List warnings) { + this.elements = elements; + this.extra = extra; + this.warnings = warnings; + } + + public ReadResult map(Function, DocumentElement> function) { + return new ReadResult( + list(function.apply(elements)), + extra, + warnings); + } + + public ReadResult toExtra() { + return new ReadResult(list(), concat(extra, elements), warnings); + } + + public ReadResult appendExtra() { + return new ReadResult(concat(elements, extra), list(), warnings); + } + + public Result> toResult() { + return new Result<>(elements, warnings); + } + + private static List concat(List first, List second) { + ImmutableList.Builder builder = ImmutableList.builder(); + builder.addAll(first); + builder.addAll(second); + return builder.build(); + } +} -- cgit v1.2.3