path: root/src/argparser.bash

#!/bin/bash
##
# argparser – command line argument parser library
# 
# Copyright © 2013  Mattias Andrée (maandree@member.fsf.org)
# 
# This library is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
# 
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
# 
# You should have received a copy of the GNU General Public License
# along with this library.  If not, see <http://www.gnu.org/licenses/>.
##



# I could make maps or Bash, but I really prefer not to, therefore
# /tmp/argparser_$$ is used instead to let the fs constructor sets.



# :int  Option takes no arguments
args_ARGUMENTLESS=0

# :int  Option takes one argument per instance
args_ARGUMENTED=1

# :int  Option consumes all following arguments
args_VARIADIC=2



# Constructor.
# The short description is printed on same line as the program name
# 
# @param  $1:str  Short, single-line, description of the program
# @param  $2:str  Formated, multi-line, usage text, empty for none
# @param  $3:str  Long, multi-line, description of the program, empty for none
# @param  $4:str  The name of the program, empty for automatic
# @param  $5:str  Output channel, by fd
function args_init
{
    test "$TERM" = linux
    args_linuxvt=$?
    args_program="$4"
    if [ "$args_program" = "" ]; then
        args_program="$0"
    fi
    args_description="$1"
    args_usage="$2"
    args_longdescription="$3"
    args_options=()
    args_opts_alts=()
    args_opts="/tmp/argparser_$$/opts"
    args_optmap="/tmp/argparser_$$/optmap"
    mkdir -p "${args_opts}"
    mkdir -p "${args_optmap}"
    args_out="$5"
    if [ "$args_out" = "" ]; then
        args_out="1"
    fi
    args_out="$(realpath "/proc/$$/fd/${args_out}")"
    args_files=()
}


