From 634fe99fa506d2ce9485a362ae8f80c370ef291b Mon Sep 17 00:00:00 2001
From: Andrey Kuzmin <hi@unsoundscapes.com>
Date: Wed, 31 Jan 2024 10:05:38 +0100
Subject: [PATCH]  Add LSP support for Elm (#7116)

Closes #4595

Release Notes:

- Added LSP support for Elm
([#4595](https://github.com/zed-industries/zed/issues/4595)).

---------

Co-authored-by: Jared M. Smith <absynce@gmail.com>
---
 crates/zed/src/languages.rs     |   7 +-
 crates/zed/src/languages/elm.rs | 122 ++++++++++++++++++++++++++++++++
 docs/src/languages/elm.md       |   2 +-
 3 files changed, 129 insertions(+), 2 deletions(-)
 create mode 100644 crates/zed/src/languages/elm.rs

diff --git a/crates/zed/src/languages.rs b/crates/zed/src/languages.rs
index 2055518df36895b0dc2cd1aff19821139367dfe9..92682d590a81660767910fe31e07c56774f6010b 100644
--- a/crates/zed/src/languages.rs
+++ b/crates/zed/src/languages.rs
@@ -14,6 +14,7 @@ mod csharp;
 mod css;
 mod deno;
 mod elixir;
+mod elm;
 mod gleam;
 mod go;
 mod haskell;
@@ -278,7 +279,11 @@ pub fn init(
             node_runtime.clone(),
         ))],
     );
-    language("elm", tree_sitter_elm::language(), vec![]);
+    language(
+        "elm",
+        tree_sitter_elm::language(),
+        vec![Arc::new(elm::ElmLspAdapter::new(node_runtime.clone()))],
+    );
     language("glsl", tree_sitter_glsl::language(), vec![]);
     language("nix", tree_sitter_nix::language(), vec![]);
     language(
diff --git a/crates/zed/src/languages/elm.rs b/crates/zed/src/languages/elm.rs
new file mode 100644
index 0000000000000000000000000000000000000000..6eb56451026c3df5825c12b1149308efeaead2c5
--- /dev/null
+++ b/crates/zed/src/languages/elm.rs
@@ -0,0 +1,122 @@
+use anyhow::{anyhow, Result};
+use async_trait::async_trait;
+use futures::StreamExt;
+use language::{LanguageServerName, LspAdapter, LspAdapterDelegate};
+use lsp::LanguageServerBinary;
+use node_runtime::NodeRuntime;
+use smol::fs;
+use std::{
+    any::Any,
+    ffi::OsString,
+    path::{Path, PathBuf},
+    sync::Arc,
+};
+use util::ResultExt;
+
+const SERVER_PATH: &'static str = "node_modules/@elm-tooling/elm-language-server/out/node/index.js";
+
+fn server_binary_arguments(server_path: &Path) -> Vec<OsString> {
+    vec![server_path.into(), "--stdio".into()]
+}
+
+pub struct ElmLspAdapter {
+    node: Arc<dyn NodeRuntime>,
+}
+
+impl ElmLspAdapter {
+    pub fn new(node: Arc<dyn NodeRuntime>) -> Self {
+        ElmLspAdapter { node }
+    }
+}
+
+#[async_trait]
+impl LspAdapter for ElmLspAdapter {
+    fn name(&self) -> LanguageServerName {
+        LanguageServerName("elm-language-server".into())
+    }
+
+    fn short_name(&self) -> &'static str {
+        "elmLS"
+    }
+
+    async fn fetch_latest_server_version(
+        &self,
+        _: &dyn LspAdapterDelegate,
+    ) -> Result<Box<dyn 'static + Any + Send>> {
+        Ok(Box::new(
+            self.node
+                .npm_package_latest_version("@elm-tooling/elm-language-server")
+                .await?,
+        ) as Box<_>)
+    }
+
+    async fn fetch_server_binary(
+        &self,
+        version: Box<dyn 'static + Send + Any>,
+        container_dir: PathBuf,
+        _: &dyn LspAdapterDelegate,
+    ) -> Result<LanguageServerBinary> {
+        let version = version.downcast::<String>().unwrap();
+        let server_path = container_dir.join(SERVER_PATH);
+
+        if fs::metadata(&server_path).await.is_err() {
+            self.node
+                .npm_install_packages(
+                    &container_dir,
+                    &[("@elm-tooling/elm-language-server", version.as_str())],
+                )
+                .await?;
+        }
+
+        Ok(LanguageServerBinary {
+            path: self.node.binary_path().await?,
+            arguments: server_binary_arguments(&server_path),
+        })
+    }
+
+    async fn cached_server_binary(
+        &self,
+        container_dir: PathBuf,
+        _: &dyn LspAdapterDelegate,
+    ) -> Option<LanguageServerBinary> {
+        get_cached_server_binary(container_dir, &*self.node).await
+    }
+
+    async fn installation_test_binary(
+        &self,
+        container_dir: PathBuf,
+    ) -> Option<LanguageServerBinary> {
+        get_cached_server_binary(container_dir, &*self.node).await
+    }
+}
+
+async fn get_cached_server_binary(
+    container_dir: PathBuf,
+    node: &dyn NodeRuntime,
+) -> Option<LanguageServerBinary> {
+    (|| async move {
+        let mut last_version_dir = None;
+        let mut entries = fs::read_dir(&container_dir).await?;
+        while let Some(entry) = entries.next().await {
+            let entry = entry?;
+            if entry.file_type().await?.is_dir() {
+                last_version_dir = Some(entry.path());
+            }
+        }
+        let last_version_dir = last_version_dir.ok_or_else(|| anyhow!("no cached binary"))?;
+        let server_path = last_version_dir.join(SERVER_PATH);
+        if server_path.exists() {
+            Ok(LanguageServerBinary {
+                path: node.binary_path().await?,
+                arguments: server_binary_arguments(&server_path),
+            })
+        } else {
+            Err(anyhow!(
+                "missing executable in directory {:?}",
+                last_version_dir
+            ))
+        }
+    })()
+    .await
+    .log_err()
+}
diff --git a/docs/src/languages/elm.md b/docs/src/languages/elm.md
index 8c12d87fa3eafffb8a7c7f345dd8b507531c7e18..ec6aa46c7a4d18238fc6507a9b128a3b30e2302b 100644
--- a/docs/src/languages/elm.md
+++ b/docs/src/languages/elm.md
@@ -1,4 +1,4 @@
 # Elm
 
 - Tree Sitter: [tree-sitter-elm](https://github.com/elm-tooling/tree-sitter-elm)
-- Language Server: N/A
+- Language Server: [elm-language-server](https://github.com/elm-tooling/elm-language-server)