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