Update TIldagon wifi network settings

update_wifi.sh

#!/bin/bash

# Script to set the WiFi SSID and password on the Tildagon
# Author: Andy Piper

#echo "This script sets the WiFi SSID and password on the Tildagon."
#echo "Usage: $0 [port] [ssid] [password]"
#echo "  port      : The USB port for the Tildagon (e.g., /dev/ttyUSB0)"
#echo "  ssid      : The WiFi SSID"
#echo "  password  : The WiFi password"
#echo ""

# Check if the correct number of arguments is provided
if [ $# -ne 3 ]; then
    echo "Error: Incorrect number of arguments."
    echo "Usage: $0 [port] [ssid] [password]"
    exit 1
fi

port=$1
ssid=$2
password=$3

# Check if the SSID is empty
if [ -z "$ssid" ]; then
    echo "Error: SSID cannot be empty."
    exit 1
fi

# Update WiFi settings on the board
output=$(mpremote connect $port exec "
import settings
try:
    settings.set('wifi_ssid', '$ssid')
    settings.set('wifi_password', '$password')
    settings.save()
    print('WiFi settings updated successfully.')
except Exception as e:
    print('Error updating WiFi settings:', str(e))
")

echo "$output"

# Check if the WiFi settings were updated successfully
if echo "$output" | grep -q "Error updating WiFi settings"; then
    echo "Failed to update WiFi settings on the Tildagon."
    exit 1
else
    echo "WiFi settings updated successfully on the Tildagon."
    # Reset the badge
    mpremote connect $port reset
    exit 0
fi