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