From 557e56021395bf008573ca2a22995216d31db72a Mon Sep 17 00:00:00 2001
From: zhiayang <gpg@tsumaran.ai>
Date: Fri, 9 Feb 2024 12:07:51 -0500
Subject: [PATCH] make setup.sh prompt for board revision for boards that have
 revisions

---
 docs/src/templates/setup.sh.mustache | 31 ++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/docs/src/templates/setup.sh.mustache b/docs/src/templates/setup.sh.mustache
index c711dbc5..09ed8c29 100644
--- a/docs/src/templates/setup.sh.mustache
+++ b/docs/src/templates/setup.sh.mustache
@@ -122,6 +122,9 @@ if [ "$keyboard_shield" == "y" ]; then
     board_ids=({{#boards}}"{{id}}" {{/boards}})
     boards_usb_only=({{#boards}}"{{#usb_only}}y{{/usb_only}}{{^usb_only}}n{{/usb_only}}" {{/boards}})
 
+    boards_revisions=({{#boards}}"{{#revisions}}{{.}} {{/revisions}}" {{/boards}})
+    boards_default_revision=({{#boards}}"{{{default_revision}}}" {{/boards}})
+
     echo ""
     echo "MCU Board Selection:"
     PS3="$prompt "
@@ -151,6 +154,34 @@ if [ "$keyboard_shield" == "y" ]; then
 
         esac
     done
+
+    if [ -n "${boards_revisions[$board_index]}" ]; then
+        read -a _valid_revisions <<< "${boards_revisions[$board_index]}"
+        for (( _i=0; _i<${#_valid_revisions}; _i++ )); do
+            if [ "${boards_default_revision[board_index]}" = "${_valid_revisions[_i]}" ]; then
+                _valid_revisions[_i]+=" (default)"
+            fi
+        done
+
+        echo ""
+        echo "MCU Board Revision:"
+        select opt in "${_valid_revisions[@]}" "Quit"; do
+            ''|*[!0-9]*) echo "Invalid option. Try another one."; continue;;
+
+            $(( ${#_valid_revisions[@]}+1 )) ) echo "Goodbye!"; exit 1;;
+            *)
+                if [ $REPLY -gt $(( ${#_valid_revisions[@]}+1 )) ] || [ $REPLY -lt 0 ]; then
+                    echo "Invalid option. Try another one."
+                    continue
+                fi
+
+                _rev_index=$(( $REPLY-1 ))
+                board="${board_ids[$board_index]}@${_valid_revisions[_rev_index]}"
+                boards=( "${board}" )
+                break
+            ;;
+        done
+    fi
 else
     board=${keyboard}
     boards=$keyboard_siblings