# Look up function for parsed arguments
# 
# @param  $1:str  Action: has, count, null or get
# @param  $2:str  Option name
# @exit           0 on success/true, 1 on false, 2 on failure
# 
# @action  has    Checks if the option is used
# @action  count  Gets the number of time the option is used, :int is returned
# @action  null   Checks if the option value at index $3:int is null
# @action  get    Gets option value of indices $3…, all values if $# = 2, :(str) is returned
function args_option
{
    local rc dir="${args_opts}/$2" i
    if [ "$1" = has ]; then
	[ -e "${dir}" ]
	return $?
    elif [ "$1" = count ]; then
	if [ -e "${dir}" ]; then
	    echo "$(wc -l < "${dir}/null")"
	    return 0
	else
	    echo 0
	    return 1
	fi
    elif [ "$1" = null ]; then
	rc="$(head -n $(( $3 + 1 )) < "${dir}/null" | tail -n 1)"
	return $(( 1 - $rc ))
    elif [ "$1" = get ]; then
	shift 2
	if [ $# = 0 ]; then
	    cat "${dir}/data"
	else
	    for i in "$@"; do
		head -n $(( $i + 1 )) < "${dir}/data" | tail -n 1
	    done
	fi
	return 0
    fi
    return 2
}


# Disposes of resources, you really should be doing this when you are done with this module
# 
# @param  $1:int  The ID of process that alloceted the resources, empty for the current process
function args_dispose
{
    local pid=$1
    if [ "$pid" = "" ]; then
        pid=$$
    fi
    rm -r "/tmp/argparser_${pid}"
}


# Gets the name of the parent process
# 
# @param   $1:int   The number of parents to walk, 0 for self, and 1 for direct parent
# @param   $2:bool  Whether to get all arguments
# @return  :str     The name of the parent process, empty if not found
# @exit             0 of success, 1 if not found
function args_parent_name
{
    local pid=$$ lvl=$1 found=0 all=$2 line rc
    if [ "${lvl}" = "" ]; then
	lvl=1
    fi
    while (( $lvl > 0 )); do
	found=0
	while read line; do
	    if [ "${line::5}" = "PPid:" ]; then
		line="${line:5}"
		line="${line/$(echo -e \\t)/}"
		pid="${line/ /}"
		(( lvl-- ))
		found=1
		break
	    fi
	done < "/proc/${pid}/status"
	if [ $found = 0 ]; then
	    return 1
	fi
    done
    if [ -e "/proc/${pid}/cmdline" ]; then
	rc="$(cat /proc/${pid}/cmdline | sed -e 's/\x00/\n/g')"
	if [ "$all" = 1 ]; then
	    echo "$rc"
	else
	    head -n 1 <<<"$rc"
	fi
	return 0
    fi
    return 1
}


# Maps up options that are alternatives to the first alternative for each option
function args_support_alternatives
{
    local alt
    ls "${args_optmap}" | while read alt; do
	if [ ! -e "${args_opts}/${alt}" ]; then
	    ln -s "${args_opts}/$(head -n 1 < "${args_optmap}/${alt}")" "${args_opts}/${alt}"
	fi
    done
}


# Checks for option conflicts
# 
# @param   $@:(str)  Exclusive options
# @exit              Whether at most one exclusive option was used
function args_test_exclusiveness
{
    local used used_use used_std opt msg i=0 n
    used=()
    opts_use=()
    opts_std=()
    
    for opt in "$@"; do
	echo "$opt"
    done > "/tmp/argparser_$$/tmp"
    
    comm -12 <(ls "${args_opts}" | sort) <(cat "/tmp/argparser_$$/tmp" | sort) |
    while read opt; do
	if [ -e "${args_opts}/${opt}" ]; then
            opts_use+=( "${opt}" )
	    opts_std+=( "$(head -n 1 < ${args_optmap}/${opt})" )
	fi
    done
    
    rm "/tmp/argparser_$$/tmp"
    
    n=${#used[@]}
    if (( $n > 1 )); then
	msg="${args_program}: conflicting options:"
	while (( $i < $n )); do
	    if [ "${used_use[$i]}" = "${used_std[$i]}" ]; then
		msg="${msg} ${used_use[$i]}"
	    else
		msg="${msg} ${used_use[$i]}(${used_std[$i]})"
	    fi
	    (( i++ ))
	done
	echo "$msg" > "${args_out}"
	return 1
    fi
    return 0
}


# Checks for out of context option usage
# 
# @param   @:(str)  Allowed options
# @exit             Whether only allowed options was used
function args_test_allowed
{
    local opt msg std rc=0
    
    for opt in "$@"; do
	echo "$opt"
    done > "/tmp/argparser_$$/tmp"
    
    comm -23 <(ls "${args_opts}" | sort) <(cat "/tmp/argparser_$$/tmp" | sort) |
    while read opt; do
	if [ -e "${args_opts}/${opt}" ]; then
	    msg="${args_program}: option used out of context: ${opt}"
	    std="$(head -n 1 < "${args_optmap}/${opt}")"
	    if [ ! "${opt}" = "${std}" ]; then
		msg="${msg}(${std})"
	    fi
	    echo "$msg" > "${args_out}"
	    rc=1
	fi
    done
    
    rm "/tmp/argparser_$$/tmp"
    return $rc
}


# Checks the correctness of the number of used non-option arguments
# 
# @param  $1:int   The minimum number of files
# @param  $2:int?  The maximum number of files, empty for unlimited
# @exit            Whether the usage was correct
function args_test_files
{
    local min=$1 max=$2 n=${#args_files[@]}
    if [ "$max" = "" ]; then
	max=$n
    fi
    (( $min <= $n )) && (( $n <= $max ))
    return $?
}


# Add option that takes no arguments
# 
# @param  $1:int     The default argument's index
# @param  $2:str     Short description, use empty to hide the option
# @param  ...:(str)  Option names
function args_add_argumentless
{
    local default="$1" help="$2" std alts alt
    shift 2
    alts=( "$@" )
    std="${alts[$default]}"
    args_options+=( ${args_ARGUMENTLESS} "" "${help}" ${#args_opts_alts[@]} $# "${std}" )
    args_opts_alts+=( "$@" )
    for alt in "${alts[@]}"; do
	echo "$std" > "${args_optmap}/${alt}"
	echo ${args_ARGUMENTLESS} >> "${args_optmap}/${alt}"
    done
}


# Add option that takes one argument
# 
# @param  $1:int     The default argument's index
# @param  $2:str     The name of the takes argument, one word
# @param  $3:str     Short description, use empty to hide the option
# @param  ...:(str)  Option names
function args_add_argumented
{
    local default="$1" arg="$2" help="$3" std alts alt
    shift 3
    if [ "${arg}" = "" ]; then
	arg="ARG"
    fi
    alts=( "$@" )
    std="${alts[$default]}"
    args_options+=( ${args_ARGUMENTED} "${arg}" "${help}" ${#args_opts_alts[@]} $# "${std}" )
    args_opts_alts+=( "$@" )
    for alt in "${alts[@]}"; do
	echo "$std" > "${args_optmap}/${alt}"
	echo ${args_ARGUMENTED} >> "${args_optmap}/${alt}"
    done
}


# Add option that takes all following arguments
# 
# @param  $1:int     The default argument's index
# @param  $2:str     The name of the takes argument, one word
# @param  $3:str     Short description, use empty to hide the option
# @param  ...:(str)  Option names
function args_add_variadic
{
    local default="$1" arg="$2" help="$3" std alts alt type arg
    shift 3
    if [ "${arg}" = "" ]; then
	arg="ARG"
    fi
    alts=( "$@" )
    std="${alts[$default]}"
    args_options+=( ${args_VARIADIC} "${arg}" "${help}" ${#args_opts_alts[@]} $# "${std}" )
    args_opts_alts+=( "$@" )
    for alt in "${alts[@]}"; do
	echo "$std" > "${args_optmap}/${alt}"
	echo ${args_VARIADIC} >> "${args_optmap}/${alt}"
    done
}


# Prints a colourful help message
function args_help
{
    local dash first last maxfirstlen=0 i=0 n help start count lines=() lens=()
    local empty="        " line l col index=0 colour _colour
    
    dash="—"
    if [ ${args_linuxvt} = 1 ]; then
	dash="-"
    fi
    echo -en "\e[01m" >> "${args_out}"
    echo -n "${args_program}" >> "${args_out}"
    echo -en "\e[21m" >> "${args_out}"
    echo " ${dash} ${args_description}" >> "${args_out}"
    
    echo >> "${args_out}"
    if [ ! "${args_longdescription}" = "" ]; then
	echo "${args_longdescription}" >> "${args_out}"
    fi
    echo >> "${args_out}"
    
    if [ ! "${args_usage}" = "" ]; then
	echo -en "\e[01mUSAGE:\e[21m" >> "${args_out}"
	first=1
	while read line; do
	    if [ $first = 1 ]; then
		first=0
	    else
		echo -n "    or" >> "${args_out}"
	    fi
	    echo -en "\t" >> "${args_out}"
	    echo "${line}" >> "${args_out}"
	done <<< "${args_usage}"
	echo >> "${args_out}"
    fi
    
    n=$(( ${#args_options[@]} / 6 ))
    while (( $i < $n )); do
	help="${args_options[(( $i * 6 + 2 ))]}"
	start=${args_options[(( $i * 6 + 3 ))]}
	count=${args_options[(( $i * 6 + 4 ))]}
	(( i++ ))
	if [ "${help}" = "" ]; then
	    continue
	fi
	if (( $count > 1 )) && (( $maxfirstlen < ${#args_opts_alts[$start]} )); then
	    maxfirstlen=${#args_opts_alts[$start]}
	fi
    done
    
    while (( ${#empty} < $maxfirstlen )); do
	empty="${empty}${empty}"
    done
    empty="${empty::$maxfirstlen}"
    
    echo -e "\e[01mSYNOPSIS:\e[21m" >> "${args_out}"
    i=0
    while (( $i < $n )); do
	 type=${args_options[(( $i * 6 + 0 ))]}
         arg="${args_options[(( $i * 6 + 1 ))]}"
	help="${args_options[(( $i * 6 + 2 ))]}"
	start=${args_options[(( $i * 6 + 3 ))]}
	count=${args_options[(( $i * 6 + 4 ))]}
	(( i++ ))
	if [ "${help}" = "" ]; then
	    continue
	fi
	first="$empty"
	last="${args_opts_alts[(( $start + $count - 1))]}"
	if (( "$count" > 1 )); then
	    first="${args_opts_alts[(( $start ))]}"
	    first="${first}${empty:${#first}}"
	fi
	line="$(echo -en "    \e[02m")${first}$(echo -en "\e[22m  ")/ARGPARSER_COLOUR/${last}"
	l=$(( ${#first} + ${#last} + 6 ))
	if [ $type = ${args_ARGUMENTED} ]; then
	    line+="$(echo -en " \e[04m")${arg}$(echo -en "\e[24m")"
	    l=$(( $l + ${#arg} + 1 ))
	elif [ $type = ${args_VARIADIC} ]; then
	    line+="$(echo -en " [\e[04m")${arg}$(echo -en "\e[24m...]")"
	    l=$(( $l + ${#arg} + 6 ))
	fi
	lines+=( "$line" )
	lens+=( $l )
    done
    
    col=0
    for i in "${lens[@]}"; do
	if (( $col < $i )); then
	    col=$i
	fi
    done
    col=$(( $col + 8 - (($col - 4) & 7) ))
    while (( ${#empty} < $col )); do
	empty="${empty}${empty}"
    done
    empty="${empty::$col}"
    i=0
    while (( $i < $n )); do
	help="${args_options[(( $i * 6 + 2 ))]}"
	(( i++ ))
	if [ "${help}" = "" ]; then
	    continue
	fi
	first=1
	colour=$(( 36 - 2 * ($index & 1) ))
	_colour="$(sed -e "s:/ARGPARSER_COLOUR/:\x1b[${colour};01m:g" <<< "${lines[$index]}")"
	echo -n "${_colour}${empty:${lens[$index]}}" >> "${args_out}"
	echo -e "${help}" | while read line; do
	    if [ $first = 1 ]; then
		first=0
		echo -n "${line}" >> "${args_out}"
		echo -e "\e[00m" >> "${args_out}"
	    else
		echo -en "${empty}\e[${colour}m" >> "${args_out}"
		echo -n "${line}" >> "${args_out}"
		echo -e "\e[00m" >> "${args_out}"
	    fi
	done
	(( index++ ))
    done
    
    echo >> "${args_out}"
}


# Parse arguments
# 
# @param  $@:(str)  The command line arguments, should exclude the execute file
# @exit             Whether no unrecognised option is used
function args_parse
{
    local nulqueue=() argqueue=() optqueue=() queue=() opt arg _arg argnull
    local dashed=0 tmpdashed=0 get=0 dontget=0 rc=0 i n more std sign type
    
    args_argcount=$#
    args_unrecognised_count=0
    args_arguments=( "$@" )
    
    while [ ! $# = 0 ]; do
	while [ ! $# = 0 ]; do
	    arg="$1"
	    shift 1
	    _arg="${arg/+/-}"
	    if [ ! $get = 0 ] && [ $dontget = 0 ]; then
		(( get-- ))
		argqueue+=( "$arg" )
		nulqueue+=( 0 )
	    elif [ $tmpdashed = 1 ]; then
		args_files+=( "$arg" )
		tmpdashed=0
	    elif [ $dashed = 1 ]; then
		args_files+=( "$arg" )
	    elif [ "$arg" = "++" ]; then
		tmpdashed=1
	    elif [ "$arg" = "--" ]; then
		dashed=1
	    elif (( ${#arg} > 1 )) && [ "${_arg::1}" = "-" ]; then
		if (( ${#arg} > 2 )) && [ "${_arg::1}" = "${_arg:1:1}" ]; then
		    if [ ! $dontget = 0 ]; then
			(( dontget-- ))
		    elif [ ! -e "${args_optmap}/${arg}" ]; then
			(( args_unrecognised_count++ ))
			if (( args_unrecognised_count <= 5 )); then
			    echo "${args_program}: warning: unrecognised option ${arg}" >> "${args_out}"
			fi
			rc=1
		    else
			type="$(tail -n 1 < "${args_optmap}/${arg}")"
			if [ $type = ${args_ARGUMENTLESS} ]; then
			    optqueue+=( "${arg}" )
			    argqueue+=( "" )
			    nulqueue+=( 1 )
			elif [ ! "${arg/=/}" = "${arg}" ]; then
			    _arg="${arg%%=*}"
			    type="$(tail -n 1 < "${args_optmap}/${_arg}")"
			    if (( $type >= ${args_ARGUMENTED} )); then
				optqueue+=( "${_arg}" )
				argqueue+=( "${arg#*=}" )
				nulqueue+=( 0 )
				if [ $type = ${args_VARIADIC} ]; then
				    dashed=1
				fi
			    else
				(( args_unrecognised_count++ ))
				if (( args_unrecognised_count <= 5 )); then
				    echo "${args_program}: warning: unrecognised option ${arg}" >> "${args_out}"
				fi
				rc=1
			    fi
			elif [ $type = ${args_ARGUMENTED} ]; then
			    optqueue+=( "${arg}" )
			    (( get++ ))
			else
			    optqueue+=( "${arg}" )
			    argqueue+=( "" )
			    nulqueue+=( 1 )
			    dashed=1
			fi
		    fi
		else
		    sign="${_arg::1}"
		    i=1
		    n=${#arg}
		    while [ ! $i = $n ]; do
			_arg="$sign${arg:$i:1}"
			(( i++ ))
			if [ -e "${args_optmap}/${_arg}" ]; then
			    type="$(tail -n 1 < "${args_optmap}/${_arg}")"
			    optqueue+=( "${_arg}" )
			    if [ $type = ${args_ARGUMENTLESS} ]; then
				argqueue+=( "" )
				nulqueue+=( 1 )
			    elif [ $type = ${args_ARGUMENTED} ]; then
				if [ ${#arg} = $i ]; then
				    (( get++ ))
				else
				    argqueue+=( "${arg:i}" )
				    nulqueue+=( 0 )
				fi
				break
			    else
				if [ ${#arg} = $i ]; then
				    argqueue+=( "" )
				    nulqueue+=( 1 )
				else
				    argqueue+=( "${arg:i}" )
				    nulqueue+=( 0 )
				fi
				dashed=1
				break
			    fi
			else
			    (( args_unrecognised_count++ ))
			    if (( args_unrecognised_count <= 5 )); then
				echo "${args_program}: warning: unrecognised option ${arg}" >> "${args_out}"
			    fi
			    rc=1
			fi
		    done
		fi
	    else
		args_files+=( "$arg" )
	    fi
	done
	set "${queue[@]}" > /dev/null
	queue=()
    done
    
    i=0
    n=${#optqueue[@]}
    while (( $i < $n )); do
	opt="${optqueue[$i]}"
	arg=""
	(( ${#argqueue[@]} > $i )) && [ ${nulqueue[$i]} = 1 ]
	argnull=$(( 1 - $? ))
	if [ $argnull = 0 ]; then
	    arg="${argqueue[$i]}"
	fi
	(( i++ ))
	opt="$(head -n 1 < "${args_optmap}/${opt}")"
	if [ ! -e "${args_opts}/${opt}" ]; then
	    mkdir -p "${args_opts}/${opt}"
	    echo -n > "${args_opts}/${opt}/data"
	    echo -n > "${args_opts}/${opt}/null"
	fi
	if (( ${#argqueue[@]} >= $i )); then
	    echo "$arg" >> "${args_opts}/${opt}/data"
	    echo "$argnull" >> "${args_opts}/${opt}/null"
	fi
    done
    
    i=0
    n=$(( ${#args_options[@]} / 6 ))
    while (( $i < $n )); do
	type=${args_options[(( $i * 6 + 0 ))]}
	 std=${args_options[(( $i * 6 + 5 ))]}
	(( i++ ))
	if [ $type = ${args_VARIADIC} ] && [ -e "${args_opts}/${std}" ]; then
	    if [ "$(head -n 1 < "${args_opts}/${std}/null")" = 1 ]; then
		rm "${args_opts}/${std}/data"
		rm "${args_opts}/${std}/null"
	    fi
	    for arg in "${args_files[@]}"; do
		echo "${arg}" >> "${args_opts}/${std}/data"
		echo 0 >> "${args_opts}/${std}/null"
	    done
	    args_files=()
	    break
	fi
    done
    
    args_message="${args_files[*]}"
    args_has_message="${#args_files[@]}"
    if [ ! $args_has_message = 0 ]; then
	args_has_message=1
    fi
    
    if (( ${args_unrecognised_count} > 5 )); then
	more=$(( ${args_unrecognised_count} - 1 ))
        echo -n "${args_program}: warning: ${more} more unrecognised " >> "${args_out}"
	if [ ! $more = 1 ]; then
	    echo "options" >> "${args_out}"
	else
	    echo "option" >> "${args_out}"
	fi
    fi
    
    return $rc
}