1use std::fs;
2use zed::LanguageServerId;
3use zed_extension_api::{self as zed, Result, settings::LspSettings};
4
5struct RuffBinary {
6 path: String,
7 args: Option<Vec<String>>,
8}
9
10struct RuffExtension {
11 cached_binary_path: Option<String>,
12}
13
14impl RuffExtension {
15 fn language_server_binary(
16 &mut self,
17 language_server_id: &LanguageServerId,
18 worktree: &zed::Worktree,
19 ) -> Result<RuffBinary> {
20 let binary_settings = LspSettings::for_worktree("ruff", worktree)
21 .ok()
22 .and_then(|lsp_settings| lsp_settings.binary);
23 let binary_args = binary_settings
24 .as_ref()
25 .and_then(|binary_settings| binary_settings.arguments.clone());
26
27 if let Some(path) = binary_settings.and_then(|binary_settings| binary_settings.path) {
28 return Ok(RuffBinary {
29 path,
30 args: binary_args,
31 });
32 }
33
34 if let Some(path) = worktree.which("ruff") {
35 return Ok(RuffBinary {
36 path,
37 args: binary_args,
38 });
39 }
40
41 if let Some(path) = &self.cached_binary_path
42 && fs::metadata(path).map_or(false, |stat| stat.is_file()) {
43 return Ok(RuffBinary {
44 path: path.clone(),
45 args: binary_args,
46 });
47 }
48
49 zed::set_language_server_installation_status(
50 language_server_id,
51 &zed::LanguageServerInstallationStatus::CheckingForUpdate,
52 );
53 let release = zed::latest_github_release(
54 "astral-sh/ruff",
55 zed::GithubReleaseOptions {
56 require_assets: true,
57 pre_release: false,
58 },
59 )?;
60
61 let (platform, arch) = zed::current_platform();
62
63 let asset_stem = format!(
64 "ruff-{arch}-{os}",
65 arch = match arch {
66 zed::Architecture::Aarch64 => "aarch64",
67 zed::Architecture::X86 => "x86",
68 zed::Architecture::X8664 => "x86_64",
69 },
70 os = match platform {
71 zed::Os::Mac => "apple-darwin",
72 zed::Os::Linux => "unknown-linux-gnu",
73 zed::Os::Windows => "pc-windows-msvc",
74 }
75 );
76 let asset_name = format!(
77 "{asset_stem}.{suffix}",
78 suffix = match platform {
79 zed::Os::Windows => "zip",
80 _ => "tar.gz",
81 }
82 );
83
84 let asset = release
85 .assets
86 .iter()
87 .find(|asset| asset.name == asset_name)
88 .ok_or_else(|| format!("no asset found matching {:?}", asset_name))?;
89
90 let version_dir = format!("ruff-{}", release.version);
91 let binary_path = match platform {
92 zed::Os::Windows => format!("{version_dir}/ruff.exe"),
93 _ => format!("{version_dir}/{asset_stem}/ruff"),
94 };
95
96 if !fs::metadata(&binary_path).map_or(false, |stat| stat.is_file()) {
97 zed::set_language_server_installation_status(
98 language_server_id,
99 &zed::LanguageServerInstallationStatus::Downloading,
100 );
101 let file_kind = match platform {
102 zed::Os::Windows => zed::DownloadedFileType::Zip,
103 _ => zed::DownloadedFileType::GzipTar,
104 };
105 zed::download_file(&asset.download_url, &version_dir, file_kind)
106 .map_err(|e| format!("failed to download file: {e}"))?;
107
108 let entries =
109 fs::read_dir(".").map_err(|e| format!("failed to list working directory {e}"))?;
110 for entry in entries {
111 let entry = entry.map_err(|e| format!("failed to load directory entry {e}"))?;
112 if entry.file_name().to_str() != Some(&version_dir) {
113 fs::remove_dir_all(entry.path()).ok();
114 }
115 }
116 }
117
118 self.cached_binary_path = Some(binary_path.clone());
119 Ok(RuffBinary {
120 path: binary_path,
121 args: binary_args,
122 })
123 }
124}
125
126impl zed::Extension for RuffExtension {
127 fn new() -> Self {
128 Self {
129 cached_binary_path: None,
130 }
131 }
132
133 fn language_server_command(
134 &mut self,
135 language_server_id: &LanguageServerId,
136 worktree: &zed::Worktree,
137 ) -> Result<zed::Command> {
138 let ruff_binary = self.language_server_binary(language_server_id, worktree)?;
139 Ok(zed::Command {
140 command: ruff_binary.path,
141 args: ruff_binary.args.unwrap_or_else(|| vec!["server".into()]),
142 env: vec![],
143 })
144 }
145
146 fn language_server_initialization_options(
147 &mut self,
148 server_id: &LanguageServerId,
149 worktree: &zed_extension_api::Worktree,
150 ) -> Result<Option<zed_extension_api::serde_json::Value>> {
151 let settings = LspSettings::for_worktree(server_id.as_ref(), worktree)
152 .ok()
153 .and_then(|lsp_settings| lsp_settings.initialization_options.clone())
154 .unwrap_or_default();
155 Ok(Some(settings))
156 }
157
158 fn language_server_workspace_configuration(
159 &mut self,
160 server_id: &LanguageServerId,
161 worktree: &zed_extension_api::Worktree,
162 ) -> Result<Option<zed_extension_api::serde_json::Value>> {
163 let settings = LspSettings::for_worktree(server_id.as_ref(), worktree)
164 .ok()
165 .and_then(|lsp_settings| lsp_settings.settings.clone())
166 .unwrap_or_default();
167 Ok(Some(settings))
168 }
169}
170
171zed::register_extension!(RuffExtension);