crates/rust-analyzer/src/caps.rs

   1 //! Advertizes the capabilities of the LSP Server.
   2 use std::env;
   3
   4 use ide::CompletionResolveCapability;
   5 use lsp_types::{
   6     CallHierarchyServerCapability, ClientCapabilities, CodeActionKind, CodeActionOptions,
   7     CodeActionProviderCapability, CodeLensOptions, CompletionOptions,
   8     DocumentOnTypeFormattingOptions, FileOperationFilter, FileOperationPattern,
   9     FileOperationPatternKind, FileOperationRegistrationOptions, FoldingRangeProviderCapability,
  10     HoverProviderCapability, ImplementationProviderCapability, OneOf, RenameOptions, SaveOptions,
  11     SelectionRangeProviderCapability, SemanticTokensFullOptions, SemanticTokensLegend,
  12     SemanticTokensOptions, ServerCapabilities, SignatureHelpOptions, TextDocumentSyncCapability,
  13     TextDocumentSyncKind, TextDocumentSyncOptions, TypeDefinitionProviderCapability,
  14     WorkDoneProgressOptions, WorkspaceFileOperationsServerCapabilities,
  15     WorkspaceServerCapabilities,
  16 };
  17 use rustc_hash::FxHashSet;
  18 use serde_json::json;
  19
  20 use crate::semantic_tokens;
  21
  22 pub fn server_capabilities(client_caps: &ClientCapabilities) -> ServerCapabilities {
  23     ServerCapabilities {
  24         text_document_sync: Some(TextDocumentSyncCapability::Options(TextDocumentSyncOptions {
  25             open_close: Some(true),
  26             change: Some(if env::var("RA_NO_INCREMENTAL_SYNC").is_ok() {
  27                 TextDocumentSyncKind::Full
  28             } else {
  29                 TextDocumentSyncKind::Incremental
  30             }),
  31             will_save: None,
  32             will_save_wait_until: None,
  33             save: Some(SaveOptions::default().into()),
  34         })),
  35         hover_provider: Some(HoverProviderCapability::Simple(true)),
  36         completion_provider: Some(CompletionOptions {
  37             resolve_provider: completions_resolve_provider(client_caps),
  38             trigger_characters: Some(vec![":".to_string(), ".".to_string()]),
  39             work_done_progress_options: WorkDoneProgressOptions { work_done_progress: None },
  40         }),
  41         signature_help_provider: Some(SignatureHelpOptions {
  42             trigger_characters: Some(vec!["(".to_string(), ",".to_string()]),
  43             retrigger_characters: None,
  44             work_done_progress_options: WorkDoneProgressOptions { work_done_progress: None },
  45         }),
  46         declaration_provider: None,
  47         definition_provider: Some(OneOf::Left(true)),
  48         type_definition_provider: Some(TypeDefinitionProviderCapability::Simple(true)),
  49         implementation_provider: Some(ImplementationProviderCapability::Simple(true)),
  50         references_provider: Some(OneOf::Left(true)),
  51         document_highlight_provider: Some(OneOf::Left(true)),
  52         document_symbol_provider: Some(OneOf::Left(true)),
  53         workspace_symbol_provider: Some(OneOf::Left(true)),
  54         code_action_provider: Some(code_action_capabilities(client_caps)),
  55         code_lens_provider: Some(CodeLensOptions { resolve_provider: Some(true) }),
  56         document_formatting_provider: Some(OneOf::Left(true)),
  57         document_range_formatting_provider: None,
  58         document_on_type_formatting_provider: Some(DocumentOnTypeFormattingOptions {
  59             first_trigger_character: "=".to_string(),
  60             more_trigger_character: Some(vec![".".to_string(), ">".to_string()]),
  61         }),
  62         selection_range_provider: Some(SelectionRangeProviderCapability::Simple(true)),
  63         semantic_highlighting: None,
  64         folding_range_provider: Some(FoldingRangeProviderCapability::Simple(true)),
  65         rename_provider: Some(OneOf::Right(RenameOptions {
  66             prepare_provider: Some(true),
  67             work_done_progress_options: WorkDoneProgressOptions { work_done_progress: None },
  68         })),
  69         linked_editing_range_provider: None,
  70         document_link_provider: None,
  71         color_provider: None,
  72         execute_command_provider: None,
  73         workspace: Some(WorkspaceServerCapabilities {
  74             workspace_folders: None,
  75             file_operations: Some(WorkspaceFileOperationsServerCapabilities {
  76                 did_create: None,
  77                 will_create: None,
  78                 did_rename: None,
  79                 will_rename: Some(FileOperationRegistrationOptions {
  80                     filters: vec![
  81                         FileOperationFilter {
  82                             scheme: Some(String::from("file")),
  83                             pattern: FileOperationPattern {
  84                                 glob: String::from("**/*.rs"),
  85                                 matches: Some(FileOperationPatternKind::File),
  86                                 options: None,
  87                             },
  88                         },
  89                         FileOperationFilter {
  90                             scheme: Some(String::from("file")),
  91                             pattern: FileOperationPattern {
  92                                 glob: String::from("**"),
  93                                 matches: Some(FileOperationPatternKind::Folder),
  94                                 options: None,
  95                             },
  96                         }],
  97                 }),
  98                 did_delete: None,
  99                 will_delete: None,
 100             }),
 101         }),
 102         call_hierarchy_provider: Some(CallHierarchyServerCapability::Simple(true)),
 103         semantic_tokens_provider: Some(
 104             SemanticTokensOptions {
 105                 legend: SemanticTokensLegend {
 106                     token_types: semantic_tokens::SUPPORTED_TYPES.to_vec(),
 107                     token_modifiers: semantic_tokens::SUPPORTED_MODIFIERS.to_vec(),
 108                 },
 109
 110                 full: Some(SemanticTokensFullOptions::Delta { delta: Some(true) }),
 111                 range: Some(true),
 112                 work_done_progress_options: Default::default(),
 113             }
 114             .into(),
 115         ),
 116         moniker_provider: None,
 117         experimental: Some(json!({
 118             "joinLines": true,
 119             "ssr": true,
 120             "onEnter": true,
 121             "parentModule": true,
 122             "runnables": {
 123                 "kinds": [ "cargo" ],
 124             },
 125         })),
 126     }
 127 }
 128
 129 fn completions_resolve_provider(client_caps: &ClientCapabilities) -> Option<bool> {
 130     if enabled_completions_resolve_capabilities(client_caps)?.is_empty() {
 131         log::info!("No `additionalTextEdits` completion resolve capability was found in the client capabilities, autoimport completion is disabled");
 132         None
 133     } else {
 134         Some(true)
 135     }
 136 }
 137
 138 /// Parses client capabilities and returns all completion resolve capabilities rust-analyzer supports.
 139 pub(crate) fn enabled_completions_resolve_capabilities(
 140     caps: &ClientCapabilities,
 141 ) -> Option<FxHashSet<CompletionResolveCapability>> {
 142     Some(
 143         caps.text_document
 144             .as_ref()?
 145             .completion
 146             .as_ref()?
 147             .completion_item
 148             .as_ref()?
 149             .resolve_support
 150             .as_ref()?
 151             .properties
 152             .iter()
 153             .filter_map(|cap_string| match cap_string.as_str() {
 154                 "additionalTextEdits" => Some(CompletionResolveCapability::AdditionalTextEdits),
 155                 "detail" => Some(CompletionResolveCapability::Detail),
 156                 "documentation" => Some(CompletionResolveCapability::Documentation),
 157                 _unsupported => None,
 158             })
 159             .collect(),
 160     )
 161 }
 162
 163 fn code_action_capabilities(client_caps: &ClientCapabilities) -> CodeActionProviderCapability {
 164     client_caps
 165         .text_document
 166         .as_ref()
 167         .and_then(|it| it.code_action.as_ref())
 168         .and_then(|it| it.code_action_literal_support.as_ref())
 169         .map_or(CodeActionProviderCapability::Simple(true), |_| {
 170             CodeActionProviderCapability::Options(CodeActionOptions {
 171                 // Advertise support for all built-in CodeActionKinds.
 172                 // Ideally we would base this off of the client capabilities
 173                 // but the client is supposed to fall back gracefully for unknown values.
 174                 code_action_kinds: Some(vec![
 175                     CodeActionKind::EMPTY,
 176                     CodeActionKind::QUICKFIX,
 177                     CodeActionKind::REFACTOR,
 178                     CodeActionKind::REFACTOR_EXTRACT,
 179                     CodeActionKind::REFACTOR_INLINE,
 180                     CodeActionKind::REFACTOR_REWRITE,
 181                 ]),
 182                 resolve_provider: Some(true),
 183                 work_done_progress_options: Default::default(),
 184             })
 185         })
 186 }