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![FileOperationFilter {
  81                         scheme: Some(String::from("file")),
  82                         pattern: FileOperationPattern {
  83                             glob: String::from("**/*.rs"),
  84                             matches: Some(FileOperationPatternKind::File),
  85                             options: None,
  86                         },
  87                     }],
  88                 }),
  89                 did_delete: None,
  90                 will_delete: None,
  91             }),
  92         }),
  93         call_hierarchy_provider: Some(CallHierarchyServerCapability::Simple(true)),
  94         semantic_tokens_provider: Some(
  95             SemanticTokensOptions {
  96                 legend: SemanticTokensLegend {
  97                     token_types: semantic_tokens::SUPPORTED_TYPES.to_vec(),
  98                     token_modifiers: semantic_tokens::SUPPORTED_MODIFIERS.to_vec(),
  99                 },
 100
 101                 full: Some(SemanticTokensFullOptions::Delta { delta: Some(true) }),
 102                 range: Some(true),
 103                 work_done_progress_options: Default::default(),
 104             }
 105             .into(),
 106         ),
 107         moniker_provider: None,
 108         experimental: Some(json!({
 109             "joinLines": true,
 110             "ssr": true,
 111             "onEnter": true,
 112             "parentModule": true,
 113             "runnables": {
 114                 "kinds": [ "cargo" ],
 115             },
 116         })),
 117     }
 118 }
 119
 120 fn completions_resolve_provider(client_caps: &ClientCapabilities) -> Option<bool> {
 121     if enabled_completions_resolve_capabilities(client_caps)?.is_empty() {
 122         log::info!("No `additionalTextEdits` completion resolve capability was found in the client capabilities, autoimport completion is disabled");
 123         None
 124     } else {
 125         Some(true)
 126     }
 127 }
 128
 129 /// Parses client capabilities and returns all completion resolve capabilities rust-analyzer supports.
 130 pub(crate) fn enabled_completions_resolve_capabilities(
 131     caps: &ClientCapabilities,
 132 ) -> Option<FxHashSet<CompletionResolveCapability>> {
 133     Some(
 134         caps.text_document
 135             .as_ref()?
 136             .completion
 137             .as_ref()?
 138             .completion_item
 139             .as_ref()?
 140             .resolve_support
 141             .as_ref()?
 142             .properties
 143             .iter()
 144             .filter_map(|cap_string| match cap_string.as_str() {
 145                 "additionalTextEdits" => Some(CompletionResolveCapability::AdditionalTextEdits),
 146                 "detail" => Some(CompletionResolveCapability::Detail),
 147                 "documentation" => Some(CompletionResolveCapability::Documentation),
 148                 _unsupported => None,
 149             })
 150             .collect(),
 151     )
 152 }
 153
 154 fn code_action_capabilities(client_caps: &ClientCapabilities) -> CodeActionProviderCapability {
 155     client_caps
 156         .text_document
 157         .as_ref()
 158         .and_then(|it| it.code_action.as_ref())
 159         .and_then(|it| it.code_action_literal_support.as_ref())
 160         .map_or(CodeActionProviderCapability::Simple(true), |_| {
 161             CodeActionProviderCapability::Options(CodeActionOptions {
 162                 // Advertise support for all built-in CodeActionKinds.
 163                 // Ideally we would base this off of the client capabilities
 164                 // but the client is supposed to fall back gracefully for unknown values.
 165                 code_action_kinds: Some(vec![
 166                     CodeActionKind::EMPTY,
 167                     CodeActionKind::QUICKFIX,
 168                     CodeActionKind::REFACTOR,
 169                     CodeActionKind::REFACTOR_EXTRACT,
 170                     CodeActionKind::REFACTOR_INLINE,
 171                     CodeActionKind::REFACTOR_REWRITE,
 172                 ]),
 173                 resolve_provider: Some(true),
 174                 work_done_progress_options: Default::default(),
 175             })
 176         })
 177 }