1use super::latest;
2use crate::wasm_host::wit::since_v0_0_4;
3use crate::wasm_host::WasmState;
4use anyhow::Result;
5use async_trait::async_trait;
6use language::{LanguageServerBinaryStatus, LspAdapterDelegate};
7use semantic_version::SemanticVersion;
8use std::sync::{Arc, OnceLock};
9use wasmtime::component::{Linker, Resource};
10
11pub const MIN_VERSION: SemanticVersion = SemanticVersion::new(0, 0, 1);
12
13wasmtime::component::bindgen!({
14 async: true,
15 path: "../extension_api/wit/since_v0.0.1",
16 with: {
17 "worktree": ExtensionWorktree,
18 "zed:extension/github": latest::zed::extension::github,
19 "zed:extension/platform": latest::zed::extension::platform,
20 },
21});
22
23pub type ExtensionWorktree = Arc<dyn LspAdapterDelegate>;
24
25pub fn linker() -> &'static Linker<WasmState> {
26 static LINKER: OnceLock<Linker<WasmState>> = OnceLock::new();
27 LINKER.get_or_init(|| super::new_linker(Extension::add_to_linker))
28}
29
30impl From<DownloadedFileType> for latest::DownloadedFileType {
31 fn from(value: DownloadedFileType) -> Self {
32 match value {
33 DownloadedFileType::Gzip => latest::DownloadedFileType::Gzip,
34 DownloadedFileType::GzipTar => latest::DownloadedFileType::GzipTar,
35 DownloadedFileType::Zip => latest::DownloadedFileType::Zip,
36 DownloadedFileType::Uncompressed => latest::DownloadedFileType::Uncompressed,
37 }
38 }
39}
40
41impl From<since_v0_0_4::LanguageServerConfig> for LanguageServerConfig {
42 fn from(value: since_v0_0_4::LanguageServerConfig) -> Self {
43 Self {
44 name: value.name,
45 language_name: value.language_name,
46 }
47 }
48}
49
50impl From<Command> for latest::Command {
51 fn from(value: Command) -> Self {
52 Self {
53 command: value.command,
54 args: value.args,
55 env: value.env,
56 }
57 }
58}
59
60#[async_trait]
61impl HostWorktree for WasmState {
62 async fn read_text_file(
63 &mut self,
64 delegate: Resource<Arc<dyn LspAdapterDelegate>>,
65 path: String,
66 ) -> wasmtime::Result<Result<String, String>> {
67 latest::HostWorktree::read_text_file(self, delegate, path).await
68 }
69
70 async fn shell_env(
71 &mut self,
72 delegate: Resource<Arc<dyn LspAdapterDelegate>>,
73 ) -> wasmtime::Result<EnvVars> {
74 latest::HostWorktree::shell_env(self, delegate).await
75 }
76
77 async fn which(
78 &mut self,
79 delegate: Resource<Arc<dyn LspAdapterDelegate>>,
80 binary_name: String,
81 ) -> wasmtime::Result<Option<String>> {
82 latest::HostWorktree::which(self, delegate, binary_name).await
83 }
84
85 fn drop(&mut self, _worktree: Resource<Worktree>) -> Result<()> {
86 Ok(())
87 }
88}
89
90#[async_trait]
91impl ExtensionImports for WasmState {
92 async fn node_binary_path(&mut self) -> wasmtime::Result<Result<String, String>> {
93 latest::nodejs::Host::node_binary_path(self).await
94 }
95
96 async fn npm_package_latest_version(
97 &mut self,
98 package_name: String,
99 ) -> wasmtime::Result<Result<String, String>> {
100 latest::nodejs::Host::npm_package_latest_version(self, package_name).await
101 }
102
103 async fn npm_package_installed_version(
104 &mut self,
105 package_name: String,
106 ) -> wasmtime::Result<Result<Option<String>, String>> {
107 latest::nodejs::Host::npm_package_installed_version(self, package_name).await
108 }
109
110 async fn npm_install_package(
111 &mut self,
112 package_name: String,
113 version: String,
114 ) -> wasmtime::Result<Result<(), String>> {
115 latest::nodejs::Host::npm_install_package(self, package_name, version).await
116 }
117
118 async fn latest_github_release(
119 &mut self,
120 repo: String,
121 options: GithubReleaseOptions,
122 ) -> wasmtime::Result<Result<GithubRelease, String>> {
123 latest::zed::extension::github::Host::latest_github_release(self, repo, options).await
124 }
125
126 async fn current_platform(&mut self) -> Result<(Os, Architecture)> {
127 latest::zed::extension::platform::Host::current_platform(self).await
128 }
129
130 async fn set_language_server_installation_status(
131 &mut self,
132 server_name: String,
133 status: LanguageServerInstallationStatus,
134 ) -> wasmtime::Result<()> {
135 let status = match status {
136 LanguageServerInstallationStatus::CheckingForUpdate => {
137 LanguageServerBinaryStatus::CheckingForUpdate
138 }
139 LanguageServerInstallationStatus::Downloading => {
140 LanguageServerBinaryStatus::Downloading
141 }
142 LanguageServerInstallationStatus::Cached
143 | LanguageServerInstallationStatus::Downloaded => LanguageServerBinaryStatus::None,
144 LanguageServerInstallationStatus::Failed(error) => {
145 LanguageServerBinaryStatus::Failed { error }
146 }
147 };
148
149 self.host
150 .language_registry
151 .update_lsp_status(language::LanguageServerName(server_name.into()), status);
152 Ok(())
153 }
154
155 async fn download_file(
156 &mut self,
157 url: String,
158 path: String,
159 file_type: DownloadedFileType,
160 ) -> wasmtime::Result<Result<(), String>> {
161 latest::ExtensionImports::download_file(self, url, path, file_type.into()).await
162 }
163}