#!/bin/bash

get_tmux_option() {
    local option default_value option_value

    option=$1
    default_value=$2
    option_value=$(tmux show-option -gqv "$option")

    if [[ -z "$option_value" ]]; then
        echo "$default_value"
    else
        echo "$option_value"
    fi
}

get_option() {
    local option=$1

    case "$option" in
        "@extrakto_key")
            echo $(get_tmux_option $option "tab")
            ;;

        "@extrakto_split_direction")
            echo $(get_tmux_option $option "a")
            ;;

        "@extrakto_split_size")
            echo $(get_tmux_option $option "7")
            ;;

        "@extrakto_grab_area")
            echo $(get_tmux_option $option "window full")
            ;;

        "@extrakto_clip_tool")
            echo $(get_tmux_option $option "auto")
            ;;

        "@extrakto_fzf_tool")
            echo $(get_tmux_option $option "fzf")
            ;;

        "@extrakto_open_tool")
            echo $(get_tmux_option $option "auto")
            ;;

        "@extrakto_copy_key")
            echo $(get_tmux_option $option "enter")
            ;;

        "@extrakto_insert_key")
            echo $(get_tmux_option $option "tab")
            ;;

        "@extrakto_filter_key")
            echo $(get_tmux_option $option "ctrl-f")
            ;;

        "@extrakto_open_key")
            echo $(get_tmux_option $option "ctrl-o")
            ;;

        "@extrakto_edit_key")
            echo $(get_tmux_option $option "ctrl-e")
            ;;

        "@extrakto_grab_key")
            echo $(get_tmux_option $option "ctrl-g")
            ;;

        "@extrakto_help_key")
            echo $(get_tmux_option $option "ctrl-h")
            ;;

        "@extrakto_clip_tool_run")
            echo $(get_tmux_option $option "bg")
            ;;

        "@extrakto_popup_size")
            echo $(get_tmux_option $option "90%")
            ;;

        "@extrakto_popup_position")
            echo $(get_tmux_option $option "C")
            ;;

        "@extrakto_fzf_layout")
            echo $(get_tmux_option $option "default")
            ;;

        "@extrakto_filter_order")
            echo $(get_tmux_option $option "word all line")
            ;;

        "@extrakto_fzf_unset_default_opts")
            echo $(get_tmux_option $option "true")
            ;;
    esac
}

# This returns the start point parameter for `tmux capture-pane`.
# The result will depend on how the user has set the grab area and grab size.
get_capture_pane_start() {
    local grab_area capture_start history_limit

    grab_area="$1"

    if [[ "$grab_area" == "recent" || "$grab_area" == "window recent" ]]; then
        capture_start="-10"

    elif [[ "$grab_area" == "full" || "$grab_area" == "window full" ]]; then
        # use the history limit, this is all the data on the pane
        # if not set just go with tmux's default
        history_limit=$(get_tmux_option "history-limit" "2000")
        capture_start="-${history_limit}"

    elif [[ "$grab_area" =~ ^window\  ]]; then
        # use the user defined limit for how much to grab from every pane in the current window
        capture_start="-${grab_area:7}"

    else
        # use the user defined limit for how much to grab from the current pane
        capture_start="-${grab_area}"
    fi

    echo "$capture_start"
}

# Implement ordered filters list as defined by @extrakto_filter_order
declare -A next_mode
declare -a modes_list

# If there is a bogus filter mode name in the list,
# fall back to the default list of "word all line"
sanitize_modes_list() {
    # in case of further "first class" filter modes implemented in the future
    # add their names to the following default list and valid_modes set
    local -a default=("word" "all" "line")
    local -A valid_modes=(["word"]=1
        ["all"]=1
        ["line"]=1
        ["url"]=1
        ["path"]=1
        ["quote"]=1
        ["s-quote"]=1)

    local invalid=false
    for mode in ${modes_list[@]}; do
        if [[ ${valid_modes[$mode]} -ne 1 ]]; then
            invalid=true
            break
        fi
    done

    if [[ $invalid == true ]]; then
        # change $modes_list to $default
        for i in ${!default[@]}; do
            modes_list[$i]=${default[$i]}
        done
    fi
}

# transform the modes_list acquired from @extrakto_filter_order into
# the more usable form of associative array
mk_next_mode_map() {
    for i in ${!modes_list[@]}; do
        if [[ $i -eq $((${#modes_list[@]} - 1)) ]]; then
            next_mode+=([${modes_list[$i]}]=${modes_list[0]})
        else
            next_mode+=([${modes_list[$i]}]=${modes_list[$((i + 1))]})
        fi
    done
}

# initialize
modes_list_init() {
    readarray -td ' ' modes_list <<< "$(get_option @extrakto_filter_order) "
    unset 'modes_list[-1]'
    sanitize_modes_list
    mk_next_mode_map
}

# get next mode in order defined by @extrakto_filter_order
get_next_mode() {
    if [[ ${#modes_list[@]} -eq 0 || ${#next_mode[@]} -le ${#modes_list[@]} ]]; then
        modes_list_init
    fi

    local next=$1
    if [ $next == "initial" ]; then
        echo ${modes_list[0]}
    else
        echo ${next_mode[$next]}
    fi
}