147 lines
4.0 KiB
Bash
147 lines
4.0 KiB
Bash
#!/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 [[ ! -e "$functions_dir" ]]; then
|
|
ln -s "$(pwd)" "$functions_dir"
|
|
echo "$functions_dir symlinked"
|
|
else
|
|
echo "$functions_dir already exists"
|
|
fi
|
|
}
|
|
|
|
|
|
_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" "$@")"
|