cortex_m_rtic_macros/
lib.rs

1#![doc(
2    html_logo_url = "https://raw.githubusercontent.com/rtic-rs/cortex-m-rtic/master/book/en/src/RTIC.svg",
3    html_favicon_url = "https://raw.githubusercontent.com/rtic-rs/cortex-m-rtic/master/book/en/src/RTIC.svg"
4)]
5//deny_warnings_placeholder_for_ci
6
7extern crate proc_macro;
8
9use proc_macro::TokenStream;
10use std::{env, fs, path::Path};
11
12use rtic_syntax::Settings;
13
14mod analyze;
15mod check;
16mod codegen;
17#[cfg(test)]
18mod tests;
19
20/// Attribute used to declare a RTIC application
21///
22/// For user documentation see the [RTIC book](https://rtic.rs)
23///
24/// # Panics
25///
26/// Should never panic, cargo feeds a path which is later converted to a string
27#[proc_macro_attribute]
28pub fn app(args: TokenStream, input: TokenStream) -> TokenStream {
29    let mut settings = Settings::default();
30    settings.optimize_priorities = false;
31    settings.parse_binds = true;
32    settings.parse_extern_interrupt = true;
33
34    let (app, analysis) = match rtic_syntax::parse(args, input, settings) {
35        Err(e) => return e.to_compile_error().into(),
36        Ok(x) => x,
37    };
38
39    let extra = match check::app(&app, &analysis) {
40        Err(e) => return e.to_compile_error().into(),
41        Ok(x) => x,
42    };
43
44    let analysis = analyze::app(analysis, &app);
45
46    let ts = codegen::app(&app, &analysis, &extra);
47
48    // Default output path: <project_dir>/target/
49    let mut out_dir = Path::new("target");
50
51    // Get output directory from Cargo environment
52    // TODO don't want to break builds if OUT_DIR is not set, is this ever the case?
53    let out_str = env::var("OUT_DIR").unwrap_or_else(|_| "".to_string());
54
55    // Assuming we are building for a thumbv* target
56    let target_triple_prefix = "thumbv";
57
58    // Check for special scenario where default target/ directory is not present
59    //
60    // This is configurable in .cargo/config:
61    //
62    // [build]
63    // target-dir = "target"
64    #[cfg(feature = "debugprint")]
65    println!("OUT_DIR\n{:#?}", out_str);
66
67    if out_dir.exists() {
68        #[cfg(feature = "debugprint")]
69        println!("\ntarget/ exists\n");
70    } else {
71        // Set out_dir to OUT_DIR
72        out_dir = Path::new(&out_str);
73
74        // Default build path, annotated below:
75        // $(pwd)/target/thumbv7em-none-eabihf/debug/build/cortex-m-rtic-<HASH>/out/
76        // <project_dir>/<target-dir>/<TARGET>/debug/build/cortex-m-rtic-<HASH>/out/
77        //
78        // traverse up to first occurrence of TARGET, approximated with starts_with("thumbv")
79        // and use the parent() of this path
80        //
81        // If no "target" directory is found, <project_dir>/<out_dir_root> is used
82        for path in out_dir.ancestors() {
83            if let Some(dir) = path.components().last() {
84                if dir
85                    .as_os_str()
86                    .to_str()
87                    .unwrap()
88                    .starts_with(target_triple_prefix)
89                {
90                    if let Some(out) = path.parent() {
91                        out_dir = out;
92                        #[cfg(feature = "debugprint")]
93                        println!("{:#?}\n", out_dir);
94                        break;
95                    }
96                    // If no parent, just use it
97                    out_dir = path;
98                    break;
99                }
100            }
101        }
102    }
103
104    // Try to write the expanded code to disk
105    if let Some(out_str) = out_dir.to_str() {
106        #[cfg(feature = "debugprint")]
107        println!("Write file:\n{}/rtic-expansion.rs\n", out_str);
108        fs::write(format!("{}/rtic-expansion.rs", out_str), ts.to_string()).ok();
109    }
110
111    ts.into()
112}
cortex_m_rtic_macros/lib.rs

cortex_m_rtic_macros/
lib.rs
