summaryrefslogtreecommitdiff
path: root/support/texlab/src/features/reference.rs
blob: f84a3f67d5c86ca0d9cb4f53744544cf4c56e52f (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
mod entry;
mod label;
mod string;

use std::sync::Arc;

use lsp_types::{Location, ReferenceParams, Url};
use rowan::TextRange;

use crate::LineIndexExt;

use self::{
    entry::find_entry_references, label::find_label_references, string::find_string_references,
};

use super::{cursor::CursorContext, FeatureRequest};

pub fn find_all_references(request: FeatureRequest<ReferenceParams>) -> Vec<Location> {
    let mut results = Vec::new();
    let context = CursorContext::new(request);
    log::debug!("[References] Cursor: {:?}", context.cursor);
    find_label_references(&context, &mut results);
    find_entry_references(&context, &mut results);
    find_string_references(&context, &mut results);

    results
        .into_iter()
        .map(|result| Location {
            uri: result.uri.as_ref().clone(),
            range: context.request.workspace.documents_by_uri[&result.uri]
                .line_index
                .line_col_lsp_range(result.range),
        })
        .collect()
}

#[derive(Debug, Clone)]
struct ReferenceResult {
    uri: Arc<Url>,
    range: TextRange,
}