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