Pico W Device Maker

#!/bin/bash
# iot-maker-pico.sh — orchestrator for Pico W stem cell provisioning
# ─────────────────────────────────────────
# 20260417  Refactored into sub-scripts
# 20260416  Initial version
# ─────────────────────────────────────────
#
# Usage: ./iot-maker-pico.sh
#
# Calls in order:
#   scripts/nuke-pico.sh          (optional)
#   scripts/flash-pico.sh
#   scripts/wait-pico.sh
#   scripts/copy-files-pico.sh
#   scripts/register-pico.sh

set -e

SCRIPTS=~/iot-device/scripts

echo ""
echo "╔══════════════════════════════════════╗"
echo "║       IoT Device Maker               ║"
echo "║       Pico W stem cell               ║"
echo "╚══════════════════════════════════════╝"
echo ""


# ── Step 1: Nuke? ─────────────────────────────────────────────

echo "► Step 1: Prepare device"
echo "  Nuking is recommended — it ensures a clean flash."
read -p "  Nuke existing firmware first? (Y/n): " DO_NUKE
echo ""

if [[ ! "$DO_NUKE" =~ ^[Nn]$ ]]; then
    bash $SCRIPTS/nuke-pico.sh
else
    echo "  ⚠️  Skipping nuke — flash may fail on non-fresh devices"
    echo ""
fi


# ── Step 2: Flash MicroPython ─────────────────────────────────

bash $SCRIPTS/flash-pico.sh


# ── Step 3: Wait for MicroPython ─────────────────────────────

bash $SCRIPTS/wait-pico.sh


# ── Step 4: Ask WiFi credentials ─────────────────────────────

echo ""
echo "► Step 4: WiFi credentials"
echo "  (leave SSID blank to skip — device will start AP mode instead)"
echo ""
read -p "  SSID: " SSID

TMP_WIFI=""
if [ -n "$SSID" ]; then
    read -s -p "  Password: " PASSWORD
    echo ""
    TMP_WIFI=$(mktemp /tmp/wifi.XXXXXX.json)
    echo "{\"ssid\": \"$SSID\", \"password\": \"$PASSWORD\"}" > $TMP_WIFI
    export TMP_WIFI
else
    echo "  ⚠️  Skipped — device will start AP mode (PicoSetup / 12345678)"
fi


# ── Step 5: Copy files ────────────────────────────────────────

bash $SCRIPTS/copy-files-pico.sh "$TMP_WIFI"

[ -n "$TMP_WIFI" ] && rm -f $TMP_WIFI


# ── Step 6-7: Wait for registration ──────────────────────────

bash $SCRIPTS/register-pico.sh

#!/bin/bash
# nuke-pico.sh — wipe Pico W flash before provisioning
# ─────────────────────────────────────────
# 20260417  Initial version
# ─────────────────────────────────────────
#
# Usage: ./scripts/nuke-pico.sh
#        or called from iot-maker-pico.sh

set -e

PICO=~/iot-device/provision/pico_w
MOUNT=/media/$USER/RPI-RP2

if [ ! -f "$PICO/flash_nuke.uf2" ]; then
    echo "  ❌ flash_nuke.uf2 not found: $PICO/flash_nuke.uf2"
    exit 1
fi

echo "► Nuke: wipe existing firmware"
echo "  Hold BOOTSEL, plug in the device, then press ENTER"
read

echo "  Waiting for RPI-RP2..."
for i in $(seq 1 20); do
    DEV=$(lsblk -o NAME,SIZE | grep "128M" | awk '{print $1}' | head -1)
    if [ -n "$DEV" ]; then
        echo "  Found: /dev/$DEV"
        break
    fi
    sleep 1
done

if [ -z "$DEV" ]; then
    echo "  ❌ Device not found — did you hold BOOTSEL?"
    exit 1
fi

sudo mkdir -p $MOUNT

# Clean up any stale mounts from previous attempts
echo "  Cleaning up stale mounts..."
sudo umount -l $MOUNT 2>/dev/null || true
sudo umount -l $MOUNT 2>/dev/null || true
sudo umount -l $MOUNT 2>/dev/null || true
sudo umount -l $MOUNT 2>/dev/null || true

sudo mount /dev/${DEV}1 $MOUNT

echo "  Waiting for filesystem to initialise..."
sleep 5

echo "  Flashing flash_nuke.uf2..."
sudo cp $PICO/flash_nuke.uf2 $MOUNT/
# Do NOT sync or umount — let the Pico accept and disconnect naturally

echo "  ✅ Nuked — device rebooting"
sleep 3

echo ""
echo "  Unplug the device, then press ENTER"
read
echo ""

#!/bin/bash
# flash-pico.sh — flash MicroPython UF2 onto Pico W
# ─────────────────────────────────────────
# 20260417  Initial version
# ─────────────────────────────────────────
#
# Usage: ./scripts/flash-pico.sh
#        or called from iot-maker-pico.sh

set -e

PICO=~/iot-device/provision/pico_w
MOUNT=/media/$USER/RPI-RP2

if [ ! -f "$PICO/MicroPython.uf2" ]; then
    echo "❌ MicroPython.uf2 not found: $PICO/MicroPython.uf2"
    exit 1
fi

echo "► Step 2: Flash MicroPython"
echo "  Hold BOOTSEL, plug in the device, then press ENTER"
read

echo "  Waiting for RPI-RP2..."
for i in $(seq 1 20); do
    DEV=$(lsblk -o NAME,SIZE | grep "128M" | awk '{print $1}' | head -1)
    if [ -n "$DEV" ]; then
        echo "  Found: /dev/$DEV"
        break
    fi
    sleep 1
done

