128 lines · 4.0 KB · Rust
RawDownload
use zed_extension_api::{    self as zed,    lsp::{Completion, CompletionKind, Symbol, SymbolKind},    settings::LspSettings,    CodeLabel, CodeLabelSpan, LanguageServerId, Result,}; pub struct RubyLspBinary {    pub path: String,    pub args: Option<Vec<String>>,} pub struct RubyLsp {} impl RubyLsp {    pub const LANGUAGE_SERVER_ID: &'static str = "ruby-lsp";     pub fn new() -> Self {        Self {}    }     pub fn language_server_command(        &mut self,        language_server_id: &LanguageServerId,        worktree: &zed::Worktree,    ) -> Result<zed::Command> {        let binary = self.language_server_binary(language_server_id, worktree)?;         Ok(zed::Command {            command: binary.path,            args: binary.args.unwrap_or_default(),            env: worktree.shell_env(),        })    }     fn language_server_binary(        &self,        _language_server_id: &LanguageServerId,        worktree: &zed::Worktree,    ) -> Result<RubyLspBinary> {        let binary_settings = LspSettings::for_worktree("ruby-lsp", worktree)            .ok()            .and_then(|lsp_settings| lsp_settings.binary);        let binary_args = binary_settings            .as_ref()            .and_then(|binary_settings| binary_settings.arguments.clone());         if let Some(path) = binary_settings.and_then(|binary_settings| binary_settings.path) {            return Ok(RubyLspBinary {                path,                args: binary_args,            });        }         if let Some(path) = worktree.which("ruby-lsp") {            return Ok(RubyLspBinary {                path,                args: binary_args,            });        }         Err(            "ruby-lsp must be installed manually. Install it with `gem install ruby-lsp`."                .to_string(),        )    }     pub fn label_for_completion(&self, completion: Completion) -> Option<CodeLabel> {        let highlight_name = match completion.kind? {            CompletionKind::Class | CompletionKind::Module => "type",            CompletionKind::Constant => "constant",            CompletionKind::Method => "function.method",            CompletionKind::Reference => "function.method",            CompletionKind::Keyword => "keyword",            _ => return None,        };         let len = completion.label.len();        let name_span = CodeLabelSpan::literal(completion.label, Some(highlight_name.to_string()));         Some(CodeLabel {            code: Default::default(),            spans: vec![name_span],            filter_range: (0..len).into(),        })    }     pub fn label_for_symbol(&self, symbol: Symbol) -> Option<CodeLabel> {        let name = &symbol.name;         return match symbol.kind {            SymbolKind::Method => {                let code = format!("def {name}; end");                let filter_range = 0..name.len();                let display_range = 4..4 + name.len();                 Some(CodeLabel {                    code,                    spans: vec![CodeLabelSpan::code_range(display_range)],                    filter_range: filter_range.into(),                })            }            SymbolKind::Class | SymbolKind::Module => {                let code = format!("class {name}; end");                let filter_range = 0..name.len();                let display_range = 6..6 + name.len();                 Some(CodeLabel {                    code,                    spans: vec![CodeLabelSpan::code_range(display_range)],                    filter_range: filter_range.into(),                })            }            SymbolKind::Constant => {                let code = name.to_uppercase().to_string();                let filter_range = 0..name.len();                let display_range = 0..name.len();                 Some(CodeLabel {                    code,                    spans: vec![CodeLabelSpan::code_range(display_range)],                    filter_range: filter_range.into(),                })            }            _ => None,        };    }}