1use anyhow::{anyhow, Result};
  2use async_trait::async_trait;
  3use futures::{future::BoxFuture, FutureExt, StreamExt};
  4use gpui::AppContext;
  5use language::{
  6    language_settings::all_language_settings, LanguageServerName, LspAdapter, LspAdapterDelegate,
  7};
  8use lsp::LanguageServerBinary;
  9use node_runtime::NodeRuntime;
 10use serde_json::Value;
 11use smol::fs;
 12use std::{
 13    any::Any,
 14    ffi::OsString,
 15    future,
 16    path::{Path, PathBuf},
 17    sync::Arc,
 18};
 19use util::ResultExt;
 20
 21const SERVER_PATH: &'static str = "node_modules/yaml-language-server/bin/yaml-language-server";
 22
 23fn server_binary_arguments(server_path: &Path) -> Vec<OsString> {
 24    vec![server_path.into(), "--stdio".into()]
 25}
 26
 27pub struct YamlLspAdapter {
 28    node: Arc<dyn NodeRuntime>,
 29}
 30
 31impl YamlLspAdapter {
 32    pub fn new(node: Arc<dyn NodeRuntime>) -> Self {
 33        YamlLspAdapter { node }
 34    }
 35}
 36
 37#[async_trait]
 38impl LspAdapter for YamlLspAdapter {
 39    async fn name(&self) -> LanguageServerName {
 40        LanguageServerName("yaml-language-server".into())
 41    }
 42
 43    fn short_name(&self) -> &'static str {
 44        "yaml"
 45    }
 46
 47    async fn fetch_latest_server_version(
 48        &self,
 49        _: &dyn LspAdapterDelegate,
 50    ) -> Result<Box<dyn 'static + Any + Send>> {
 51        Ok(Box::new(
 52            self.node
 53                .npm_package_latest_version("yaml-language-server")
 54                .await?,
 55        ) as Box<_>)
 56    }
 57
 58    async fn fetch_server_binary(
 59        &self,
 60        version: Box<dyn 'static + Send + Any>,
 61        container_dir: PathBuf,
 62        _: &dyn LspAdapterDelegate,
 63    ) -> Result<LanguageServerBinary> {
 64        let version = version.downcast::<String>().unwrap();
 65        let server_path = container_dir.join(SERVER_PATH);
 66
 67        if fs::metadata(&server_path).await.is_err() {
 68            self.node
 69                .npm_install_packages(
 70                    &container_dir,
 71                    &[("yaml-language-server", version.as_str())],
 72                )
 73                .await?;
 74        }
 75
 76        Ok(LanguageServerBinary {
 77            path: self.node.binary_path().await?,
 78            arguments: server_binary_arguments(&server_path),
 79        })
 80    }
 81
 82    async fn cached_server_binary(
 83        &self,
 84        container_dir: PathBuf,
 85        _: &dyn LspAdapterDelegate,
 86    ) -> Option<LanguageServerBinary> {
 87        get_cached_server_binary(container_dir, &*self.node).await
 88    }
 89
 90    async fn installation_test_binary(
 91        &self,
 92        container_dir: PathBuf,
 93    ) -> Option<LanguageServerBinary> {
 94        get_cached_server_binary(container_dir, &*self.node).await
 95    }
 96    fn workspace_configuration(
 97        &self,
 98        _workspace_root: &Path,
 99        cx: &mut AppContext,
100    ) -> BoxFuture<'static, Value> {
101        let tab_size = all_language_settings(None, cx)
102            .language(Some("YAML"))
103            .tab_size;
104
105        future::ready(serde_json::json!({
106            "yaml": {
107                "keyOrdering": false
108            },
109            "[yaml]": {
110                "editor.tabSize": tab_size,
111            }
112        }))
113        .boxed()
114    }
115}
116
117async fn get_cached_server_binary(
118    container_dir: PathBuf,
119    node: &dyn NodeRuntime,
120) -> Option<LanguageServerBinary> {
121    (|| async move {
122        let mut last_version_dir = None;
123        let mut entries = fs::read_dir(&container_dir).await?;
124        while let Some(entry) = entries.next().await {
125            let entry = entry?;
126            if entry.file_type().await?.is_dir() {
127                last_version_dir = Some(entry.path());
128            }
129        }
130        let last_version_dir = last_version_dir.ok_or_else(|| anyhow!("no cached binary"))?;
131        let server_path = last_version_dir.join(SERVER_PATH);
132        if server_path.exists() {
133            Ok(LanguageServerBinary {
134                path: node.binary_path().await?,
135                arguments: server_binary_arguments(&server_path),
136            })
137        } else {
138            Err(anyhow!(
139                "missing executable in directory {:?}",
140                last_version_dir
141            ))
142        }
143    })()
144    .await
145    .log_err()
146}