inject-phase-timings.nix

{ lib
, phases ? [
    "Patch"
    "Unpack"
    "Configure"
    "Build"
    "Install"
    "Check"
    "InstallCheck"
  ]
}:

drv: let
  timePhase = phaseName:
    final: prev: let
      pre = "pre${phaseName}";
      post = "post${phaseName}";
    in {
      ${pre} = ''
        __${phaseName}Start=$(date +%s%3N)
      '' + (prev.${pre} or "");
      ${post} = (prev.${post} or "") + ''
        __${phaseName}End=$(date +%s%3N)
        echo -e "\e[1;33m[TIMINGS]\e[0m $(( __${phaseName}End - __${phaseName}Start ))ms for ${phaseName} phase"
      '';
    };

  overrides = lib.composeManyExtensions
    (builtins.map timePhase phases);
in drv.overrideAttrs overrides