Add pre/post hooks to custom phases

anduril · Aug 21, 2024 · 893b0f6 · 893b0f6
1 parent 3b47152
commit 893b0f6
Show file tree

Hide file tree

Showing 7 changed files with 28 additions and 0 deletions.
diff --git a/modules/default.nix b/modules/default.nix
@@ -235,9 +235,13 @@ in
           name = "l4t-core-wrapper";
           phases = [ "installPhase" ];
           installPhase = ''
+            runHook preInstall
+
             mkdir -p $out/lib
             ln -s ${l4t-core}/lib/libnvrm_gpu.so $out/lib/libnvrm_gpu.so
             ln -s ${l4t-core}/lib/libnvrm_mem.so $out/lib/libnvrm_mem.so
+
+            runHook postInstall
           '';
         };
       in

diff --git a/pkgs/board-automation/default.nix b/pkgs/board-automation/default.nix
@@ -27,7 +27,11 @@ stdenv.mkDerivation {
   '';
 
   installPhase = ''
+    runHook preInstall
+
     mkdir -p $out/bin
     cp tools/board_automation/* $out/bin
+
+    runHook postInstall
   '';
 }
diff --git a/pkgs/cuda-packages/default.nix b/pkgs/cuda-packages/default.nix
@@ -152,7 +152,11 @@ let
       '';
 
       installPhase = ''
+        runHook preInstall
+
         cp -r . $out
+
+        runHook postInstall
       '';
 
       meta = {

diff --git a/pkgs/flash-tools/default.nix b/pkgs/flash-tools/default.nix
@@ -94,8 +94,12 @@ let
     noDumpEnvVars = true;
 
     installPhase = ''
+      runHook preInstall
+
       mkdir -p $out
       cp -r . $out/
+
+      runHook postInstall
     '';
 
     # Stuff to put into PATH for flash.sh

diff --git a/pkgs/jetson-benchmarks/default.nix b/pkgs/jetson-benchmarks/default.nix
@@ -95,6 +95,8 @@ stdenvNoCC.mkDerivation {
     substituteInPlace utils/utilities.py --replace "/usr/src/tensorrt" "${cudaPackages.tensorrt}"
   '';
   installPhase = ''
+    runHook preInstall
+
     mkdir -p models
     cp -r $models/. models
 
@@ -104,6 +106,8 @@ stdenvNoCC.mkDerivation {
     mkdir -p $out/bin
     cp -r ${./scripts}/. $out/bin
     chmod +x $out/bin/*
+
+    runHook postInstall
   '';
   postFixup = ''
     wrapProgram $out/bin/run-jetson-benchmarks \

diff --git a/pkgs/l4t/default.nix b/pkgs/l4t/default.nix
@@ -84,7 +84,11 @@ let
       '';
 
       installPhase = ''
+        runHook preInstall
+
         cp -r . $out
+
+        runHook postInstall
       '';
 
       meta = {

diff --git a/pkgs/ota-utils/default.nix b/pkgs/ota-utils/default.nix
@@ -10,6 +10,8 @@ stdenvNoCC.mkDerivation {
   dontBuild = true;
 
   installPhase = ''
+    runHook preInstall
+
     mkdir -p $out/bin $out/share
     cp ${./ota-setup-efivars.sh} $out/bin/ota-setup-efivars
     cp ${./ota-apply-capsule-update.sh} $out/bin/ota-apply-capsule-update
@@ -28,5 +30,7 @@ stdenvNoCC.mkDerivation {
 
     # patchShebangs does not seem to work here for some reason
     substituteInPlace $out/bin/* --replace '#!/usr/bin/env bash' '#!${bash}/bin/bash'
+
+    runHook postInstall
   '';
 }