OctoPrint CLI flasher for Prusa MK2, MK3 and MMU2

flash.sh

#!/usr/bin/env bash

#
# This script was created to flash firmwares for the MK2, MK3 and the MMU2,
# all from the command line of your OctoPrint installation.
#
# The script assumes that you have avrdude and jq installed on your raspbian and that you
# have filled the API key below.
#
# Usage: flash.sh mk2|mk3|mmu2 firmware.hex
#
# If you are flashing a printer, it will use the API to disconnect it first if necessary
# (and reconnect it afterwards)
#
# The MMU2 can only be flashed while in bootloader mode, which only 5 lasts seconds.
# That's why this script is started first, it then waits until you go press the reset button
# and then starts flashing as soon as the MMU2 reconnects.
#
# I have tested this on all my devices (MK2.5S, MK3 and MMU2) and it works, but:
# THIS SCRIPT COMES WITH NO WARRANTY WHATSOEVER, USE AT YOUR OWN RISK!
#

API_KEY=<YOUR_API_KEY_HERE>

function device {
    searching=true
    while [ $searching == true ]; do
        # https://unix.stackexchange.com/a/144735
        for sysdevpath in $(find /sys/bus/usb/devices/usb*/ -name dev); do
            # Get rid of "dev"
            syspath="${sysdevpath%/dev}"
            # Translate from sys to dev
            devname="$(udevadm info -q name -p $syspath)"
            # Only keep devices, we don't want controllers/hubs
            [[ "$devname" == "bus/"* ]] && continue
            # Export the device properties as environment settings
            eval "$(udevadm info -q property --export -p $syspath)"
            [[ -z "$ID_SERIAL" ]] && continue
            if [[ $ID_SERIAL == *${1}* ]]; then
                echo /dev/$devname
                searching=false
            fi
        done
        if [ $searching == true ]; then
            sleep 1
        fi
    done
}

function api {
    # http://docs.octoprint.org/en/master/api/connection.html
    if [ "$2" == POST ]; then
        curl -s --header "X-Api-Key: $API_KEY" --header "Content-Type: application/json" --data "$3" http://127.0.0.1:5000/api/$1
    else
        curl -s --header "X-Api-Key: $API_KEY" http://127.0.0.1:5000/api/$1
    fi
}

if [ "$1" == MMU ] || [ "$1" == mmu ] || [ "$1" == MMU2 ] || [ "$1" == mmu2 ]; then
    echo Flashing MMU2
    echo Beware: If you have modem manager installed, flashing will fail
    echo "(see https://github.com/prusa3d/MM-control-01#linux-1)"

    DEV=$(device Multi_Material_2.0)
    # Count USB DEVICES
    ORIGINAL_USB_DEVICES=$(lsusb | wc -l)
    USB_DEVICES=$ORIGINAL_USB_DEVICES
    echo Device found at $DEV, please hit the reset button now...
    while [ $USB_DEVICES -eq $ORIGINAL_USB_DEVICES ]; do USB_DEVICES=$(lsusb | wc -l) ; done
    while [ $USB_DEVICES -ne $ORIGINAL_USB_DEVICES ]; do USB_DEVICES=$(lsusb | wc -l) ; done

    DEV=$(device Multi_Material_2.0)
    echo Reset detected, device now at $DEV
    echo Starting flash process...
    avrdude -v -p atmega32u4 -c avr109 -P ${DEV} -b 57600 -D -U flash:w:${2}:i

elif [ $1 == MK3 ] || [ $1 == mk3 ] || [ $1 == MK2 ] || [ $1 == mk2 ]; then

    if [ $1 == MK3 ] || [ $1 == mk3 ]; then
        NAME=MK3
        DEV=$(device MK3)
    elif [ $1 == MK2 ] || [ $1 == mk2 ]; then
        NAME=MK2
        DEV=$(device MK2)
    fi

    echo Flashing $NAME, found at $DEV

    # Get the printer connection status
    CONNECTION=$(api connection)
    STATUS=$(echo $CONNECTION | jq -r .current.state)

    if [ "$STATUS" != Closed ]; then
        # Make sure the printer is disconnected before we do anything
        echo Printer is connected, disconnecting
        api connection POST '{"command": "disconnect"}'
    else
        echo "Printer not connected, disconnect not needed (will not auto-reconnect after flash)"
    fi

    echo Starting flash process via $DEV...
    avrdude -v -p atmega2560 -c wiring -P ${DEV} -D -U flash:w:${2}:i

    if [ $? -eq 0 ] && [ "$STATUS" == Operational ]; then
        echo Flash successful, reconnecting $NAME
        BAUDRATE=$(echo $CONNECTION | jq -r .current.baudrate)
        PROFILE=$(echo $CONNECTION | jq -r .current.printerProfile)
        COMMAND="{\"command\":\"connect\",\"port\":\"$DEV\",\"baudrate\":$BAUDRATE,\"printerProfile\":\"$PROFILE\",\"save\":true,\"autoconnect\":true}"
        api connection POST "$COMMAND"
    fi
fi