From 580ba6a62b3d93364b758d0eecb24b117363632e Mon Sep 17 00:00:00 2001
From: Evan Pratten <evan@ewpratten.com>
Date: Thu, 9 Nov 2023 13:11:42 -0500
Subject: [PATCH] Add a pre-check

---
 scripts/qmk-helper | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/scripts/qmk-helper b/scripts/qmk-helper
index 2305283..ec3821d 100755
--- a/scripts/qmk-helper
+++ b/scripts/qmk-helper
@@ -140,7 +140,7 @@ def main() -> int:
 
     # Refresh the QMK repository
     refresh_qmk_repo()
-    
+
     # Figure out the keymap name
     keymap = args.keyboard.split("/")[0]
 
@@ -151,6 +151,14 @@ def main() -> int:
     if args.mode == "build":
         build_keymap(args.keyboard)
     elif args.mode == "flash":
+        # Make sure that the flash mode is valid
+        if args.keyboard == "ferris-sweep" and args.flash_mode not in ["dfu-split-left", "dfu-split-right"]:
+            logger.error(
+                "Invalid flash mode. Must be one of: dfu-split-left, dfu-split-right"
+            )
+            return 1
+        
+        # Flash
         flash_keymap(args.keyboard, args.flash_mode)
 
     return 0