From 4e48a73f9c342544e4eabd1c1cd31cdfb6a6e5e3 Mon Sep 17 00:00:00 2001
From: Edwin Cheng <edwin0cheng@gmail.com>
Date: Fri, 21 Feb 2020 10:04:03 +0800
Subject: Improve server version info

---
 crates/rust-analyzer/build.rs | 30 +++++++++++++++++++++++++++++-
 1 file changed, 29 insertions(+), 1 deletion(-)

(limited to 'crates')

diff --git a/crates/rust-analyzer/build.rs b/crates/rust-analyzer/build.rs
index 05f9772c0..cda8833d1 100644
--- a/crates/rust-analyzer/build.rs
+++ b/crates/rust-analyzer/build.rs
@@ -1,12 +1,40 @@
 //! Just embed git-hash to `--version`
 
-use std::process::Command;
+use std::{env, path::PathBuf, process::Command};
 
 fn main() {
+    let _ = set_rerun();
+
     let rev = rev().unwrap_or_else(|| "???????".to_string());
     println!("cargo:rustc-env=REV={}", rev)
 }
 
+fn set_rerun() {
+    let mut manifest_dir = PathBuf::from(
+        env::var("CARGO_MANIFEST_DIR").expect("`CARGO_MANIFEST_DIR` is always set by cargo."),
+    );
+
+    while manifest_dir.parent().is_some() {
+        if manifest_dir.join(".git/HEAD").exists() {
+            let git_dir = manifest_dir.join(".git");
+
+            println!("cargo:rerun-if-changed={}", git_dir.join("HEAD").display());
+            // current branch ref
+            if let Ok(output) =
+                Command::new("git").args(&["rev-parse", "--symbolic-full-name", "HEAD"]).output()
+            {
+                if let Ok(ref_link) = String::from_utf8(output.stdout) {
+                    println!("cargo:rerun-if-changed={}", git_dir.join(ref_link).display());
+                }
+            }
+            return;
+        }
+
+        manifest_dir.pop();
+    }
+    println!("cargo:warning=Could not find `.git/HEAD` from manifest dir!");
+}
+
 fn rev() -> Option<String> {
     let output = Command::new("git").args(&["rev-parse", "HEAD"]).output().ok()?;
     let stdout = String::from_utf8(output.stdout).ok()?;
-- 
cgit v1.2.3