summaryrefslogtreecommitdiff
path: root/support/texlab/src/symbol/latex_section/enumeration.rs
blob: 34f4dff72fbd9eeb607b8f4c79c974ee2f245ee1 (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
use super::{label_name, selection_range};
use crate::range::RangeExt;
use crate::symbol::{LatexSymbol, LatexSymbolKind};
use crate::syntax::*;
use crate::workspace::*;
use lsp_types::Range;

pub fn symbols(view: &DocumentView, tree: &LatexSyntaxTree) -> Vec<LatexSymbol> {
    let mut symbols = Vec::new();
    for environment in &tree.env.environments {
        if environment.left.is_enum() {
            symbols.push(make_symbol(view, tree, environment));
        }
    }
    symbols
}

fn make_symbol(
    view: &DocumentView,
    tree: &LatexSyntaxTree,
    enumeration: &LatexEnvironment,
) -> LatexSymbol {
    let name = titlelize(enumeration.left.name().unwrap().text());

    let items: Vec<_> = tree
        .structure
        .items
        .iter()
        .filter(|item| tree.is_enumeration_item(enumeration, item))
        .collect();

    let mut children = Vec::new();
    for i in 0..items.len() {
        let start = items[i].start();
        let end = items
            .get(i + 1)
            .map(|item| item.start())
            .unwrap_or_else(|| enumeration.right.start());
        let range = Range::new(start, end);

        let label = find_item_label(tree, range);

        let number = items[i].name().or_else(|| {
            label
                .as_ref()
                .and_then(|label| OutlineContext::find_number(view, label))
        });

        let name = number.unwrap_or_else(|| "Item".into());
        children.push(LatexSymbol {
            name,
            label: label_name(label),
            kind: LatexSymbolKind::EnumerationItem,
            deprecated: false,
            full_range: range,
            selection_range: selection_range(items[i].range(), label),
            children: Vec::new(),
        });
    }

    LatexSymbol {
        name,
        label: None,
        kind: LatexSymbolKind::Enumeration,
        deprecated: false,
        full_range: enumeration.range(),
        selection_range: enumeration.range(),
        children,
    }
}

fn find_item_label(tree: &LatexSyntaxTree, item_range: Range) -> Option<&LatexLabel> {
    let label = tree.find_label_by_range(item_range)?;
    if tree
        .env
        .environments
        .iter()
        .filter(|env| item_range.contains(env.start()))
        .all(|env| !env.range().contains(label.start()))
    {
        Some(label)
    } else {
        None
    }
}