Skip to content

Commit

Permalink
feat: argc run-tool helps you input json data (#37)
Browse files Browse the repository at this point in the history
  • Loading branch information
sigoden authored Jun 8, 2024
1 parent 0ac5193 commit 57e8799
Showing 1 changed file with 106 additions and 4 deletions.
110 changes: 106 additions & 4 deletions Argcfile.sh
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@ set -e
# @meta dotenv

BIN_DIR=bin
TMP_DIR="cache/tmp"

LANG_CMDS=( \
"sh:bash" \
Expand All @@ -19,6 +20,15 @@ run-tool() {
if _is_win; then
ext=".cmd"
fi
if [[ -z "$argc_json" ]]; then
declaration="$(jq --arg name "$argc_cmd" '.[] | select(.name == $name)' functions.json)"
if [[ -n "$declaration" ]]; then
_ask_json_data "$declaration"
fi
fi
if [[ -z "$argc_json" ]]; then
_die "error: no JSON data"
fi
"$BIN_DIR/$argc_cmd$ext" "$argc_json"
}

Expand All @@ -34,7 +44,15 @@ build() {
# Example:
# get_current_weather.sh
# may_execute_js_code.js
# @arg tools*[`_choice_tool`] The tool filenames
build-tools() {
if [[ "${#argc_tools[@]}" -gt 0 ]]; then
mkdir -p "$TMP_DIR"
argc_names_file="$TMP_DIR/tools.txt"
printf "%s\n" "${argc_tools[@]}" > "$argc_names_file"
else
argc clean-tools
fi
argc build-tools-json --names-file "${argc_names_file}" --declarations-file "${argc_declarations_file}"
argc build-tools-bin --names-file "${argc_names_file}"
}
Expand Down Expand Up @@ -139,10 +157,9 @@ test() {

# @cmd Test tools
test-tools() {
tmp_dir="cache/tmp"
mkdir -p "$tmp_dir"
names_file="$tmp_dir/tools.txt"
declarations_file="$tmp_dir/functions.json"
mkdir -p "$TMP_DIR"
names_file="$TMP_DIR/tools.txt"
declarations_file="$TMP_DIR/functions.json"
argc list-tools > "$names_file"
argc build-tools --names-file "$names_file" --declarations-file "$declarations_file"
test-tools-execute-lang
Expand Down Expand Up @@ -201,6 +218,12 @@ test-tools-demo() {
done
}


# @cmd Clean tools
clean-tools() {
_choice_tool | sed 's/\.\([a-z]\+\)$//' | xargs -I{} rm -rf "$BIN_DIR/{}"
}

# @cmd Install this repo to aichat functions_dir
install() {
functions_dir="$(aichat --info | grep functions_dir | awk '{print $2}')"
Expand Down Expand Up @@ -266,6 +289,85 @@ $run "%script_dir%scripts\run-tool.$lang" "%script_name%.$lang" %*
EOF
}

_ask_json_data() {
declaration="$1"
echo 'Missing the JSON data but here are its properties:'
echo "$declaration" | _inspect_declaration_params | sed 's/^/> /'
echo 'Generate placeholder data:'
data="$(echo "$declaration" | _generate_data_according_declaration)"
echo "> $data"
read -r -p 'Use the generated data? (y/n) ' res
case "$res" in
[yY][eE][sS]|[yY])
argc_json="$data"
;;
*)
read -r -p "Please enter the data: " data
argc_json="$data"
;;
esac
}

_inspect_declaration_params() {
jq -r '
def get_indicator:
.value.type as $type |
[
{ condition: ($type == "array" and .required), result: "+" },
{ condition: ($type == "array"), result: "*" },
{ condition: .required, result: "!" },
{ condition: true, result: "" }
] | map(select(.condition) | .result) | first;
def get_kind:
.value.type as $type |
(.value.enum // []) as $enum |
([
{ condition: ($type == "array"), result: "string[]" },
{ condition: ($type == "string" and ($enum | length > 0)), result: ($enum | join("|")) },
{ condition: ($type == "string"), result: "" },
{ condition: true, result: $type }
] | map(select(.condition) | .result) | first) as $kind |
if $kind != "" then "(\($kind))" else "" end;
def print_property:
.key as $key |
(.value.description | split("\n")[0]) as $description |
(. | get_kind) as $kind |
(. | get_indicator) as $indicator |
"\($key)\($kind)\($indicator): \($description)";
.parameters |
.required as $requiredProperties |
.properties | to_entries[] |
.key as $key | .+ { "required": ($requiredProperties | index($key) != null) } |
print_property
'
}

_generate_data_according_declaration() {
jq -c '
def convert_string:
if has("enum") then .enum[0] else "foo" end;
def convert_property:
.key as $key |
.value.type as $type |
[
{ condition: ($type == "string"), result: { $key: (.value | convert_string) }},
{ condition: ($type == "boolean"), result: { $key: false }},
{ condition: ($type == "integer"), result: { $key: 42 }},
{ condition: ($type == "number"), result: { $key: 3.14 }},
{ condition: ($type == "array"), result: { $key: [ "v1" ] } }
] | map(select(.condition) | .result) | first;
.parameters |
[
.properties | to_entries[] | convert_property
] | add // {}
'
}

_normalize_path() {
if _is_win; then
cygpath -w "$1"
Expand Down

0 comments on commit 57e8799

Please sign in to comment.