1use std::fs;
2use zed::LanguageServerId;
3use zed_extension_api::{self as zed, Result};
4
5struct ZigExtension {
6 cached_binary_path: Option<String>,
7}
8
9#[derive(Clone)]
10struct ZlsBinary {
11 path: String,
12 environment: Option<Vec<(String, String)>>,
13}
14
15impl ZigExtension {
16 fn language_server_binary(
17 &mut self,
18 language_server_id: &LanguageServerId,
19 worktree: &zed::Worktree,
20 ) -> Result<ZlsBinary> {
21 if let Some(path) = worktree.which("zls") {
22 let environment = worktree.shell_env();
23 return Ok(ZlsBinary {
24 path,
25 environment: Some(environment),
26 });
27 }
28
29 if let Some(path) = &self.cached_binary_path {
30 if fs::metadata(&path).map_or(false, |stat| stat.is_file()) {
31 return Ok(ZlsBinary {
32 path: path.clone(),
33 environment: None,
34 });
35 }
36 }
37
38 zed::set_language_server_installation_status(
39 &language_server_id,
40 &zed::LanguageServerInstallationStatus::CheckingForUpdate,
41 );
42 // We're pinning ZLS to a release that has `.tar.gz` assets, since the latest release does not have
43 // them, at time of writing.
44 //
45 // ZLS tracking issue: https://github.com/zigtools/zls/issues/1879
46 // let release = zed::github_release_by_tag_name("zigtools/zls", "0.11.0")?;
47
48 let release = zed::latest_github_release(
49 "zigtools/zls",
50 zed::GithubReleaseOptions {
51 require_assets: true,
52 pre_release: false,
53 },
54 )?;
55
56 let (platform, arch) = zed::current_platform();
57 let asset_name = format!(
58 "zls-{arch}-{os}.{extension}",
59 arch = match arch {
60 zed::Architecture::Aarch64 => "aarch64",
61 zed::Architecture::X86 => "x86",
62 zed::Architecture::X8664 => "x86_64",
63 },
64 os = match platform {
65 zed::Os::Mac => "macos",
66 zed::Os::Linux => "linux",
67 zed::Os::Windows => "windows",
68 },
69 extension = match platform {
70 zed::Os::Mac | zed::Os::Linux => "tar.gz",
71 zed::Os::Windows => "zip",
72 }
73 );
74
75 let asset = release
76 .assets
77 .iter()
78 .find(|asset| asset.name == asset_name)
79 .ok_or_else(|| format!("no asset found matching {:?}", asset_name))?;
80
81 let version_dir = format!("zls-{}", release.version);
82 let binary_path = format!("{version_dir}/bin/zls");
83
84 if !fs::metadata(&binary_path).map_or(false, |stat| stat.is_file()) {
85 zed::set_language_server_installation_status(
86 &language_server_id,
87 &zed::LanguageServerInstallationStatus::Downloading,
88 );
89
90 zed::download_file(
91 &asset.download_url,
92 &version_dir,
93 match platform {
94 zed::Os::Mac | zed::Os::Linux => zed::DownloadedFileType::GzipTar,
95 zed::Os::Windows => zed::DownloadedFileType::Zip,
96 },
97 )
98 .map_err(|e| format!("failed to download file: {e}"))?;
99
100 zed::make_file_executable(&binary_path)?;
101
102 let entries =
103 fs::read_dir(".").map_err(|e| format!("failed to list working directory {e}"))?;
104 for entry in entries {
105 let entry = entry.map_err(|e| format!("failed to load directory entry {e}"))?;
106 if entry.file_name().to_str() != Some(&version_dir) {
107 fs::remove_dir_all(&entry.path()).ok();
108 }
109 }
110 }
111
112 self.cached_binary_path = Some(binary_path.clone());
113 Ok(ZlsBinary {
114 path: binary_path,
115 environment: None,
116 })
117 }
118}
119
120impl zed::Extension for ZigExtension {
121 fn new() -> Self {
122 Self {
123 cached_binary_path: None,
124 }
125 }
126
127 fn language_server_command(
128 &mut self,
129 language_server_id: &LanguageServerId,
130 worktree: &zed::Worktree,
131 ) -> Result<zed::Command> {
132 let zls_binary = self.language_server_binary(language_server_id, worktree)?;
133 Ok(zed::Command {
134 command: zls_binary.path,
135 args: vec![],
136 env: zls_binary.environment.unwrap_or_default(),
137 })
138 }
139}
140
141zed::register_extension!(ZigExtension);