summaryrefslogtreecommitdiff
path: root/support/texlab/src/db/diagnostics/bib.rs
blob: 4f5dfc55a1f16294124a6d2cadef7d89271a451b (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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
use lsp_types::DiagnosticSeverity;
use rowan::{ast::AstNode, TextRange};

use crate::{
    db::document::Document,
    syntax::bibtex::{self, HasDelims, HasEq, HasName, HasType, HasValue},
    util::line_index_ext::LineIndexExt,
    Db,
};

use super::{BibCode, Diagnostic, DiagnosticCode};

#[salsa::tracked(return_ref)]
pub fn collect(db: &dyn Db, document: Document) -> Vec<Diagnostic> {
    let mut results = Vec::new();
    let data = match document.parse(db).as_bib() {
        Some(data) => data,
        None => return results,
    };

    for node in data.root(db).descendants() {
        analyze_entry(db, document, node.clone(), &mut results)
            .or_else(|| analyze_field(db, document, node, &mut results));
    }

    results
}

fn analyze_entry(
    db: &dyn Db,
    document: Document,
    node: bibtex::SyntaxNode,
    results: &mut Vec<Diagnostic>,
) -> Option<()> {
    let line_index = document.line_index(db);

    let entry = bibtex::Entry::cast(node)?;
    if entry.left_delim_token().is_none() {
        let code = BibCode::ExpectingLCurly;
        results.push(Diagnostic {
            severity: DiagnosticSeverity::ERROR,
            range: line_index.line_col_lsp_range(entry.type_token()?.text_range()),
            code: DiagnosticCode::Bib(code),
            message: String::from(code),
        });

        return Some(());
    }

    if entry.name_token().is_none() {
        let code = BibCode::ExpectingKey;
        results.push(Diagnostic {
            severity: DiagnosticSeverity::ERROR,
            range: line_index.line_col_lsp_range(entry.left_delim_token()?.text_range()),
            code: DiagnosticCode::Bib(code),
            message: String::from(code),
        });

        return Some(());
    }

    if entry.right_delim_token().is_none() {
        let code = BibCode::ExpectingRCurly;
        results.push(Diagnostic {
            severity: DiagnosticSeverity::ERROR,
            range: line_index
                .line_col_lsp_range(TextRange::empty(entry.syntax().text_range().end())),
            code: DiagnosticCode::Bib(code),
            message: String::from(code),
        });

        return Some(());
    }

    Some(())
}

fn analyze_field(
    db: &dyn Db,
    document: Document,
    node: bibtex::SyntaxNode,
    results: &mut Vec<Diagnostic>,
) -> Option<()> {
    let line_index = document.line_index(db);

    let field = bibtex::Field::cast(node)?;
    if field.eq_token().is_none() {
        let code = BibCode::ExpectingEq;
        results.push(Diagnostic {
            severity: DiagnosticSeverity::ERROR,
            range: line_index.line_col_lsp_range(field.name_token()?.text_range()),
            code: DiagnosticCode::Bib(code),
            message: String::from(code),
        });

        return Some(());
    }

    if field.value().is_none() {
        let code = BibCode::ExpectingFieldValue;
        results.push(Diagnostic {
            severity: DiagnosticSeverity::ERROR,
            range: line_index.line_col_lsp_range(field.name_token()?.text_range()),
            code: DiagnosticCode::Bib(code),
            message: String::from(code),
        });

        return Some(());
    }

    Some(())
}