From c4101205da7d497dc90d19ff5641fcb939b53d2b Mon Sep 17 00:00:00 2001
From: sadasilv <sadasilv@redhat.com>
Date: Tue, 3 Dec 2024 11:05:30 -0500
Subject: [PATCH] utils/fwutils.py: Add fixboard setup to fw reflash

---
 utils/fwutils.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/utils/fwutils.py b/utils/fwutils.py
index 6989afa..f63747a 100644
--- a/utils/fwutils.py
+++ b/utils/fwutils.py
@@ -48,8 +48,9 @@ def __init__(
                 "clean_up_imc",
                 "flash_ssd_image",
                 "flash_spi_image",
+                "apply_fixboard",
             ]
-        self.steps_to_run = steps_to_run
+        self.steps_to_run = steps_to_run if not dry_run else []
         if self.dry_run:
             self.logger.info(
                 "DRY RUN, This is just a preview of the actions that will be taken"
@@ -138,6 +139,15 @@ def reflash_ipu(self) -> None:
         else:
             self.logger.info("Skipping flash_spi_image")
 
+        self.logger.info("Step 5: apply_fixboard")
+        if self.should_run("apply_fixboard"):
+            if self.fixboard_is_needed():
+                self.logger.info("Applying fixboard!")
+                self.apply_fixboard()
+            else:
+                self.logger.info("Fixboard not needed!")
+        else:
+            self.logger.info("Skipping applying_fixboard")
         # Step 5: Reboot IMC
         self.logger.info("Done!")
         self.logger.info(f"Please cold reboot IMC at {self.imc_address}")