if [ -z "$DEV" ]; then
    echo "  ❌ Device not found — did you hold BOOTSEL?"
    exit 1
fi

sudo mkdir -p $MOUNT

# Clean up any stale mounts from previous attempts
echo "  Cleaning up stale mounts..."
sudo umount -l $MOUNT 2>/dev/null || true
sudo umount -l $MOUNT 2>/dev/null || true
sudo umount -l $MOUNT 2>/dev/null || true
sudo umount -l $MOUNT 2>/dev/null || true

sudo mount /dev/${DEV}1 $MOUNT

echo "  Waiting for filesystem to initialise..."
sleep 5

echo "  Flashing MicroPython.uf2..."
sudo cp $PICO/MicroPython.uf2 $MOUNT/
# Do NOT sync or umount — let the Pico accept the UF2 and disconnect naturally
# Forcing umount causes incomplete writes and boot loops

echo "  ✅ MicroPython flashed — waiting for device to reboot..."
sleep 3

#!/bin/bash
# wait-pico.sh — wait for Pico W to appear as MicroPython serial device (ttyACM0)
# ─────────────────────────────────────────
# 20260417  Watch for ttyACM0 specifically before attempting mpremote
# ─────────────────────────────────────────
#
# Usage: ./scripts/wait-pico.sh
#        or called from iot-maker-pico.sh

set -e

echo "► Step 3: Waiting for MicroPython..."
echo "  (watching for ttyACM0)"

# Step 1: Wait for ttyACM0 to appear
ACM_FOUND=0
for i in $(seq 1 40); do
    if [ -e /dev/ttyACM0 ]; then
        echo "  ✅ ttyACM0 detected"
        ACM_FOUND=1
        break
    fi
    echo "  ... waiting for ttyACM0 ($i/40)"
    sleep 1
done

if [ $ACM_FOUND -eq 0 ]; then
    echo "  ❌ ttyACM0 never appeared"
    echo "  Tip: check 'dmesg | tail -20' to see if device is detected"
    exit 1
fi

# Step 2: Give MicroPython a moment to settle, then verify with mpremote
sleep 5
if mpremote ls : > /dev/null 2>&1; then
    echo "  ✅ Device ready"
    exit 0
else
    echo "  ❌ ttyACM0 present but mpremote not responding"
    echo "  Tip: try 'mpremote ls :' manually"
    exit 1
fi

#!/bin/bash
# copy-files-pico.sh — copy provisioning files to Pico W in one mpremote session
# ─────────────────────────────────────────
# 20260417  Initial version
# ─────────────────────────────────────────
#
# Usage: ./scripts/copy-files-pico.sh [/tmp/wifi.json]
#        or called from iot-maker-pico.sh
#
# boot.py is copied LAST so it doesn't run during the session.
# Reset is included at the end of the mpremote chain.

set -e

PICO=~/iot-device/provision/pico_w
COMMON=~/iot-device/provision/common
TMP_WIFI=$1

if [ ! -f "$PICO/boot.py" ]; then
    echo "❌ boot.py not found: $PICO/boot.py"
    exit 1
fi
if [ ! -f "$COMMON/microdot.py" ]; then
    echo "❌ microdot.py not found: $COMMON/microdot.py"
    exit 1
fi

echo "► Step 5: Copying files..."

if [ -n "$TMP_WIFI" ] && [ -f "$TMP_WIFI" ]; then
    mpremote \
        cp $COMMON/microdot.py :microdot.py + \
        cp $TMP_WIFI :wifi.json + \
        cp $PICO/boot.py :boot.py + \
        reset 2>/dev/null || true
    echo "  ✅ microdot.py copied"
    echo "  ✅ wifi.json copied"
else
    mpremote \
        cp $COMMON/microdot.py :microdot.py + \
        cp $PICO/boot.py :boot.py + \
        reset 2>/dev/null || true
    echo "  ✅ microdot.py copied"
    echo "  ⚠️  wifi.json skipped — device will start AP mode"
fi

echo "  ✅ boot.py copied (last)"
echo "     MD5: $(md5sum $PICO/boot.py | awk '{print $1}')"

echo ""
echo "► Step 6: Device rebooting..."
echo "  ✅ Reset sent"

#!/bin/bash
# register-pico.sh — wait for newly provisioned Pico W to appear in database
# ─────────────────────────────────────────
# 20260417  Initial version
# ─────────────────────────────────────────
#
# Usage: ./scripts/register-pico.sh
#        or called from iot-maker-pico.sh
#
# Press Ctrl+C to skip if you don't want to wait.

DB_HOST=192.168.100.51
DB_USER=rforssen
DB_NAME=iot_registry

echo ""
echo "► Step 7: Waiting for device to register..."
echo "  (press Ctrl+C to skip)"
echo ""
read -s -p "  DB password: " DB_PASS
echo ""

REGISTERED=""
for i in $(seq 1 30); do
    RESULT=$(mysql -h $DB_HOST -u $DB_USER -p$DB_PASS $DB_NAME \
        -se "SELECT device_id, last_seen FROM devices ORDER BY registered_at DESC LIMIT 1;" 2>/dev/null)
    if [ -n "$RESULT" ]; then
        REGISTERED=$RESULT
        break
    fi
    echo "  ... waiting ($i/30)"
    sleep 2
done

echo ""
if [ -n "$REGISTERED" ]; then
    echo "╔══════════════════════════════════════╗"
    echo "║     ✅ Device provisioned            ║"
    echo "║     $REGISTERED"
    echo "╚══════════════════════════════════════╝"
else
    echo "╔══════════════════════════════════════╗"
    echo "║     ⚠️  Device not seen in DB yet    ║"
    echo "║     Check manually in a minute       ║"
    echo "╚══════════════════════════════════════╝"
fi