1use anyhow::Context;
2use gpui::AppContext;
3pub use language::*;
4use node_runtime::NodeRuntime;
5use rust_embed::RustEmbed;
6use settings::Settings;
7use std::{str, sync::Arc};
8use util::asset_str;
9
10use crate::{elixir::elixir_task_context, rust::RustContextProvider};
11
12use self::{deno::DenoSettings, elixir::ElixirSettings};
13
14mod astro;
15mod c;
16mod clojure;
17mod csharp;
18mod css;
19mod dart;
20mod deno;
21mod dockerfile;
22mod elixir;
23mod elm;
24mod erlang;
25mod go;
26mod haskell;
27mod html;
28mod json;
29mod lua;
30mod nu;
31mod ocaml;
32mod php;
33mod prisma;
34mod purescript;
35mod python;
36mod ruby;
37mod rust;
38mod tailwind;
39mod terraform;
40mod toml;
41mod typescript;
42mod vue;
43mod yaml;
44mod zig;
45
46// 1. Add tree-sitter-{language} parser to zed crate
47// 2. Create a language directory in zed/crates/zed/src/languages and add the language to init function below
48// 3. Add config.toml to the newly created language directory using existing languages as a template
49// 4. Copy highlights from tree sitter repo for the language into a highlights.scm file.
50// Note: github highlights take the last match while zed takes the first
51// 5. Add indents.scm, outline.scm, and brackets.scm to implement indent on newline, outline/breadcrumbs,
52// and autoclosing brackets respectively
53// 6. If the language has injections add an injections.scm query file
54
55#[derive(RustEmbed)]
56#[folder = "src/"]
57#[exclude = "*.rs"]
58struct LanguageDir;
59
60pub fn init(
61 languages: Arc<LanguageRegistry>,
62 node_runtime: Arc<dyn NodeRuntime>,
63 cx: &mut AppContext,
64) {
65 ElixirSettings::register(cx);
66 DenoSettings::register(cx);
67
68 languages.register_native_grammars([
69 ("astro", tree_sitter_astro::language()),
70 ("bash", tree_sitter_bash::language()),
71 ("c", tree_sitter_c::language()),
72 ("c_sharp", tree_sitter_c_sharp::language()),
73 ("clojure", tree_sitter_clojure::language()),
74 ("cpp", tree_sitter_cpp::language()),
75 ("css", tree_sitter_css::language()),
76 ("dockerfile", tree_sitter_dockerfile::language()),
77 ("elixir", tree_sitter_elixir::language()),
78 ("elm", tree_sitter_elm::language()),
79 (
80 "embedded_template",
81 tree_sitter_embedded_template::language(),
82 ),
83 ("erlang", tree_sitter_erlang::language()),
84 ("glsl", tree_sitter_glsl::language()),
85 ("go", tree_sitter_go::language()),
86 ("gomod", tree_sitter_gomod::language()),
87 ("gowork", tree_sitter_gowork::language()),
88 ("haskell", tree_sitter_haskell::language()),
89 ("hcl", tree_sitter_hcl::language()),
90 ("heex", tree_sitter_heex::language()),
91 ("html", tree_sitter_html::language()),
92 ("jsdoc", tree_sitter_jsdoc::language()),
93 ("json", tree_sitter_json::language()),
94 ("lua", tree_sitter_lua::language()),
95 ("markdown", tree_sitter_markdown::language()),
96 ("nix", tree_sitter_nix::language()),
97 ("nu", tree_sitter_nu::language()),
98 ("ocaml", tree_sitter_ocaml::language_ocaml()),
99 (
100 "ocaml_interface",
101 tree_sitter_ocaml::language_ocaml_interface(),
102 ),
103 ("php", tree_sitter_php::language_php()),
104 ("prisma", tree_sitter_prisma_io::language()),
105 ("proto", tree_sitter_proto::language()),
106 ("purescript", tree_sitter_purescript::language()),
107 ("python", tree_sitter_python::language()),
108 ("racket", tree_sitter_racket::language()),
109 ("regex", tree_sitter_regex::language()),
110 ("ruby", tree_sitter_ruby::language()),
111 ("rust", tree_sitter_rust::language()),
112 ("scheme", tree_sitter_scheme::language()),
113 ("toml", tree_sitter_toml::language()),
114 ("tsx", tree_sitter_typescript::language_tsx()),
115 ("typescript", tree_sitter_typescript::language_typescript()),
116 ("vue", tree_sitter_vue::language()),
117 ("yaml", tree_sitter_yaml::language()),
118 ("zig", tree_sitter_zig::language()),
119 ("dart", tree_sitter_dart::language()),
120 ]);
121
122 macro_rules! language {
123 ($name:literal) => {
124 let config = load_config($name);
125 languages.register_language(
126 config.name.clone(),
127 config.grammar.clone(),
128 config.matcher.clone(),
129 move || {
130 Ok((
131 config.clone(),
132 load_queries($name),
133 Some(Arc::new(language::SymbolContextProvider)),
134 ))
135 },
136 );
137 };
138 ($name:literal, $adapters:expr) => {
139 let config = load_config($name);
140 // typeck helper
141 let adapters: Vec<Arc<dyn LspAdapter>> = $adapters;
142 for adapter in adapters {
143 languages.register_lsp_adapter(config.name.clone(), adapter);
144 }
145 languages.register_language(
146 config.name.clone(),
147 config.grammar.clone(),
148 config.matcher.clone(),
149 move || {
150 Ok((
151 config.clone(),
152 load_queries($name),
153 Some(Arc::new(language::SymbolContextProvider)),
154 ))
155 },
156 );
157 };
158 ($name:literal, $adapters:expr, $context_provider:expr) => {
159 let config = load_config($name);
160 // typeck helper
161 let adapters: Vec<Arc<dyn LspAdapter>> = $adapters;
162 for adapter in adapters {
163 languages.register_lsp_adapter(config.name.clone(), adapter);
164 }
165 languages.register_language(
166 config.name.clone(),
167 config.grammar.clone(),
168 config.matcher.clone(),
169 move || {
170 Ok((
171 config.clone(),
172 load_queries($name),
173 Some(Arc::new($context_provider)),
174 ))
175 },
176 );
177 };
178 }
179 language!(
180 "astro",
181 vec![
182 Arc::new(astro::AstroLspAdapter::new(node_runtime.clone())),
183 Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
184 ]
185 );
186 language!("bash");
187 language!("c", vec![Arc::new(c::CLspAdapter) as Arc<dyn LspAdapter>]);
188 language!("clojure", vec![Arc::new(clojure::ClojureLspAdapter)]);
189 language!("cpp", vec![Arc::new(c::CLspAdapter)]);
190 language!("csharp", vec![Arc::new(csharp::OmniSharpAdapter {})]);
191 language!(
192 "css",
193 vec![
194 Arc::new(css::CssLspAdapter::new(node_runtime.clone())),
195 Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
196 ]
197 );
198
199 language!(
200 "dockerfile",
201 vec![Arc::new(dockerfile::DockerfileLspAdapter::new(
202 node_runtime.clone(),
203 ))]
204 );
205
206 match &ElixirSettings::get(None, cx).lsp {
207 elixir::ElixirLspSetting::ElixirLs => {
208 language!(
209 "elixir",
210 vec![
211 Arc::new(elixir::ElixirLspAdapter),
212 Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
213 ],
214 elixir_task_context()
215 );
216 }
217 elixir::ElixirLspSetting::NextLs => {
218 language!(
219 "elixir",
220 vec![Arc::new(elixir::NextLspAdapter)],
221 elixir_task_context()
222 );
223 }
224 elixir::ElixirLspSetting::Local { path, arguments } => {
225 language!(
226 "elixir",
227 vec![Arc::new(elixir::LocalLspAdapter {
228 path: path.clone(),
229 arguments: arguments.clone(),
230 })],
231 elixir_task_context()
232 );
233 }
234 }
235 language!("erlang", vec![Arc::new(erlang::ErlangLspAdapter)]);
236 language!("go", vec![Arc::new(go::GoLspAdapter)]);
237 language!("gomod");
238 language!("gowork");
239 language!("zig", vec![Arc::new(zig::ZlsAdapter)]);
240 language!(
241 "heex",
242 vec![
243 Arc::new(elixir::ElixirLspAdapter),
244 Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
245 ]
246 );
247 language!(
248 "json",
249 vec![Arc::new(json::JsonLspAdapter::new(
250 node_runtime.clone(),
251 languages.clone(),
252 ))]
253 );
254 language!("markdown");
255 language!(
256 "python",
257 vec![Arc::new(python::PythonLspAdapter::new(
258 node_runtime.clone(),
259 ))]
260 );
261 language!(
262 "rust",
263 vec![Arc::new(rust::RustLspAdapter)],
264 RustContextProvider
265 );
266 language!("toml", vec![Arc::new(toml::TaploLspAdapter)]);
267 match &DenoSettings::get(None, cx).enable {
268 true => {
269 language!(
270 "tsx",
271 vec![
272 Arc::new(deno::DenoLspAdapter::new()),
273 Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
274 ]
275 );
276 language!("typescript", vec![Arc::new(deno::DenoLspAdapter::new())]);
277 language!(
278 "javascript",
279 vec![
280 Arc::new(deno::DenoLspAdapter::new()),
281 Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
282 ]
283 );
284 language!("jsdoc", vec![Arc::new(deno::DenoLspAdapter::new())]);
285 }
286 false => {
287 language!(
288 "tsx",
289 vec![
290 Arc::new(typescript::TypeScriptLspAdapter::new(node_runtime.clone())),
291 Arc::new(typescript::EsLintLspAdapter::new(node_runtime.clone())),
292 Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
293 ]
294 );
295 language!(
296 "typescript",
297 vec![
298 Arc::new(typescript::TypeScriptLspAdapter::new(node_runtime.clone())),
299 Arc::new(typescript::EsLintLspAdapter::new(node_runtime.clone())),
300 ]
301 );
302 language!(
303 "javascript",
304 vec![
305 Arc::new(typescript::TypeScriptLspAdapter::new(node_runtime.clone())),
306 Arc::new(typescript::EsLintLspAdapter::new(node_runtime.clone())),
307 Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
308 ]
309 );
310 language!(
311 "jsdoc",
312 vec![Arc::new(typescript::TypeScriptLspAdapter::new(
313 node_runtime.clone(),
314 ))]
315 );
316 }
317 }
318
319 language!("haskell", vec![Arc::new(haskell::HaskellLanguageServer {})]);
320 language!(
321 "html",
322 vec![
323 Arc::new(html::HtmlLspAdapter::new(node_runtime.clone())),
324 Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
325 ]
326 );
327 language!("ruby", vec![Arc::new(ruby::RubyLanguageServer)]);
328 language!(
329 "erb",
330 vec![
331 Arc::new(ruby::RubyLanguageServer),
332 Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
333 ]
334 );
335 language!("scheme");
336 language!("racket");
337 language!("regex");
338 language!("lua", vec![Arc::new(lua::LuaLspAdapter)]);
339 language!(
340 "yaml",
341 vec![Arc::new(yaml::YamlLspAdapter::new(node_runtime.clone()))]
342 );
343 language!(
344 "php",
345 vec![
346 Arc::new(php::IntelephenseLspAdapter::new(node_runtime.clone())),
347 Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
348 ]
349 );
350 language!(
351 "purescript",
352 vec![Arc::new(purescript::PurescriptLspAdapter::new(
353 node_runtime.clone(),
354 ))]
355 );
356 language!(
357 "elm",
358 vec![Arc::new(elm::ElmLspAdapter::new(node_runtime.clone()))]
359 );
360 language!("glsl");
361 language!("nix");
362 language!("nu", vec![Arc::new(nu::NuLanguageServer {})]);
363 language!("ocaml", vec![Arc::new(ocaml::OCamlLspAdapter)]);
364 language!("ocaml-interface", vec![Arc::new(ocaml::OCamlLspAdapter)]);
365 language!(
366 "vue",
367 vec![Arc::new(vue::VueLspAdapter::new(node_runtime.clone()))]
368 );
369 language!("proto");
370 language!("terraform", vec![Arc::new(terraform::TerraformLspAdapter)]);
371 language!(
372 "terraform-vars",
373 vec![Arc::new(terraform::TerraformLspAdapter)]
374 );
375 language!("hcl", vec![]);
376 language!(
377 "prisma",
378 vec![Arc::new(prisma::PrismaLspAdapter::new(
379 node_runtime.clone(),
380 ))]
381 );
382 language!("dart", vec![Arc::new(dart::DartLanguageServer {})]);
383
384 languages.register_secondary_lsp_adapter(
385 "Svelte".into(),
386 Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
387 );
388}
389
390#[cfg(any(test, feature = "test-support"))]
391pub fn language(name: &str, grammar: tree_sitter::Language) -> Arc<Language> {
392 Arc::new(
393 Language::new(load_config(name), Some(grammar))
394 .with_queries(load_queries(name))
395 .unwrap(),
396 )
397}
398
399fn load_config(name: &str) -> LanguageConfig {
400 let config_toml = String::from_utf8(
401 LanguageDir::get(&format!("{}/config.toml", name))
402 .unwrap()
403 .data
404 .to_vec(),
405 )
406 .unwrap();
407
408 ::toml::from_str(&config_toml)
409 .with_context(|| format!("failed to load config.toml for language {name:?}"))
410 .unwrap()
411}
412
413fn load_queries(name: &str) -> LanguageQueries {
414 let mut result = LanguageQueries::default();
415 for path in LanguageDir::iter() {
416 if let Some(remainder) = path.strip_prefix(name).and_then(|p| p.strip_prefix('/')) {
417 if !remainder.ends_with(".scm") {
418 continue;
419 }
420 for (name, query) in QUERY_FILENAME_PREFIXES {
421 if remainder.starts_with(name) {
422 let contents = asset_str::<LanguageDir>(path.as_ref());
423 match query(&mut result) {
424 None => *query(&mut result) = Some(contents),
425 Some(r) => r.to_mut().push_str(contents.as_ref()),
426 }
427 }
428 }
429 }
430 }
431 result
432}