rust/xtask/src/dist.rs
2020-10-17 02:42:42 +02:00

123 lines
3.2 KiB
Rust

use std::{
env,
fs::File,
io,
path::{Path, PathBuf},
};
use anyhow::Result;
use flate2::{write::GzEncoder, Compression};
use xshell::{cmd, cp, mkdir_p, pushd, read_file, rm_rf, write_file};
use crate::{date_iso, project_root};
pub struct DistCmd {
pub nightly: bool,
pub client_version: Option<String>,
}
impl DistCmd {
pub fn run(self) -> Result<()> {
let dist = project_root().join("dist");
rm_rf(&dist)?;
mkdir_p(&dist)?;
if let Some(version) = self.client_version {
let release_tag = if self.nightly { "nightly".to_string() } else { date_iso()? };
dist_client(&version, &release_tag)?;
}
dist_server()?;
Ok(())
}
}
fn dist_client(version: &str, release_tag: &str) -> Result<()> {
let _d = pushd("./editors/code")?;
let nightly = release_tag == "nightly";
let mut patch = Patch::new("./package.json")?;
patch
.replace(r#""version": "0.4.0-dev""#, &format!(r#""version": "{}""#, version))
.replace(r#""releaseTag": null"#, &format!(r#""releaseTag": "{}""#, release_tag));
if nightly {
patch.replace(
r#""displayName": "rust-analyzer""#,
r#""displayName": "rust-analyzer (nightly)""#,
);
}
if !nightly {
patch.replace(r#""enableProposedApi": true,"#, r#""#);
}
patch.commit()?;
cmd!("npm ci").run()?;
cmd!("npx vsce package -o ../../dist/rust-analyzer.vsix").run()?;
Ok(())
}
fn dist_server() -> Result<()> {
if cfg!(target_os = "linux") {
env::set_var("CC", "clang");
}
cmd!("cargo build --manifest-path ./crates/rust-analyzer/Cargo.toml --bin rust-analyzer --release").run()?;
let (src, dst) = if cfg!(target_os = "linux") {
("./target/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")
};
let src = Path::new(src);
let dst = Path::new(dst);
cp(&src, &dst)?;
gzip(&src, &dst.with_extension("gz"))?;
Ok(())
}
fn gzip(src_path: &Path, dest_path: &Path) -> Result<()> {
let mut encoder = GzEncoder::new(File::create(dest_path)?, Compression::best());
let mut input = io::BufReader::new(File::open(src_path)?);
io::copy(&mut input, &mut encoder)?;
encoder.finish()?;
Ok(())
}
struct Patch {
path: PathBuf,
original_contents: String,
contents: String,
}
impl Patch {
fn new(path: impl Into<PathBuf>) -> Result<Patch> {
let path = path.into();
let contents = read_file(&path)?;
Ok(Patch { path, original_contents: contents.clone(), contents })
}
fn replace(&mut self, from: &str, to: &str) -> &mut Patch {
assert!(self.contents.contains(from));
self.contents = self.contents.replace(from, to);
self
}
fn commit(&self) -> Result<()> {
write_file(&self.path, &self.contents)?;
Ok(())
}
}
impl Drop for Patch {
fn drop(&mut self) {
write_file(&self.path, &self.original_contents).unwrap();
}
}