#!/usr/bin/env bash
#
# Can inject correctly formatted version strings/numbers in all the various
# project metadata files. Can also back them up and restore them.

set -eu

SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
cd "$SCRIPT_DIR"

COMMAND="$1"
shift 1

INCLUDED_CRATES=(
    "mullvad-daemon"
    "mullvad-cli"
    "mullvad-problem-report"
    "mullvad-setup"
    "mullvad-exclude"
    "talpid-openvpn-plugin"
)
MANIFESTS=( "${INCLUDED_CRATES[@]/%//Cargo.toml}" )

function inject_version {
    # Regex that only matches valid Mullvad VPN versions. It also captures
    # relevant values into capture groups, read out via BASH_REMATCH[x].
    local VERSION_REGEX="^20([0-9]{2})\.([1-9][0-9]?)(-beta([1-9][0-9]?))?(-dev-[0-9a-f]+)?$"
    local product_version="$1"

    if [[ ! $product_version =~ $VERSION_REGEX ]]; then
        echo "Invalid version format. Please specify version as:"
        echo "<YEAR>.<NUMBER>[-beta<NUMBER>]"
        return 1
    fi

    local semver_version=$(echo "$product_version" | sed -Ee 's/($|-.*)/.0\1/g')
    local semver_major="20${BASH_REMATCH[1]}"
    local semver_minor=${BASH_REMATCH[2]}
    local semver_patch="0"

    if [[ "${2:-""}" != "--only-android" ]]; then
        # Electron GUI
        cp gui/package.json gui/package.json.bak
        cp gui/package-lock.json gui/package-lock.json.bak
        (cd gui/ && npm version "$semver_version" --no-git-tag-version --allow-same-version)

        # Windows C++
        cp dist-assets/windows/version.h dist-assets/windows/version.h.bak
        cat <<EOF > dist-assets/windows/version.h
#define MAJOR_VERSION $semver_major
#define MINOR_VERSION $semver_minor
#define PATCH_VERSION $semver_patch
#define PRODUCT_VERSION "$product_version"
EOF
    fi

    # Rust crates
    sed -i.bak -Ee "s/^version = \"[^\"]+\"\$/version = \"$semver_version\"/g" \
        "${MANIFESTS[@]}"

    # Android
    if [[ ("$(uname -s)" == "Linux") ]]; then
        local version_year=$(printf "%02d" "${BASH_REMATCH[1]}")
        local version_number=$(printf "%02d" "${BASH_REMATCH[2]}")
        local version_patch="00" # Not used for now.
        local version_beta=$(printf "%02d" "${BASH_REMATCH[4]:-99}")
        local android_version_code=${version_year}${version_number}${version_patch}${version_beta}

        cp android/build.gradle android/build.gradle.bak
        sed -i -Ee "s/versionCode [0-9]+/versionCode $android_version_code/g" \
            android/build.gradle
        sed -i -Ee "s/versionName \"[^\"]+\"/versionName \"$product_version\"/g" \
            android/build.gradle
    fi
}

function restore_backup {
    set +e

    if [[ "${1:-""}" != "--only-android" ]]; then
        # Electron GUI
        mv gui/package.json.bak gui/package.json
        mv gui/package-lock.json.bak gui/package-lock.json
        # Windows C++
        mv dist-assets/windows/version.h.bak dist-assets/windows/version.h
    fi

    # Rust crates
    for toml in "${MANIFESTS[@]}"; do
        mv "${toml}.bak" "${toml}"
    done
    # Android
    if [[ ("$(uname -s)" == "Linux") ]]; then
        mv android/build.gradle.bak android/build.gradle
    fi
    set -e
}

function delete_backup {
    set +e

    if [[ "${1:-""}" != "--only-android" ]]; then
        # Electron GUI
        rm gui/package.json.bak
        rm gui/package-lock.json.bak
        # Windows C++
        rm dist-assets/windows/version.h.bak
    fi

    # Rust crates
    for toml in "${MANIFESTS[@]}"; do
        rm "${toml}.bak"
    done
    # Android
    if [[ ("$(uname -s)" == "Linux") ]]; then
        rm android/build.gradle.bak
    fi
    set -e
}

case "$COMMAND" in
    "inject")
        inject_version "$@"
        ;;
    "restore-backup")
        restore_backup "$@"
        ;;
    "delete-backup")
        delete_backup "$@"
        ;;
    *)
        echo "Invalid command"
        exit 1
        ;;
esac