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