5 files changed, 109 insertions, 10 deletions

diff --git a/xtask/src/dist.rs b/xtask/src/dist.rs
new file mode 100644
index 000000000..737751ae8
--- /dev/null
+++ b/xtask/src/dist.rs
@@ -0,0 +1,79 @@
+use std::path::PathBuf;
+
+use anyhow::Result;
+
+use crate::{
+    not_bash::{fs2, pushd, pwd, rm_rf, run},
+    project_root,
+};
+
+pub fn run_dist(nightly: bool) -> Result<()> {
+    let dist = project_root().join("dist");
+    rm_rf(&dist)?;
+    fs2::create_dir_all(&dist)?;
+
+    if cfg!(target_os = "linux") {
+        dist_client(nightly)?;
+    }
+    dist_server()?;
+    Ok(())
+}
+
+fn dist_client(nightly: bool) -> Result<()> {
+    let _d = pushd("./editors/code");
+
+    let package_json_path = pwd().join("package.json");
+    let original_package_json = fs2::read_to_string(&package_json_path)?;
+    let _restore =
+        Restore { path: package_json_path.clone(), contents: original_package_json.clone() };
+
+    let mut package_json = original_package_json.replace(r#""enableProposedApi": true,"#, r#""#);
+
+    if nightly {
+        package_json = package_json.replace(
+            r#""displayName": "rust-analyzer""#,
+            r#""displayName": "rust-analyzer nightly""#,
+        );
+    } else {
+        package_json = original_package_json.replace(r#""enableProposedApi": true,"#, r#""#);
+    }
+    fs2::write(package_json_path, package_json)?;
+
+    run!("npx vsce package -o ../../dist/rust-analyzer.vsix")?;
+    Ok(())
+}
+
+fn dist_server() -> Result<()> {
+    if cfg!(target_os = "linux") {
+        std::env::set_var("CC", "clang");
+        run!("cargo build --package rust-analyzer --bin rust-analyzer --release --target x86_64-unknown-linux-musl")?;
+        run!("strip ./target/x86_64-unknown-linux-musl/release/rust-analyzer")?;
+    } else {
+        run!("cargo build --package rust-analyzer --bin rust-analyzer --release")?;
+    }
+
+    let (src, dst) = if cfg!(target_os = "linux") {
+        ("./target/x86_64-unknown-linux-musl/release/rust-analyzer", "./dist/rust-analyzer-linux")
+    } else if cfg!(target_os = "windows") {
+        ("/target/release/rust-analyzer.exe", "./dist/rust-analyzer-windows.exe")
+    } else if cfg!(target_os = "macos") {
+        ("/target/release/rust-analyzer", "./dist/rust-analyzer-mac")
+    } else {
+        panic!("Unsupported OS")
+    };
+
+    fs2::copy(src, dst)?;
+
+    Ok(())
+}
+
+struct Restore {
+    path: PathBuf,
+    contents: String,
+}
+
+impl Drop for Restore {
+    fn drop(&mut self) {
+        fs2::write(&self.path, &self.contents).unwrap();
+    }
+}
diff --git a/xtask/src/install.rs b/xtask/src/install.rs
index f76467cac..d0d745b05 100644
--- a/xtask/src/install.rs
+++ b/xtask/src/install.rs
@@ -4,10 +4,7 @@ use std::{env, path::PathBuf, str};
 
 use anyhow::{bail, format_err, Context, Result};
 
-use crate::{
-    not_bash::{pushd, run},
-    project_root,
-};
+use crate::not_bash::{pushd, run};
 
 // Latest stable, feel free to send a PR if this lags behind.
 const REQUIRED_RUST_VERSION: u32 = 41;
@@ -27,7 +24,6 @@ pub struct ServerOpt {
 
 impl InstallCmd {
     pub fn run(self) -> Result<()> {
-        let _dir = pushd(project_root());
         let both = self.server.is_some() && self.client.is_some();
         if cfg!(target_os = "macos") {
             fix_path_for_mac().context("Fix path for mac")?
diff --git a/xtask/src/lib.rs b/xtask/src/lib.rs
index f48045d17..014b61b37 100644
--- a/xtask/src/lib.rs
+++ b/xtask/src/lib.rs
@@ -4,6 +4,7 @@
 
 pub mod not_bash;
 pub mod install;
+pub mod dist;
 pub mod pre_commit;
 
 pub mod codegen;
@@ -90,7 +91,6 @@ pub fn run_clippy() -> Result<()> {
 
     let allowed_lints = [
         "clippy::collapsible_if",
-        "clippy::map_clone", // FIXME: remove when Iterator::copied stabilizes (1.36.0)
         "clippy::needless_pass_by_value",
         "clippy::nonminimal_bool",
         "clippy::redundant_pattern_matching",
diff --git a/xtask/src/main.rs b/xtask/src/main.rs
index a7dffe2cc..7c8ea9001 100644
--- a/xtask/src/main.rs
+++ b/xtask/src/main.rs
@@ -13,8 +13,11 @@ use std::env;
 use pico_args::Arguments;
 use xtask::{
     codegen::{self, Mode},
+    dist::run_dist,
     install::{ClientOpt, InstallCmd, ServerOpt},
-    pre_commit, run_clippy, run_fuzzer, run_pre_cache, run_release, run_rustfmt, Result,
+    not_bash::pushd,
+    pre_commit, project_root, run_clippy, run_fuzzer, run_pre_cache, run_release, run_rustfmt,
+    Result,
 };
 
 fn main() -> Result<()> {
@@ -22,6 +25,8 @@ fn main() -> Result<()> {
         return pre_commit::run_hook();
     }
 
+    let _d = pushd(project_root());
+
     let mut args = Arguments::from_env();
     let subcommand = args.subcommand()?.unwrap_or_default();
 
@@ -97,6 +102,11 @@ FLAGS:
             args.finish()?;
             run_release(dry_run)
         }
+        "dist" => {
+            let nightly = args.contains("--nightly");
+            args.finish()?;
+            run_dist(nightly)
+        }
         _ => {
             eprintln!(
                 "\
@@ -112,7 +122,8 @@ SUBCOMMANDS:
     fuzz-tests
     codegen
     install
-    lint"
+    lint
+    dist"
             );
             Ok(())
         }
diff --git a/xtask/src/not_bash.rs b/xtask/src/not_bash.rs
index 40f706d9f..1697b7fcd 100644
--- a/xtask/src/not_bash.rs
+++ b/xtask/src/not_bash.rs
@@ -19,6 +19,11 @@ pub mod fs2 {
         fs::read_dir(path).with_context(|| format!("Failed to read {}", path.display()))
     }
 
+    pub fn read_to_string<P: AsRef<Path>>(path: P) -> Result<String> {
+        let path = path.as_ref();
+        fs::read_to_string(path).with_context(|| format!("Failed to read {}", path.display()))
+    }
+
     pub fn write<P: AsRef<Path>, C: AsRef<[u8]>>(path: P, contents: C) -> Result<()> {
         let path = path.as_ref();
         fs::write(path, contents).with_context(|| format!("Failed to write {}", path.display()))
@@ -40,6 +45,11 @@ pub mod fs2 {
         let path = path.as_ref();
         fs::remove_dir_all(path).with_context(|| format!("Failed to remove dir {}", path.display()))
     }
+
+    pub fn create_dir_all<P: AsRef<Path>>(path: P) -> Result<()> {
+        let path = path.as_ref();
+        fs::create_dir_all(path).with_context(|| format!("Failed to create dir {}", path.display()))
+    }
 }
 
 macro_rules! _run {
@@ -61,6 +71,10 @@ pub fn pushd(path: impl Into<PathBuf>) -> Pushd {
     Pushd { _p: () }
 }
 
+pub fn pwd() -> PathBuf {
+    Env::with(|env| env.cwd())
+}
+
 impl Drop for Pushd {
     fn drop(&mut self) {
         Env::with(|env| env.popd())
@@ -85,7 +99,6 @@ pub fn run_process(cmd: String, echo: bool) -> Result<String> {
 }
 
 fn run_process_inner(cmd: &str, echo: bool) -> Result<String> {
-    let cwd = Env::with(|env| env.cwd());
     let mut args = shelx(cmd);
     let binary = args.remove(0);
 
@@ -95,7 +108,7 @@ fn run_process_inner(cmd: &str, echo: bool) -> Result<String> {
 
     let output = Command::new(binary)
         .args(args)
-        .current_dir(cwd)
+        .current_dir(pwd())
         .stdin(Stdio::null())
         .stderr(Stdio::inherit())
         .output()?;