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