summaryrefslogtreecommitdiff
path: root/src/main/java/org/zwobble/mammoth/internal/docx/Numbering.java
blob: 90ea5c144022fc600b2ef33a9b4a1930f04785e4 (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
package org.zwobble.mammoth.internal.docx;

import org.zwobble.mammoth.internal.documents.NumberingLevel;

import java.util.Map;
import java.util.Optional;

import static org.zwobble.mammoth.internal.util.Maps.lookup;
import static org.zwobble.mammoth.internal.util.Maps.map;

public class Numbering {
    public static class AbstractNum {
        private final Map<String, NumberingLevel> levels;
        private final Optional<String> numStyleLink;

        public AbstractNum(Map<String, NumberingLevel> levels, Optional<String> numStyleLink) {
            this.levels = levels;
            this.numStyleLink = numStyleLink;
        }
    }

    public static class Num {
        private final Optional<String> abstractNumId;

        public Num(Optional<String> abstractNumId) {
            this.abstractNumId = abstractNumId;
        }
    }

    public static final Numbering EMPTY = new Numbering(map(), map(), Styles.EMPTY);

    private final Map<String, AbstractNum> abstractNums;
    private final Map<String, Num> nums;
    private final Styles styles;

    public Numbering(
        Map<String, AbstractNum> abstractNums,
        Map<String, Num> nums,
        Styles styles
    ) {
        this.abstractNums = abstractNums;
        this.nums = nums;
        this.styles = styles;
    }

    public Optional<NumberingLevel> findLevel(String numId, String level) {
        return lookup(nums, numId)
            .flatMap(num -> num.abstractNumId)
            .flatMap(abstractNumId -> lookup(this.abstractNums, abstractNumId))
            .flatMap(abstractNum -> {
                if (abstractNum.numStyleLink.isPresent()) {
                    return abstractNum.numStyleLink
                        .flatMap(numStyleLink -> styles.findNumberingStyleById(numStyleLink))
                        .flatMap(style -> style.getNumId())
                        .flatMap(linkedNumId -> findLevel(linkedNumId, level));
                } else {
                    return lookup(abstractNum.levels, level);
                }
            });
    }
}