ZIm/crates/zed/src/languages.rs

use anyhow::Context;
pub use language::*;
use node_runtime::NodeRuntime;
use rust_embed::RustEmbed;
use std::{borrow::Cow, str, sync::Arc};
use theme::ThemeRegistry;

mod c;
mod elixir;
mod go;
mod html;
mod json;
mod language_plugin;
mod lua;
mod python;
mod ruby;
mod rust;
mod typescript;
mod yaml;

// 1. Add tree-sitter-{language} parser to zed crate
// 2. Create a language directory in zed/crates/zed/src/languages and add the language to init function below
// 3. Add config.toml to the newly created language directory using existing languages as a template
// 4. Copy highlights from tree sitter repo for the language into a highlights.scm file.
//      Note: github highlights take the last match while zed takes the first
// 5. Add indents.scm, outline.scm, and brackets.scm to implement indent on newline, outline/breadcrumbs,
//    and autoclosing brackets respectively
// 6. If the language has injections add an injections.scm query file

#[derive(RustEmbed)]
#[folder = "src/languages"]
#[exclude = "*.rs"]
struct LanguageDir;

pub fn init(
    languages: Arc<LanguageRegistry>,
    themes: Arc<ThemeRegistry>,
    node_runtime: Arc<NodeRuntime>,
) {
    fn adapter_arc(adapter: impl LspAdapter) -> Arc<dyn LspAdapter> {
        Arc::new(adapter)
    }

    let languages_list = [
        (
            "c",
            tree_sitter_c::language(),
            vec![adapter_arc(c::CLspAdapter)],
        ),
        (
            "cpp",
            tree_sitter_cpp::language(),
            vec![adapter_arc(c::CLspAdapter)],
        ),
        ("css", tree_sitter_css::language(), vec![]),
        (
            "elixir",
            tree_sitter_elixir::language(),
            vec![adapter_arc(elixir::ElixirLspAdapter)],
        ),
        (
            "go",
            tree_sitter_go::language(),
            vec![adapter_arc(go::GoLspAdapter)],
        ),
        (
            "json",
            tree_sitter_json::language(),
            vec![adapter_arc(json::JsonLspAdapter::new(
                node_runtime.clone(),
                languages.clone(),
                themes.clone(),
            ))],
        ),
        ("markdown", tree_sitter_markdown::language(), vec![]),
        (
            "python",
            tree_sitter_python::language(),
            vec![adapter_arc(python::PythonLspAdapter::new(
                node_runtime.clone(),
            ))],
        ),
        (
            "rust",
            tree_sitter_rust::language(),
            vec![adapter_arc(rust::RustLspAdapter)],
        ),
        ("toml", tree_sitter_toml::language(), vec![]),
        (
            "tsx",
            tree_sitter_typescript::language_tsx(),
            vec![adapter_arc(typescript::TypeScriptLspAdapter::new(
                node_runtime.clone(),
            ))],
        ),
        (
            "typescript",
            tree_sitter_typescript::language_typescript(),
            vec![adapter_arc(typescript::TypeScriptLspAdapter::new(
                node_runtime.clone(),
            ))],
        ),
        (
            "javascript",
            tree_sitter_typescript::language_tsx(),
            vec![adapter_arc(typescript::TypeScriptLspAdapter::new(
                node_runtime.clone(),
            ))],
        ),
        (
            "html",
            tree_sitter_html::language(),
            vec![adapter_arc(html::HtmlLspAdapter::new(node_runtime.clone()))],
        ),
        (
            "ruby",
            tree_sitter_ruby::language(),
            vec![adapter_arc(ruby::RubyLanguageServer)],
        ),
        (
            "erb",
            tree_sitter_embedded_template::language(),
            vec![adapter_arc(ruby::RubyLanguageServer)],
        ),
        ("scheme", tree_sitter_scheme::language(), vec![]),
        ("racket", tree_sitter_racket::language(), vec![]),
        (
            "lua",
            tree_sitter_lua::language(),
            vec![adapter_arc(lua::LuaLspAdapter)],
        ),
        (
            "yaml",
            tree_sitter_yaml::language(),
            vec![adapter_arc(yaml::YamlLspAdapter::new(node_runtime.clone()))],
        ),
    ];

    for (name, grammar, lsp_adapters) in languages_list {
        languages.register(name, load_config(name), grammar, lsp_adapters, load_queries);
    }
}

#[cfg(any(test, feature = "test-support"))]
pub async fn language(
    name: &str,
    grammar: tree_sitter::Language,
    lsp_adapter: Option<Arc<dyn LspAdapter>>,
) -> Arc<Language> {
    Arc::new(
        Language::new(load_config(name), Some(grammar))
            .with_lsp_adapters(lsp_adapter.into_iter().collect())
            .await
            .with_queries(load_queries(name))
            .unwrap(),
    )
}

fn load_config(name: &str) -> LanguageConfig {
    toml::from_slice(
        &LanguageDir::get(&format!("{}/config.toml", name))
            .unwrap()
            .data,
    )
    .with_context(|| format!("failed to load config.toml for language {name:?}"))
    .unwrap()
}

fn load_queries(name: &str) -> LanguageQueries {
    LanguageQueries {
        highlights: load_query(name, "/highlights"),
        brackets: load_query(name, "/brackets"),
        indents: load_query(name, "/indents"),
        outline: load_query(name, "/outline"),
        injections: load_query(name, "/injections"),
        overrides: load_query(name, "/overrides"),
    }
}

fn load_query(name: &str, filename_prefix: &str) -> Option<Cow<'static, str>> {
    let mut result = None;
    for path in LanguageDir::iter() {
        if let Some(remainder) = path.strip_prefix(name) {
            if remainder.starts_with(filename_prefix) {
                let contents = match LanguageDir::get(path.as_ref()).unwrap().data {
                    Cow::Borrowed(s) => Cow::Borrowed(str::from_utf8(s).unwrap()),
                    Cow::Owned(s) => Cow::Owned(String::from_utf8(s).unwrap()),
                };
                match &mut result {
                    None => result = Some(contents),
                    Some(r) => r.to_mut().push_str(contents.as_ref()),
                }
            }
        }
    }
    result
}