1 files changed, 105 insertions, 0 deletions
diff --git a/src/bootstrap/channel.rs b/src/bootstrap/channel.rs
new file mode 100644
index 000000000..1932a0017
--- /dev/null
+++ b/src/bootstrap/channel.rs
@@ -0,0 +1,105 @@
+//! Build configuration for Rust's release channels.
+//!
+//! Implements the stable/beta/nightly channel distinctions by setting various
+//! flags like the `unstable_features`, calculating variables like `release` and
+//! `package_vers`, and otherwise indicating to the compiler what it should
+//! print out as part of its version information.
+
+use std::path::Path;
+use std::process::Command;
+
+use crate::util::output;
+use crate::Build;
+
+pub enum GitInfo {
+    /// This is not a git repository.
+    Absent,
+    /// This is a git repository.
+    /// If the info should be used (`ignore_git` is false), this will be
+    /// `Some`, otherwise it will be `None`.
+    Present(Option<Info>),
+}
+
+pub struct Info {
+    commit_date: String,
+    sha: String,
+    short_sha: String,
+}
+
+impl GitInfo {
+    pub fn new(ignore_git: bool, dir: &Path) -> GitInfo {
+        // See if this even begins to look like a git dir
+        if !dir.join(".git").exists() {
+            return GitInfo::Absent;
+        }
+
+        // Make sure git commands work
+        match Command::new("git").arg("rev-parse").current_dir(dir).output() {
+            Ok(ref out) if out.status.success() => {}
+            _ => return GitInfo::Absent,
+        }
+
+        // If we're ignoring the git info, we don't actually need to collect it, just make sure this
+        // was a git repo in the first place.
+        if ignore_git {
+            return GitInfo::Present(None);
+        }
+
+        // Ok, let's scrape some info
+        let ver_date = output(
+            Command::new("git")
+                .current_dir(dir)
+                .arg("log")
+                .arg("-1")
+                .arg("--date=short")
+                .arg("--pretty=format:%cd"),
+        );
+        let ver_hash = output(Command::new("git").current_dir(dir).arg("rev-parse").arg("HEAD"));
+        let short_ver_hash = output(
+            Command::new("git").current_dir(dir).arg("rev-parse").arg("--short=9").arg("HEAD"),
+        );
+        GitInfo::Present(Some(Info {
+            commit_date: ver_date.trim().to_string(),
+            sha: ver_hash.trim().to_string(),
+            short_sha: short_ver_hash.trim().to_string(),
+        }))
+    }
+
+    fn info(&self) -> Option<&Info> {
+        match self {
+            GitInfo::Present(info) => info.as_ref(),
+            GitInfo::Absent => None,
+        }
+    }
+
+    pub fn sha(&self) -> Option<&str> {
+        self.info().map(|s| &s.sha[..])
+    }
+
+    pub fn sha_short(&self) -> Option<&str> {
+        self.info().map(|s| &s.short_sha[..])
+    }
+
+    pub fn commit_date(&self) -> Option<&str> {
+        self.info().map(|s| &s.commit_date[..])
+    }
+
+    pub fn version(&self, build: &Build, num: &str) -> String {
+        let mut version = build.release(num);
+        if let Some(ref inner) = self.info() {
+            version.push_str(" (");
+            version.push_str(&inner.short_sha);
+            version.push(' ');
+            version.push_str(&inner.commit_date);
+            version.push(')');
+        }
+        version
+    }
+
+    pub fn is_git(&self) -> bool {
+        match self {
+            GitInfo::Absent => false,
+            GitInfo::Present(_) => true,
+        }
+    }
+}