1use serde_json::json;
  2use std::fs;
  3use zed::LanguageServerId;
  4use zed_extension_api::{self as zed, Result, settings::LspSettings};
  5
  6struct SnippetExtension {
  7    cached_binary_path: Option<String>,
  8}
  9
 10impl SnippetExtension {
 11    fn language_server_binary_path(
 12        &mut self,
 13        language_server_id: &LanguageServerId,
 14        worktree: &zed::Worktree,
 15    ) -> Result<String> {
 16        if let Some(path) = worktree.which("simple-completion-language-server") {
 17            return Ok(path);
 18        }
 19
 20        if let Some(path) = &self.cached_binary_path
 21            && fs::metadata(path).map_or(false, |stat| stat.is_file()) {
 22                return Ok(path.clone());
 23            }
 24
 25        zed::set_language_server_installation_status(
 26            language_server_id,
 27            &zed::LanguageServerInstallationStatus::CheckingForUpdate,
 28        );
 29        let release = zed::latest_github_release(
 30            "zed-industries/simple-completion-language-server",
 31            zed::GithubReleaseOptions {
 32                require_assets: true,
 33                pre_release: false,
 34            },
 35        )?;
 36
 37        let (platform, arch) = zed::current_platform();
 38        let asset_name = format!(
 39            "simple-completion-language-server-{arch}-{os}.tar.gz",
 40            arch = match arch {
 41                zed::Architecture::Aarch64 => "aarch64",
 42                zed::Architecture::X86 => "x86",
 43                zed::Architecture::X8664 => "x86_64",
 44            },
 45            os = match platform {
 46                zed::Os::Mac => "apple-darwin",
 47                zed::Os::Linux => "unknown-linux-gnu",
 48                zed::Os::Windows => "pc-windows-msvc",
 49            },
 50        );
 51
 52        let asset = release
 53            .assets
 54            .iter()
 55            .find(|asset| asset.name == asset_name)
 56            .ok_or_else(|| format!("no asset found matching {:?}", asset_name))?;
 57
 58        let version_dir = format!("simple-completion-language-server-{}", release.version);
 59        let binary_path = format!("{version_dir}/simple-completion-language-server");
 60
 61        if !fs::metadata(&binary_path).map_or(false, |stat| stat.is_file()) {
 62            zed::set_language_server_installation_status(
 63                language_server_id,
 64                &zed::LanguageServerInstallationStatus::Downloading,
 65            );
 66
 67            zed::download_file(
 68                &asset.download_url,
 69                &version_dir,
 70                zed::DownloadedFileType::GzipTar,
 71            )
 72            .map_err(|e| format!("failed to download file: {e}"))?;
 73
 74            let entries =
 75                fs::read_dir(".").map_err(|e| format!("failed to list working directory {e}"))?;
 76            for entry in entries {
 77                let entry = entry.map_err(|e| format!("failed to load directory entry {e}"))?;
 78                if entry.file_name().to_str() != Some(&version_dir) {
 79                    fs::remove_dir_all(entry.path()).ok();
 80                }
 81            }
 82        }
 83
 84        self.cached_binary_path = Some(binary_path.clone());
 85        Ok(binary_path)
 86    }
 87}
 88
 89impl zed::Extension for SnippetExtension {
 90    fn new() -> Self {
 91        Self {
 92            cached_binary_path: None,
 93        }
 94    }
 95
 96    fn language_server_command(
 97        &mut self,
 98        language_server_id: &LanguageServerId,
 99        worktree: &zed::Worktree,
100    ) -> Result<zed::Command> {
101        Ok(zed::Command {
102            command: self.language_server_binary_path(language_server_id, worktree)?,
103            args: vec![],
104            env: vec![("SCLS_CONFIG_SUBDIRECTORY".to_owned(), "zed".to_owned())],
105        })
106    }
107
108    fn language_server_workspace_configuration(
109        &mut self,
110        server_id: &LanguageServerId,
111        worktree: &zed_extension_api::Worktree,
112    ) -> Result<Option<zed_extension_api::serde_json::Value>> {
113        let settings = LspSettings::for_worktree(server_id.as_ref(), worktree)
114            .ok()
115            .and_then(|lsp_settings| lsp_settings.settings.clone())
116            .unwrap_or_else(|| {
117                json!({
118                    "max_completion_items": 20,
119                    "snippets_first": true,
120                    "feature_words": false,
121                    "feature_snippets": true,
122                    "feature_paths": true
123                })
124            });
125        Ok(Some(settings))
126    }
127}
128
129zed::register_extension!(SnippetExtension);