#!/usr/bin/env bash

# TODO: Validate checksum
function atfile.update() {
    cmd="$1"
    unset error
    is_git=0

    if [ -x "$(command -v git)" ] && [[ -d "$_prog_dir/.git" ]] && [[ "$(atfile.util.get_realpath "$(pwd)")" == "$_prog_dir" ]]; then
        is_git=1
    fi

    if [[ "$cmd" == "check-only" ]]; then
        # shellcheck disable=SC2154
        [[ $_disable_update_checking == 1 ]] && return
        # shellcheck disable=SC2154
        [[ $_disable_update_command == 1 ]] && return
        # shellcheck disable=SC2154
        [[ $is_git == 1 && $_enable_update_git_clobber == 0 ]] && return
        # shellcheck disable=SC2154
        [[ $_output_json == 1 ]] && return

        last_checked="$(atfile.cache.get "update-check")"
        current_checked="$(atfile.util.get_date "" "%s")"
        check_sleep=3600
        next_check=$(( last_checked + check_sleep ))

        atfile.say.debug "Checking for last update check...\n↳ Last: $last_checked\n↳ Cur.: $current_checked\n↳ Next: $next_check"

        if [[ $(( next_check < current_checked )) == 0 ]]; then
            return
        else
            last_checked="$(atfile.cache.set "update-check" "$current_checked")"
        fi
    fi

    [[ $_output_json == 1 ]] && atfile.die "Command not available as JSON"

    # shellcheck disable=SC2154
    update_did="$_devel_dist_username"

    atfile.util.override_actor "$update_did"

    atfile.say.debug "Getting latest release..."
    latest_release_record="$(com.atproto.repo.getRecord "$update_did" "self.atfile.latest" "self")"
    error="$(atfile.util.get_xrpc_error $? "$latest_release_record")"

    [[ -n "$error" ]] && atfile.die "Unable to get latest version" "$error"

    latest_version="$(echo "$latest_release_record" | jq -r '.value.version')"
    latest_version_commit="$(echo "$latest_release_record" | jq -r '.value.commit')"
    latest_version_date="$(echo "$latest_release_record" | jq -r '.value.releasedAt')"
    parsed_latest_version="$(atfile.util.parse_version "$latest_version")"
    parsed_running_version="$(atfile.util.parse_version "$_version")"
    latest_version_record_id="atfile-$parsed_latest_version"
    update_available=0
    
    atfile.say.debug "Checking version...\n↳ Latest: $latest_version ($parsed_latest_version)\n ↳ Date: $latest_version_date\n ↳ Commit: $latest_version_commit\n↳ Running: $_version ($parsed_running_version)"

    if [[ $(( parsed_latest_version > parsed_running_version )) == 1 ]]; then
        update_available=1
    fi

    case "$cmd" in
        "check-only")
            if [[ $update_available == 0 ]]; then
                atfile.say.debug "No updates found"
                return
            fi

            echo "---"
            if [[ $_os == "haiku" ]]; then
                # BUG: Haiku Terminal has issues with emojis
                # shellcheck disable=SC2154
                atfile.say "Update available ($latest_version)\n↳ Run \`$_prog update\` to update"
            else
                atfile.say "😎 Update available ($latest_version)\n  ↳ Run \`$_prog update\` to update"
            fi
            ;;
        "install")
            [[ $is_git == 1 && $_enable_update_git_clobber == 0 ]] &&\
                atfile.die "Cannot update in Git repository"
            [[ $_disable_update_command == 1 ]] &&\
                atfile.die "Cannot update system-managed version: update from your package manager"

            if [[ $update_available == 0 ]]; then
                atfile.say "No updates found"
                return
            fi

            # shellcheck disable=SC2154
            temp_updated_path="$_prog_dir/${_prog}-${latest_version}.tmp"
            
            atfile.say.debug "Touching temporary path ($temp_updated_path)..."
            touch "$temp_updated_path"
            # shellcheck disable=SC2181
            [[ $? != 0 ]] && atfile.die "Unable to create temporary file (do you have permission?)"
            
            atfile.say.debug "Getting blob URL for $latest_version ($latest_version_record_id)..."
            blob_url="$(atfile.get_url "$latest_version_record_id")"
            # shellcheck disable=SC2181
            [[ $? != 0 ]] && atfile.die "Unable to get blob URL"

            atfile.say.debug "Downloading latest release..."
            curl -H "User-Agent: $(atfile.util.get_uas)" -s -o "$temp_updated_path" "$blob_url"
            # shellcheck disable=SC2181
            if [[ $? == 0 ]]; then
                # shellcheck disable=SC2154
                mv "$temp_updated_path" "$_prog_path"
                # shellcheck disable=SC2181
                if [[ $? != 0 ]]; then
                    atfile.die "Unable to update (do you have permission?)"
                else
                    chmod +x "$_prog_path"

                    if [[ $_os == "haiku" ]]; then
                        atfile.say "Updated to $latest_version!" # BUG: Haiku Terminal has issues with emojis
                    else
                        atfile.say "😎 Updated to $latest_version!"
                    fi

                    last_checked="$(atfile.cache.set "update-check" "$(atfile.util.get_date "" "%s")")"

                    return
                fi
            else
                atfile.die "Unable to download latest version"
            fi
            ;;
    esac
}