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

# @meta dotenv

BIN_DIR=bin

LANG_CMDS=( \
    "sh:bash" \
    "js:node" \
    "py:python" \
    "rb:ruby" \
)

# @cmd Call the function
# @arg func![`_choice_func`] The function name
# @arg args~[?`_choice_func_args`] The function args
call() {
    basename="${argc_func%.*}"
    lang="${argc_func##*.}"
    func_path="./$lang/$basename.$lang"
    if [[ ! -e "$func_path" ]]; then
        _die "error: not found $argc_func"
    fi
    if [[ "$lang" == "sh" ]]; then
        "$func_path" "${argc_args[@]}"
    else
       "$(_lang_to_cmd "$lang")" "./cmd/cmd.$lang" "$argc_func"
    fi
}

# @cmd Build the project
# @option --names-file=functions.txt Path to a file containing function filenames, one per line.
# This file specifies which function files to build. 
# Example:
#   get_current_weather.sh
#   may_execute_js_code.js
build() {
    argc build-declarations-json --names-file "${argc_names_file}"
    argc build-bin --names-file "${argc_names_file}"
}

# @cmd Build bin dir 
# @option --names-file=functions.txt Path to a file containing function filenames, one per line.
build-bin() {
    if [[ ! -f "$argc_names_file" ]]; then
        _die "no found "$argc_names_file""
    fi
    mkdir -p "$BIN_DIR"
    rm -rf "$BIN_DIR"/*
    names=($(cat "$argc_names_file"))
    invalid_names=()
    for name in "${names[@]}"; do
        basename="${name%.*}"
        lang="${name##*.}"
        func_file="$lang/$name"
        if [[  -f "$func_file" ]]; then
            if _is_win; then
                bin_file="$BIN_DIR/$basename.cmd" 
                if [[ "$lang" == sh ]]; then
                    _build_win_sh > "$bin_file"
                else
                    _build_win_lang $lang "$(_lang_to_cmd "$lang")"  > "$bin_file"
                fi
            else
                bin_file="$BIN_DIR/$basename" 
                if [[ "$lang" == sh ]]; then
                    ln -s "$PWD/$func_file" "$bin_file"
                else
                    ln -s "$PWD/cmd/cmd.$lang" "$bin_file"
                fi
            fi
        else
            invalid_names+=("$name")
        fi
    done
    if [[ -n "$invalid_names" ]]; then
        _die "error: missing following functions: ${invalid_names[*]}"
    fi
    echo "Build bin"
}

# @cmd Build declarations.json
# @option --output=functions.json <FILE> Path to a json file to save function declarations
# @option --names-file=functions.txt Path to a file containing function filenames, one per line.
# @arg funcs*[`_choice_func`] The function filenames
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 function for building declarations.json"
    fi
    json_list=()
    not_found_funcs=()
    build_failed_funcs=()
    for name in "${names[@]}"; do
        lang="${name##*.}"
        func_file="$lang/$name"
        if [[ ! -f "$func_file" ]]; then
            not_found_funcs+=("$name")
            continue;
        fi
        json_data="$("build-single-declaration" "$name")" || {
            build_failed_funcs+=("$name")
        }
        json_list+=("$json_data")
    done
    if [[ -n "$not_found_funcs" ]]; then
        _die "error: not found functions: ${not_found_funcs[*]}"
    fi
    if [[ -n "$build_failed_funcs" ]]; then
        _die "error: invalid functions: ${build_failed_funcs[*]}"
    fi
    echo "Build $argc_output"
    echo "["$(IFS=,; echo "${json_list[*]}")"]"  | jq '.' > "$argc_output"
}


# @cmd Build single declaration
# @arg func![`_choice_func`] The function name
build-single-declaration() {
    func="$1"
    lang="${func##*.}"
    cmd="$(_lang_to_cmd "$lang")"
    if [[ "$lang" == sh ]]; then
        argc --argc-export "$lang/$func" | _parse_argc_declaration
    else
        LLM_FUNCTION_DECLARATE=1 "$cmd" "cmd/cmd.$lang" "$func"
    fi
}

# @cmd List functions that can be put into functions.txt
# Examples:
#      argc --list-functions > functions.txt
#      argc --list-functions search_duckduckgo.sh >> functions.txt
# @arg funcs*[`_choice_func`] The function filenames, list all available functions if not provided
list-functions() {
    _choice_func
}

# @cmd Test the project
test() {
    func_names_file=functions.txt.test
    argc list-functions > "$func_names_file"
    argc build --names-file "$func_names_file"
    argc test-call-functions
    rm -rf "$func_names_file"
}


# @cmd Test call functions
test-call-functions() {
    if _is_win; then
        ext=".cmd"
    fi
    "./bin/may_execute_command$ext" --command 'echo "bash works"'
    argc call may_execute_command.sh --command 'echo "bash works"'

    if command -v node &> /dev/null; then
        export LLM_FUNCTION_DATA='{"code":"console.log(\"javascript works\")"}'
        "./bin/may_execute_js_code$ext"
        argc call may_execute_js_code.js 
    fi

    if command -v python &> /dev/null; then
        export LLM_FUNCTION_DATA='{"code":"print(\"python works\")"}' 
        "./bin/may_execute_py_code$ext"
        argc call may_execute_py_code.py
    fi

    if command -v ruby &> /dev/null; then
        export LLM_FUNCTION_DATA='{"code":"puts \"ruby works\""}' 
        "./bin/may_execute_rb_code$ext"
        argc call may_execute_rb_code.rb
    fi
}

# @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_argc_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")])
    }'
}

_lang_to_cmd() {
    match_lang="$1"
    for item in "${LANG_CMDS[@]}"; do
        lang="${item%:*}"
        if [[ "$lang" == "$match_lang" ]]; then
            echo "${item#*:}"
        fi
    done
}

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

set "bin_dir=%~dp0"
for %%i in ("%bin_dir:~0,-1%") do set "script_dir=%%~dpi"
set "script_name=%~n0"
set "script_name=%script_name%.sh"
for /f "delims=" %%a in ('argc --argc-shell-path') do set "_bash_prog=%%a"

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

_build_win_lang() {
    lang="$1"
    cmd="$2"
    cat <<-EOF
@echo off
setlocal

set "bin_dir=%~dp0"
for %%i in ("%bin_dir:~0,-1%") do set "script_dir=%%~dpi"
set "script_name=%~n0"

$cmd "%script_dir%cmd\cmd.$lang" "%script_name%.$lang" %*
EOF
}

_is_win() {
    if [[ "$OS" == "Windows_NT" ]]; then
        return 0
    else
        return 1
    fi
}

_choice_func() {
    for item in "${LANG_CMDS[@]}"; do
        lang="${item%:*}"
        cmd="${item#*:}"
        if command -v "$cmd" &> /dev/null; then
            ls -1 $lang  | grep "\.$lang$"
        fi
    done
}

_choice_func_args() {
    args=( "${argc__positionals[@]}" )
    if [[ "${args[0]}" == *.sh ]]; then
        argc --argc-compgen generic "sh/${args[0]}" "${args[@]}"
    fi
}

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

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