#!/usr/bin/env bash

if [ -z "$BASH_IT_COMMAND_DURATION" ] || [ "$BASH_IT_COMMAND_DURATION" != true ]; then
    _command_duration() {
        echo -n
    }
    return
fi

# Define tmp dir and file
COMMAND_DURATION_TMPDIR="${TMPDIR:-/tmp}"
COMMAND_DURATION_FILE="${COMMAND_DURATION_FILE:-$COMMAND_DURATION_TMPDIR/bashit_theme_execution_$BASHPID}"

COMMAND_DURATION_ICON=${COMMAND_DURATION_ICON:-'  '}
COMMAND_DURATION_MIN_SECONDS=${COMMAND_DURATION_MIN_SECONDS:-'1'}

trap _command_duration_delete_temp_file EXIT HUP INT TERM

_command_duration_delete_temp_file() {
    if [[ -f "$COMMAND_DURATION_FILE" ]]; then
        rm -f "$COMMAND_DURATION_FILE"
    fi
}

_command_duration_pre_exec() {
    date +%s.%1N > "$COMMAND_DURATION_FILE"
}

_command_duration() {
    local command_duration command_start current_time
    local minutes seconds deciseconds
    local command_start_sseconds current_time_seconds command_start_deciseconds current_time_deciseconds
    current_time=$(date +%s.%1N)

    if [[ -f "$COMMAND_DURATION_FILE" ]]; then
        command_start=$(< "$COMMAND_DURATION_FILE")
        command_start_sseconds=${command_start%.*}
        current_time_seconds=${current_time%.*}

        command_start_deciseconds=$((10#${command_start#*.}))
        current_time_deciseconds=$((10#${current_time#*.}))

        # seconds
        command_duration=$(( current_time_seconds - command_start_sseconds ))

        if (( current_time_deciseconds >= command_start_deciseconds )); then
            deciseconds=$(( (current_time_deciseconds - command_start_deciseconds) ))
        else
            ((command_duration-=1))
            deciseconds=$(( 10 - ( (command_start_deciseconds - current_time_deciseconds) ) ))
        fi
        command rm "$COMMAND_DURATION_FILE"
    else
        command_duration=0
    fi

    if (( command_duration > 0 )); then
        minutes=$(( command_duration / 60 ))
        seconds=$(( command_duration % 60 ))
    fi

    if (( minutes > 0 )); then
        printf "%s%s%dm %ds" "$COMMAND_DURATION_ICON" "$COMMAND_DURATION_COLOR" "$minutes" "$seconds"
    elif (( seconds >= COMMAND_DURATION_MIN_SECONDS )); then
        printf "%s%s%d.%01ds" "$COMMAND_DURATION_ICON" "$COMMAND_DURATION_COLOR" "$seconds" "$deciseconds"
    fi
}

preexec() (
    _command_duration_pre_exec
)

preexec_install