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