#!/usr/bin/env bash

set -e

echo "Upgrading Karrio. This will cause a few minutes of downtime."
read -r -p "Do you want to upgarde Karrio? [y/N] " response
if [[ "$response" =~ ^([yY][eE][sS]|[yY])+$ ]]; then
    echo "OK!"
else
    exit
fi

echo "Checking for named postgres volumes to avoid data loss when upgrading"
if docker volume ls | grep -e 'postgres-data'; then
    DOCKER_VOLUMES_MISSING=FALSE
    echo "Found postgres volume, proceeding..."
else
    DOCKER_VOLUMES_MISSING=TRUE
    echo ""
    echo ""
    echo "🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨"
    echo "🚨🚨🚨🚨🚨 WARNING: POTENTIAL DATA LOSS 🚨🚨🚨🚨🚨🚨"
    echo "🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨"
    echo ""
    echo ""
    echo "We were unable to find named postgres volume."
    echo ""
    echo "WE STRONGLY RECOMMEND YOU:"
    echo ""
    echo "🛑 Stop this script and do not proceed"
    echo "✅ Back up your entire environment/installation (vm, host, etc.), including all docker containers and volumes:"
    echo "✅ Specifically back up the contents of :"
    echo "  ☑ /var/lib/postgresql/data in the db (*_db_1) container"
    echo "and be ready to check/recopy the data before you boot Karrio next."
    read -r -p "Do you want to proceed anyway? [y/N] " response
    if [[ "$response" =~ ^([yY][eE][sS]|[yY])+$ ]]; then
        echo "OK!"
    else
        exit
    fi
fi

[[ -f ".env" ]] && export $(cat .env | xargs) || (echo "No .env file found. Please create it with SECRET_KEY and DOMAIN set." && exit 1)
export KARRIO_TAG="${KARRIO_TAG:-2025.5rc31}"

# get karrio scripts
mkdir -p ./karrio
cd karrio

echo "Downloading the latest karrio's installation files"
curl https://raw.githubusercontent.com/karrioapi/karrio/HEAD/docker/docker-compose.hobby.yml -o docker-compose.hobby.yml

cd -

# Upgrade Docker Compose to version 2.13.0
echo "Setting up Docker Compose"
sudo rm /usr/local/bin/docker-compose
sudo curl -L "https://github.com/docker/compose/releases/download/v2.13.0/docker-compose-$(uname -s)-$(uname -m)" -o /usr/local/bin/docker-compose || true
sudo chmod +x /usr/local/bin/docker-compose

rm -f docker-compose.yml
cp karrio/docker-compose.hobby.yml docker-compose.yml.tmpl
envsubst <docker-compose.yml.tmpl >docker-compose.yml
rm docker-compose.yml.tmpl

docker-compose pull

echo "Stopping the stack!"
docker-compose stop

# Patch existing Caddyfile to include X-Forwarded-Proto header if missing
if [[ -f "Caddyfile" ]]; then
    echo "Checking existing Caddyfile for HTTPS header forwarding..."
    TS="$(date +%Y%m%d_%H%M%S)"
    cp Caddyfile "Caddyfile.bak.${TS}" || true

    if ! grep -q "reverse_proxy http://api:5002 {" Caddyfile; then
        # Insert header block for API upstream
        sed -i.bak "s|reverse_proxy http://api:5002$|reverse_proxy http://api:5002 {\\n        header_up X-Forwarded-Proto https\\n    }|" Caddyfile || true
    fi

    if ! grep -q "reverse_proxy http://dashboard:3002 {" Caddyfile; then
        # Insert header block for Dashboard upstream
        sed -i.bak "s|reverse_proxy http://dashboard:3002$|reverse_proxy http://dashboard:3002 {\\n        header_up X-Forwarded-Proto https\\n    }|" Caddyfile || true
    fi

    echo "Caddyfile check complete. A backup was saved as Caddyfile.bak.${TS}"
fi

if [ ${DOCKER_VOLUMES_MISSING} == 'TRUE' ]; then
    echo ""
    echo ""
    echo "🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨"
    echo "🚨🚨🚨🚨🚨WARNING: LAST CHANCE TO AVOID DATA LOSS 🚨🚨🚨🚨🚨🚨"
    echo "🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨🚨"
    echo ""
    echo ""
    echo "Before we restart the stack, you should restore data you have backed up from the previous warning."
    echo ""
    echo ""
fi

read -r -p "Do you want to restart the Karrio stack now ? (docker-compose up) [y/N] " response
if [[ "$response" =~ ^([yY][eE][sS]|[yY])+$ ]]; then
    echo "OK, Restarting the stack!"
    sudo -E docker-compose up -d
else
    echo "OK, we are leaving the stack OFFLINE. Run 'sudo -E docker-compose up -d' when you are ready to start it."
    exit
fi

echo "Karrio upgraded successfully!"