add build_rs to Generation struct

Author: jacobrosenthal

src/lib.rs

@@ -449,10 +449,12 @@ mod generate;
 mod util;
 
 pub use util::Target;
+use util::build_rs;
 
 pub struct Generation {
     pub lib_rs: String,
     pub device_x: String,
+    pub build_rs: String,
 }
 
 type Result<T> = std::result::Result<T, SvdError>;
@@ -482,6 +484,7 @@ pub fn generate(xml: &str, target: &Target, nightly: bool) -> Result<Generation>
     Ok(Generation {
         lib_rs: lib_rs,
         device_x: device_x,
+        build_rs: build_rs().to_string(),
     })
 }
 

src/main.rs

@@ -23,7 +23,7 @@ use quote::Tokens;
 use clap::{App, Arg};
 
 use errors::*;
-use util::Target;
+use util::{buld_rs, Target};
 
 fn run() -> Result<()> {
     use std::io::Read;
@@ -122,28 +122,3 @@ fn main() {
         process::exit(1);
     }
 }
-
-fn build_rs() -> Tokens {
-    quote! {
-        use std::env;
-        use std::fs::File;
-        use std::io::Write;
-        use std::path::PathBuf;
-
-        fn main() {
-            if env::var_os("CARGO_FEATURE_RT").is_some() {
-                // Put the linker script somewhere the linker can find it
-                let out = &PathBuf::from(env::var_os("OUT_DIR").unwrap());
-                File::create(out.join("device.x"))
-                    .unwrap()
-                    .write_all(include_bytes!("device.x"))
-                    .unwrap();
-                println!("cargo:rustc-link-search={}", out.display());
-
-                println!("cargo:rerun-if-changed=device.x");
-            }
-
-            println!("cargo:rerun-if-changed=build.rs");
-        }
-    }
-}

src/util.rs

@@ -311,3 +311,28 @@ pub fn only_registers(ercs: &[Either<Register, Cluster>]) -> Vec<&Register> {
         .collect();
     registers
 }
+
+pub fn build_rs() -> Tokens {
+    quote! {
+        use std::env;
+        use std::fs::File;
+        use std::io::Write;
+        use std::path::PathBuf;
+
+        fn main() {
+            if env::var_os("CARGO_FEATURE_RT").is_some() {
+                // Put the linker script somewhere the linker can find it
+                let out = &PathBuf::from(env::var_os("OUT_DIR").unwrap());
+                File::create(out.join("device.x"))
+                    .unwrap()
+                    .write_all(include_bytes!("device.x"))
+                    .unwrap();
+                println!("cargo:rustc-link-search={}", out.display());
+
+                println!("cargo:rerun-if-changed=device.x");
+            }
+
+            println!("cargo:rerun-if-changed=build.rs");
+        }
+    }
+}