Make it possible to pass the flash commands

Signed-off-by: Brian McGillion <[email protected]>
anduril · Dec 19, 2024 · 4dc2542 · 4dc2542
1 parent f3c691d
commit 4dc2542
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/modules/flash-script.nix b/modules/flash-script.nix
@@ -335,6 +335,18 @@ in
           default = [ ];
           description = "A list of paths to compiled .dtbo files to include with the UEFI image while flashing. These overlays are applied by UEFI at runtime";
         };
+
+        preFlashCommands = mkOption {
+          type = types.lines;
+          default = "";
+          description = "Additional commands to be added to the flash script while it is being constructed.";
+        };
+
+        postFlashCommands = mkOption {
+          type = types.lines;
+          default = "";
+          description = "Additional commands to be added to the flash script while it is being constructed.";
+        };
       };
 
       flashScript = mkOption {

diff --git a/overlay-with-config.nix b/overlay-with-config.nix
@@ -107,7 +107,7 @@ final: prev: (
       mkFlashScript = flash-tools: args: import ./device-pkgs/flash-script.nix ({
         inherit lib flash-tools;
         inherit (cfg.firmware) eksFile;
-        inherit (cfg.flashScriptOverrides) flashArgs partitionTemplate;
+        inherit (cfg.flashScriptOverrides) flashArgs partitionTemplate preFlashCommands postFlashCommands;
         inherit (finalJetpack) tosImage socType uefi-firmware;
 
         additionalDtbOverlays = args.additionalDtbOverlays or cfg.flashScriptOverrides.additionalDtbOverlays;