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
use std::cell::RefCell;
use shader_sense::{
position::{ShaderFilePosition, ShaderRange},
shader_error::ShaderError,
symbols::symbols::{ShaderSymbolData, ShaderSymbolMode},
};
use lsp_types::{GotoDefinitionResponse, Position, Url};
use crate::server::common::{shader_range_to_lsp_range, ServerLanguageError};
use crate::server::ServerLanguage;
impl ServerLanguage {
pub fn recolt_goto(
&mut self,
uri: &Url,
position: Position,
) -> Result<Option<GotoDefinitionResponse>, ServerLanguageError> {
let cached_file = self.get_cachable_file(&uri)?;
let language_data = self
.language_data
.get(&cached_file.shading_language)
.unwrap();
let file_path = uri.to_file_path().unwrap();
let shader_position = ShaderFilePosition::new(
file_path.clone(),
position.line as u32,
position.character as u32,
);
let symbol_list = self.watched_files.get_all_symbols(uri);
match language_data.symbol_provider.get_word_range_at_position(
&RefCell::borrow(&cached_file.shader_module),
&shader_position.position,
) {
Ok(word) => {
let matching_symbols =
word.find_symbol_from_parent(file_path.clone(), &symbol_list);
Ok(Some(GotoDefinitionResponse::Link(
matching_symbols
.iter()
.filter_map(|symbol| {
if let ShaderSymbolData::Include { target } = &symbol.data {
match &symbol.mode {
// _runtime.range here should be equal to selected_range.
ShaderSymbolMode::Runtime(_runtime) => {
Some(lsp_types::LocationLink {
origin_selection_range: Some(
shader_range_to_lsp_range(&word.get_range()),
),
target_uri: Url::from_file_path(&target).unwrap(),
target_range: shader_range_to_lsp_range(
&ShaderRange::zero(),
),
target_selection_range: shader_range_to_lsp_range(
&ShaderRange::zero(),
),
})
}
_ => None,
}
} else {
match &symbol.mode {
ShaderSymbolMode::Runtime(runtime) => {
Some(lsp_types::LocationLink {
origin_selection_range: Some(
shader_range_to_lsp_range(&word.get_range()),
),
target_uri: Url::from_file_path(&runtime.file_path)
.unwrap(),
target_range: shader_range_to_lsp_range(&runtime.range),
target_selection_range: shader_range_to_lsp_range(
&runtime.range,
),
})
}
_ => None,
}
}
})
.collect(),
)))
}
Err(err) => {
if let ShaderError::NoSymbol = err {
Ok(None)
} else {
Err(err.into())
}
}
}
}
}