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 smol::fs;
  8use std::{
  9    any::Any,
 10    ffi::OsString,
 11    path::{Path, PathBuf},
 12    sync::Arc,
 13};
 14use util::{async_maybe, ResultExt};
 15
 16const SERVER_PATH: &'static str = "node_modules/.bin/prisma-language-server";
 17
 18fn server_binary_arguments(server_path: &Path) -> Vec<OsString> {
 19    vec![server_path.into(), "--stdio".into()]
 20}
 21
 22pub struct PrismaLspAdapter {
 23    node: Arc<dyn NodeRuntime>,
 24}
 25
 26impl PrismaLspAdapter {
 27    pub fn new(node: Arc<dyn NodeRuntime>) -> Self {
 28        PrismaLspAdapter { node }
 29    }
 30}
 31
 32#[async_trait]
 33impl LspAdapter for PrismaLspAdapter {
 34    fn name(&self) -> LanguageServerName {
 35        LanguageServerName("prisma-language-server".into())
 36    }
 37
 38    fn short_name(&self) -> &'static str {
 39        "prisma-language-server"
 40    }
 41
 42    async fn fetch_latest_server_version(
 43        &self,
 44        _: &dyn LspAdapterDelegate,
 45    ) -> Result<Box<dyn 'static + Any + Send>> {
 46        Ok(Box::new(
 47            self.node
 48                .npm_package_latest_version("@prisma/language-server")
 49                .await?,
 50        ) as Box<_>)
 51    }
 52
 53    async fn fetch_server_binary(
 54        &self,
 55        version: Box<dyn 'static + Send + Any>,
 56        container_dir: PathBuf,
 57        _: &dyn LspAdapterDelegate,
 58    ) -> Result<LanguageServerBinary> {
 59        let version = version.downcast::<String>().unwrap();
 60        let server_path = container_dir.join(SERVER_PATH);
 61
 62        if fs::metadata(&server_path).await.is_err() {
 63            self.node
 64                .npm_install_packages(
 65                    &container_dir,
 66                    &[("@prisma/language-server", version.as_str())],
 67                )
 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        None
 95    }
 96}
 97
 98async fn get_cached_server_binary(
 99    container_dir: PathBuf,
100    node: &dyn NodeRuntime,
101) -> Option<LanguageServerBinary> {
102    async_maybe!({
103        let mut last_version_dir = None;
104        let mut entries = fs::read_dir(&container_dir).await?;
105        while let Some(entry) = entries.next().await {
106            let entry = entry?;
107            if entry.file_type().await?.is_dir() {
108                last_version_dir = Some(entry.path());
109            }
110        }
111        let last_version_dir = last_version_dir.ok_or_else(|| anyhow!("no cached binary"))?;
112        let server_path = last_version_dir.join(SERVER_PATH);
113        if server_path.exists() {
114            Ok(LanguageServerBinary {
115                path: node.binary_path().await?,
116                env: None,
117                arguments: server_binary_arguments(&server_path),
118            })
119        } else {
120            Err(anyhow!(
121                "missing executable in directory {:?}",
122                last_version_dir
123            ))
124        }
125    })
126    .await
127    .log_err()
128}