autoload -U add-zsh-hook
add-zsh-hook zshaddhistory _hishtory_add
add-zsh-hook precmd _hishtory_precmd

_hishtory_first_prompt=1

function _hishtory_add() {
    # Runs after <ENTER>, but before the command is executed
    # $1 contains the command that was run 
    _hishtory_command=$1
    _hishtory_start_time=`hishtory getTimestamp`
    if ! [ -z "$_hishtory_command " ]; then
        (hishtory presaveHistoryEntry zsh "$_hishtory_command" $_hishtory_start_time &)  # Background Run
        # hishtory presaveHistoryEntry zsh "$_hishtory_command" $_hishtory_start_time  # Foreground Run
    fi
}

function _hishtory_precmd() {
    # Runs after the command is executed in order to render the prompt
    # $? contains the exit code 
    _hishtory_exit_code=$?
    if [ -n "${_hishtory_first_prompt:-}" ]; then
        unset _hishtory_first_prompt
        return
    fi
    (hishtory saveHistoryEntry zsh $_hishtory_exit_code "$_hishtory_command" $_hishtory_start_time &)  # Background Run
    # hishtory saveHistoryEntry zsh $_hishtory_exit_code "$_hishtory_command" $_hishtory_start_time  # Foreground Run
    (hishtory updateLocalDbFromRemote &)
}

_hishtory_widget() {
    BUFFER=$(HISHTORY_TERM_INTEGRATION=1 hishtory tquery $BUFFER)
    CURSOR=${#BUFFER}
    zle reset-prompt
}

_hishtory_bind_control_r() {
    zle     -N   _hishtory_widget
    bindkey '^R' _hishtory_widget
}

[ "$(hishtory config-get enable-control-r)" = true ] && _hishtory_bind_control_r

# If running in a test environment, force loading of compinit so that shell completions work.
# Otherwise, we respect the user's choice and only run compdef if the user has loaded compinit.
if [ -n "${HISHTORY_TEST:-}" ]; then
    autoload -Uz compinit
    compinit
fi

source <(hishtory completion zsh); which compdef 2>&1 >/dev/null && compdef _hishtory hishtory