Use buildrs for conditional compilation

1 files changed, 5 insertions, 0 deletions

diff --git a/build.rs b/build.rs
index 14c3d24..fee1a6a 100644
--- a/build.rs
+++ b/build.rs
@@ -1,8 +1,13 @@
 use std::env;
+use version_check;
 
 fn main() {
     let target = env::var("TARGET").unwrap();
 
+    if version_check::Channel::read().unwrap().is_nightly() {
+        println!("cargo:rustc-cfg=rustc_is_nightly")
+    }
+
     if target.starts_with("thumbv6m") {
         println!("cargo:rustc-cfg=armv6m")
     }