lib.rs

  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| {
126        let config = load_config(asset_dir_name);
127        languages.register_language(
128            config.name.clone(),
129            config.grammar.clone(),
130            config.matcher.clone(),
131            adapters,
132            move || Ok((config.clone(), load_queries(asset_dir_name))),
133        )
134    };
135
136    language(
137        "astro",
138        vec![
139            Arc::new(astro::AstroLspAdapter::new(node_runtime.clone())),
140            Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
141        ],
142    );
143    language("bash", vec![]);
144    language("c", vec![Arc::new(c::CLspAdapter) as Arc<dyn LspAdapter>]);
145    language("clojure", vec![Arc::new(clojure::ClojureLspAdapter)]);
146    language("cpp", vec![Arc::new(c::CLspAdapter)]);
147    language("csharp", vec![Arc::new(csharp::OmniSharpAdapter {})]);
148    language(
149        "css",
150        vec![
151            Arc::new(css::CssLspAdapter::new(node_runtime.clone())),
152            Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
153        ],
154    );
155
156    language(
157        "dockerfile",
158        vec![Arc::new(dockerfile::DockerfileLspAdapter::new(
159            node_runtime.clone(),
160        ))],
161    );
162
163    match &ElixirSettings::get(None, cx).lsp {
164        elixir::ElixirLspSetting::ElixirLs => language(
165            "elixir",
166            vec![
167                Arc::new(elixir::ElixirLspAdapter),
168                Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
169            ],
170        ),
171        elixir::ElixirLspSetting::NextLs => {
172            language("elixir", vec![Arc::new(elixir::NextLspAdapter)])
173        }
174        elixir::ElixirLspSetting::Local { path, arguments } => language(
175            "elixir",
176            vec![Arc::new(elixir::LocalLspAdapter {
177                path: path.clone(),
178                arguments: arguments.clone(),
179            })],
180        ),
181    }
182    language("gitcommit", vec![]);
183    language("erlang", vec![Arc::new(erlang::ErlangLspAdapter)]);
184
185    language("gleam", vec![Arc::new(gleam::GleamLspAdapter)]);
186    language("go", vec![Arc::new(go::GoLspAdapter)]);
187    language("gomod", vec![]);
188    language("gowork", vec![]);
189    language("zig", vec![Arc::new(zig::ZlsAdapter)]);
190    language(
191        "heex",
192        vec![
193            Arc::new(elixir::ElixirLspAdapter),
194            Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
195        ],
196    );
197    language(
198        "json",
199        vec![Arc::new(json::JsonLspAdapter::new(
200            node_runtime.clone(),
201            languages.clone(),
202        ))],
203    );
204    language("markdown", vec![]);
205    language(
206        "python",
207        vec![Arc::new(python::PythonLspAdapter::new(
208            node_runtime.clone(),
209        ))],
210    );
211    language("rust", vec![Arc::new(rust::RustLspAdapter)]);
212    language("toml", vec![Arc::new(toml::TaploLspAdapter)]);
213    match &DenoSettings::get(None, cx).enable {
214        true => {
215            language(
216                "tsx",
217                vec![
218                    Arc::new(deno::DenoLspAdapter::new()),
219                    Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
220                ],
221            );
222            language("typescript", vec![Arc::new(deno::DenoLspAdapter::new())]);
223            language(
224                "javascript",
225                vec![
226                    Arc::new(deno::DenoLspAdapter::new()),
227                    Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
228                ],
229            );
230        }
231        false => {
232            language(
233                "tsx",
234                vec![
235                    Arc::new(typescript::TypeScriptLspAdapter::new(node_runtime.clone())),
236                    Arc::new(typescript::EsLintLspAdapter::new(node_runtime.clone())),
237                    Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
238                ],
239            );
240            language(
241                "typescript",
242                vec![
243                    Arc::new(typescript::TypeScriptLspAdapter::new(node_runtime.clone())),
244                    Arc::new(typescript::EsLintLspAdapter::new(node_runtime.clone())),
245                ],
246            );
247            language(
248                "javascript",
249                vec![
250                    Arc::new(typescript::TypeScriptLspAdapter::new(node_runtime.clone())),
251                    Arc::new(typescript::EsLintLspAdapter::new(node_runtime.clone())),
252                    Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
253                ],
254            );
255        }
256    }
257
258    language("haskell", vec![Arc::new(haskell::HaskellLanguageServer {})]);
259    language(
260        "html",
261        vec![
262            Arc::new(html::HtmlLspAdapter::new(node_runtime.clone())),
263            Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
264        ],
265    );
266    language("ruby", vec![Arc::new(ruby::RubyLanguageServer)]);
267    language(
268        "erb",
269        vec![
270            Arc::new(ruby::RubyLanguageServer),
271            Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
272        ],
273    );
274    language("scheme", vec![]);
275    language("racket", vec![]);
276    language("lua", vec![Arc::new(lua::LuaLspAdapter)]);
277    language(
278        "yaml",
279        vec![Arc::new(yaml::YamlLspAdapter::new(node_runtime.clone()))],
280    );
281    language(
282        "svelte",
283        vec![
284            Arc::new(svelte::SvelteLspAdapter::new(node_runtime.clone())),
285            Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
286        ],
287    );
288    language(
289        "php",
290        vec![
291            Arc::new(php::IntelephenseLspAdapter::new(node_runtime.clone())),
292            Arc::new(tailwind::TailwindLspAdapter::new(node_runtime.clone())),
293        ],
294    );
295    language(
296        "purescript",
297        vec![Arc::new(purescript::PurescriptLspAdapter::new(
298            node_runtime.clone(),
299        ))],
300    );
301    language(
302        "elm",
303        vec![Arc::new(elm::ElmLspAdapter::new(node_runtime.clone()))],
304    );
305    language("glsl", vec![]);
306    language("nix", vec![]);
307    language("nu", vec![Arc::new(nu::NuLanguageServer {})]);
308    language("ocaml", vec![Arc::new(ocaml::OCamlLspAdapter)]);
309    language("ocaml-interface", vec![Arc::new(ocaml::OCamlLspAdapter)]);
310    language(
311        "vue",
312        vec![Arc::new(vue::VueLspAdapter::new(node_runtime.clone()))],
313    );
314    language("uiua", vec![Arc::new(uiua::UiuaLanguageServer {})]);
315    language("proto", vec![]);
316    language("terraform", vec![Arc::new(terraform::TerraformLspAdapter)]);
317    language(
318        "terraform-vars",
319        vec![Arc::new(terraform::TerraformLspAdapter)],
320    );
321    language("hcl", vec![]);
322    language(
323        "prisma",
324        vec![Arc::new(prisma::PrismaLspAdapter::new(
325            node_runtime.clone(),
326        ))],
327    );
328    language("dart", vec![Arc::new(dart::DartLanguageServer {})]);
329}
330
331#[cfg(any(test, feature = "test-support"))]
332pub async fn language(
333    name: &str,
334    grammar: tree_sitter::Language,
335    lsp_adapter: Option<Arc<dyn LspAdapter>>,
336) -> Arc<Language> {
337    Arc::new(
338        Language::new(load_config(name), Some(grammar))
339            .with_lsp_adapters(lsp_adapter.into_iter().collect())
340            .await
341            .with_queries(load_queries(name))
342            .unwrap(),
343    )
344}
345
346fn load_config(name: &str) -> LanguageConfig {
347    let config_toml = String::from_utf8(
348        LanguageDir::get(&format!("{}/config.toml", name))
349            .unwrap()
350            .data
351            .to_vec(),
352    )
353    .unwrap();
354
355    ::toml::from_str(&config_toml)
356        .with_context(|| format!("failed to load config.toml for language {name:?}"))
357        .unwrap()
358}
359
360fn load_queries(name: &str) -> LanguageQueries {
361    let mut result = LanguageQueries::default();
362    for path in LanguageDir::iter() {
363        if let Some(remainder) = path.strip_prefix(name).and_then(|p| p.strip_prefix('/')) {
364            if !remainder.ends_with(".scm") {
365                continue;
366            }
367            for (name, query) in QUERY_FILENAME_PREFIXES {
368                if remainder.starts_with(name) {
369                    let contents = asset_str::<LanguageDir>(path.as_ref());
370                    match query(&mut result) {
371                        None => *query(&mut result) = Some(contents),
372                        Some(r) => r.to_mut().push_str(contents.as_ref()),
373                    }
374                }
375            }
376        }
377    }
378    result
379}