Expand - Support #[cfg] on use statements

1 files changed, 9 insertions, 0 deletions

diff --git a/src/expand/cfg.cpp b/src/expand/cfg.cpp
index 70ac6ac1..2a629432 100644
--- a/src/expand/cfg.cpp
+++ b/src/expand/cfg.cpp
@@ -117,6 +117,15 @@ class CCfgHandler:
             mac.clear();
         }
     }
+    void handle(const Span& sp, const AST::MetaItem& mi, ::AST::Crate& crate, AST::UseStmt& use) const override {
+        DEBUG("#[cfg] use - " << mi);
+        if( check_cfg(sp, mi) ) {
+            // Leave as is
+        }
+        else {
+            use.path = AST::Path();
+        }
+    }
     void handle(const Span& sp, const AST::MetaItem& mi, ::AST::Crate& crate, const AST::Path& path, AST::Module& mod, AST::Item&i) const override {
         DEBUG("#[cfg] item - " << mi);
         if( check_cfg(sp, mi) ) {