From a738555f02626685f119fe332d4e2e6e9f2581f4 Mon Sep 17 00:00:00 2001
From: Boris Kolpackov <boris@codesynthesis.com>
Date: Wed, 29 Nov 2023 09:51:43 +0200
Subject: Add rule for extracting C and C++ predefs

---
 libbuild2/c/init.cxx | 59 ++++++++++++++++++++++++++++++++++++++++++----------
 1 file changed, 48 insertions(+), 11 deletions(-)

(limited to 'libbuild2/c/init.cxx')

diff --git a/libbuild2/c/init.cxx b/libbuild2/c/init.cxx
index ebf0631..cb0469c 100644
--- a/libbuild2/c/init.cxx
+++ b/libbuild2/c/init.cxx
@@ -479,11 +479,11 @@ namespace build2
 
     bool
     as_cpp_init (scope& rs,
-                scope& bs,
-                const location& loc,
-                bool,
-                bool,
-                module_init_extra&)
+                 scope& bs,
+                 const location& loc,
+                 bool,
+                 bool,
+                 module_init_extra&)
     {
       tracer trace ("c::as_cpp_init");
       l5 ([&]{trace << "for " << bs;});
@@ -513,17 +513,54 @@ namespace build2
       return true;
     }
 
+    bool
+    predefs_init (scope& rs,
+                  scope& bs,
+                  const location& loc,
+                  bool,
+                  bool,
+                  module_init_extra&)
+    {
+      tracer trace ("c::predefs_init");
+      l5 ([&]{trace << "for " << bs;});
+
+      // We only support root loading (which means there can only be one).
+      //
+      if (rs != bs)
+        fail (loc) << "c.predefs module must be loaded in project root";
+
+      module* mod (rs.find_module<module> ("c"));
+
+      if (mod == nullptr)
+        fail (loc) << "c.predefs module must be loaded after c module";
+
+      // Register the c.predefs rule.
+      //
+      // Why invent a separate module instead of just always registering it in
+      // the c module? The reason is performance: this rule will be called for
+      // every C header.
+      //
+      cc::predefs_rule& r (*mod);
+
+      rs.insert_rule<h> (perform_update_id,   r.rule_name, r);
+      rs.insert_rule<h> (perform_clean_id,    r.rule_name, r);
+      rs.insert_rule<h> (configure_update_id, r.rule_name, r);
+
+      return true;
+    }
+
     static const module_functions mod_functions[] =
     {
       // NOTE: don't forget to also update the documentation in init.hxx if
       //       changing anything here.
 
-      {"c.guess",  nullptr, guess_init},
-      {"c.config", nullptr, config_init},
-      {"c.objc",   nullptr, objc_init},
-      {"c.as-cpp", nullptr, as_cpp_init},
-      {"c",        nullptr, init},
-      {nullptr,    nullptr, nullptr}
+      {"c.guess",   nullptr, guess_init},
+      {"c.config",  nullptr, config_init},
+      {"c.objc",    nullptr, objc_init},
+      {"c.as-cpp",  nullptr, as_cpp_init},
+      {"c.predefs", nullptr, predefs_init},
+      {"c",         nullptr, init},
+      {nullptr,     nullptr, nullptr}
     };
 
     const module_functions*
-- 
cgit v1.1