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