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