common.sh

# Set the nuspec version field for PACKAGE_NAME to VERSION
function bump-nuspec-version() {
    package="${1}"
    new_version="${2}"
    new_version_without_date=$(echo "${new_version}" | sed -r 's|\.[0-9]{8}$||') # Strip the YYYYMMDD suffix if present
    new_maj_min="${new_version_without_date%.*}"

    if [ -z "${package}" ] || [ -z "${new_version}" ]; then
        echo "Usage: ${FUNCNAME[0]} PACKAGE_NAME VERSION"
        return 1
    fi

    current_version=$(sed -nr 's|<version>(.+)</version>|\1|p' "${package}.nuspec" | xargs)
    current_version_without_date=$(echo "${current_version}" | sed -r 's|\.[0-9]{8}$||') # Strip the YYYYMMDD suffix if present
    current_maj_min="${current_version_without_date%.*}"

    echo "Current package version: ${current_version}"
    echo "Current version: ${current_version_without_date}"
    echo "Major minor: ${current_maj_min}"
    echo
    echo "New version: ${new_version}"
    echo "New version: ${new_version_without_date}"
    echo "Major minor: ${new_maj_min}"
    echo

    # Make sure we actually stripped a segment
    # And that we didn't strip too much
    if [ "${current_maj_min}" == "${current_version}" ] ||
    [ "${current_maj_min}" == "${current_maj_min%.*}" ]; then
        echo "Issues parsing current version"
        echo
        return 1
    fi

    # Replace the full version
    # Replace the major.minor version
    # TODO: This will replace the release note version with a version.YYYYMMDD if we're doing a Chocolatey-only release.
    sed -r -i "s|${current_version}|${new_version}|g" "${package}.nuspec"
    sed -r -i "s|${current_version_without_date}|${new_version_without_date}|g" "${package}.nuspec"
    sed -r -i "s|${current_maj_min}|${new_maj_min}|g" "${package}.nuspec"
}

function replace-checksums() {
    checksum="${1}"
    checksum_64="${2}"

    if [ -z "${checksum}" ] || [ -z "${checksum_64}" ]; then
        echo "Usage: ${FUNCNAME[0]} CHECKSUM CHECKSUM_64"
    fi

    echo "x86 Checksum: ${checksum}"
    echo "x64 Checksum: ${checksum_64}"
    echo

    replace-checksum checksum "${checksum}"
    replace-checksum checksum64 "${checksum_64}"
}

function replace-checksum() {
    key="${1}"
    checksum="${2}"

    if [ -z "${key}" ] || [ -z "${checksum}" ]; then
        echo "Usage: ${FUNCNAME[0]} KEY_NAME CHECKSUM"
    fi

    sed -r -i "s|(${key}[ ]+=[ ]+)\"[^\"]*\"|\1\"${checksum}\"|" tools/chocolateyInstall.ps1
}

function package-and-test() {
    package="${1}"
    version="${2}"

    if [ -z "${package}" ]; then
        echo "Usage: ${FUNCNAME[0]} PACKAGE_NAME"
        return 1
    fi

    windows_dir=$(powershell.exe -c 'echo "${PWD}"')
    echo "Windows Directory: ${windows_dir}"
    echo

    # Assume choco is on WSL path as well
    choco.exe pack "${package}.nuspec"

    script="
    choco install -y '${package}' --version '${version}' --source . --x86 ;
    echo '' ;
    echo '' ;
    choco uninstall -y --removedependencies '${package}' ;
    echo '' ;
    echo '' ;
    choco install -y '${package}' --version '${version}' --source . ;
    echo '' ;
    echo '' ;
    choco uninstall -y --removedependencies '${package}' ;
    echo '' ;
    echo '' ;
    pause ;
    "
    echo "Testing x86 install, x86 uninstall, x64 install, x64 uninstall..."

    # Start an admin PowerShell to take care of the install
    powershell.exe -c "Start-Process powershell.exe -ArgumentList \"
    echo ${windows_dir} ;
    cd ${windows_dir} ;
    ${script}
    \" -Verb RunAs"

    read -p "Waiting for any key..."
}

function choco-push() {
    package="${1}"
    version="${2}"

    if [ -z "${package}" ] || [ -z "${version}" ]; then
        echo "Usage: ${FUNCNAME[0]} PACKAGE_NAME VERSION"
        return 1
    fi

    choco.exe push -source https://push.chocolatey.org/ "${package}.${version}.nupkg"
}

function commit-and-push() {
    package="${1}"
    version="${2}"

    if [ -z "${package}" ] || [ -z "${version}" ]; then
        echo "Usage: ${FUNCNAME[0]} PACKAGE_NAME VERSION"
        return 1
    fi

    git add .
    git diff --cached

    echo
    read -p "Waiting for confirmation..."
    echo

    git commit -m "Release ${package} ${version}"
    git tag "${package}-${version}"
    git push --tags origin main
}

function calc-sha256() {
    binary="${1}"

    if [ -z "${binary}" ]; then
        echo "Usage: ${FUNCNAME[0]} PATH_TO_BINARY"
        return 1
    fi

    sha256sum -b "${binary}" | cut -d ' ' -f 1
}