1mod language_servers;
 2
 3use zed_extension_api::{self as zed, Result};
 4
 5use crate::language_servers::{ErlangLanguagePlatform, ErlangLs};
 6
 7struct ErlangExtension {
 8    erlang_ls: Option<ErlangLs>,
 9    erlang_language_platform: Option<ErlangLanguagePlatform>,
10}
11
12impl zed::Extension for ErlangExtension {
13    fn new() -> Self {
14        Self {
15            erlang_ls: None,
16            erlang_language_platform: None,
17        }
18    }
19
20    fn language_server_command(
21        &mut self,
22        language_server_id: &zed::LanguageServerId,
23        worktree: &zed::Worktree,
24    ) -> Result<zed::Command> {
25        match language_server_id.as_ref() {
26            ErlangLs::LANGUAGE_SERVER_ID => {
27                let erlang_ls = self.erlang_ls.get_or_insert_with(ErlangLs::new);
28
29                Ok(zed::Command {
30                    command: erlang_ls.language_server_binary_path(language_server_id, worktree)?,
31                    args: vec![],
32                    env: Default::default(),
33                })
34            }
35            ErlangLanguagePlatform::LANGUAGE_SERVER_ID => {
36                let erlang_language_platform = self
37                    .erlang_language_platform
38                    .get_or_insert_with(ErlangLanguagePlatform::new);
39                erlang_language_platform.language_server_command(language_server_id, worktree)
40            }
41            language_server_id => Err(format!("unknown language server: {language_server_id}")),
42        }
43    }
44}
45
46zed::register_extension!(ErlangExtension);