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
|
use base_db::{
semantics::tex::LabelKind,
util::{render_label, RenderedObject},
DocumentData,
};
use rowan::{ast::AstNode, TextRange};
use syntax::latex;
use crate::util::{cursor::CursorContext, lsp_enums::Structure};
use super::builder::CompletionBuilder;
pub fn complete<'db>(
context: &'db CursorContext,
builder: &mut CompletionBuilder<'db>,
) -> Option<()> {
let (range, is_math) = find_reference(context).or_else(|| find_reference_range(context))?;
for document in &context.project.documents {
let DocumentData::Tex(data) = &document.data else { continue };
for label in data
.semantics
.labels
.iter()
.filter(|label| label.kind == LabelKind::Definition)
{
match render_label(context.workspace, &context.project, label) {
Some(rendered_label) => {
let kind = match &rendered_label.object {
RenderedObject::Section { .. } => Structure::Section,
RenderedObject::Float { .. } => Structure::Float,
RenderedObject::Theorem { .. } => Structure::Theorem,
RenderedObject::Equation => Structure::Equation,
RenderedObject::EnumItem => Structure::Item,
};
if is_math && kind != Structure::Equation {
continue;
}
let header = rendered_label.detail();
let footer = match &rendered_label.object {
RenderedObject::Float { caption, .. } => Some(caption.clone()),
_ => None,
};
let text = format!("{} {}", label.name.text, rendered_label.reference());
builder.label(range, &label.name.text, kind, header, footer, text);
}
None => {
let kind = Structure::Label;
let header = None;
let footer = None;
let text = label.name.text.clone();
builder.label(range, &label.name.text, kind, header, footer, text);
}
}
}
}
Some(())
}
fn find_reference(context: &CursorContext) -> Option<(TextRange, bool)> {
let (_, range, group) = context.find_curly_group_word_list()?;
let reference = latex::LabelReference::cast(group.syntax().parent()?)?;
let is_math = reference.command()?.text() == "\\eqref";
Some((range, is_math))
}
fn find_reference_range(context: &CursorContext) -> Option<(TextRange, bool)> {
let (_, range, group) = context.find_curly_group_word()?;
latex::LabelReferenceRange::cast(group.syntax().parent()?)?;
Some((range, false))
}
|