1use super::{latest, since_v0_1_0};
2use crate::wasm_host::WasmState;
3use anyhow::Result;
4use async_trait::async_trait;
5use language::LspAdapterDelegate;
6use semantic_version::SemanticVersion;
7use std::sync::{Arc, OnceLock};
8use wasmtime::component::{Linker, Resource};
9
10pub const MIN_VERSION: SemanticVersion = SemanticVersion::new(0, 0, 6);
11
12wasmtime::component::bindgen!({
13 async: true,
14 trappable_imports: true,
15 path: "../extension_api/wit/since_v0.0.6",
16 with: {
17 "worktree": ExtensionWorktree,
18 "zed:extension/github": latest::zed::extension::github,
19 "zed:extension/lsp": since_v0_1_0::zed::extension::lsp,
20 "zed:extension/nodejs": latest::zed::extension::nodejs,
21 "zed:extension/platform": latest::zed::extension::platform,
22 },
23});
24
25mod settings {
26 include!(concat!(env!("OUT_DIR"), "/since_v0.0.6/settings.rs"));
27}
28
29pub type ExtensionWorktree = Arc<dyn LspAdapterDelegate>;
30
31pub fn linker() -> &'static Linker<WasmState> {
32 static LINKER: OnceLock<Linker<WasmState>> = OnceLock::new();
33 LINKER.get_or_init(|| super::new_linker(Extension::add_to_linker))
34}
35
36impl From<Command> for latest::Command {
37 fn from(value: Command) -> Self {
38 Self {
39 command: value.command,
40 args: value.args,
41 env: value.env,
42 }
43 }
44}
45
46impl From<SettingsLocation> for latest::SettingsLocation {
47 fn from(value: SettingsLocation) -> Self {
48 Self {
49 worktree_id: value.worktree_id,
50 path: value.path,
51 }
52 }
53}
54
55impl From<LanguageServerInstallationStatus> for latest::LanguageServerInstallationStatus {
56 fn from(value: LanguageServerInstallationStatus) -> Self {
57 match value {
58 LanguageServerInstallationStatus::None => Self::None,
59 LanguageServerInstallationStatus::Downloading => Self::Downloading,
60 LanguageServerInstallationStatus::CheckingForUpdate => Self::CheckingForUpdate,
61 LanguageServerInstallationStatus::Failed(message) => Self::Failed(message),
62 }
63 }
64}
65
66impl From<DownloadedFileType> for latest::DownloadedFileType {
67 fn from(value: DownloadedFileType) -> Self {
68 match value {
69 DownloadedFileType::Gzip => Self::Gzip,
70 DownloadedFileType::GzipTar => Self::GzipTar,
71 DownloadedFileType::Zip => Self::Zip,
72 DownloadedFileType::Uncompressed => Self::Uncompressed,
73 }
74 }
75}
76
77impl From<Range> for latest::Range {
78 fn from(value: Range) -> Self {
79 Self {
80 start: value.start,
81 end: value.end,
82 }
83 }
84}
85
86impl From<CodeLabelSpan> for latest::CodeLabelSpan {
87 fn from(value: CodeLabelSpan) -> Self {
88 match value {
89 CodeLabelSpan::CodeRange(range) => Self::CodeRange(range.into()),
90 CodeLabelSpan::Literal(literal) => Self::Literal(literal.into()),
91 }
92 }
93}
94
95impl From<CodeLabelSpanLiteral> for latest::CodeLabelSpanLiteral {
96 fn from(value: CodeLabelSpanLiteral) -> Self {
97 Self {
98 text: value.text,
99 highlight_name: value.highlight_name,
100 }
101 }
102}
103
104impl From<CodeLabel> for latest::CodeLabel {
105 fn from(value: CodeLabel) -> Self {
106 Self {
107 code: value.code,
108 spans: value.spans.into_iter().map(Into::into).collect(),
109 filter_range: value.filter_range.into(),
110 }
111 }
112}
113
114#[async_trait]
115impl HostWorktree for WasmState {
116 async fn id(
117 &mut self,
118 delegate: Resource<Arc<dyn LspAdapterDelegate>>,
119 ) -> wasmtime::Result<u64> {
120 latest::HostWorktree::id(self, delegate).await
121 }
122
123 async fn root_path(
124 &mut self,
125 delegate: Resource<Arc<dyn LspAdapterDelegate>>,
126 ) -> wasmtime::Result<String> {
127 latest::HostWorktree::root_path(self, delegate).await
128 }
129
130 async fn read_text_file(
131 &mut self,
132 delegate: Resource<Arc<dyn LspAdapterDelegate>>,
133 path: String,
134 ) -> wasmtime::Result<Result<String, String>> {
135 latest::HostWorktree::read_text_file(self, delegate, path).await
136 }
137
138 async fn shell_env(
139 &mut self,
140 delegate: Resource<Arc<dyn LspAdapterDelegate>>,
141 ) -> wasmtime::Result<EnvVars> {
142 latest::HostWorktree::shell_env(self, delegate).await
143 }
144
145 async fn which(
146 &mut self,
147 delegate: Resource<Arc<dyn LspAdapterDelegate>>,
148 binary_name: String,
149 ) -> wasmtime::Result<Option<String>> {
150 latest::HostWorktree::which(self, delegate, binary_name).await
151 }
152
153 fn drop(&mut self, _worktree: Resource<Worktree>) -> Result<()> {
154 // We only ever hand out borrows of worktrees.
155 Ok(())
156 }
157}
158
159#[async_trait]
160impl ExtensionImports for WasmState {
161 async fn get_settings(
162 &mut self,
163 location: Option<self::SettingsLocation>,
164 category: String,
165 key: Option<String>,
166 ) -> wasmtime::Result<Result<String, String>> {
167 latest::ExtensionImports::get_settings(
168 self,
169 location.map(|location| location.into()),
170 category,
171 key,
172 )
173 .await
174 }
175
176 async fn set_language_server_installation_status(
177 &mut self,
178 server_name: String,
179 status: LanguageServerInstallationStatus,
180 ) -> wasmtime::Result<()> {
181 latest::ExtensionImports::set_language_server_installation_status(
182 self,
183 server_name,
184 status.into(),
185 )
186 .await
187 }
188
189 async fn download_file(
190 &mut self,
191 url: String,
192 path: String,
193 file_type: DownloadedFileType,
194 ) -> wasmtime::Result<Result<(), String>> {
195 latest::ExtensionImports::download_file(self, url, path, file_type.into()).await
196 }
197
198 async fn make_file_executable(&mut self, path: String) -> wasmtime::Result<Result<(), String>> {
199 latest::ExtensionImports::make_file_executable(self, path).await
200 }
201}