1 //! Conversion of rust-analyzer specific types to lsp_types equivalents.
5 sync::atomic::{AtomicU32, Ordering},
9 Annotation, AnnotationKind, Assist, AssistKind, CallInfo, Cancelable, CompletionItem,
10 CompletionItemKind, CompletionRelevance, Documentation, FileId, FileRange, FileSystemEdit,
11 Fold, FoldKind, Highlight, HlMod, HlOperator, HlPunct, HlRange, HlTag, Indel, InlayHint,
12 InlayKind, InsertTextFormat, Markup, NavigationTarget, ReferenceAccess, RenameError, Runnable,
13 Severity, SourceChange, StructureNodeKind, SymbolKind, TextEdit, TextRange, TextSize,
15 use itertools::Itertools;
16 use serde_json::to_value;
19 cargo_target_spec::CargoTargetSpec,
20 global_state::GlobalStateSnapshot,
21 line_index::{LineEndings, LineIndex, OffsetEncoding},
22 lsp_ext, semantic_tokens, Result,
25 pub(crate) fn position(line_index: &LineIndex, offset: TextSize) -> lsp_types::Position {
26 let line_col = line_index.index.line_col(offset);
27 match line_index.encoding {
28 OffsetEncoding::Utf8 => lsp_types::Position::new(line_col.line, line_col.col),
29 OffsetEncoding::Utf16 => {
30 let line_col = line_index.index.to_utf16(line_col);
31 lsp_types::Position::new(line_col.line, line_col.col)
36 pub(crate) fn range(line_index: &LineIndex, range: TextRange) -> lsp_types::Range {
37 let start = position(line_index, range.start());
38 let end = position(line_index, range.end());
39 lsp_types::Range::new(start, end)
42 pub(crate) fn symbol_kind(symbol_kind: SymbolKind) -> lsp_types::SymbolKind {
44 SymbolKind::Function => lsp_types::SymbolKind::Function,
45 SymbolKind::Struct => lsp_types::SymbolKind::Struct,
46 SymbolKind::Enum => lsp_types::SymbolKind::Enum,
47 SymbolKind::Variant => lsp_types::SymbolKind::EnumMember,
48 SymbolKind::Trait => lsp_types::SymbolKind::Interface,
49 SymbolKind::Macro => lsp_types::SymbolKind::Function,
50 SymbolKind::Module => lsp_types::SymbolKind::Module,
51 SymbolKind::TypeAlias | SymbolKind::TypeParam => lsp_types::SymbolKind::TypeParameter,
52 SymbolKind::Field => lsp_types::SymbolKind::Field,
53 SymbolKind::Static => lsp_types::SymbolKind::Constant,
54 SymbolKind::Const => lsp_types::SymbolKind::Constant,
55 SymbolKind::ConstParam => lsp_types::SymbolKind::Constant,
56 SymbolKind::Impl => lsp_types::SymbolKind::Object,
58 | SymbolKind::SelfParam
59 | SymbolKind::LifetimeParam
60 | SymbolKind::ValueParam
61 | SymbolKind::Label => lsp_types::SymbolKind::Variable,
62 SymbolKind::Union => lsp_types::SymbolKind::Struct,
66 pub(crate) fn structure_node_kind(kind: StructureNodeKind) -> lsp_types::SymbolKind {
68 StructureNodeKind::SymbolKind(symbol) => symbol_kind(symbol),
69 StructureNodeKind::Region => lsp_types::SymbolKind::Namespace,
73 pub(crate) fn document_highlight_kind(
74 reference_access: ReferenceAccess,
75 ) -> lsp_types::DocumentHighlightKind {
76 match reference_access {
77 ReferenceAccess::Read => lsp_types::DocumentHighlightKind::Read,
78 ReferenceAccess::Write => lsp_types::DocumentHighlightKind::Write,
82 pub(crate) fn diagnostic_severity(severity: Severity) -> lsp_types::DiagnosticSeverity {
84 Severity::Error => lsp_types::DiagnosticSeverity::Error,
85 Severity::WeakWarning => lsp_types::DiagnosticSeverity::Hint,
89 pub(crate) fn documentation(documentation: Documentation) -> lsp_types::Documentation {
90 let value = crate::markdown::format_docs(documentation.as_str());
91 let markup_content = lsp_types::MarkupContent { kind: lsp_types::MarkupKind::Markdown, value };
92 lsp_types::Documentation::MarkupContent(markup_content)
95 pub(crate) fn insert_text_format(
96 insert_text_format: InsertTextFormat,
97 ) -> lsp_types::InsertTextFormat {
98 match insert_text_format {
99 InsertTextFormat::Snippet => lsp_types::InsertTextFormat::Snippet,
100 InsertTextFormat::PlainText => lsp_types::InsertTextFormat::PlainText,
104 pub(crate) fn completion_item_kind(
105 completion_item_kind: CompletionItemKind,
106 ) -> lsp_types::CompletionItemKind {
107 match completion_item_kind {
108 CompletionItemKind::Attribute => lsp_types::CompletionItemKind::EnumMember,
109 CompletionItemKind::Binding => lsp_types::CompletionItemKind::Variable,
110 CompletionItemKind::BuiltinType => lsp_types::CompletionItemKind::Struct,
111 CompletionItemKind::Keyword => lsp_types::CompletionItemKind::Keyword,
112 CompletionItemKind::Method => lsp_types::CompletionItemKind::Method,
113 CompletionItemKind::Snippet => lsp_types::CompletionItemKind::Snippet,
114 CompletionItemKind::UnresolvedReference => lsp_types::CompletionItemKind::Reference,
115 CompletionItemKind::SymbolKind(symbol) => match symbol {
116 SymbolKind::Const => lsp_types::CompletionItemKind::Constant,
117 SymbolKind::ConstParam => lsp_types::CompletionItemKind::TypeParameter,
118 SymbolKind::Enum => lsp_types::CompletionItemKind::Enum,
119 SymbolKind::Field => lsp_types::CompletionItemKind::Field,
120 SymbolKind::Function => lsp_types::CompletionItemKind::Function,
121 SymbolKind::Impl => lsp_types::CompletionItemKind::Text,
122 SymbolKind::Label => lsp_types::CompletionItemKind::Variable,
123 SymbolKind::LifetimeParam => lsp_types::CompletionItemKind::TypeParameter,
124 SymbolKind::Local => lsp_types::CompletionItemKind::Variable,
125 SymbolKind::Macro => lsp_types::CompletionItemKind::Method,
126 SymbolKind::Module => lsp_types::CompletionItemKind::Module,
127 SymbolKind::SelfParam => lsp_types::CompletionItemKind::Value,
128 SymbolKind::Static => lsp_types::CompletionItemKind::Value,
129 SymbolKind::Struct => lsp_types::CompletionItemKind::Struct,
130 SymbolKind::Trait => lsp_types::CompletionItemKind::Interface,
131 SymbolKind::TypeAlias => lsp_types::CompletionItemKind::Struct,
132 SymbolKind::TypeParam => lsp_types::CompletionItemKind::TypeParameter,
133 SymbolKind::Union => lsp_types::CompletionItemKind::Struct,
134 SymbolKind::ValueParam => lsp_types::CompletionItemKind::Value,
135 SymbolKind::Variant => lsp_types::CompletionItemKind::EnumMember,
140 pub(crate) fn text_edit(line_index: &LineIndex, indel: Indel) -> lsp_types::TextEdit {
141 let range = range(line_index, indel.delete);
142 let new_text = match line_index.endings {
143 LineEndings::Unix => indel.insert,
144 LineEndings::Dos => indel.insert.replace('\n', "\r\n"),
146 lsp_types::TextEdit { range, new_text }
149 pub(crate) fn completion_text_edit(
150 line_index: &LineIndex,
151 insert_replace_support: Option<lsp_types::Position>,
153 ) -> lsp_types::CompletionTextEdit {
154 let text_edit = text_edit(line_index, indel);
155 match insert_replace_support {
156 Some(cursor_pos) => lsp_types::InsertReplaceEdit {
157 new_text: text_edit.new_text,
158 insert: lsp_types::Range { start: text_edit.range.start, end: cursor_pos },
159 replace: text_edit.range,
162 None => text_edit.into(),
166 pub(crate) fn snippet_text_edit(
167 line_index: &LineIndex,
170 ) -> lsp_ext::SnippetTextEdit {
171 let text_edit = text_edit(line_index, indel);
172 let insert_text_format =
173 if is_snippet { Some(lsp_types::InsertTextFormat::Snippet) } else { None };
174 lsp_ext::SnippetTextEdit {
175 range: text_edit.range,
176 new_text: text_edit.new_text,
182 pub(crate) fn text_edit_vec(
183 line_index: &LineIndex,
185 ) -> Vec<lsp_types::TextEdit> {
186 text_edit.into_iter().map(|indel| self::text_edit(line_index, indel)).collect()
189 pub(crate) fn snippet_text_edit_vec(
190 line_index: &LineIndex,
193 ) -> Vec<lsp_ext::SnippetTextEdit> {
196 .map(|indel| self::snippet_text_edit(line_index, is_snippet, indel))
200 pub(crate) fn completion_item(
201 insert_replace_support: Option<lsp_types::Position>,
202 line_index: &LineIndex,
203 item: CompletionItem,
204 ) -> Vec<lsp_types::CompletionItem> {
205 let mut additional_text_edits = Vec::new();
206 let mut text_edit = None;
207 // LSP does not allow arbitrary edits in completion, so we have to do a
208 // non-trivial mapping here.
209 let source_range = item.source_range();
210 for indel in item.text_edit().iter() {
211 if indel.delete.contains_range(source_range) {
212 text_edit = Some(if indel.delete == source_range {
213 self::completion_text_edit(line_index, insert_replace_support, indel.clone())
215 assert!(source_range.end() == indel.delete.end());
216 let range1 = TextRange::new(indel.delete.start(), source_range.start());
217 let range2 = source_range;
218 let indel1 = Indel::replace(range1, String::new());
219 let indel2 = Indel::replace(range2, indel.insert.clone());
220 additional_text_edits.push(self::text_edit(line_index, indel1));
221 self::completion_text_edit(line_index, insert_replace_support, indel2)
224 assert!(source_range.intersect(indel.delete).is_none());
225 let text_edit = self::text_edit(line_index, indel.clone());
226 additional_text_edits.push(text_edit);
229 let text_edit = text_edit.unwrap();
231 let mut lsp_item = lsp_types::CompletionItem {
232 label: item.label().to_string(),
233 detail: item.detail().map(|it| it.to_string()),
234 filter_text: Some(item.lookup().to_string()),
235 kind: item.kind().map(completion_item_kind),
236 text_edit: Some(text_edit),
237 additional_text_edits: Some(additional_text_edits),
238 documentation: item.documentation().map(documentation),
239 deprecated: Some(item.deprecated()),
243 fn set_score(res: &mut lsp_types::CompletionItem, relevance: CompletionRelevance) {
244 if relevance.is_relevant() {
245 res.preselect = Some(true);
247 // The relevance needs to be inverted to come up with a sort score
248 // because the client will sort ascending.
249 let sort_score = relevance.score() ^ 0xFF_FF_FF_FF;
250 // Zero pad the string to ensure values can be properly sorted
251 // by the client. Hex format is used because it is easier to
252 // visually compare very large values, which the sort text
253 // tends to be since it is the opposite of the score.
254 res.sort_text = Some(format!("{:08x}", sort_score));
257 set_score(&mut lsp_item, item.relevance());
259 if item.deprecated() {
260 lsp_item.tags = Some(vec![lsp_types::CompletionItemTag::Deprecated])
263 if item.trigger_call_info() {
264 lsp_item.command = Some(command::trigger_parameter_hints());
267 let mut res = match item.ref_match() {
268 Some((mutability, relevance)) => {
269 let mut lsp_item_with_ref = lsp_item.clone();
270 set_score(&mut lsp_item_with_ref, relevance);
271 lsp_item_with_ref.label =
272 format!("&{}{}", mutability.as_keyword_for_ref(), lsp_item_with_ref.label);
273 if let Some(lsp_types::CompletionTextEdit::Edit(it)) = &mut lsp_item_with_ref.text_edit
275 it.new_text = format!("&{}{}", mutability.as_keyword_for_ref(), it.new_text);
277 vec![lsp_item_with_ref, lsp_item]
279 None => vec![lsp_item],
282 for lsp_item in res.iter_mut() {
283 lsp_item.insert_text_format = Some(insert_text_format(item.insert_text_format()));
288 pub(crate) fn signature_help(
292 ) -> lsp_types::SignatureHelp {
293 let (label, parameters) = match (concise, label_offsets) {
295 let params = call_info
297 .map(|label| lsp_types::ParameterInformation {
298 label: lsp_types::ParameterLabel::Simple(label.to_string()),
301 .collect::<Vec<_>>();
303 if concise { call_info.parameter_labels().join(", ") } else { call_info.signature };
307 let params = call_info
310 .map(|it| [u32::from(it.start()), u32::from(it.end())])
311 .map(|label_offsets| lsp_types::ParameterInformation {
312 label: lsp_types::ParameterLabel::LabelOffsets(label_offsets),
315 .collect::<Vec<_>>();
316 (call_info.signature, params)
319 let mut params = Vec::new();
320 let mut label = String::new();
321 let mut first = true;
322 for param in call_info.parameter_labels() {
324 label.push_str(", ");
327 let start = label.len() as u32;
328 label.push_str(param);
329 let end = label.len() as u32;
330 params.push(lsp_types::ParameterInformation {
331 label: lsp_types::ParameterLabel::LabelOffsets([start, end]),
340 let documentation = if concise {
343 call_info.doc.map(|doc| {
344 lsp_types::Documentation::MarkupContent(lsp_types::MarkupContent {
345 kind: lsp_types::MarkupKind::Markdown,
351 let active_parameter = call_info.active_parameter.map(|it| it as u32);
353 let signature = lsp_types::SignatureInformation {
356 parameters: Some(parameters),
359 lsp_types::SignatureHelp {
360 signatures: vec![signature],
361 active_signature: None,
366 pub(crate) fn inlay_hint(line_index: &LineIndex, inlay_hint: InlayHint) -> lsp_ext::InlayHint {
368 label: inlay_hint.label.to_string(),
369 range: range(line_index, inlay_hint.range),
370 kind: match inlay_hint.kind {
371 InlayKind::ParameterHint => lsp_ext::InlayKind::ParameterHint,
372 InlayKind::TypeHint => lsp_ext::InlayKind::TypeHint,
373 InlayKind::ChainingHint => lsp_ext::InlayKind::ChainingHint,
378 static TOKEN_RESULT_COUNTER: AtomicU32 = AtomicU32::new(1);
380 pub(crate) fn semantic_tokens(
382 line_index: &LineIndex,
383 highlights: Vec<HlRange>,
384 ) -> lsp_types::SemanticTokens {
385 let id = TOKEN_RESULT_COUNTER.fetch_add(1, Ordering::SeqCst).to_string();
386 let mut builder = semantic_tokens::SemanticTokensBuilder::new(id);
388 for highlight_range in highlights {
389 if highlight_range.highlight.is_empty() {
392 let (type_, mods) = semantic_token_type_and_modifiers(highlight_range.highlight);
393 let token_index = semantic_tokens::type_index(type_);
394 let modifier_bitset = mods.0;
396 for mut text_range in line_index.index.lines(highlight_range.range) {
397 if text[text_range].ends_with('\n') {
399 TextRange::new(text_range.start(), text_range.end() - TextSize::of('\n'));
401 let range = range(&line_index, text_range);
402 builder.push(range, token_index, modifier_bitset);
409 pub(crate) fn semantic_token_delta(
410 previous: &lsp_types::SemanticTokens,
411 current: &lsp_types::SemanticTokens,
412 ) -> lsp_types::SemanticTokensDelta {
413 let result_id = current.result_id.clone();
414 let edits = semantic_tokens::diff_tokens(&previous.data, ¤t.data);
415 lsp_types::SemanticTokensDelta { result_id, edits }
418 fn semantic_token_type_and_modifiers(
419 highlight: Highlight,
420 ) -> (lsp_types::SemanticTokenType, semantic_tokens::ModifierSet) {
421 let mut mods = semantic_tokens::ModifierSet::default();
422 let type_ = match highlight.tag {
423 HlTag::Symbol(symbol) => match symbol {
424 SymbolKind::Module => lsp_types::SemanticTokenType::NAMESPACE,
425 SymbolKind::Impl => lsp_types::SemanticTokenType::TYPE,
426 SymbolKind::Field => lsp_types::SemanticTokenType::PROPERTY,
427 SymbolKind::TypeParam => lsp_types::SemanticTokenType::TYPE_PARAMETER,
428 SymbolKind::ConstParam => semantic_tokens::CONST_PARAMETER,
429 SymbolKind::LifetimeParam => semantic_tokens::LIFETIME,
430 SymbolKind::Label => semantic_tokens::LABEL,
431 SymbolKind::ValueParam => lsp_types::SemanticTokenType::PARAMETER,
432 SymbolKind::SelfParam => semantic_tokens::SELF_KEYWORD,
433 SymbolKind::Local => lsp_types::SemanticTokenType::VARIABLE,
434 SymbolKind::Function => {
435 if highlight.mods.contains(HlMod::Associated) {
436 lsp_types::SemanticTokenType::METHOD
438 lsp_types::SemanticTokenType::FUNCTION
441 SymbolKind::Const => {
442 mods |= semantic_tokens::CONSTANT;
443 mods |= lsp_types::SemanticTokenModifier::STATIC;
444 lsp_types::SemanticTokenType::VARIABLE
446 SymbolKind::Static => {
447 mods |= lsp_types::SemanticTokenModifier::STATIC;
448 lsp_types::SemanticTokenType::VARIABLE
450 SymbolKind::Struct => lsp_types::SemanticTokenType::STRUCT,
451 SymbolKind::Enum => lsp_types::SemanticTokenType::ENUM,
452 SymbolKind::Variant => lsp_types::SemanticTokenType::ENUM_MEMBER,
453 SymbolKind::Union => semantic_tokens::UNION,
454 SymbolKind::TypeAlias => semantic_tokens::TYPE_ALIAS,
455 SymbolKind::Trait => lsp_types::SemanticTokenType::INTERFACE,
456 SymbolKind::Macro => lsp_types::SemanticTokenType::MACRO,
458 HlTag::Attribute => semantic_tokens::ATTRIBUTE,
459 HlTag::BoolLiteral => semantic_tokens::BOOLEAN,
460 HlTag::BuiltinType => semantic_tokens::BUILTIN_TYPE,
461 HlTag::ByteLiteral | HlTag::NumericLiteral => lsp_types::SemanticTokenType::NUMBER,
462 HlTag::CharLiteral => semantic_tokens::CHAR_LITERAL,
463 HlTag::Comment => lsp_types::SemanticTokenType::COMMENT,
464 HlTag::EscapeSequence => semantic_tokens::ESCAPE_SEQUENCE,
465 HlTag::FormatSpecifier => semantic_tokens::FORMAT_SPECIFIER,
466 HlTag::Keyword => lsp_types::SemanticTokenType::KEYWORD,
467 HlTag::None => semantic_tokens::GENERIC,
468 HlTag::Operator(op) => match op {
469 HlOperator::Bitwise => semantic_tokens::BITWISE,
470 HlOperator::Arithmetic => semantic_tokens::ARITHMETIC,
471 HlOperator::Logical => semantic_tokens::LOGICAL,
472 HlOperator::Comparison => semantic_tokens::COMPARISON,
473 HlOperator::Other => semantic_tokens::OPERATOR,
475 HlTag::StringLiteral => lsp_types::SemanticTokenType::STRING,
476 HlTag::UnresolvedReference => semantic_tokens::UNRESOLVED_REFERENCE,
477 HlTag::Punctuation(punct) => match punct {
478 HlPunct::Bracket => semantic_tokens::BRACKET,
479 HlPunct::Brace => semantic_tokens::BRACE,
480 HlPunct::Parenthesis => semantic_tokens::PARENTHESIS,
481 HlPunct::Angle => semantic_tokens::ANGLE,
482 HlPunct::Comma => semantic_tokens::COMMA,
483 HlPunct::Dot => semantic_tokens::DOT,
484 HlPunct::Colon => semantic_tokens::COLON,
485 HlPunct::Semi => semantic_tokens::SEMICOLON,
486 HlPunct::Other => semantic_tokens::PUNCTUATION,
490 for modifier in highlight.mods.iter() {
491 let modifier = match modifier {
492 HlMod::Attribute => semantic_tokens::ATTRIBUTE_MODIFIER,
493 HlMod::Definition => lsp_types::SemanticTokenModifier::DECLARATION,
494 HlMod::Documentation => lsp_types::SemanticTokenModifier::DOCUMENTATION,
495 HlMod::Injected => semantic_tokens::INJECTED,
496 HlMod::ControlFlow => semantic_tokens::CONTROL_FLOW,
497 HlMod::Mutable => semantic_tokens::MUTABLE,
498 HlMod::Consuming => semantic_tokens::CONSUMING,
499 HlMod::Unsafe => semantic_tokens::UNSAFE,
500 HlMod::Callable => semantic_tokens::CALLABLE,
501 HlMod::Static => lsp_types::SemanticTokenModifier::STATIC,
502 HlMod::IntraDocLink => semantic_tokens::INTRA_DOC_LINK,
503 HlMod::Trait => semantic_tokens::TRAIT_MODIFIER,
504 HlMod::Associated => continue,
512 pub(crate) fn folding_range(
514 line_index: &LineIndex,
515 line_folding_only: bool,
517 ) -> lsp_types::FoldingRange {
518 let kind = match fold.kind {
519 FoldKind::Comment => Some(lsp_types::FoldingRangeKind::Comment),
520 FoldKind::Imports => Some(lsp_types::FoldingRangeKind::Imports),
521 FoldKind::Region => Some(lsp_types::FoldingRangeKind::Region),
527 | FoldKind::Array => None,
530 let range = range(line_index, fold.range);
532 if line_folding_only {
533 // Clients with line_folding_only == true (such as VSCode) will fold the whole end line
534 // even if it contains text not in the folding range. To prevent that we exclude
535 // range.end.line from the folding region if there is more text after range.end
537 let has_more_text_on_end_line = text[TextRange::new(fold.range.end(), TextSize::of(text))]
539 .take_while(|it| *it != '\n')
540 .any(|it| !it.is_whitespace());
542 let end_line = if has_more_text_on_end_line {
543 range.end.line.saturating_sub(1)
548 lsp_types::FoldingRange {
549 start_line: range.start.line,
550 start_character: None,
556 lsp_types::FoldingRange {
557 start_line: range.start.line,
558 start_character: Some(range.start.character),
559 end_line: range.end.line,
560 end_character: Some(range.end.character),
566 pub(crate) fn url(snap: &GlobalStateSnapshot, file_id: FileId) -> lsp_types::Url {
567 snap.file_id_to_url(file_id)
570 /// Returns a `Url` object from a given path, will lowercase drive letters if present.
571 /// This will only happen when processing windows paths.
573 /// When processing non-windows path, this is essentially the same as `Url::from_file_path`.
574 pub(crate) fn url_from_abs_path(path: &Path) -> lsp_types::Url {
575 assert!(path.is_absolute());
576 let url = lsp_types::Url::from_file_path(path).unwrap();
577 match path.components().next() {
578 Some(path::Component::Prefix(prefix))
579 if matches!(prefix.kind(), path::Prefix::Disk(_) | path::Prefix::VerbatimDisk(_)) =>
581 // Need to lowercase driver letter
586 let driver_letter_range = {
587 let (scheme, drive_letter, _rest) = match url.as_str().splitn(3, ':').collect_tuple() {
591 let start = scheme.len() + ':'.len_utf8();
592 start..(start + drive_letter.len())
595 // Note: lowercasing the `path` itself doesn't help, the `Url::parse`
596 // machinery *also* canonicalizes the drive letter. So, just massage the
598 let mut url = url.into_string();
599 url[driver_letter_range].make_ascii_lowercase();
600 lsp_types::Url::parse(&url).unwrap()
603 pub(crate) fn optional_versioned_text_document_identifier(
604 snap: &GlobalStateSnapshot,
606 ) -> lsp_types::OptionalVersionedTextDocumentIdentifier {
607 let url = url(snap, file_id);
608 let version = snap.url_file_version(&url);
609 lsp_types::OptionalVersionedTextDocumentIdentifier { uri: url, version }
612 pub(crate) fn location(
613 snap: &GlobalStateSnapshot,
615 ) -> Result<lsp_types::Location> {
616 let url = url(snap, frange.file_id);
617 let line_index = snap.file_line_index(frange.file_id)?;
618 let range = range(&line_index, frange.range);
619 let loc = lsp_types::Location::new(url, range);
623 /// Perefer using `location_link`, if the client has the cap.
624 pub(crate) fn location_from_nav(
625 snap: &GlobalStateSnapshot,
626 nav: NavigationTarget,
627 ) -> Result<lsp_types::Location> {
628 let url = url(snap, nav.file_id);
629 let line_index = snap.file_line_index(nav.file_id)?;
630 let range = range(&line_index, nav.full_range);
631 let loc = lsp_types::Location::new(url, range);
635 pub(crate) fn location_link(
636 snap: &GlobalStateSnapshot,
637 src: Option<FileRange>,
638 target: NavigationTarget,
639 ) -> Result<lsp_types::LocationLink> {
640 let origin_selection_range = match src {
642 let line_index = snap.file_line_index(src.file_id)?;
643 let range = range(&line_index, src.range);
648 let (target_uri, target_range, target_selection_range) = location_info(snap, target)?;
649 let res = lsp_types::LocationLink {
650 origin_selection_range,
653 target_selection_range,
659 snap: &GlobalStateSnapshot,
660 target: NavigationTarget,
661 ) -> Result<(lsp_types::Url, lsp_types::Range, lsp_types::Range)> {
662 let line_index = snap.file_line_index(target.file_id)?;
664 let target_uri = url(snap, target.file_id);
665 let target_range = range(&line_index, target.full_range);
666 let target_selection_range =
667 target.focus_range.map(|it| range(&line_index, it)).unwrap_or(target_range);
668 Ok((target_uri, target_range, target_selection_range))
671 pub(crate) fn goto_definition_response(
672 snap: &GlobalStateSnapshot,
673 src: Option<FileRange>,
674 targets: Vec<NavigationTarget>,
675 ) -> Result<lsp_types::GotoDefinitionResponse> {
676 if snap.config.location_link() {
679 .map(|nav| location_link(snap, src, nav))
680 .collect::<Result<Vec<_>>>()?;
683 let locations = targets
686 location(snap, FileRange { file_id: nav.file_id, range: nav.focus_or_full_range() })
688 .collect::<Result<Vec<_>>>()?;
693 fn outside_workspace_annotation_id() -> String {
694 String::from("OutsideWorkspace")
697 pub(crate) fn snippet_text_document_edit(
698 snap: &GlobalStateSnapshot,
702 ) -> Result<lsp_ext::SnippetTextDocumentEdit> {
703 let text_document = optional_versioned_text_document_identifier(snap, file_id);
704 let line_index = snap.file_line_index(file_id)?;
705 let mut edits: Vec<_> =
706 edit.into_iter().map(|it| snippet_text_edit(&line_index, is_snippet, it)).collect();
708 if snap.analysis.is_library_file(file_id)? && snap.config.change_annotation_support() {
709 for edit in &mut edits {
710 edit.annotation_id = Some(outside_workspace_annotation_id())
713 Ok(lsp_ext::SnippetTextDocumentEdit { text_document, edits })
716 pub(crate) fn snippet_text_document_ops(
717 snap: &GlobalStateSnapshot,
718 file_system_edit: FileSystemEdit,
719 ) -> Cancelable<Vec<lsp_ext::SnippetDocumentChangeOperation>> {
720 let mut ops = Vec::new();
721 match file_system_edit {
722 FileSystemEdit::CreateFile { dst, initial_contents } => {
723 let uri = snap.anchored_path(&dst);
724 let create_file = lsp_types::ResourceOp::Create(lsp_types::CreateFile {
729 ops.push(lsp_ext::SnippetDocumentChangeOperation::Op(create_file));
730 if !initial_contents.is_empty() {
732 lsp_types::OptionalVersionedTextDocumentIdentifier { uri, version: None };
733 let text_edit = lsp_ext::SnippetTextEdit {
734 range: lsp_types::Range::default(),
735 new_text: initial_contents,
736 insert_text_format: Some(lsp_types::InsertTextFormat::PlainText),
740 lsp_ext::SnippetTextDocumentEdit { text_document, edits: vec![text_edit] };
741 ops.push(lsp_ext::SnippetDocumentChangeOperation::Edit(edit_file));
744 FileSystemEdit::MoveFile { src, dst } => {
745 let old_uri = snap.file_id_to_url(src);
746 let new_uri = snap.anchored_path(&dst);
747 let mut rename_file =
748 lsp_types::RenameFile { old_uri, new_uri, options: None, annotation_id: None };
749 if snap.analysis.is_library_file(src) == Ok(true)
750 && snap.config.change_annotation_support()
752 rename_file.annotation_id = Some(outside_workspace_annotation_id())
754 ops.push(lsp_ext::SnippetDocumentChangeOperation::Op(lsp_types::ResourceOp::Rename(
762 pub(crate) fn snippet_workspace_edit(
763 snap: &GlobalStateSnapshot,
764 source_change: SourceChange,
765 ) -> Result<lsp_ext::SnippetWorkspaceEdit> {
766 let mut document_changes: Vec<lsp_ext::SnippetDocumentChangeOperation> = Vec::new();
768 for op in source_change.file_system_edits {
769 let ops = snippet_text_document_ops(snap, op)?;
770 document_changes.extend_from_slice(&ops);
772 for (file_id, edit) in source_change.source_file_edits {
773 let edit = snippet_text_document_edit(&snap, source_change.is_snippet, file_id, edit)?;
774 document_changes.push(lsp_ext::SnippetDocumentChangeOperation::Edit(edit));
776 let mut workspace_edit = lsp_ext::SnippetWorkspaceEdit {
778 document_changes: Some(document_changes),
779 change_annotations: None,
781 if snap.config.change_annotation_support() {
782 workspace_edit.change_annotations = Some(
784 outside_workspace_annotation_id(),
785 lsp_types::ChangeAnnotation {
786 label: String::from("Edit outside of the workspace"),
787 needs_confirmation: Some(true),
788 description: Some(String::from(
789 "This edit lies outside of the workspace and may affect dependencies",
799 pub(crate) fn workspace_edit(
800 snap: &GlobalStateSnapshot,
801 source_change: SourceChange,
802 ) -> Result<lsp_types::WorkspaceEdit> {
803 assert!(!source_change.is_snippet);
804 snippet_workspace_edit(snap, source_change).map(|it| it.into())
807 impl From<lsp_ext::SnippetWorkspaceEdit> for lsp_types::WorkspaceEdit {
808 fn from(snippet_workspace_edit: lsp_ext::SnippetWorkspaceEdit) -> lsp_types::WorkspaceEdit {
809 lsp_types::WorkspaceEdit {
811 document_changes: snippet_workspace_edit.document_changes.map(|changes| {
812 lsp_types::DocumentChanges::Operations(
815 .map(|change| match change {
816 lsp_ext::SnippetDocumentChangeOperation::Op(op) => {
817 lsp_types::DocumentChangeOperation::Op(op)
819 lsp_ext::SnippetDocumentChangeOperation::Edit(edit) => {
820 lsp_types::DocumentChangeOperation::Edit(
821 lsp_types::TextDocumentEdit {
822 text_document: edit.text_document,
823 edits: edit.edits.into_iter().map(From::from).collect(),
831 change_annotations: snippet_workspace_edit.change_annotations,
836 impl From<lsp_ext::SnippetTextEdit>
837 for lsp_types::OneOf<lsp_types::TextEdit, lsp_types::AnnotatedTextEdit>
840 lsp_ext::SnippetTextEdit { annotation_id, insert_text_format:_, new_text, range }: lsp_ext::SnippetTextEdit,
842 match annotation_id {
843 Some(annotation_id) => lsp_types::OneOf::Right(lsp_types::AnnotatedTextEdit {
844 text_edit: lsp_types::TextEdit { range, new_text },
847 None => lsp_types::OneOf::Left(lsp_types::TextEdit { range, new_text }),
852 pub(crate) fn call_hierarchy_item(
853 snap: &GlobalStateSnapshot,
854 target: NavigationTarget,
855 ) -> Result<lsp_types::CallHierarchyItem> {
856 let name = target.name.to_string();
857 let detail = target.description.clone();
858 let kind = target.kind.map(symbol_kind).unwrap_or(lsp_types::SymbolKind::Function);
859 let (uri, range, selection_range) = location_info(snap, target)?;
860 Ok(lsp_types::CallHierarchyItem {
872 pub(crate) fn code_action_kind(kind: AssistKind) -> lsp_types::CodeActionKind {
874 AssistKind::None | AssistKind::Generate => lsp_types::CodeActionKind::EMPTY,
875 AssistKind::QuickFix => lsp_types::CodeActionKind::QUICKFIX,
876 AssistKind::Refactor => lsp_types::CodeActionKind::REFACTOR,
877 AssistKind::RefactorExtract => lsp_types::CodeActionKind::REFACTOR_EXTRACT,
878 AssistKind::RefactorInline => lsp_types::CodeActionKind::REFACTOR_INLINE,
879 AssistKind::RefactorRewrite => lsp_types::CodeActionKind::REFACTOR_REWRITE,
883 pub(crate) fn code_action(
884 snap: &GlobalStateSnapshot,
886 resolve_data: Option<(usize, lsp_types::CodeActionParams)>,
887 ) -> Result<lsp_ext::CodeAction> {
888 let mut res = lsp_ext::CodeAction {
889 title: assist.label.to_string(),
890 group: assist.group.filter(|_| snap.config.code_action_group()).map(|gr| gr.0),
891 kind: Some(code_action_kind(assist.id.1)),
896 match (assist.source_change, resolve_data) {
897 (Some(it), _) => res.edit = Some(snippet_workspace_edit(snap, it)?),
898 (None, Some((index, code_action_params))) => {
899 res.data = Some(lsp_ext::CodeActionData {
900 id: format!("{}:{}:{}", assist.id.0, assist.id.1.name(), index),
905 stdx::never!("assist should always be resolved if client can't do lazy resolving")
911 pub(crate) fn runnable(
912 snap: &GlobalStateSnapshot,
914 ) -> Result<lsp_ext::Runnable> {
915 let config = snap.config.runnables();
916 let spec = CargoTargetSpec::for_file(snap, runnable.nav.file_id)?;
917 let workspace_root = spec.as_ref().map(|it| it.workspace_root.clone());
918 let target = spec.as_ref().map(|s| s.target.clone());
919 let (cargo_args, executable_args) =
920 CargoTargetSpec::runnable_args(snap, spec, &runnable.kind, &runnable.cfg)?;
921 let label = runnable.label(target);
922 let location = location_link(snap, None, runnable.nav)?;
924 Ok(lsp_ext::Runnable {
926 location: Some(location),
927 kind: lsp_ext::RunnableKind::Cargo,
928 args: lsp_ext::CargoRunnable {
929 workspace_root: workspace_root.map(|it| it.into()),
930 override_cargo: config.override_cargo,
932 cargo_extra_args: config.cargo_extra_args,
939 pub(crate) fn code_lens(
940 snap: &GlobalStateSnapshot,
941 annotation: Annotation,
942 ) -> Result<lsp_types::CodeLens> {
943 match annotation.kind {
944 AnnotationKind::Runnable { debug, runnable: run } => {
945 let line_index = snap.file_line_index(run.nav.file_id)?;
946 let annotation_range = range(&line_index, annotation.range);
948 let action = run.action();
949 let r = runnable(&snap, run)?;
951 let command = if debug {
952 command::debug_single(&r)
954 let title = action.run_title.to_string();
955 command::run_single(&r, &title)
958 Ok(lsp_types::CodeLens { range: annotation_range, command: Some(command), data: None })
960 AnnotationKind::HasImpls { position: file_position, data } => {
961 let line_index = snap.file_line_index(file_position.file_id)?;
962 let annotation_range = range(&line_index, annotation.range);
963 let url = url(snap, file_position.file_id);
965 let position = position(&line_index, file_position.offset);
967 let id = lsp_types::TextDocumentIdentifier { uri: url.clone() };
969 let doc_pos = lsp_types::TextDocumentPositionParams::new(id, position);
971 let goto_params = lsp_types::request::GotoImplementationParams {
972 text_document_position_params: doc_pos,
973 work_done_progress_params: Default::default(),
974 partial_result_params: Default::default(),
977 let command = data.map(|ranges| {
978 let locations: Vec<lsp_types::Location> = ranges
980 .filter_map(|target| {
983 FileRange { file_id: target.file_id, range: target.full_range },
989 command::show_references(
990 implementation_title(locations.len()),
997 Ok(lsp_types::CodeLens {
998 range: annotation_range,
1000 data: Some(to_value(lsp_ext::CodeLensResolveData::Impls(goto_params)).unwrap()),
1003 AnnotationKind::HasReferences { position: file_position, data } => {
1004 let line_index = snap.file_line_index(file_position.file_id)?;
1005 let annotation_range = range(&line_index, annotation.range);
1006 let url = url(snap, file_position.file_id);
1008 let position = position(&line_index, file_position.offset);
1010 let id = lsp_types::TextDocumentIdentifier { uri: url.clone() };
1012 let doc_pos = lsp_types::TextDocumentPositionParams::new(id, position);
1014 let command = data.map(|ranges| {
1015 let locations: Vec<lsp_types::Location> =
1016 ranges.into_iter().filter_map(|range| location(snap, range).ok()).collect();
1018 command::show_references(
1019 reference_title(locations.len()),
1026 Ok(lsp_types::CodeLens {
1027 range: annotation_range,
1029 data: Some(to_value(lsp_ext::CodeLensResolveData::References(doc_pos)).unwrap()),
1035 pub(crate) mod command {
1036 use ide::{FileRange, NavigationTarget};
1037 use serde_json::to_value;
1040 global_state::GlobalStateSnapshot,
1042 to_proto::{location, location_link},
1045 pub(crate) fn show_references(
1047 uri: &lsp_types::Url,
1048 position: lsp_types::Position,
1049 locations: Vec<lsp_types::Location>,
1050 ) -> lsp_types::Command {
1051 // We cannot use the 'editor.action.showReferences' command directly
1052 // because that command requires vscode types which we convert in the handler
1053 // on the client side.
1055 lsp_types::Command {
1057 command: "rust-analyzer.showReferences".into(),
1058 arguments: Some(vec![
1059 to_value(uri).unwrap(),
1060 to_value(position).unwrap(),
1061 to_value(locations).unwrap(),
1066 pub(crate) fn run_single(runnable: &lsp_ext::Runnable, title: &str) -> lsp_types::Command {
1067 lsp_types::Command {
1068 title: title.to_string(),
1069 command: "rust-analyzer.runSingle".into(),
1070 arguments: Some(vec![to_value(runnable).unwrap()]),
1074 pub(crate) fn debug_single(runnable: &lsp_ext::Runnable) -> lsp_types::Command {
1075 lsp_types::Command {
1076 title: "Debug".into(),
1077 command: "rust-analyzer.debugSingle".into(),
1078 arguments: Some(vec![to_value(runnable).unwrap()]),
1082 pub(crate) fn goto_location(
1083 snap: &GlobalStateSnapshot,
1084 nav: &NavigationTarget,
1085 ) -> Option<lsp_types::Command> {
1086 let value = if snap.config.location_link() {
1087 let link = location_link(snap, None, nav.clone()).ok()?;
1088 to_value(link).ok()?
1090 let range = FileRange { file_id: nav.file_id, range: nav.focus_or_full_range() };
1091 let location = location(snap, range).ok()?;
1092 to_value(location).ok()?
1095 Some(lsp_types::Command {
1096 title: nav.name.to_string(),
1097 command: "rust-analyzer.gotoLocation".into(),
1098 arguments: Some(vec![value]),
1102 pub(crate) fn trigger_parameter_hints() -> lsp_types::Command {
1103 lsp_types::Command {
1104 title: "triggerParameterHints".into(),
1105 command: "editor.action.triggerParameterHints".into(),
1111 pub(crate) fn implementation_title(count: usize) -> String {
1113 "1 implementation".into()
1115 format!("{} implementations", count)
1119 pub(crate) fn reference_title(count: usize) -> String {
1121 "1 reference".into()
1123 format!("{} references", count)
1127 pub(crate) fn markup_content(markup: Markup) -> lsp_types::MarkupContent {
1128 let value = crate::markdown::format_docs(markup.as_str());
1129 lsp_types::MarkupContent { kind: lsp_types::MarkupKind::Markdown, value }
1132 pub(crate) fn rename_error(err: RenameError) -> crate::LspError {
1133 crate::LspError { code: lsp_server::ErrorCode::InvalidParams as i32, message: err.to_string() }
1141 use ide_db::helpers::{
1142 insert_use::{InsertUseConfig, PrefixKind},
1149 fn test_completion_with_ref() {
1152 fn foo(arg: &Foo) {}
1158 let (offset, text) = test_utils::extract_offset(fixture);
1159 let line_index = LineIndex {
1160 index: Arc::new(ide::LineIndex::new(&text)),
1161 endings: LineEndings::Unix,
1162 encoding: OffsetEncoding::Utf16,
1164 let (analysis, file_id) = Analysis::from_single_file(text);
1165 let completions: Vec<(String, Option<String>)> = analysis
1167 &ide::CompletionConfig {
1168 enable_postfix_completions: true,
1169 enable_imports_on_the_fly: true,
1170 add_call_parenthesis: true,
1171 add_call_argument_snippets: true,
1172 snippet_cap: SnippetCap::new(true),
1173 insert_use: InsertUseConfig {
1175 prefix_kind: PrefixKind::Plain,
1179 ide_db::base_db::FilePosition { file_id, offset },
1184 .filter(|c| c.label().ends_with("arg"))
1185 .map(|c| completion_item(None, &line_index, c))
1186 .flat_map(|comps| comps.into_iter().map(|c| (c.label, c.sort_text)))
1188 expect_test::expect![[r#"
1204 .assert_debug_eq(&completions);
1208 fn conv_fold_line_folding_only_fixup() {
1209 let text = r#"mod a;
1221 let (analysis, file_id) = Analysis::from_single_file(text.to_string());
1222 let folds = analysis.folding_ranges(file_id).unwrap();
1223 assert_eq!(folds.len(), 4);
1225 let line_index = LineIndex {
1226 index: Arc::new(ide::LineIndex::new(&text)),
1227 endings: LineEndings::Unix,
1228 encoding: OffsetEncoding::Utf16,
1230 let converted: Vec<lsp_types::FoldingRange> =
1231 folds.into_iter().map(|it| folding_range(&text, &line_index, true, it)).collect();
1233 let expected_lines = [(0, 2), (4, 10), (5, 6), (7, 9)];
1234 assert_eq!(converted.len(), expected_lines.len());
1235 for (folding_range, (start_line, end_line)) in converted.iter().zip(expected_lines.iter()) {
1236 assert_eq!(folding_range.start_line, *start_line);
1237 assert_eq!(folding_range.start_character, None);
1238 assert_eq!(folding_range.end_line, *end_line);
1239 assert_eq!(folding_range.end_character, None);
1243 // `Url` is not able to parse windows paths on unix machines.
1245 #[cfg(target_os = "windows")]
1246 fn test_lowercase_drive_letter_with_drive() {
1247 let url = url_from_abs_path(Path::new("C:\\Test"));
1248 assert_eq!(url.to_string(), "file:///c:/Test");
1252 #[cfg(target_os = "windows")]
1253 fn test_drive_without_colon_passthrough() {
1254 let url = url_from_abs_path(Path::new(r#"\\localhost\C$\my_dir"#));
1255 assert_eq!(url.to_string(), "file://localhost/C$/my_dir");