run.sh

#!/bin/sh

# find connected Trezor T device
find_trezor_usb() {

    # get Trezor T USB path    
    TREZOR="$(lsusb -d $1 | awk '{print $2,$4}' | tr ' ' '/' | tr -d ':')"
    TREZOR_USB_PATH="/dev/bus/usb/${TREZOR}"

    # check if Trezor T is connected to pc
    if [ -z $TREZOR ]; then
        echo "[-][ERROR] Please connect Trezor T"
        exit 0;
    fi

    # check if path to USB exists
    if ! [ -e $TREZOR_USB_PATH ]; then
        echo "[-][ERROR] Path $TREZOR_USB_PATH to USB device not found."
        echo "[-][ERROR] Please change TREZOR_USB_PATH so it reflect your OS.\n"
        exit 0;
    fi

    # export variable for docker-compose
    export TREZOR_USB_PATH
}

# upload firmwate with Tesos baking support to Trezor T device 
upload_firmware(){

        # swith Trezor T to bootloader mode
        # ask for confirmation device will be wiped out
        
        echo "Please turn on bootloader mode on your Trezor T : "
        echo "    1. disconnect Trezor."
        echo "    2. hold your Trezor with your finger already on the screen at the bottom."
        echo "    3. plug the cable in."
        echo "    4. slide the finger up.\n"

        echo "After 5+ minutes you will be requested to confirm new firmware on you Trezor T.\n"

        echo "\033[1;33mWARNING:\e[0m Device will be completly ereased. Please backup your mnemonic."
        read -p "Are you sure you want to upload new firmware ? [Y/n] " RESPONSE
        if ! [ "$RESPONSE" = "y" ] && ! [ "$RESPONSE" = "" ]; then
            exit 0
        fi
        
        # find connected Trezor T device in bootloader mode 
        find_trezor_usb "1209:53c0"

        # launch docker-compose
        docker-compose -f docker-compose.firmware.yml pull &&
        docker-compose -f docker-compose.firmware.yml up
}


initialize() {

    echo "1. download faucets from https://faucet.tzalpha.net/"
    echo "2. save it to ./tezos-client/faucet"
    echo "3. transfer XTZ from activated accounts to delegate "
    echo "4. register delegate for baking\n"

    # find connected Trezor T device
    find_trezor_usb "1209:53c1"

    # launch docker-compose
    docker-compose -f docker-compose.initialize.yml pull &&
    docker-compose -f docker-compose.initialize.yml up
}

baking_start() {

    # read env variables from config 
    export $(grep -v '^#' config.baking.env | xargs -d '\n')

    # find connected Trezor T device
    find_trezor_usb "1209:53c1"

    # launch docker-compose
    docker-compose -f docker-compose.baking-start.yml pull &&
    docker-compose -f docker-compose.baking-start.yml up
}

baking_stop() {

    # read env variables from config 
    export $(grep -v '^#' config.baking.env | xargs -d '\n')

    # find connected Trezor T device
    find_trezor_usb "1209:53c1"

    # launch docker-compose
    # docker-compose -f docker-compose.baking-stop.yml pull &&
    docker-compose -f docker-compose.baking-stop.yml up
}


baking_start_debug() {

    # read env variables from config 
    export $(grep -v '^#' config.debug.env | xargs -d '\n')

    # use Trezor emulator form trezor-core
   
    # find connected Trezor T device
    find_trezor_usb "1209:53c1"

    # launch docker-compose
    docker-compose -f docker-compose.debug.yml pull &&
    docker-compose -f docker-compose.debug.yml up
}


#  cli
while :; do
  case $1 in

    -u|--upload-firmware)
        echo "\033[1;37mUpload firmware to Trezor T with support for Tezos baking\e[0m\n";
        upload_firmware
        ;;

    -i|--initialize)
        echo "\033[1;37mInitialize baking on Tezos\e[0m\n";
        initialize
        ;;

    -b|--start)
        echo "\033[1;37mStart banking & endorsing\e[0m\n";
        baking_start
        ;;

    -s|--stop)
        echo "\033[1;37mStop banking mode\e[0m\n";
        baking_stop
        ;;

    -d|--debug)
        echo "\033[1;37mStart banking & endorsing in debug mode \e[0m\n";
        baking_start_debug
        ;;

    -h|--help)
        echo "Usage:"
        echo "run.sh [OPTION]\n"
        echo "Set of tools for baking on Tezos with Trezor T support \n"
        echo " -u,  --upload-firmware   upload firmware with support for Tezos baking"
        echo " -i,  --initialize        activate faucets, transfer XTZ, register delegate"
        echo " -b,  --start             start baking and endorsing"
        echo " -s,  --stop              stop baking mode"
        echo " -d,  --debug             debug mode suited for development"
        echo " -h,  --help              display this message"
        exit 0
        ;;
    
    --)              
        shift
        break   
        ;;

    -?*)
        printf 'Unknown option: %s\n' "$1" >&2
        echo "(run $0 -h for help)\n"
        ;;
  
    ?*)
        echo "Missing option"
        echo "(run $0 -h for help)\n"
        ;;
  
    *)
        break
    
    esac

    shift
done