Merge ~alextu/pc-enablement/+git/prepare-checkbox-sanity:interactive-wifi-ssid-setting into ~oem-solutions-engineers/pc-enablement/+git/prepare-checkbox-sanity:master

[Alex Tu]

[Yuan-Chen Cheng (ycheng-twn)]

can you make sure that it runs well?

review: Needs Information

[Yuan-Chen Cheng (ycheng-twn)]

if modify the config file is simple enough, and the temporary config won't be saved...

Shall we consider just ask user to modify the config file.

diff --git a/usr/sbin/checkbox-run-plan b/usr/sbin/checkbox-run-plan
index 48f07c1..bc2859b 100755
--- a/usr/sbin/checkbox-run-plan
+++ b/usr/sbin/checkbox-run-plan
@@ -3,6 +3,7 @@
 EXCLUDE=""
 DEF_ENV="/etc/default/prepare-checkbox-sanity.conf"
 source $DEF_ENV
+WORKING_DIR="$(mktemp -d)"
 
 usage() {
 cat << EOF
@@ -82,6 +83,11 @@ done
 
 [ -z "$PLAN" ] && error
 
+declare -A WPA_SSID_ARRAY
+declare -A WPA_PWD_ARRAY
+declare -A OPEN_SSID_ARRAY
+supported_wpa_wifi="BG N AC"
+supported_open_wifi="BG N AC"
 interactive(){
     echo "do you want to set the path of checkbox.conf? (refer to https://checkbox.readthedocs.io/en/latest/launcher-tutorial.html)"
     read -p "(Y/N):" set_checkbox_conf
@@ -90,6 +96,32 @@ interactive(){
     else
         echo "Keep the checkbox.conf as default from $DEF_ENV"
     fi
+
+    for wifi_proto in $supported_wpa_wifi; do
+        echo "do you want to set your WPA_""$wifi_proto""_SSID for wifi testing with password?(e.g. ubuntu-cert-{wifi protocal}-wpa-tpelab)"
+        read -p "(Y/N):" set_ssid
+        if [[ $set_ssid == [Yy] ]] ; then
+            read -p "WPA_""$wifi_proto""_SSID(wifi SSID):" WPA_SSID_ARRAY[$wifi_proto]
+            read -p "WPA_""$wifi_proto""_PSK(password):" WPA_PWD_ARRAY[$wifi_proto]
+        fi
+
+        echo "do you want to set your OPEN_""$wifi_proto""_SSID for wifi testing without password?(e.g. ubuntu-cert-{wifi protocal}-open-tpelab)"
+        read -p "(Y/N):" set_ssid
+        if [[ $set_ssid == [Yy] ]] ; then
+            read -p "OPEN_""$wifi_proto""_SSID(wifi SSID):" OPEN_SSID_ARRAY[$wifi_proto]
+        fi
+    done
+    if [ "${#WPA_SSID_ARRAY[@]}" != 0 ] ||[ "${#OPEN_SSID_ARRAY[@]}" != 0 ] ;then
+        echo "[environment]" > "$WORKING_DIR/wifi_env_vars"
+        for a_key in ${!WPA_SSID_ARRAY[@]}; do
+            echo "WPA_""$a_key""_SSID = ""${WPA_SSID_ARRAY[$a_key]}" >> "$WORKING_DIR/wifi_env_vars"
+            echo "WPA_""$a_key""_PSK = ""${WPA_PWD_ARRAY[$a_key]}" >> "$WORKING_DIR/wifi_env_vars"
+        done
+
+        for a_key in ${!OPEN_SSID_ARRAY[@]}; do
+            echo "OPEN_""$a_key""_SSID = ""${OPEN_SSID_ARRAY[$a_key]}" >> "$WORKING_DIR/wifi_env_vars"
+        done
+    fi
 }
 [ "$SILENCE" == "TRUE"  ] || interactive
 if [ -n "$CHECKBOX_CONF" ] && [ -z "${CHECKBOX_CONF##http*}" ]; then
@@ -97,7 +129,7 @@ if [ -n "$CHECKBOX_CONF" ] && [ -z "${CHECKBOX_CONF##http*}" ]; then
     if wget --waitretry=5 -T 5 -t 3 $CHECKBOX_CONF -O $HOME/.cache/prepare-checkbox/checkbox.conf; then
         CHECKBOX_CONF="$HOME/.cache/prepare-checkbox/checkbox.conf"
     else
-        CHECKBOX_CONF="usr/lib/prepare-checkbox-sanity/fall-back-checkbox.conf"
+        CHECKBOX_CONF="/usr/lib/prepare-checkbox-sanity/fall-back-checkbox.conf"
     fi
 fi
 
@@ -128,6 +160,8 @@ exclude = $EXCLUDE
 
 [ui]
 type = silent
+
+$(cat "$WORKING_DIR/wifi_env_vars")
 EOF
 echo "========= the checkbox launcher will be run ========="
 cat "$HOME/generated_launcher"