#!/usr/bin/env bash
set -e

# @meta dotenv

# @cmd Call the function
# @arg func![`_choice_func`] The function name
# @arg args~[?`_choice_func_args`] The function args
call() {
    "./bin/$argc_func" "${argc_args[@]}"
}

# @cmd Build all artifacts
build() {
    if [[ -f functions.txt ]]; then
        argc build-declarations-json
    fi
    if [[ "$OS" = "Windows_NT" ]]; then
        argc build-win-shims
    fi
}

# @cmd Build declarations for specific functions
# @option --output=functions.json <FILE> Specify a file path to save the function declarations
# @option --names-file=functions.txt Specify a file containing function names
# @arg funcs*[`_choice_func`] The function names
build-declarations-json() {
    if [[ "${#argc_funcs[@]}" -gt 0 ]]; then
        names=("${argc_funcs[@]}" )
    elif [[ -f "$argc_names_file" ]]; then
        names=($(cat "$argc_names_file"))
    fi
    if [[ -z "$names" ]]; then
        _die "error: no specific function"
    fi
    result=()
    for name in "${names[@]}"; do
        result+=("$(build-func-declaration "$name")")
    done
    echo "["$(IFS=,; echo "${result[*]}")"]"  | jq '.' > "$argc_output"
    echo "Build $argc_output"
}

# @cmd Build declaration for a single function
# @arg func![`_choice_func`] The function name
build-func-declaration() {
    argc --argc-export bin/$1 | _parse_declaration
}

# @cmd Build shims for the functions
# Because Windows OS can't run bash scripts directly, we need to make a shim for each function
#
# @flag --clear Clear the shims
build-win-shims() {
    funcs=($(_choice_func))
    for func in "${funcs[@]}"; do
        echo "Shim bin/${func}.cmd"
        _win_shim > "bin/${func}.cmd"
    done
}

# @cmd Install this repo to aichat functions_dir
install() {
    functions_dir="$(aichat --info | grep functions_dir | awk '{print $2}')"
    if [[ -z "$functions_dir" ]]; then
        _die "error: your aichat version don't support function calling"
    fi
    if [[ ! -e "$functions_dir" ]]; then
        ln -s "$(pwd)" "$functions_dir" 
        echo "$functions_dir symlinked"
    else
        echo "$functions_dir already exists"
    fi
}

# @cmd Show versions of required tools for bug reports.
version() {
    argc --argc-version
    jq --version
    curl --version | head -n 1
}

_parse_declaration() {
    jq -r '
    def parse_description(flag_option):
        if flag_option.describe == "" then
            {}
        else
            { "description": flag_option.describe }
        end;

    def parse_enum(flag_option):
        if flag_option.choice.type == "Values" then
            { "enum": flag_option.choice.data }
        else
            {}
        end;

    def parse_property(flag_option):
        [
            { condition: (flag_option.flag == true), result: { type: "boolean" } },
            { condition: (flag_option.multiple_occurs == true), result: { type: "array", items: { type: "string" } } },
            { condition: (flag_option.notations[0] == "INT"), result: { type: "integer" } },
            { condition: (flag_option.notations[0] == "NUM"), result: { type: "number" } },
            { condition: true, result: { type: "string" } } ]
        | map(select(.condition) | .result) | first 
        | (. + parse_description(flag_option))
        | (. + parse_enum(flag_option))
        ;


    def parse_parameter(flag_options):
        {
            type: "object",
            properties: (reduce flag_options[] as $item ({}; . + { ($item.id | sub("-"; "_"; "g")): parse_property($item) })),
            required: [flag_options[] | select(.required == true) | .id],
        };

    {
        name: (.name | sub("-"; "_"; "g")),
        description: .describe,
        parameters: parse_parameter([.flag_options[] | select(.id != "help" and .id != "version")])
    }'
}

_win_shim() {
    cat <<-'EOF'
@echo off
setlocal

set "script_dir=%~dp0"
set "script_name=%~n0"
for /f "delims=" %%a in ('argc --argc-shell-path') do set "_bash_prog=%%a"

"%_bash_prog%" --noprofile --norc "%script_dir%\%script_name%" %*
EOF
}

_choice_func() {
    ls -1 bin | grep -v '\.cmd'
}

_choice_func_args() {
    args=( "${argc__positionals[@]}" )
    argc --argc-compgen generic "bin/${args[0]}" "${args[@]}"
}

_die() {
    echo "$*"
    exit 1
}

# See more details at https://github.com/sigoden/argc
eval "$(argc --argc-eval "$0" "$@")"