From c3cfd0875994335b1fb4a6a89656e742337d82d4 Mon Sep 17 00:00:00 2001
From: ROMemories <romain.fouquet@inria.fr>
Date: Thu, 18 Apr 2024 15:38:46 +0200
Subject: [PATCH] refactor(arch): use `cfg_if` to select the arch module

This constructs is a bit longer, but it makes it more obvious what needs
to be done when adding a new arch module.
It also provides a better error message if the architecture is not
supported.
---
 src/riot-rs-embassy/src/lib.rs | 25 +++++++++++++++++--------
 1 file changed, 17 insertions(+), 8 deletions(-)

diff --git a/src/riot-rs-embassy/src/lib.rs b/src/riot-rs-embassy/src/lib.rs
index 251a38788..0a849512c 100644
--- a/src/riot-rs-embassy/src/lib.rs
+++ b/src/riot-rs-embassy/src/lib.rs
@@ -6,14 +6,23 @@
 
 pub mod define_peripherals;
 
-#[cfg_attr(context = "nrf", path = "arch/nrf.rs")]
-#[cfg_attr(context = "rp2040", path = "arch/rp2040.rs")]
-#[cfg_attr(context = "esp", path = "arch/esp.rs")]
-#[cfg_attr(
-    not(any(context = "nrf", context = "rp2040", context = "esp")),
-    path = "arch/dummy.rs"
-)]
-pub mod arch;
+cfg_if::cfg_if! {
+    if #[cfg(context = "nrf")] {
+        #[path = "arch/nrf.rs"]
+        pub mod arch;
+    } else if #[cfg(context = "rp2040")] {
+        #[path = "arch/rp2040.rs"]
+        pub mod arch;
+    } else if #[cfg(context = "esp")] {
+        #[path = "arch/esp.rs"]
+        pub mod arch;
+    } else if #[cfg(context = "riot-rs")] {
+        compile_error!("this architecture is not supported");
+    } else {
+        #[path = "arch/dummy.rs"]
+        pub mod arch;
+    }
+}
 
 #[cfg(feature = "usb")]
 pub mod usb;