1 files changed, 65 insertions, 8 deletions

diff --git a/src/expand/cfg.cpp b/src/expand/cfg.cpp
index d178b9ae..3def0150 100644
--- a/src/expand/cfg.cpp
+++ b/src/expand/cfg.cpp
@@ -3,11 +3,68 @@
 #include <parse/tokentree.hpp>
 #include <parse/lex.hpp>
 #include <parse/common.hpp>
+#include "cfg.hpp"
 
-bool check_cfg(const ::AST::MetaItem& mi) {
-    // TODO: Handle cfg conditions
-    throw ::std::runtime_error("TODO: Handle #[cfg] or cfg! conditions");
-    return true;
+#include <map>
+#include <set>
+
+::std::map< ::std::string, ::std::string>   g_cfg_values;
+::std::set< ::std::string >   g_cfg_flags;
+
+void Cfg_SetFlag(::std::string name) {
+    g_cfg_flags.insert( mv$(name) );
+}
+void Cfg_SetValue(::std::string name, ::std::string val) {
+    g_cfg_values.insert( ::std::make_pair(mv$(name), mv$(name)) );
+}
+
+bool check_cfg(Span sp, const ::AST::MetaItem& mi) {
+    
+    if( mi.has_sub_items() ) {
+        // Must be `any`/`not`/`all`
+        if( mi.name() == "any" || mi.name() == "cfg" ) {
+            for(const auto& si : mi.items()) {
+                if( check_cfg(sp, si) )
+                    return true;
+            }
+            return false;
+        }
+        else if( mi.name() == "not" ) {
+            if( mi.items().size() != 1 )
+                ERROR(sp, E0000, "cfg(not()) with != 1 argument");
+            return !check_cfg(sp, mi.items()[0]);
+        }
+        else if( mi.name() == "all" ) {
+            for(const auto& si : mi.items()) {
+                if( ! check_cfg(sp, si) )
+                    return false;
+            }
+            return true;
+        }
+        else {
+            // oops
+            ERROR(sp, E0000, "Unknown cfg() function - " << mi.name());
+        }
+    }
+    else if( mi.has_string() ) {
+        // Equaliy
+        auto it = g_cfg_values.find(mi.name());
+        if( it != g_cfg_values.end() )
+        {
+            return it->second == mi.string();
+        }
+        else
+        {
+            WARNING(sp, W0000, "Unknown cfg() param '" << mi.name() << "'");
+            return false;
+        }
+    }
+    else {
+        // Flag
+        auto it = g_cfg_flags.find(mi.name());
+        return (it != g_cfg_flags.end());
+    }
+    BUG(sp, "Fell off the end of check_cfg");
 }
 
 class CCfgExpander:
@@ -25,7 +82,7 @@ class CCfgExpander:
         auto attrs = Parse_MetaItem(lex);
         DEBUG("cfg!() - " << attrs);
         
-        if( check_cfg(attrs) ) {
+        if( check_cfg(sp, attrs) ) {
             return box$( TTStreamO(TokenTree(TOK_RWORD_TRUE )) );
         }
         else {
@@ -42,7 +99,7 @@ class CCfgHandler:
     
     
     void handle(const AST::MetaItem& mi, AST::Crate& crate, AST::MacroInvocation& mac) const override {
-        if( check_cfg(mi) ) {
+        if( check_cfg(mac.span(), mi) ) {
             // Leave as is
         }
         else {
@@ -50,7 +107,7 @@ class CCfgHandler:
         }
     }
     void handle(const AST::MetaItem& mi, AST::Crate& crate, const AST::Path& path, AST::Module& mod, AST::Item&i) const override {
-        if( check_cfg(mi) ) {
+        if( check_cfg(Span(), mi) ) {
             // Leave
         }
         else {
@@ -58,7 +115,7 @@ class CCfgHandler:
         }
     }
     void handle(const AST::MetaItem& mi, ::std::unique_ptr<AST::ExprNode>& expr) const override {
-        if( check_cfg(mi) ) {
+        if( check_cfg(Span(expr->get_pos()), mi) ) {
             // Leave
         }
         else {