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