optionally pass env to AddBuildMiddleware callback

platformio · Aug 5, 2022 · bd8009d · bd8009d
1 parent 17a7293
commit bd8009d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/platformio/builder/tools/piobuild.py b/platformio/builder/tools/piobuild.py
@@ -292,7 +292,10 @@ def CollectBuildFiles(
         for callback, pattern in middlewares:
             if pattern and not fnmatch.fnmatch(node.srcnode().get_path(), pattern):
                 continue
-            new_node = callback(new_node)
+            if callback.__code__.co_argcount == 2:
+                new_node = callback(env, new_node)
+            else:
+                new_node = callback(new_node)
         if new_node:
             new_sources.append(new_node)