#!/bin/bash -e

echo Stopping daemons
snapctl stop --disable solana.daemon-drone
snapctl stop --disable solana.daemon-leader
snapctl stop --disable solana.daemon-validator

mode="$(snapctl get mode)"
if [[ -z "$mode" ]]; then
  exit 0
fi

$SNAP/bin/setup.sh "$(snapctl get num-tokens)"

case $mode in
leader+drone)
  snapctl start --enable solana.daemon-leader
  snapctl start --enable solana.daemon-drone
  ;;
leader)
  snapctl start --enable solana.daemon-leader
  ;;
validator)
  snapctl start --enable solana.daemon-validator
  ;;
*)
  echo "Error: Unknown mode: $mode"
  exit 1
  ;;
esac