mirror of
https://github.com/denoland/deno.git
synced 2024-12-25 00:29:09 -05:00
276 lines
8 KiB
Rust
276 lines
8 KiB
Rust
// Copyright 2018-2020 the Deno authors. All rights reserved. MIT license.
|
|
|
|
//! This module provides feature to upgrade deno executable
|
|
//!
|
|
//! At the moment it is only consumed using CLI but in
|
|
//! the future it can be easily extended to provide
|
|
//! the same functions as ops available in JS runtime.
|
|
|
|
use crate::http_util::fetch_once;
|
|
use crate::http_util::FetchOnceResult;
|
|
use crate::AnyError;
|
|
use deno_core::error::custom_error;
|
|
use deno_core::futures::FutureExt;
|
|
use deno_core::url::Url;
|
|
use deno_fetch::reqwest;
|
|
use deno_fetch::reqwest::redirect::Policy;
|
|
use deno_fetch::reqwest::Client;
|
|
use regex::Regex;
|
|
use semver_parser::version::parse as semver_parse;
|
|
use semver_parser::version::Version;
|
|
use std::fs;
|
|
use std::future::Future;
|
|
use std::io::prelude::*;
|
|
use std::path::Path;
|
|
use std::path::PathBuf;
|
|
use std::pin::Pin;
|
|
use std::process::Command;
|
|
use std::process::Stdio;
|
|
use std::string::String;
|
|
use tempfile::TempDir;
|
|
|
|
lazy_static! {
|
|
static ref ARCHIVE_NAME: String = format!("deno-{}.zip", env!("TARGET"));
|
|
}
|
|
|
|
async fn get_latest_version(client: &Client) -> Result<Version, AnyError> {
|
|
println!("Checking for latest version");
|
|
let body = client
|
|
.get(Url::parse(
|
|
"https://github.com/denoland/deno/releases/latest",
|
|
)?)
|
|
.send()
|
|
.await?
|
|
.text()
|
|
.await?;
|
|
let v = find_version(&body)?;
|
|
Ok(semver_parse(&v).unwrap())
|
|
}
|
|
|
|
/// Asynchronously updates deno executable to greatest version
|
|
/// if greatest version is available.
|
|
pub async fn upgrade_command(
|
|
dry_run: bool,
|
|
force: bool,
|
|
version: Option<String>,
|
|
output: Option<PathBuf>,
|
|
ca_file: Option<String>,
|
|
) -> Result<(), AnyError> {
|
|
let mut client_builder = Client::builder().redirect(Policy::none());
|
|
|
|
// If we have been provided a CA Certificate, add it into the HTTP client
|
|
if let Some(ca_file) = ca_file {
|
|
let buf = std::fs::read(ca_file);
|
|
let cert = reqwest::Certificate::from_pem(&buf.unwrap())?;
|
|
client_builder = client_builder.add_root_certificate(cert);
|
|
}
|
|
|
|
let client = client_builder.build()?;
|
|
|
|
let current_version = semver_parse(crate::version::DENO).unwrap();
|
|
|
|
let install_version = match version {
|
|
Some(passed_version) => match semver_parse(&passed_version) {
|
|
Ok(ver) => {
|
|
if !force && current_version == ver {
|
|
println!("Version {} is already installed", &ver);
|
|
return Ok(());
|
|
} else {
|
|
ver
|
|
}
|
|
}
|
|
Err(_) => {
|
|
eprintln!("Invalid semver passed");
|
|
std::process::exit(1)
|
|
}
|
|
},
|
|
None => {
|
|
let latest_version = get_latest_version(&client).await?;
|
|
|
|
if !force && current_version >= latest_version {
|
|
println!(
|
|
"Local deno version {} is the most recent release",
|
|
&crate::version::DENO
|
|
);
|
|
return Ok(());
|
|
} else {
|
|
latest_version
|
|
}
|
|
}
|
|
};
|
|
|
|
let archive_data = download_package(
|
|
&compose_url_to_exec(&install_version)?,
|
|
client,
|
|
&install_version,
|
|
)
|
|
.await?;
|
|
let old_exe_path = std::env::current_exe()?;
|
|
let new_exe_path = unpack(archive_data)?;
|
|
let permissions = fs::metadata(&old_exe_path)?.permissions();
|
|
fs::set_permissions(&new_exe_path, permissions)?;
|
|
check_exe(&new_exe_path, &install_version)?;
|
|
|
|
if !dry_run {
|
|
match output {
|
|
Some(path) => {
|
|
fs::rename(&new_exe_path, &path)
|
|
.or_else(|_| fs::copy(&new_exe_path, &path).map(|_| ()))?;
|
|
}
|
|
None => replace_exe(&new_exe_path, &old_exe_path)?,
|
|
}
|
|
}
|
|
|
|
println!("Upgrade done successfully");
|
|
|
|
Ok(())
|
|
}
|
|
|
|
fn download_package(
|
|
url: &Url,
|
|
client: Client,
|
|
version: &Version,
|
|
) -> Pin<Box<dyn Future<Output = Result<Vec<u8>, AnyError>>>> {
|
|
println!("downloading {}", url);
|
|
let url = url.clone();
|
|
let version = version.clone();
|
|
let fut = async move {
|
|
match fetch_once(client.clone(), &url, None).await {
|
|
Ok(result) => {
|
|
println!(
|
|
"Version has been found\nDeno is upgrading to version {}",
|
|
&version
|
|
);
|
|
match result {
|
|
FetchOnceResult::Code(source, _) => Ok(source),
|
|
FetchOnceResult::NotModified => unreachable!(),
|
|
FetchOnceResult::Redirect(_url, _) => {
|
|
download_package(&_url, client, &version).await
|
|
}
|
|
}
|
|
}
|
|
Err(_) => {
|
|
println!("Version has not been found, aborting");
|
|
std::process::exit(1)
|
|
}
|
|
}
|
|
};
|
|
fut.boxed_local()
|
|
}
|
|
|
|
fn compose_url_to_exec(version: &Version) -> Result<Url, AnyError> {
|
|
let s = format!(
|
|
"https://github.com/denoland/deno/releases/download/v{}/{}",
|
|
version, *ARCHIVE_NAME
|
|
);
|
|
Url::parse(&s).map_err(AnyError::from)
|
|
}
|
|
|
|
fn find_version(text: &str) -> Result<String, AnyError> {
|
|
let re = Regex::new(r#"v([^\?]+)?""#)?;
|
|
if let Some(_mat) = re.find(text) {
|
|
let mat = _mat.as_str();
|
|
return Ok(mat[1..mat.len() - 1].to_string());
|
|
}
|
|
Err(custom_error("NotFound", "Cannot read latest tag version"))
|
|
}
|
|
|
|
fn unpack(archive_data: Vec<u8>) -> Result<PathBuf, std::io::Error> {
|
|
// We use into_path so that the tempdir is not automatically deleted. This is
|
|
// useful for debugging upgrade, but also so this function can return a path
|
|
// to the newly uncompressed file without fear of the tempdir being deleted.
|
|
let temp_dir = TempDir::new()?.into_path();
|
|
let exe_ext = if cfg!(windows) { "exe" } else { "" };
|
|
let exe_path = temp_dir.join("deno").with_extension(exe_ext);
|
|
assert!(!exe_path.exists());
|
|
|
|
let archive_ext = Path::new(&*ARCHIVE_NAME)
|
|
.extension()
|
|
.and_then(|ext| ext.to_str())
|
|
.unwrap();
|
|
let unpack_status = match archive_ext {
|
|
"gz" => {
|
|
let exe_file = fs::File::create(&exe_path)?;
|
|
let mut cmd = Command::new("gunzip")
|
|
.arg("-c")
|
|
.stdin(Stdio::piped())
|
|
.stdout(Stdio::from(exe_file))
|
|
.spawn()?;
|
|
cmd.stdin.as_mut().unwrap().write_all(&archive_data)?;
|
|
cmd.wait()?
|
|
}
|
|
"zip" if cfg!(windows) => {
|
|
let archive_path = temp_dir.join("deno.zip");
|
|
fs::write(&archive_path, &archive_data)?;
|
|
Command::new("powershell.exe")
|
|
.arg("-NoLogo")
|
|
.arg("-NoProfile")
|
|
.arg("-NonInteractive")
|
|
.arg("-Command")
|
|
.arg(
|
|
"& {
|
|
param($Path, $DestinationPath)
|
|
trap { $host.ui.WriteErrorLine($_.Exception); exit 1 }
|
|
Add-Type -AssemblyName System.IO.Compression.FileSystem
|
|
[System.IO.Compression.ZipFile]::ExtractToDirectory(
|
|
$Path,
|
|
$DestinationPath
|
|
);
|
|
}",
|
|
)
|
|
.arg("-Path")
|
|
.arg(format!("'{}'", &archive_path.to_str().unwrap()))
|
|
.arg("-DestinationPath")
|
|
.arg(format!("'{}'", &temp_dir.to_str().unwrap()))
|
|
.spawn()?
|
|
.wait()?
|
|
}
|
|
"zip" => {
|
|
let archive_path = temp_dir.join("deno.zip");
|
|
fs::write(&archive_path, &archive_data)?;
|
|
Command::new("unzip")
|
|
.current_dir(&temp_dir)
|
|
.arg(archive_path)
|
|
.spawn()?
|
|
.wait()?
|
|
}
|
|
ext => panic!("Unsupported archive type: '{}'", ext),
|
|
};
|
|
assert!(unpack_status.success());
|
|
assert!(exe_path.exists());
|
|
Ok(exe_path)
|
|
}
|
|
|
|
fn replace_exe(new: &Path, old: &Path) -> Result<(), std::io::Error> {
|
|
if cfg!(windows) {
|
|
// On windows you cannot replace the currently running executable.
|
|
// so first we rename it to deno.old.exe
|
|
fs::rename(old, old.with_extension("old.exe"))?;
|
|
} else {
|
|
fs::remove_file(old)?;
|
|
}
|
|
// Windows cannot rename files across device boundaries, so if rename fails,
|
|
// we try again with copy.
|
|
fs::rename(new, old).or_else(|_| fs::copy(new, old).map(|_| ()))?;
|
|
Ok(())
|
|
}
|
|
|
|
fn check_exe(
|
|
exe_path: &Path,
|
|
expected_version: &Version,
|
|
) -> Result<(), AnyError> {
|
|
let output = Command::new(exe_path)
|
|
.arg("-V")
|
|
.stderr(std::process::Stdio::inherit())
|
|
.output()?;
|
|
let stdout = String::from_utf8(output.stdout)?;
|
|
assert!(output.status.success());
|
|
assert_eq!(stdout.trim(), format!("deno {}", expected_version));
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn test_find_version() {
|
|
let url = "<html><body>You are being <a href=\"https://github.com/denoland/deno/releases/tag/v0.36.0\">redirected</a>.</body></html>";
|
|
assert_eq!(find_version(url).unwrap(), "0.36.0".to_string());
|
|
}
|