1use anyhow::{anyhow, Result};
2use async_trait::async_trait;
3use collections::HashMap;
4use gpui::AsyncAppContext;
5use language::{LanguageServerName, LspAdapter, LspAdapterDelegate};
6use lsp::{CodeActionKind, LanguageServerBinary};
7use node_runtime::NodeRuntime;
8use serde_json::{json, Value};
9use std::{
10 any::Any,
11 ffi::OsString,
12 path::{Path, PathBuf},
13 sync::Arc,
14};
15use util::{maybe, ResultExt};
16
17fn typescript_server_binary_arguments(server_path: &Path) -> Vec<OsString> {
18 vec![server_path.into(), "--stdio".into()]
19}
20
21pub struct VtslsLspAdapter {
22 node: Arc<dyn NodeRuntime>,
23}
24
25impl VtslsLspAdapter {
26 const SERVER_PATH: &'static str = "node_modules/@vtsls/language-server/bin/vtsls.js";
27
28 pub fn new(node: Arc<dyn NodeRuntime>) -> Self {
29 VtslsLspAdapter { node }
30 }
31}
32
33struct TypeScriptVersions {
34 typescript_version: String,
35 server_version: String,
36}
37
38#[async_trait(?Send)]
39impl LspAdapter for VtslsLspAdapter {
40 fn name(&self) -> LanguageServerName {
41 LanguageServerName("vtsls".into())
42 }
43
44 async fn fetch_latest_server_version(
45 &self,
46 _: &dyn LspAdapterDelegate,
47 ) -> Result<Box<dyn 'static + Send + Any>> {
48 Ok(Box::new(TypeScriptVersions {
49 typescript_version: self.node.npm_package_latest_version("typescript").await?,
50 server_version: self
51 .node
52 .npm_package_latest_version("@vtsls/language-server")
53 .await?,
54 }) as Box<_>)
55 }
56
57 async fn fetch_server_binary(
58 &self,
59 latest_version: Box<dyn 'static + Send + Any>,
60 container_dir: PathBuf,
61 _: &dyn LspAdapterDelegate,
62 ) -> Result<LanguageServerBinary> {
63 let latest_version = latest_version.downcast::<TypeScriptVersions>().unwrap();
64 let server_path = container_dir.join(Self::SERVER_PATH);
65 let package_name = "typescript";
66
67 let should_install_language_server = self
68 .node
69 .should_install_npm_package(
70 package_name,
71 &server_path,
72 &container_dir,
73 latest_version.typescript_version.as_str(),
74 )
75 .await;
76
77 if should_install_language_server {
78 self.node
79 .npm_install_packages(
80 &container_dir,
81 &[
82 (package_name, latest_version.typescript_version.as_str()),
83 (
84 "@vtsls/language-server",
85 latest_version.server_version.as_str(),
86 ),
87 ],
88 )
89 .await?;
90 }
91
92 Ok(LanguageServerBinary {
93 path: self.node.binary_path().await?,
94 env: None,
95 arguments: typescript_server_binary_arguments(&server_path),
96 })
97 }
98
99 async fn cached_server_binary(
100 &self,
101 container_dir: PathBuf,
102 _: &dyn LspAdapterDelegate,
103 ) -> Option<LanguageServerBinary> {
104 get_cached_ts_server_binary(container_dir, &*self.node).await
105 }
106
107 async fn installation_test_binary(
108 &self,
109 container_dir: PathBuf,
110 ) -> Option<LanguageServerBinary> {
111 get_cached_ts_server_binary(container_dir, &*self.node).await
112 }
113
114 fn code_action_kinds(&self) -> Option<Vec<CodeActionKind>> {
115 Some(vec![
116 CodeActionKind::QUICKFIX,
117 CodeActionKind::REFACTOR,
118 CodeActionKind::REFACTOR_EXTRACT,
119 CodeActionKind::SOURCE,
120 ])
121 }
122
123 async fn label_for_completion(
124 &self,
125 item: &lsp::CompletionItem,
126 language: &Arc<language::Language>,
127 ) -> Option<language::CodeLabel> {
128 use lsp::CompletionItemKind as Kind;
129 let len = item.label.len();
130 let grammar = language.grammar()?;
131 let highlight_id = match item.kind? {
132 Kind::CLASS | Kind::INTERFACE | Kind::ENUM => grammar.highlight_id_for_name("type"),
133 Kind::CONSTRUCTOR => grammar.highlight_id_for_name("type"),
134 Kind::CONSTANT => grammar.highlight_id_for_name("constant"),
135 Kind::FUNCTION | Kind::METHOD => grammar.highlight_id_for_name("function"),
136 Kind::PROPERTY | Kind::FIELD => grammar.highlight_id_for_name("property"),
137 Kind::VARIABLE => grammar.highlight_id_for_name("variable"),
138 _ => None,
139 }?;
140
141 let text = if let Some(description) = item
142 .label_details
143 .as_ref()
144 .and_then(|label_details| label_details.description.as_ref())
145 {
146 format!("{} {}", item.label, description)
147 } else if let Some(detail) = &item.detail {
148 format!("{} {}", item.label, detail)
149 } else {
150 item.label.clone()
151 };
152
153 Some(language::CodeLabel {
154 text,
155 runs: vec![(0..len, highlight_id)],
156 filter_range: 0..len,
157 })
158 }
159
160 async fn initialization_options(
161 self: Arc<Self>,
162 _: &Arc<dyn LspAdapterDelegate>,
163 ) -> Result<Option<serde_json::Value>> {
164 Ok(Some(json!({
165 "typescript": {
166 "tsdk": "node_modules/typescript/lib",
167 "format": {
168 "enable": true
169 },
170 "inlayHints": {
171 "parameterNames": {
172 "enabled": "all",
173 "suppressWhenArgumentMatchesName": false,
174 },
175 "parameterTypes": {
176 "enabled": true
177 },
178 "variableTypes": {
179 "enabled": true,
180 "suppressWhenTypeMatchesName": false,
181 },
182 "propertyDeclarationTypes": {
183 "enabled": true,
184 },
185 "functionLikeReturnTypes": {
186 "enabled": true,
187 },
188 "enumMemberValues": {
189 "enabled": true,
190 }
191 }
192 },
193 "vtsls": {
194 "experimental": {
195 "completion": {
196 "enableServerSideFuzzyMatch": true,
197 "entriesLimit": 5000,
198 }
199 }
200 }
201 })))
202 }
203
204 async fn workspace_configuration(
205 self: Arc<Self>,
206 _: &Arc<dyn LspAdapterDelegate>,
207 _cx: &mut AsyncAppContext,
208 ) -> Result<Value> {
209 Ok(json!({
210 "typescript": {
211 "suggest": {
212 "completeFunctionCalls": true
213 },
214 "tsdk": "node_modules/typescript/lib",
215 "format": {
216 "enable": true
217 },
218 "inlayHints": {
219 "parameterNames": {
220 "enabled": "all",
221 "suppressWhenArgumentMatchesName": false,
222 },
223 "parameterTypes": {
224 "enabled": true
225 },
226 "variableTypes": {
227 "enabled": true,
228 "suppressWhenTypeMatchesName": false,
229 },
230 "propertyDeclarationTypes": {
231 "enabled": true,
232 },
233 "functionLikeReturnTypes": {
234 "enabled": true,
235 },
236 "enumMemberValues": {
237 "enabled": true,
238 }
239 }
240 },
241 "vtsls": {
242 "experimental": {
243 "completion": {
244 "enableServerSideFuzzyMatch": true,
245 "entriesLimit": 5000,
246 }
247 }
248 }
249 }))
250 }
251
252 fn language_ids(&self) -> HashMap<String, String> {
253 HashMap::from_iter([
254 ("TypeScript".into(), "typescript".into()),
255 ("JavaScript".into(), "javascript".into()),
256 ("TSX".into(), "typescriptreact".into()),
257 ])
258 }
259}
260
261async fn get_cached_ts_server_binary(
262 container_dir: PathBuf,
263 node: &dyn NodeRuntime,
264) -> Option<LanguageServerBinary> {
265 maybe!(async {
266 let server_path = container_dir.join(VtslsLspAdapter::SERVER_PATH);
267 if server_path.exists() {
268 Ok(LanguageServerBinary {
269 path: node.binary_path().await?,
270 env: None,
271 arguments: typescript_server_binary_arguments(&server_path),
272 })
273 } else {
274 Err(anyhow!(
275 "missing executable in directory {:?}",
276 container_dir
277 ))
278 }
279 })
280 .await
281 .log_err()
282}