149 lines
4.3 KiB
Rust
149 lines
4.3 KiB
Rust
// Copyright 2014-2018 The Rust Project Developers. See the COPYRIGHT
|
|
// file at the top-level directory of this distribution.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
// option. This file may not be copied, modified, or distributed
|
|
// except according to those terms.
|
|
|
|
|
|
use std::env;
|
|
|
|
#[macro_export]
|
|
macro_rules! get_version_info {
|
|
() => {{
|
|
let major = env!("CARGO_PKG_VERSION_MAJOR").parse::<u8>().unwrap();
|
|
let minor = env!("CARGO_PKG_VERSION_MINOR").parse::<u8>().unwrap();
|
|
let patch = env!("CARGO_PKG_VERSION_PATCH").parse::<u16>().unwrap();
|
|
let crate_name = String::from(env!("CARGO_PKG_NAME"));
|
|
|
|
let host_compiler = $crate::get_channel();
|
|
let commit_hash = option_env!("GIT_HASH").map(|s| s.to_string());
|
|
let commit_date = option_env!("COMMIT_DATE").map(|s| s.to_string());
|
|
|
|
VersionInfo {
|
|
major,
|
|
minor,
|
|
patch,
|
|
host_compiler,
|
|
commit_hash,
|
|
commit_date,
|
|
crate_name,
|
|
}
|
|
}};
|
|
}
|
|
|
|
// some code taken and adapted from RLS and cargo
|
|
pub struct VersionInfo {
|
|
pub major: u8,
|
|
pub minor: u8,
|
|
pub patch: u16,
|
|
pub host_compiler: Option<String>,
|
|
pub commit_hash: Option<String>,
|
|
pub commit_date: Option<String>,
|
|
pub crate_name: String,
|
|
}
|
|
|
|
impl std::fmt::Display for VersionInfo {
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
if self.commit_hash.is_some() {
|
|
write!(
|
|
f,
|
|
"{} {}.{}.{} ({} {})",
|
|
self.crate_name,
|
|
self.major,
|
|
self.minor,
|
|
self.patch,
|
|
self.commit_hash.clone().unwrap_or_default().trim(),
|
|
self.commit_date.clone().unwrap_or_default().trim(),
|
|
)?;
|
|
} else {
|
|
write!(f, "{} {}.{}.{}", self.crate_name, self.major, self.minor, self.patch)?;
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
impl std::fmt::Debug for VersionInfo {
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
write!(
|
|
f,
|
|
"VersionInfo {{ crate_name: \"{}\", major: {}, minor: {}, patch: {}",
|
|
self.crate_name, self.major, self.minor, self.patch,
|
|
)?;
|
|
if self.commit_hash.is_some() {
|
|
write!(
|
|
f,
|
|
", commit_hash: \"{}\", commit_date: \"{}\" }}",
|
|
self.commit_hash.clone().unwrap_or_default().trim(),
|
|
self.commit_date.clone().unwrap_or_default().trim()
|
|
)?;
|
|
} else {
|
|
write!(f, " }}")?;
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
pub fn get_channel() -> Option<String> {
|
|
if let Ok(channel) = env::var("CFG_RELEASE_CHANNEL") {
|
|
Some(channel)
|
|
} else {
|
|
// we could ask ${RUSTC} -Vv and do some parsing and find out
|
|
Some(String::from("nightly"))
|
|
}
|
|
}
|
|
|
|
pub fn get_commit_hash() -> Option<String> {
|
|
std::process::Command::new("git")
|
|
.args(&["rev-parse", "--short", "HEAD"])
|
|
.output()
|
|
.ok()
|
|
.and_then(|r| String::from_utf8(r.stdout).ok())
|
|
}
|
|
|
|
pub fn get_commit_date() -> Option<String> {
|
|
std::process::Command::new("git")
|
|
.args(&["log", "-1", "--date=short", "--pretty=format:%cd"])
|
|
.output()
|
|
.ok()
|
|
.and_then(|r| String::from_utf8(r.stdout).ok())
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn test_struct_local() {
|
|
let vi = get_version_info!();
|
|
assert_eq!(vi.major, 0);
|
|
assert_eq!(vi.minor, 1);
|
|
assert_eq!(vi.patch, 0);
|
|
assert_eq!(vi.crate_name, "rustc_tools_util");
|
|
// hard to make positive tests for these since they will always change
|
|
assert!(vi.commit_hash.is_none());
|
|
assert!(vi.commit_date.is_none());
|
|
}
|
|
|
|
#[test]
|
|
fn test_display_local() {
|
|
let vi = get_version_info!();
|
|
assert_eq!(vi.to_string(), "rustc_tools_util 0.1.0");
|
|
}
|
|
|
|
#[test]
|
|
fn test_debug_local() {
|
|
let vi = get_version_info!();
|
|
let s = format!("{:?}", vi);
|
|
assert_eq!(
|
|
s,
|
|
"VersionInfo { crate_name: \"rustc_tools_util\", major: 0, minor: 1, patch: 0 }"
|
|
);
|
|
}
|
|
|
|
}
|