feat: add sorbet Ruby LSP server option

Author: notchairmk

extension.toml

@@ -18,6 +18,10 @@ languages = ["Ruby", "ERB"]
 name = "Rubocop"
 languages = ["Ruby"]
 
+[language_servers.sorbet]
+name = "Sorbet"
+languages = ["Ruby"]
+
 [grammars.ruby]
 repository = "https://github.com/tree-sitter/tree-sitter-ruby"
 commit = "71bd32fb7607035768799732addba884a37a6210"

src/language_servers/mod.rs

@@ -2,8 +2,10 @@ mod language_server;
 mod rubocop;
 mod ruby_lsp;
 mod solargraph;
+mod sorbet;
 
 pub use language_server::LanguageServer;
 pub use rubocop::*;
 pub use ruby_lsp::*;
 pub use solargraph::*;
+pub use sorbet::*;

src/language_servers/sorbet.rs

@@ -0,0 +1,53 @@
+use super::LanguageServer;
+
+pub struct Sorbet {}
+
+impl LanguageServer for Sorbet {
+    const SERVER_ID: &str = "sorbet";
+    const EXECUTABLE_NAME: &str = "srb";
+    const GEM_NAME: &str = "sorbet";
+
+    fn get_executable_args() -> Vec<String> {
+        [
+            "tc",
+            "--lsp",
+            "--enable-experimental-lsp-document-highlight",
+        ]
+        .iter()
+        .map(|s| s.to_string())
+        .collect()
+    }
+}
+
+impl Sorbet {
+    pub fn new() -> Self {
+        Self {}
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use crate::language_servers::{LanguageServer, Sorbet};
+
+    #[test]
+    fn test_server_id() {
+        assert_eq!(Sorbet::SERVER_ID, "sorbet");
+    }
+
+    #[test]
+    fn test_executable_name() {
+        assert_eq!(Sorbet::EXECUTABLE_NAME, "srb");
+    }
+
+    #[test]
+    fn test_executable_args() {
+        assert_eq!(
+            Sorbet::get_executable_args(),
+            vec![
+                "tc",
+                "--lsp",
+                "--enable-experimental-lsp-document-highlight"
+            ]
+        );
+    }
+}

src/ruby.rs

@@ -2,7 +2,7 @@ mod bundler;
 mod command_executor;
 mod gemset;
 mod language_servers;
-use language_servers::{LanguageServer, Rubocop, RubyLsp, Solargraph};
+use language_servers::{LanguageServer, Rubocop, RubyLsp, Solargraph, Sorbet};
 
 use zed::lsp::{Completion, Symbol};
 use zed::settings::LspSettings;
@@ -14,6 +14,7 @@ struct RubyExtension {
     solargraph: Option<Solargraph>,
     ruby_lsp: Option<RubyLsp>,
     rubocop: Option<Rubocop>,
+    sorbet: Option<Sorbet>,
 }
 
 impl zed::Extension for RubyExtension {
@@ -39,6 +40,10 @@ impl zed::Extension for RubyExtension {
                 let rubocop = self.rubocop.get_or_insert_with(Rubocop::new);
                 rubocop.language_server_command(language_server_id, worktree)
             }
+            Sorbet::SERVER_ID => {
+                let sorbet = self.sorbet.get_or_insert_with(Sorbet::new);
+                sorbet.language_server_command(language_server_id, worktree)
+            }
             language_server_id => Err(format!("unknown language server: {language_server_id}")),
         }
     }