bash v5: working version
This commit is contained in:
435
barg
435
barg
@@ -1,435 +0,0 @@
|
||||
#!/usr/bin/env bash
|
||||
|
||||
# Bash Argument Parser (BARG)
|
||||
# Requires Bash 4.3 or higher for associative arrays
|
||||
|
||||
# Check Bash version
|
||||
if [ "${BASH_VERSINFO:-0}" -lt 4 ]; then
|
||||
echo "Error: This script requires Bash 4.0 or higher (current: $BASH_VERSION)" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Main parsing function - all logic contained here
|
||||
parse_arguments() {
|
||||
local -a spec_input=("$@")
|
||||
local input="${spec_input[-1]}"
|
||||
unset 'spec_input[-1]'
|
||||
|
||||
# Local storage (no globals!)
|
||||
local -A schema
|
||||
local -A aliases
|
||||
local -a tokens
|
||||
|
||||
# Tokenize input into tagged tokens (format: tag::value)
|
||||
tokenize() {
|
||||
local input_str="$1"
|
||||
local -a args
|
||||
tokens=()
|
||||
|
||||
# Split input into array
|
||||
read -ra args <<<"$input_str"
|
||||
|
||||
# Root command
|
||||
local root="${args[0]}"
|
||||
tokens+=("root::${root}")
|
||||
|
||||
local i=1
|
||||
while [[ $i -lt ${#args[@]} ]]; do
|
||||
local token="${args[$i]}"
|
||||
|
||||
# Rest arguments (everything after --)
|
||||
if [[ "$token" == "--" ]]; then
|
||||
((i++))
|
||||
local rest_args=""
|
||||
while [[ $i -lt ${#args[@]} ]]; do
|
||||
if [[ -n "$rest_args" ]]; then
|
||||
rest_args="${rest_args} ${args[$i]}"
|
||||
else
|
||||
rest_args="${args[$i]}"
|
||||
fi
|
||||
((i++))
|
||||
done
|
||||
if [[ -n "$rest_args" ]]; then
|
||||
tokens+=("rest::${rest_args}")
|
||||
fi
|
||||
break
|
||||
fi
|
||||
|
||||
# Long option: --key or --key=value
|
||||
if [[ "$token" =~ ^--(.+)$ ]]; then
|
||||
local full="${BASH_REMATCH[1]}"
|
||||
if [[ "$full" =~ ^([^=]+)=(.*)$ ]]; then
|
||||
tokens+=("narg::${BASH_REMATCH[1]}")
|
||||
tokens+=("value::${BASH_REMATCH[2]}")
|
||||
else
|
||||
tokens+=("narg::${full}")
|
||||
fi
|
||||
((i++))
|
||||
continue
|
||||
fi
|
||||
|
||||
# Short option: -k or -abc or -k=value
|
||||
if [[ "$token" =~ ^-(.+)$ ]]; then
|
||||
local full="${BASH_REMATCH[1]}"
|
||||
if [[ "$full" =~ ^([^=]+)=(.*)$ ]]; then
|
||||
local flags="${BASH_REMATCH[1]}"
|
||||
local value="${BASH_REMATCH[2]}"
|
||||
# Split flags into individual characters
|
||||
for ((j = 0; j < ${#flags}; j++)); do
|
||||
tokens+=("narg::${flags:$j:1}")
|
||||
done
|
||||
tokens+=("value::${value}")
|
||||
else
|
||||
# Split all flags
|
||||
for ((j = 0; j < ${#full}; j++)); do
|
||||
tokens+=("narg::${full:$j:1}")
|
||||
done
|
||||
fi
|
||||
((i++))
|
||||
continue
|
||||
fi
|
||||
|
||||
# Unknown: could be command, positional, or value
|
||||
tokens+=("unk::${token}")
|
||||
((i++))
|
||||
done
|
||||
}
|
||||
|
||||
# Generate schema from spec
|
||||
generate_schema() {
|
||||
local -a spec_array=("$@")
|
||||
local entry_id=100
|
||||
local level=-1
|
||||
local -a cmd_stack=()
|
||||
local -a pos_stack=()
|
||||
|
||||
for entry in "${spec_array[@]}"; do
|
||||
# Split entry by semicolons
|
||||
IFS=';' read -ra elements <<<"$entry"
|
||||
local entry_type="${elements[0]}"
|
||||
|
||||
case "$entry_type" in
|
||||
command)
|
||||
local id="$entry_id"
|
||||
local aliases_str="${elements[1]}"
|
||||
local help_text="${elements[2]}"
|
||||
|
||||
# Schema entries
|
||||
schema["${id}.entryType"]="command"
|
||||
schema["${id}.name"]="${aliases_str%%,*}" # First alias
|
||||
schema["${id}.help"]="$help_text"
|
||||
schema["${id}.args"]=""
|
||||
|
||||
# Register aliases
|
||||
if [[ ${#cmd_stack[@]} -eq 0 ]]; then
|
||||
aliases["cmd::root"]="$id"
|
||||
else
|
||||
IFS=',' read -ra alias_list <<<"$aliases_str"
|
||||
for alias in "${alias_list[@]}"; do
|
||||
local parent="${cmd_stack[-1]}"
|
||||
aliases["${parent}.cmd::${alias}"]="$id"
|
||||
done
|
||||
fi
|
||||
|
||||
# Push to stacks
|
||||
((level++))
|
||||
cmd_stack+=("$id")
|
||||
pos_stack+=(0)
|
||||
((entry_id++))
|
||||
;;
|
||||
|
||||
end)
|
||||
((level--))
|
||||
unset 'cmd_stack[-1]'
|
||||
unset 'pos_stack[-1]'
|
||||
;;
|
||||
|
||||
argument)
|
||||
local id="$entry_id"
|
||||
local aliases_str="${elements[1]}"
|
||||
local arg_name="${aliases_str%%,*}"
|
||||
|
||||
# Default attributes
|
||||
schema["${id}.entryType"]="argument"
|
||||
schema["${id}.name"]="$arg_name"
|
||||
schema["${id}.dest"]="$arg_name"
|
||||
schema["${id}.required"]="false"
|
||||
schema["${id}.type"]="positional"
|
||||
|
||||
# Parse additional attributes
|
||||
for ((i = 2; i < ${#elements[@]}; i++)); do
|
||||
local element="${elements[$i]}"
|
||||
if [[ "$element" =~ ^([^:]+):(.*)$ ]]; then
|
||||
local attr_key="${BASH_REMATCH[1]}"
|
||||
local attr_value="${BASH_REMATCH[2]}"
|
||||
if [[ "$attr_key" == "default" ]]; then
|
||||
schema["${id}.value"]="$attr_value"
|
||||
else
|
||||
schema["${id}.${attr_key}"]="$attr_value"
|
||||
fi
|
||||
elif [[ "$element" =~ ^([^:]+)$ ]]; then
|
||||
local attr_key="${BASH_REMATCH[1]}"
|
||||
schema["${id}.${attr_key}"]="true"
|
||||
fi
|
||||
done
|
||||
|
||||
# Add to parent command's args list
|
||||
local parent="${cmd_stack[-1]}"
|
||||
local cmd_args="${schema["${parent}.args"]}"
|
||||
if [[ -n "$cmd_args" ]]; then
|
||||
schema["${parent}.args"]="${cmd_args},${id}"
|
||||
else
|
||||
schema["${parent}.args"]="$id"
|
||||
fi
|
||||
|
||||
# Register aliases based on type
|
||||
local arg_type="${schema["${id}.type"]}"
|
||||
case "$arg_type" in
|
||||
positional)
|
||||
local pos="${pos_stack[-1]}"
|
||||
aliases["${parent}.pos::${pos}"]="$id"
|
||||
((pos_stack[-1] = pos + 1))
|
||||
;;
|
||||
rest)
|
||||
aliases["${parent}.rest"]="$id"
|
||||
;;
|
||||
option | flag)
|
||||
IFS=',' read -ra alias_list <<<"$aliases_str"
|
||||
for alias in "${alias_list[@]}"; do
|
||||
aliases["${parent}.narg::${alias}"]="$id"
|
||||
done
|
||||
;;
|
||||
esac
|
||||
|
||||
((entry_id++))
|
||||
;;
|
||||
|
||||
*)
|
||||
echo "Error: Invalid entry type: $entry_type" >&2
|
||||
return 1
|
||||
;;
|
||||
esac
|
||||
done
|
||||
}
|
||||
|
||||
# Parse tokens based on schema
|
||||
process_tokens() {
|
||||
local -a cmd_stack=()
|
||||
local -a pos_stack=()
|
||||
local token_idx=0
|
||||
|
||||
# Bootstrap: Pre-populate with root command
|
||||
local root_id="${aliases["cmd::root"]}"
|
||||
cmd_stack+=("$root_id")
|
||||
pos_stack+=(0)
|
||||
|
||||
# Process each token - all follow same bubble-up pattern
|
||||
while [[ $token_idx -lt ${#tokens[@]} ]]; do
|
||||
local tagged_token="${tokens[$token_idx]}"
|
||||
local token_tag="${tagged_token%%::*}"
|
||||
local token="${tagged_token#*::}"
|
||||
|
||||
# Special case: root token just sets the name, skip matching
|
||||
if [[ "$token_tag" == "root" ]]; then
|
||||
schema["${root_id}.name"]="$token"
|
||||
((token_idx++))
|
||||
continue
|
||||
fi
|
||||
|
||||
local found=false
|
||||
local level
|
||||
|
||||
# Bubble up: try matching from current level to root
|
||||
for ((level = ${#cmd_stack[@]} - 1; level >= 0; level--)); do
|
||||
local entry_id=""
|
||||
local current_cmd="${cmd_stack[$level]}"
|
||||
|
||||
case "$token_tag" in
|
||||
rest)
|
||||
entry_id="${aliases["${current_cmd}.rest"]}"
|
||||
;;
|
||||
narg)
|
||||
entry_id="${aliases["${current_cmd}.narg::${token}"]}"
|
||||
;;
|
||||
unk)
|
||||
# Try as command
|
||||
entry_id="${aliases["${current_cmd}.cmd::${token}"]}"
|
||||
|
||||
# Try as positional (only on last level)
|
||||
if [[ -z "$entry_id" && $level -eq $((${#cmd_stack[@]} - 1)) ]]; then
|
||||
local pos="${pos_stack[$level]}"
|
||||
entry_id="${aliases["${current_cmd}.pos::${pos}"]}"
|
||||
if [[ -n "$entry_id" ]]; then
|
||||
((pos_stack[level] = pos + 1))
|
||||
fi
|
||||
fi
|
||||
;;
|
||||
esac
|
||||
|
||||
# No match at this level, try parent
|
||||
if [[ -z "$entry_id" ]]; then
|
||||
continue
|
||||
fi
|
||||
|
||||
# Found match - process it
|
||||
local entry_type="${schema["${entry_id}.entryType"]}"
|
||||
|
||||
case "$entry_type" in
|
||||
command)
|
||||
cmd_stack+=("$entry_id")
|
||||
pos_stack+=(0)
|
||||
;;
|
||||
|
||||
argument)
|
||||
local arg_type="${schema["${entry_id}.type"]}"
|
||||
local value=""
|
||||
|
||||
case "$arg_type" in
|
||||
rest | positional)
|
||||
value="$token"
|
||||
;;
|
||||
flag)
|
||||
value="true"
|
||||
;;
|
||||
option)
|
||||
# Next token should be the value
|
||||
local next_idx=$((token_idx + 1))
|
||||
|
||||
if [[ $next_idx -ge ${#tokens[@]} ]]; then
|
||||
echo "Error: No value provided for option: $token" >&2
|
||||
return 1
|
||||
fi
|
||||
|
||||
local next_tagged="${tokens[$next_idx]}"
|
||||
local next_tag="${next_tagged%%::*}"
|
||||
local next_token="${next_tagged#*::}"
|
||||
|
||||
if [[ "$next_tag" != "value" && "$next_tag" != "unk" ]]; then
|
||||
echo "Error: Expected value for option, got: $next_token ($next_tag)" >&2
|
||||
return 1
|
||||
fi
|
||||
|
||||
value="$next_token"
|
||||
# Skip next token
|
||||
((token_idx++))
|
||||
;;
|
||||
esac
|
||||
|
||||
schema["${entry_id}.value"]="$value"
|
||||
;;
|
||||
esac
|
||||
|
||||
found=true
|
||||
break
|
||||
done
|
||||
|
||||
if [[ "$found" == "false" ]]; then
|
||||
echo "Warning: Invalid argument: \"$token\" ($token_tag). Skipping..." >&2
|
||||
fi
|
||||
|
||||
((token_idx++))
|
||||
done
|
||||
|
||||
# Extract and output results
|
||||
local -a commands=()
|
||||
local -A values
|
||||
|
||||
for cmd_id in "${cmd_stack[@]}"; do
|
||||
local cmd_name="${schema["${cmd_id}.name"]}"
|
||||
local cmd_args="${schema["${cmd_id}.args"]}"
|
||||
commands+=("$cmd_name")
|
||||
|
||||
if [[ -z "$cmd_args" ]]; then
|
||||
continue
|
||||
fi
|
||||
|
||||
IFS=',' read -ra arg_ids <<<"$cmd_args"
|
||||
for arg_id in "${arg_ids[@]}"; do
|
||||
local name="${schema["${arg_id}.name"]}"
|
||||
local dest="${schema["${arg_id}.dest"]}"
|
||||
local value="${schema["${arg_id}.value"]}"
|
||||
local required="${schema["${arg_id}.required"]}"
|
||||
|
||||
# Check required
|
||||
if [[ -z "$value" && "$required" == "true" ]]; then
|
||||
echo "Error: Argument \"$name\" is required in command \"$cmd_name\"" >&2
|
||||
return 1
|
||||
fi
|
||||
|
||||
# Only collect if value was actually set
|
||||
if [[ -n "$value" ]]; then
|
||||
# Handle duplicate destinations
|
||||
if [[ -n "${values[$dest]}" ]]; then
|
||||
values["${cmd_name}_${dest}"]="$value"
|
||||
else
|
||||
values["$dest"]="$value"
|
||||
fi
|
||||
fi
|
||||
done
|
||||
done
|
||||
|
||||
# Output results
|
||||
echo "Commands: ${commands[*]}"
|
||||
echo "Values:"
|
||||
for key in "${!values[@]}"; do
|
||||
echo " ${key}=${values[$key]}"
|
||||
done
|
||||
}
|
||||
|
||||
# Execute the pipeline
|
||||
tokenize "$input"
|
||||
generate_schema "${spec_input[@]}" || return 1
|
||||
process_tokens
|
||||
}
|
||||
|
||||
# Wrapper to handle errors gracefully
|
||||
parse_input_wrap() {
|
||||
parse_arguments "$@" 2>&1
|
||||
}
|
||||
|
||||
# Example spec
|
||||
SPEC=(
|
||||
"command;barg;Barg - Bash Argument Parser"
|
||||
"argument;global;type:flag"
|
||||
"command;build;Build a dev container"
|
||||
"argument;from;type:option;dest:fromName;help:Name of the base container"
|
||||
"argument;name;type:positional;required;help:Name of the container"
|
||||
"argument;nametwo;help:Name of the container"
|
||||
"argument;image,i;type:option;required;dest:imageName;help:Base image"
|
||||
"argument;include,I;type:option;repeatable;dest:includePaths;help:Include paths"
|
||||
"argument;verbose,v;type:flag;repeatable;default:0;help:Increase verbosity (-v, -vv, ...)"
|
||||
"argument;quiet,q;type:flag;default:false;help:Suppress all output"
|
||||
"argument;cmd;type:rest;help:Command and args to run inside the container"
|
||||
"argument;name;dest:nameOpt;type:option;help:Building container name"
|
||||
"command;container;Container building"
|
||||
"argument;dev;type:flag;help:Is dev or not"
|
||||
"argument;name;required;help:Building container name"
|
||||
"argument;cmd;type:rest;dest:cmdRest;help:Command and args to run inside the container"
|
||||
"end"
|
||||
"end"
|
||||
"command;stop;Stop a dev container"
|
||||
"argument;name;type:positional;required;help:Name of the container"
|
||||
"argument;kill,k;type:flag;default:false;help:Force kill the container"
|
||||
"end"
|
||||
"end"
|
||||
)
|
||||
|
||||
# Test inputs
|
||||
INPUTS=(
|
||||
"dev build -i myimage buildtwo --from base-dev --image node:18 --include src --include test -v -v -q mycontainer -- npm run start"
|
||||
"dev build buildname container -i myimage --dev --name webapp externalContainer"
|
||||
"dev --global build --image debian mybox container --dev --name webapp externalContainer -- echo hi"
|
||||
"dev stop mycontainer --kill"
|
||||
"dev stop mycontainer --kill -- not specified one"
|
||||
"dev build --from alpine --image node:20 mybox -- echo hi"
|
||||
"dev build --aaaaa --from alpine --image node:20 --bbbbb orb mybox ccccc -- echo hi"
|
||||
"dev build --global --aaaaa --from alpine -qvvi node:20 --bbbbb orb mybox ccccc "
|
||||
)
|
||||
|
||||
# Run tests
|
||||
for input in "${INPUTS[@]}"; do
|
||||
echo ""
|
||||
echo "------------------------"
|
||||
echo "> $input"
|
||||
parse_input_wrap "${SPEC[@]}" "$input"
|
||||
done
|
||||
Reference in New Issue
Block a user