From fab3337f072919c59cd873ec2aa0512bee74dfb4 Mon Sep 17 00:00:00 2001
From: Antoine Viallon <antoine@lesviallon.fr>
Date: Fri, 17 Mar 2023 00:26:52 +0100
Subject: [PATCH] [Optimizations] Add a way to override some build dependencies

---
 optimizations.nix | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/optimizations.nix b/optimizations.nix
index 3c13503..fae1c77 100644
--- a/optimizations.nix
+++ b/optimizations.nix
@@ -72,6 +72,19 @@ let
     if (hasAttr "overrideAttrs" pkg) then
       let
         optimizedAttrs = _optimizeAttrs (attrs // {inherit level; go = (hasAttr "GOARCH" pkg); });
+        _nativeBuildInputs = filter (p: ! isNull p) (pkg.nativeBuildInputs or []);
+        _nativeBuildInputsOverriden = forEach _nativeBuildInputs (_pkg:
+          let
+            _pkgName = getName _pkg;
+            hasOverride = any (n: n == _pkgName) (attrNames cfg.overrideMap);
+            _overridePkg = if hasOverride then cfg.overrideMap.${_pkgName} else null;
+          in
+            if hasOverride then
+              warn "Replacing build dependency '${_pkgName}' by '${getName _overridePkg}'" _overridePkg
+            else
+              _pkg
+        );
+              
         _buildInputs = filter (p: ! isNull p ) (pkg.buildInputs or []);
         _buildInputsOverriden = forEach _buildInputs (_pkg:
           if (any (n: n == getName _pkg) cfg.blacklist) then
@@ -89,6 +102,7 @@ let
             pkg.overrideAttrs (old: {}
               // {
                 buildInputs = _buildInputsOverriden;
+                nativeBuildInputs = _nativeBuildInputsOverriden;
               }
               // optionalAttrs (hasAttr "CFLAGS" old) {
                 CFLAGS = if (! isList old.CFLAGS ) then [ old.CFLAGS ] else old.CFLAGS;
@@ -129,6 +143,19 @@ in
       description = "Blacklist specific packages from optimizations";
       type = types.listOf types.str;
     };
+    overrideMap = mkOption {
+      type = with types; attrsOf package;
+      default = {
+      };
+      example = literalExpression
+        ''
+          {
+            ninja = pkgs.ninja-samurai;
+            cmake = pkgs.my-cmake-override;
+          }
+        '';
+      description = mdDoc "Allow overriding packages found in `nativeBuildInputs` with custom packages.";
+    };
   };
 
   config = mkIf cfg.enable {