| // 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 { |
| let hash = self.commit_hash.clone().unwrap_or_default(); |
| let hash_trimmed = hash.trim(); |
| |
| let date = self.commit_date.clone().unwrap_or_default(); |
| let date_trimmed = date.trim(); |
| |
| if (hash_trimmed.len() + date_trimmed.len()) > 0 { |
| write!( |
| f, |
| "{} {}.{}.{} ({} {})", |
| self.crate_name, self.major, self.minor, self.patch, hash_trimmed, date_trimmed, |
| )?; |
| } 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, 1); |
| 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.1"); |
| } |
| |
| #[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: 1 }" |
| ); |
| } |
| |
| } |