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(&self, cx: &mut AppContext) -> BoxFuture<'static, Value> {
 97        let tab_size = all_language_settings(None, cx)
 98            .language(Some("YAML"))
 99            .tab_size;
100
101        future::ready(serde_json::json!({
102            "yaml": {
103                "keyOrdering": false
104            },
105            "[yaml]": {
106                "editor.tabSize": tab_size,
107            }
108        }))
109        .boxed()
110    }
111}
112
113async fn get_cached_server_binary(
114    container_dir: PathBuf,
115    node: &dyn NodeRuntime,
116) -> Option<LanguageServerBinary> {
117    (|| async move {
118        let mut last_version_dir = None;
119        let mut entries = fs::read_dir(&container_dir).await?;
120        while let Some(entry) = entries.next().await {
121            let entry = entry?;
122            if entry.file_type().await?.is_dir() {
123                last_version_dir = Some(entry.path());
124            }
125        }
126        let last_version_dir = last_version_dir.ok_or_else(|| anyhow!("no cached binary"))?;
127        let server_path = last_version_dir.join(SERVER_PATH);
128        if server_path.exists() {
129            Ok(LanguageServerBinary {
130                path: node.binary_path().await?,
131                arguments: server_binary_arguments(&server_path),
132            })
133        } else {
134            Err(anyhow!(
135                "missing executable in directory {:?}",
136                last_version_dir
137            ))
138        }
139    })()
140    .await
141    .log_err()
142}