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