Integrate with PowerShellEditorServices

Signed-off-by: Thanabodee Charoenpiriyakij <[email protected]>

Author: wingyplus

README.md

@@ -1 +1,17 @@
 # PowerShell extension for Zed editor
+
+## Setting up language server
+
+Currently, the extension didn't support automatic installation (it'll soon land). You need
+to set the bundle path through `lsp` configuration:
+
+```json
+{
+  "lsp": {
+    "powershell-es": {
+      "binary": {
+        "path": "<path to PowerShellEditorServices>"
+      }
+    }
+  }
+}

extension.toml

@@ -9,3 +9,7 @@ repository = "https://github.com/zed-industries/zed"
 [grammars.powershell]
 repository = "https://github.com/airbus-cert/tree-sitter-powershell"
 commit = "804d86fd4ad286bd0cc1c1f0f7b28bd7af6755ad"
+
+[language_servers.powershell-es]
+name = "PowerShell Editor Services"
+languages = ["PowerShell"]

src/language_server.rs

@@ -0,0 +1,39 @@
+use zed_extension_api::{self as zed, Result};
+
+pub struct PowerShellEditorServices;
+
+impl PowerShellEditorServices {
+    /// Install the PowerShellEditorServices.
+    ///
+    /// Returns the bundle path after installed.
+    pub fn install(language_server_id: &zed_extension_api::LanguageServerId) -> Result<String> {
+        let release = zed::latest_github_release(
+            "PowerShell/PowerShellEditorServices",
+            zed::GithubReleaseOptions {
+                require_assets: true,
+                pre_release: false,
+            },
+        )?;
+
+        let asset = release
+            .assets
+            .iter()
+            .find(|asset| asset.name == "PowerShellEditorServices.zip")
+            .ok_or_else(|| format!("no PowerShellEditorServices.zip found"))?;
+
+        let path = format!("powershell-es-{}", release.version);
+
+        // TODO: do download when no bundle path found.
+        // TODO: cache the bundle path.
+
+        zed::set_language_server_installation_status(
+            &language_server_id,
+            &zed::LanguageServerInstallationStatus::Downloading,
+        );
+
+        zed::download_file(&asset.download_url, &path, zed::DownloadedFileType::Zip)
+            .map_err(|err| format!("download error {}", err))?;
+
+        Ok(path.to_string())
+    }
+}

src/powershell.rs

@@ -1,21 +1,56 @@
-use zed_extension_api::{self as zed};
+mod language_server;
 
-struct PowerShellExtension;
+use zed_extension_api::{self as zed, settings::LspSettings};
+
+use crate::language_server::PowerShellEditorServices as EditorServices;
+
+struct PowerShellExtension {
+    /// The PowerShell binary, default to `pwsh`.
+    // TODO: allow to configure via Zed settings.
+    powershell_bin: Option<String>,
+}
 
 impl zed::Extension for PowerShellExtension {
     fn new() -> Self
     where
         Self: Sized,
     {
-        Self
+        Self {
+            powershell_bin: Some("pwsh".to_string()),
+        }
     }
 
     fn language_server_command(
         &mut self,
-        _language_server_id: &zed_extension_api::LanguageServerId,
-        _worktree: &zed_extension_api::Worktree,
+        language_server_id: &zed_extension_api::LanguageServerId,
+        worktree: &zed_extension_api::Worktree,
     ) -> zed_extension_api::Result<zed_extension_api::Command> {
-        todo!()
+        let pwsh_bin = worktree
+            .which(self.powershell_bin.clone().unwrap().as_str())
+            .ok_or_else(|| "No PowerShell command found")?;
+
+        let bundle_path = LspSettings::for_worktree("powershell-es", worktree)
+            .ok()
+            .and_then(|lsp_settings| lsp_settings.binary)
+            .and_then(|binary| binary.path)
+            .unwrap();
+
+        // TODO: make remote install works.
+        // let bundle_path = EditorServices::install(language_server_id)
+        //     .map_err(|err| format!("failed to get editor services: {}", err))?;
+
+        let command = format!("{bundle_path}/PowerShellEditorServices/Start-EditorServices.ps1 -BundledModulesPath {bundle_path} -Stdio -SessionDetailsPath {bundle_path}/powershell-es.session.json  -LogPath {bundle_path}/logs -FeatureFlags @() -AdditionalModules @() -HostName zed -HostProfileId 0 -HostVersion 1.0.0 -LogLevel Diagnostic");
+
+        Ok(zed::Command {
+            command: pwsh_bin,
+            args: vec![
+                "-NoLogo".to_string(),
+                "-NoProfile".to_string(),
+                "-Command".to_string(),
+                command.to_string(),
+            ],
+            env: Default::default(),
+        })
     }
 }