#!/bin/bash
#
# > make a file executable
# chmod +x ./install.sh
set -e

DEFAULT_DIRNAME="$HOME/suno"
TEMPDIR=$(mktemp -d)

cleanup() {
    rm -rf "$TEMPDIR"
}
trap cleanup EXIT

#
# SUNO binary Installation Steps:
#
# 1. Check user architecture
# 2. Ask user for installation path
# 3. Fetch the latest version from GitHub releases
# 4. Download and install suno
#
read -p "> Enter SUNO installation path [default: $DEFAULT_DIRNAME]: " DIRNAME </dev/tty
# Shorthand for "if empty, use default"
DIRNAME="${DIRNAME:-$DEFAULT_DIRNAME}"

echo "✔︎ Output directory $DIRNAME"

# Determine the architecture and set the filename accordingly
ARCH=$(uname -m)
case $ARCH in
    x86_64) TARGET=x86_64-unknown-linux-gnu ;;
    arm64)  TARGET=aarch64-apple-darwin ;;
    *)         echo "ERROR: Unknown architecture $ARCH"; exit 1 ;;
esac

# Fetch the latest version from GitHub releases
LATEST_VERSION=$(curl -L -s -H 'Accept: application/json' https://github.com/turboflakes/suno/releases/latest | sed -e 's/.*"tag_name":"\([^"]*\)".*/\1/')

if [ -z "$LATEST_VERSION" ] || [ "$LATEST_VERSION" == "null" ]; then
    echo "ERROR: Could not fetch latest version info"
    exit 1
fi

TARBALL_FILENAME=suno-$TARGET.tar.gz
TARBALL_FILENAME_SHA256=suno-$TARGET.tar.gz.sha256
URI="https://github.com/turboflakes/suno/releases/download/$LATEST_VERSION/$TARBALL_FILENAME"
URI_SHA256="https://github.com/turboflakes/suno/releases/download/$LATEST_VERSION/$TARBALL_FILENAME_SHA256"

echo "✔︎ Downloading suno $LATEST_VERSION"
cd "$TEMPDIR"

wget -q --show-progress "$URI" -O $TARBALL_FILENAME || { echo "ERROR: Failed to download the $TARBALL_FILENAME file"; exit 1; }
wget -q --show-progress "$URI_SHA256" -O $TARBALL_FILENAME_SHA256 || { echo "ERROR: Failed to download $TARBALL_FILENAME_SHA256 file"; exit 1; }

# Cross-platform checksum (Mac/Linux)
CHECK_CMD="sha256sum"
if ! command -v sha256sum &> /dev/null; then CHECK_CMD="shasum -a 256"; fi

if $CHECK_CMD -c "$TARBALL_FILENAME_SHA256" 2>&1 | grep -q 'OK'; then
    echo "✔︎ Checksum verified"
    mkdir -p "$DIRNAME"

    # Backup existing binary
    FILENAME="$DIRNAME/suno"
    if [[ -f "$FILENAME" ]]; then
        mv "$FILENAME" "$FILENAME.backup"
        echo "✔︎ Existing binary backed up to $FILENAME.backup"
    fi

    # Extract the tarball
    tar xzf $TEMPDIR/$TARBALL_FILENAME suno
    echo "✔︎ Checking if suno exists: $(ls -l)"

    if [[ ! -f suno ]]; then
        echo "ERROR: Binary suno does not exist"; exit 1
    fi

    # Install suno at $DIRNAME´
    install -m 755 suno "$FILENAME"

    echo "✔︎ Successfully installed suno $LATEST_VERSION at $FILENAME"

else
    echo "ERROR: SHA256 checksum verification failed";
fi

# Configuration Installation Steps:
#
# 1. Ask user if wants to install
# 2. Install a default configuration template if the user accepts
#
DEFAULT_CONFIG="$DIRNAME/.config.yaml"
read -p "> Would you like to install the DEFAULT configuration file? [y/N]: " INSTALL_CONFIG </dev/tty
if [[ "$INSTALL_CONFIG" == "y" || "$INSTALL_CONFIG" == "Y" ]]; then
    read -p "> Enter the configuration path [default: $DEFAULT_CONFIG]: " CONFIG </dev/tty
    CONFIG="${CONFIG:-$DEFAULT_CONFIG}"

    if [ ! -f "$CONFIG" ]; then
        echo "✔︎ Writing $CONFIG template"
        cat <<EOF > "$CONFIG"
chains:
  - polkadot:
      rpc_url: "ws://10.10.10.1:9944"
      validators:
        - "5GTD7ZeD823BjpmZBCSzBQp7cvHR1Gunq7oDkurZr9zUev2n"
  - asset_hub_polkadot:
      rpc_url: "ws://10.10.10.2:9944"
  - people_polkadot:
      rpc_url: "ws://10.10.10.3:9944"

features:
  enable_validators: true

# Builtin themes: ["Suno Dark", "Suno Light"]
themes:
  active: "Suno Dark"
  path: "./themes"

# Uncomment to use a signer proxy
# signer:
#   proxy_path: ".proxy_account.json"

explorer:
  url: "https://polkadot.js.org/apps/?rpc=wss://{chain}.rpc.turboflakes.io#/explorer/query/{block_hash}"
  # NOTE: Other explorers
  # url: "https://dev.papi.how/explorer/{block_hash}#networkId=localhost&endpoint=wss://{chain}.rpc.turboflakes.io"
  # url: "https://polkadot.chainconsole.com/apps/?rpc=wss://{chain}.rpc.turboflakes.io#/explorer/query/{block_hash}"
EOF
    echo "✔︎ Config file saved at $CONFIG. ";
    echo "==> Next edit the config file and replace STASHES and RPC endpoints as you wish.";
    else
        echo "✔︎ Config file $CONFIG already exists. Configuration skipped";
    fi
else
    echo "✔︎ Configuration skipped";
fi

echo "✔︎ Installation complete";
echo "— Enjoy suno $LATEST_VERSION";