blob: 814f450e6664891ee01f18a2d9c92aeb980fd524 [file] [log] [blame]
kate.ward60c25982008-11-12 20:12:44 +00001# $Id$
2# vim:et:ft=sh:sts=2:sw=2
3#
kate.wardf51c6162008-06-17 16:38:35 +00004# Copyright 2008 Kate Ward. All Rights Reserved.
kate.ward29ddd4e2008-06-18 19:05:43 +00005# Released under the LGPL (GNU Lesser General Public License)
kate.wardf51c6162008-06-17 16:38:35 +00006#
kate.wardc5210682009-03-30 18:54:36 +00007# shFlags -- Advanced command-line flag library for Unix shell scripts.
8# http://code.google.com/p/shflags/
kate.wardf51c6162008-06-17 16:38:35 +00009#
kate.ward29ddd4e2008-06-18 19:05:43 +000010# Author: kate.ward@forestent.com (Kate Ward)
11#
kate.warda8953952008-06-17 16:48:34 +000012# This module implements something like the google-gflags library available
kate.ward29ddd4e2008-06-18 19:05:43 +000013# from http://code.google.com/p/google-gflags/.
kate.wardf51c6162008-06-17 16:38:35 +000014#
15# FLAG TYPES: This is a list of the DEFINE_*'s that you can do. All flags take
16# a name, default value, help-string, and optional 'short' name (one-letter
17# name). Some flags have other arguments, which are described with the flag.
18#
19# DEFINE_string: takes any input, and intreprets it as a string.
20#
21# DEFINE_boolean: typically does not take any argument: say --myflag to set
22# FLAGS_myflag to true, or --nomyflag to set FLAGS_myflag to false.
23# Alternately, you can say
kate.ward60c25982008-11-12 20:12:44 +000024# --myflag=true or --myflag=t or --myflag=0 or
25# --myflag=false or --myflag=f or --myflag=1
kate.wardf51c6162008-06-17 16:38:35 +000026# Passing an option has the same affect as passing the option once.
27#
28# DEFINE_float: takes an input and intreprets it as a floating point number. As
29# shell does not support floats per-se, the input is merely validated as
30# being a valid floating point value.
31#
32# DEFINE_integer: takes an input and intreprets it as an integer.
33#
34# SPECIAL FLAGS: There are a few flags that have special meaning:
35# --help (or -?) prints a list of all the flags in a human-readable fashion
kate.ward39913d12009-04-01 14:09:23 +000036# --flagfile=foo read flags from foo. (not implemented yet)
kate.wardf51c6162008-06-17 16:38:35 +000037# -- as in getopt(), terminates flag-processing
38#
39# EXAMPLE USAGE:
40#
kate.wardc5210682009-03-30 18:54:36 +000041# -- begin hello.sh --
kate.wardf51c6162008-06-17 16:38:35 +000042# #! /bin/sh
43# . ./shflags
kate.wardc5210682009-03-30 18:54:36 +000044# DEFINE_string name 'world' "somebody's name" n
45# FLAGS "$@" || exit $?
46# eval set -- "${FLAGS_ARGV}"
kate.wardf51c6162008-06-17 16:38:35 +000047# echo "Hello, ${FLAGS_name}."
kate.wardc5210682009-03-30 18:54:36 +000048# -- end hello.sh --
49#
50# $ ./hello.sh -n Kate
51# Hello, Kate.
kate.ward29ddd4e2008-06-18 19:05:43 +000052#
kate.ward1cb79602011-06-10 11:15:49 +000053# CUSTOMIZABLE BEHAVIOR:
54#
55# A script can override the default 'getopt' command by providing the path to
56# an alternate implementation by defining the FLAGS_GETOPT_CMD variable.
57#
58# NOTES:
59#
60# * Not all systems include a getopt version that supports long flags. On these
61# systems, only short flags are recognized.
kate.wardf51c6162008-06-17 16:38:35 +000062
63#==============================================================================
kate.warda8953952008-06-17 16:48:34 +000064# shFlags
kate.wardf51c6162008-06-17 16:38:35 +000065#
66# Shared attributes:
67# flags_error: last error message
68# flags_return: last return value
69#
70# __flags_longNames: list of long names for all flags
71# __flags_shortNames: list of short names for all flags
72# __flags_boolNames: list of boolean flag names
73#
kate.ward60c25982008-11-12 20:12:44 +000074# __flags_opts: options parsed by getopt
75#
kate.wardf51c6162008-06-17 16:38:35 +000076# Per-flag attributes:
77# FLAGS_<flag_name>: contains value of flag named 'flag_name'
78# __flags_<flag_name>_default: the default flag value
79# __flags_<flag_name>_help: the flag help string
80# __flags_<flag_name>_short: the flag short name
81# __flags_<flag_name>_type: the flag type
82#
83# Notes:
84# - lists of strings are space separated, and a null value is the '~' char.
85
86# return if FLAGS already loaded
kate.wardd3d3d082008-07-10 20:00:42 +000087[ -n "${FLAGS_VERSION:-}" ] && return 0
kate.ward9a183b82010-03-18 00:25:34 +000088FLAGS_VERSION='1.0.4pre'
kate.wardf51c6162008-06-17 16:38:35 +000089
kate.warde10dd532013-01-04 21:52:23 +000090# return values that scripts can use
91FLAGS_TRUE=0
92FLAGS_FALSE=1
93FLAGS_ERROR=2
94
95# determine some reasonable command defaults
kate.ward4ae73d82012-12-17 23:34:29 +000096__FLAGS_UNAME_S=`uname -s`
97case "${__FLAGS_UNAME_S}" in
98 BSD) __FLAGS_EXPR_CMD='gexpr' ;;
99 *) __FLAGS_EXPR_CMD='expr' ;;
kate.warda589b9f2012-12-17 23:27:51 +0000100esac
101
kate.warde10dd532013-01-04 21:52:23 +0000102# commands a user can override if needed
kate.ward4ae73d82012-12-17 23:34:29 +0000103FLAGS_EXPR_CMD=${FLAGS_EXPR_CMD:-${__FLAGS_EXPR_CMD}}
kate.ward1cb79602011-06-10 11:15:49 +0000104FLAGS_GETOPT_CMD=${FLAGS_GETOPT_CMD:-getopt}
105
kate.wardf51c6162008-06-17 16:38:35 +0000106# specific shell checks
107if [ -n "${ZSH_VERSION:-}" ]; then
kate.ward1d0ecc42008-07-11 15:33:23 +0000108 setopt |grep "^shwordsplit$" >/dev/null
109 if [ $? -ne ${FLAGS_TRUE} ]; then
110 _flags_fatal 'zsh shwordsplit option is required for proper zsh operation'
kate.wardf51c6162008-06-17 16:38:35 +0000111 fi
112 if [ -z "${FLAGS_PARENT:-}" ]; then
113 _flags_fatal "zsh does not pass \$0 through properly. please declare' \
kate.warda8953952008-06-17 16:48:34 +0000114\"FLAGS_PARENT=\$0\" before calling shFlags"
kate.wardf51c6162008-06-17 16:38:35 +0000115 fi
116fi
117
kate.warde10dd532013-01-04 21:52:23 +0000118# can we use built-ins?
119( echo "${FLAGS_TRUE#0}"; ) >/dev/null 2>&1
120if [ $? -eq ${FLAGS_TRUE} ]; then
kate.wardb0759202013-01-05 13:57:46 +0000121 __FLAGS_FX_MATH='_flags_mathBuiltin'
kate.warde10dd532013-01-04 21:52:23 +0000122 __FLAGS_FX_VALID_FLOAT='_flags_validFloatBuiltin'
123 __FLAGS_FX_VALID_INT='_flags_validIntBuiltin'
124else
kate.wardb0759202013-01-05 13:57:46 +0000125 __FLAGS_FX_MATH='_flags_mathExpr'
kate.warde10dd532013-01-04 21:52:23 +0000126 __FLAGS_FX_VALID_FLOAT='_flags_validFloatExpr'
127 __FLAGS_FX_VALID_INT='_flags_validIntExpr'
128fi
129
kate.wardf51c6162008-06-17 16:38:35 +0000130#
131# constants
132#
133
kate.ward1cb79602011-06-10 11:15:49 +0000134# reserved flag names
135__FLAGS_RESERVED_LIST=' ARGC ARGV ERROR FALSE GETOPT_CMD HELP PARENT TRUE '
136__FLAGS_RESERVED_LIST="${__FLAGS_RESERVED_LIST} VERSION "
137
kate.wardf51c6162008-06-17 16:38:35 +0000138# getopt version
139__FLAGS_GETOPT_VERS_STD=0
140__FLAGS_GETOPT_VERS_ENH=1
141__FLAGS_GETOPT_VERS_BSD=2
142
kate.ward1cb79602011-06-10 11:15:49 +0000143${FLAGS_GETOPT_CMD} >/dev/null 2>&1
kate.wardf51c6162008-06-17 16:38:35 +0000144case $? in
145 0) __FLAGS_GETOPT_VERS=${__FLAGS_GETOPT_VERS_STD} ;; # bsd getopt
146 2)
kate.wardc5210682009-03-30 18:54:36 +0000147 # TODO(kward): look into '-T' option to test the internal getopt() version
kate.ward1cb79602011-06-10 11:15:49 +0000148 if [ "`${FLAGS_GETOPT_CMD} --version`" = '-- ' ]; then
kate.wardf51c6162008-06-17 16:38:35 +0000149 __FLAGS_GETOPT_VERS=${__FLAGS_GETOPT_VERS_STD}
150 else
151 __FLAGS_GETOPT_VERS=${__FLAGS_GETOPT_VERS_ENH}
152 fi
153 ;;
kate.ward1cb79602011-06-10 11:15:49 +0000154 *) _flags_fatal 'unable to determine getopt version' ;;
kate.wardf51c6162008-06-17 16:38:35 +0000155esac
156
157# getopt optstring lengths
158__FLAGS_OPTSTR_SHORT=0
159__FLAGS_OPTSTR_LONG=1
160
161__FLAGS_NULL='~'
162
163# flag info strings
kate.wardfb0e7182009-05-10 16:59:13 +0000164__FLAGS_INFO_DEFAULT='default'
165__FLAGS_INFO_HELP='help'
166__FLAGS_INFO_SHORT='short'
167__FLAGS_INFO_TYPE='type'
kate.wardf51c6162008-06-17 16:38:35 +0000168
169# flag lengths
170__FLAGS_LEN_SHORT=0
171__FLAGS_LEN_LONG=1
172
173# flag types
174__FLAGS_TYPE_NONE=0
175__FLAGS_TYPE_BOOLEAN=1
176__FLAGS_TYPE_FLOAT=2
177__FLAGS_TYPE_INTEGER=3
178__FLAGS_TYPE_STRING=4
179
180# set the constants readonly
kate.wardd9672402008-11-15 16:55:51 +0000181__flags_constants=`set |awk -F= '/^FLAGS_/ || /^__FLAGS_/ {print $1}'`
182for __flags_const in ${__flags_constants}; do
183 # skip certain flags
184 case ${__flags_const} in
kate.ward39913d12009-04-01 14:09:23 +0000185 FLAGS_HELP) continue ;;
kate.wardd9672402008-11-15 16:55:51 +0000186 FLAGS_PARENT) continue ;;
187 esac
188 # set flag readonly
189 if [ -z "${ZSH_VERSION:-}" ]; then
190 readonly ${__flags_const}
191 else # handle zsh
kate.ward1d0ecc42008-07-11 15:33:23 +0000192 case ${ZSH_VERSION} in
kate.wardd9672402008-11-15 16:55:51 +0000193 [123].*) readonly ${__flags_const} ;;
194 *) readonly -g ${__flags_const} ;; # declare readonly constants globally
kate.ward1d0ecc42008-07-11 15:33:23 +0000195 esac
kate.wardf51c6162008-06-17 16:38:35 +0000196 fi
197done
kate.wardd9672402008-11-15 16:55:51 +0000198unset __flags_const __flags_constants
kate.wardf51c6162008-06-17 16:38:35 +0000199
200#
201# internal variables
202#
203
kate.wardda8d2c02011-06-28 13:26:02 +0000204# space separated lists
205__flags_boolNames=' ' # boolean flag names
206__flags_longNames=' ' # long flag names
207__flags_shortNames=' ' # short flag names
208__flags_definedNames=' ' # defined flag names (used for validation)
kate.wardf51c6162008-06-17 16:38:35 +0000209
kate.ward14b33bf2008-11-15 20:03:18 +0000210__flags_columns='' # screen width in columns
kate.wardd9672402008-11-15 16:55:51 +0000211__flags_opts='' # temporary storage for parsed getopt flags
kate.ward60c25982008-11-12 20:12:44 +0000212
kate.wardf51c6162008-06-17 16:38:35 +0000213#------------------------------------------------------------------------------
214# private functions
215#
216
kate.warde10dd532013-01-04 21:52:23 +0000217# logging functions
218_flags_debug() { echo "flags:DEBUG $@" >&2; }
219_flags_warn() { echo "flags:WARN $@" >&2; }
220_flags_error() { echo "flags:ERROR $@" >&2; }
221_flags_fatal() { echo "flags:FATAL $@" >&2; exit ${FLAGS_ERROR}; }
222
kate.wardcc145682008-11-14 00:32:42 +0000223# Define a flag.
224#
225# Calling this function will define the following info variables for the
226# specified flag:
kate.wardf51c6162008-06-17 16:38:35 +0000227# FLAGS_flagname - the name for this flag (based upon the long flag name)
228# __flags_<flag_name>_default - the default value
229# __flags_flagname_help - the help string
230# __flags_flagname_short - the single letter alias
231# __flags_flagname_type - the type of flag (one of __FLAGS_TYPE_*)
232#
233# Args:
234# _flags__type: integer: internal type of flag (__FLAGS_TYPE_*)
235# _flags__name: string: long flag name
236# _flags__default: default flag value
237# _flags__help: string: help string
238# _flags__short: string: (optional) short flag name
239# Returns:
240# integer: success of operation, or error
241_flags_define()
242{
243 if [ $# -lt 4 ]; then
244 flags_error='DEFINE error: too few arguments'
245 flags_return=${FLAGS_ERROR}
246 _flags_error "${flags_error}"
247 return ${flags_return}
248 fi
249
kate.ward60c25982008-11-12 20:12:44 +0000250 _flags_type_=$1
251 _flags_name_=$2
252 _flags_default_=$3
253 _flags_help_=$4
254 _flags_short_=${5:-${__FLAGS_NULL}}
kate.wardf51c6162008-06-17 16:38:35 +0000255
kate.ward60c25982008-11-12 20:12:44 +0000256 _flags_return_=${FLAGS_TRUE}
kate.wardda8d2c02011-06-28 13:26:02 +0000257 _flags_usName_=`_flags_underscoreName ${_flags_name_}`
kate.wardf51c6162008-06-17 16:38:35 +0000258
kate.ward39913d12009-04-01 14:09:23 +0000259 # check whether the flag name is reserved
kate.wardda8d2c02011-06-28 13:26:02 +0000260 _flags_itemInList ${_flags_usName_} "${__FLAGS_RESERVED_LIST}"
kate.ward929261f2010-03-28 23:12:17 +0000261 if [ $? -eq ${FLAGS_TRUE} ]; then
kate.ward39913d12009-04-01 14:09:23 +0000262 flags_error="flag name (${_flags_name_}) is reserved"
263 _flags_return_=${FLAGS_ERROR}
264 fi
265
kate.wardf51c6162008-06-17 16:38:35 +0000266 # require short option for getopt that don't support long options
kate.ward60c25982008-11-12 20:12:44 +0000267 if [ ${_flags_return_} -eq ${FLAGS_TRUE} \
kate.wardf51c6162008-06-17 16:38:35 +0000268 -a ${__FLAGS_GETOPT_VERS} -ne ${__FLAGS_GETOPT_VERS_ENH} \
kate.ward60c25982008-11-12 20:12:44 +0000269 -a "${_flags_short_}" = "${__FLAGS_NULL}" ]
kate.wardf51c6162008-06-17 16:38:35 +0000270 then
kate.ward5b48fc12008-11-13 00:42:43 +0000271 flags_error="short flag required for (${_flags_name_}) on this platform"
kate.ward60c25982008-11-12 20:12:44 +0000272 _flags_return_=${FLAGS_ERROR}
kate.wardf51c6162008-06-17 16:38:35 +0000273 fi
274
275 # check for existing long name definition
kate.ward60c25982008-11-12 20:12:44 +0000276 if [ ${_flags_return_} -eq ${FLAGS_TRUE} ]; then
kate.wardda8d2c02011-06-28 13:26:02 +0000277 if _flags_itemInList ${_flags_usName_} ${__flags_definedNames}; then
278 flags_error="definition for ([no]${_flags_name_}) already exists"
kate.ward20d06782008-10-21 19:57:19 +0000279 _flags_warn "${flags_error}"
kate.ward60c25982008-11-12 20:12:44 +0000280 _flags_return_=${FLAGS_FALSE}
kate.wardf51c6162008-06-17 16:38:35 +0000281 fi
282 fi
283
284 # check for existing short name definition
kate.ward60c25982008-11-12 20:12:44 +0000285 if [ ${_flags_return_} -eq ${FLAGS_TRUE} \
286 -a "${_flags_short_}" != "${__FLAGS_NULL}" ]
kate.wardf51c6162008-06-17 16:38:35 +0000287 then
kate.ward60c25982008-11-12 20:12:44 +0000288 if _flags_itemInList "${_flags_short_}" ${__flags_shortNames}; then
289 flags_error="flag short name (${_flags_short_}) already defined"
kate.ward20d06782008-10-21 19:57:19 +0000290 _flags_warn "${flags_error}"
kate.ward60c25982008-11-12 20:12:44 +0000291 _flags_return_=${FLAGS_FALSE}
kate.wardf51c6162008-06-17 16:38:35 +0000292 fi
293 fi
294
295 # handle default value. note, on several occasions the 'if' portion of an
296 # if/then/else contains just a ':' which does nothing. a binary reversal via
297 # '!' is not done because it does not work on all shells.
kate.ward60c25982008-11-12 20:12:44 +0000298 if [ ${_flags_return_} -eq ${FLAGS_TRUE} ]; then
299 case ${_flags_type_} in
kate.wardf51c6162008-06-17 16:38:35 +0000300 ${__FLAGS_TYPE_BOOLEAN})
kate.warde10dd532013-01-04 21:52:23 +0000301 if _flags_validBool "${_flags_default_}"; then
kate.ward60c25982008-11-12 20:12:44 +0000302 case ${_flags_default_} in
303 true|t|0) _flags_default_=${FLAGS_TRUE} ;;
304 false|f|1) _flags_default_=${FLAGS_FALSE} ;;
kate.wardf51c6162008-06-17 16:38:35 +0000305 esac
306 else
kate.ward60c25982008-11-12 20:12:44 +0000307 flags_error="invalid default flag value '${_flags_default_}'"
308 _flags_return_=${FLAGS_ERROR}
kate.wardf51c6162008-06-17 16:38:35 +0000309 fi
310 ;;
311
312 ${__FLAGS_TYPE_FLOAT})
kate.warde10dd532013-01-04 21:52:23 +0000313 if ${__FLAGS_FX_VALID_FLOAT} "${_flags_default_}"; then
kate.wardf51c6162008-06-17 16:38:35 +0000314 :
315 else
kate.ward60c25982008-11-12 20:12:44 +0000316 flags_error="invalid default flag value '${_flags_default_}'"
317 _flags_return_=${FLAGS_ERROR}
kate.wardf51c6162008-06-17 16:38:35 +0000318 fi
319 ;;
320
321 ${__FLAGS_TYPE_INTEGER})
kate.warde10dd532013-01-04 21:52:23 +0000322 if ${__FLAGS_FX_VALID_INT} "${_flags_default_}"; then
kate.wardf51c6162008-06-17 16:38:35 +0000323 :
324 else
kate.ward60c25982008-11-12 20:12:44 +0000325 flags_error="invalid default flag value '${_flags_default_}'"
326 _flags_return_=${FLAGS_ERROR}
kate.wardf51c6162008-06-17 16:38:35 +0000327 fi
328 ;;
329
330 ${__FLAGS_TYPE_STRING}) ;; # everything in shell is a valid string
331
332 *)
kate.ward60c25982008-11-12 20:12:44 +0000333 flags_error="unrecognized flag type '${_flags_type_}'"
334 _flags_return_=${FLAGS_ERROR}
kate.wardf51c6162008-06-17 16:38:35 +0000335 ;;
336 esac
337 fi
338
kate.ward60c25982008-11-12 20:12:44 +0000339 if [ ${_flags_return_} -eq ${FLAGS_TRUE} ]; then
kate.wardf51c6162008-06-17 16:38:35 +0000340 # store flag information
kate.wardda8d2c02011-06-28 13:26:02 +0000341 eval "FLAGS_${_flags_usName_}='${_flags_default_}'"
342 eval "__flags_${_flags_usName_}_${__FLAGS_INFO_TYPE}=${_flags_type_}"
343 eval "__flags_${_flags_usName_}_${__FLAGS_INFO_DEFAULT}=\
kate.ward60c25982008-11-12 20:12:44 +0000344\"${_flags_default_}\""
kate.wardda8d2c02011-06-28 13:26:02 +0000345 eval "__flags_${_flags_usName_}_${__FLAGS_INFO_HELP}=\"${_flags_help_}\""
346 eval "__flags_${_flags_usName_}_${__FLAGS_INFO_SHORT}='${_flags_short_}'"
kate.wardf51c6162008-06-17 16:38:35 +0000347
kate.wardda8d2c02011-06-28 13:26:02 +0000348 # append flag names to name lists
kate.ward60c25982008-11-12 20:12:44 +0000349 __flags_shortNames="${__flags_shortNames}${_flags_short_} "
kate.wardda8d2c02011-06-28 13:26:02 +0000350 __flags_longNames="${__flags_longNames}${_flags_name_} "
kate.ward60c25982008-11-12 20:12:44 +0000351 [ ${_flags_type_} -eq ${__FLAGS_TYPE_BOOLEAN} ] && \
352 __flags_boolNames="${__flags_boolNames}no${_flags_name_} "
kate.wardda8d2c02011-06-28 13:26:02 +0000353
354 # append flag names to defined names for later validation checks
355 __flags_definedNames="${__flags_definedNames}${_flags_usName_} "
356 [ ${_flags_type_} -eq ${__FLAGS_TYPE_BOOLEAN} ] && \
357 __flags_definedNames="${__flags_definedNames}no${_flags_usName_} "
kate.wardf51c6162008-06-17 16:38:35 +0000358 fi
359
kate.ward60c25982008-11-12 20:12:44 +0000360 flags_return=${_flags_return_}
kate.wardda8d2c02011-06-28 13:26:02 +0000361 unset _flags_default_ _flags_help_ _flags_name_ _flags_return_ \
362 _flags_short_ _flags_type_ _flags_usName_
kate.wardf51c6162008-06-17 16:38:35 +0000363 [ ${flags_return} -eq ${FLAGS_ERROR} ] && _flags_error "${flags_error}"
364 return ${flags_return}
365}
366
kate.wardda8d2c02011-06-28 13:26:02 +0000367# Underscore a flag name by replacing dashes with underscores.
368#
369# Args:
370# unnamed: string: log flag name
371# Output:
372# string: underscored name
373_flags_underscoreName()
374{
375 echo $1 |tr '-' '_'
376}
377
kate.wardcc145682008-11-14 00:32:42 +0000378# Return valid getopt options using currently defined list of long options.
379#
380# This function builds a proper getopt option string for short (and long)
381# options, using the current list of long options for reference.
kate.wardf51c6162008-06-17 16:38:35 +0000382#
383# Args:
kate.wardcc145682008-11-14 00:32:42 +0000384# _flags_optStr: integer: option string type (__FLAGS_OPTSTR_*)
kate.wardf51c6162008-06-17 16:38:35 +0000385# Output:
386# string: generated option string for getopt
387# Returns:
388# boolean: success of operation (always returns True)
389_flags_genOptStr()
390{
kate.wardcc145682008-11-14 00:32:42 +0000391 _flags_optStrType_=$1
kate.wardf51c6162008-06-17 16:38:35 +0000392
kate.wardcc145682008-11-14 00:32:42 +0000393 _flags_opts_=''
kate.wardf51c6162008-06-17 16:38:35 +0000394
kate.wardda8d2c02011-06-28 13:26:02 +0000395 for _flags_name_ in ${__flags_longNames}; do
396 _flags_usName_=`_flags_underscoreName ${_flags_name_}`
397 _flags_type_=`_flags_getFlagInfo ${_flags_usName_} ${__FLAGS_INFO_TYPE}`
kate.ward1cb79602011-06-10 11:15:49 +0000398 [ $? -eq ${FLAGS_TRUE} ] || _flags_fatal 'call to _flags_type_ failed'
kate.wardcc145682008-11-14 00:32:42 +0000399 case ${_flags_optStrType_} in
kate.wardf51c6162008-06-17 16:38:35 +0000400 ${__FLAGS_OPTSTR_SHORT})
kate.wardcc145682008-11-14 00:32:42 +0000401 _flags_shortName_=`_flags_getFlagInfo \
kate.wardda8d2c02011-06-28 13:26:02 +0000402 ${_flags_usName_} ${__FLAGS_INFO_SHORT}`
kate.wardcc145682008-11-14 00:32:42 +0000403 if [ "${_flags_shortName_}" != "${__FLAGS_NULL}" ]; then
404 _flags_opts_="${_flags_opts_}${_flags_shortName_}"
kate.wardc5210682009-03-30 18:54:36 +0000405 # getopt needs a trailing ':' to indicate a required argument
kate.wardcc145682008-11-14 00:32:42 +0000406 [ ${_flags_type_} -ne ${__FLAGS_TYPE_BOOLEAN} ] && \
407 _flags_opts_="${_flags_opts_}:"
kate.wardf51c6162008-06-17 16:38:35 +0000408 fi
409 ;;
410
411 ${__FLAGS_OPTSTR_LONG})
kate.wardda8d2c02011-06-28 13:26:02 +0000412 _flags_opts_="${_flags_opts_:+${_flags_opts_},}${_flags_name_}"
kate.wardc5210682009-03-30 18:54:36 +0000413 # getopt needs a trailing ':' to indicate a required argument
kate.wardcc145682008-11-14 00:32:42 +0000414 [ ${_flags_type_} -ne ${__FLAGS_TYPE_BOOLEAN} ] && \
415 _flags_opts_="${_flags_opts_}:"
kate.wardf51c6162008-06-17 16:38:35 +0000416 ;;
417 esac
418 done
419
kate.wardcc145682008-11-14 00:32:42 +0000420 echo "${_flags_opts_}"
kate.wardda8d2c02011-06-28 13:26:02 +0000421 unset _flags_name_ _flags_opts_ _flags_optStrType_ _flags_shortName_ \
422 _flags_type_ _flags_usName_
kate.wardf51c6162008-06-17 16:38:35 +0000423 return ${FLAGS_TRUE}
424}
425
426# Returns flag details based on a flag name and flag info.
427#
428# Args:
kate.wardda8d2c02011-06-28 13:26:02 +0000429# string: underscored flag name
kate.wardf51c6162008-06-17 16:38:35 +0000430# string: flag info (see the _flags_define function for valid info types)
431# Output:
432# string: value of dereferenced flag variable
433# Returns:
434# integer: one of FLAGS_{TRUE|FALSE|ERROR}
435_flags_getFlagInfo()
436{
kate.wardda8d2c02011-06-28 13:26:02 +0000437 # note: adding gFI to variable names to prevent naming conflicts with calling
438 # functions
439 _flags_gFI_usName_=$1
440 _flags_gFI_info_=$2
kate.wardf51c6162008-06-17 16:38:35 +0000441
kate.wardda8d2c02011-06-28 13:26:02 +0000442 _flags_infoVar_="__flags_${_flags_gFI_usName_}_${_flags_gFI_info_}"
443 _flags_strToEval_="_flags_infoValue_=\"\${${_flags_infoVar_}:-}\""
kate.ward1b600c52008-11-12 21:26:05 +0000444 eval "${_flags_strToEval_}"
kate.wardda8d2c02011-06-28 13:26:02 +0000445 if [ -n "${_flags_infoValue_}" ]; then
kate.wardf51c6162008-06-17 16:38:35 +0000446 flags_return=${FLAGS_TRUE}
447 else
kate.wardda8d2c02011-06-28 13:26:02 +0000448 # see if the _flags_gFI_usName_ variable is a string as strings can be
449 # empty...
kate.ward1b600c52008-11-12 21:26:05 +0000450 # note: the DRY principle would say to have this function call itself for
451 # the next three lines, but doing so results in an infinite loop as an
452 # invalid _flags_name_ will also not have the associated _type variable.
453 # Because it doesn't (it will evaluate to an empty string) the logic will
454 # try to find the _type variable of the _type variable, and so on. Not so
455 # good ;-)
kate.wardda8d2c02011-06-28 13:26:02 +0000456 _flags_typeVar_="__flags_${_flags_gFI_usName_}_${__FLAGS_INFO_TYPE}"
kate.ward929261f2010-03-28 23:12:17 +0000457 _flags_strToEval_="_flags_typeValue_=\"\${${_flags_typeVar_}:-}\""
kate.ward1b600c52008-11-12 21:26:05 +0000458 eval "${_flags_strToEval_}"
kate.ward929261f2010-03-28 23:12:17 +0000459 if [ "${_flags_typeValue_}" = "${__FLAGS_TYPE_STRING}" ]; then
kate.ward437639d2008-10-19 17:21:41 +0000460 flags_return=${FLAGS_TRUE}
461 else
462 flags_return=${FLAGS_ERROR}
kate.wardda8d2c02011-06-28 13:26:02 +0000463 flags_error="missing flag info variable (${_flags_infoVar_})"
kate.ward437639d2008-10-19 17:21:41 +0000464 fi
kate.wardf51c6162008-06-17 16:38:35 +0000465 fi
466
kate.wardda8d2c02011-06-28 13:26:02 +0000467 echo "${_flags_infoValue_}"
468 unset _flags_gFI_usName_ _flags_gfI_info_ _flags_infoValue_ _flags_infoVar_ \
kate.ward929261f2010-03-28 23:12:17 +0000469 _flags_strToEval_ _flags_typeValue_ _flags_typeVar_
kate.wardf51c6162008-06-17 16:38:35 +0000470 [ ${flags_return} -eq ${FLAGS_ERROR} ] && _flags_error "${flags_error}"
471 return ${flags_return}
472}
473
kate.wardda8d2c02011-06-28 13:26:02 +0000474# Check for presense of item in a list.
475#
476# Passed a string (e.g. 'abc'), this function will determine if the string is
477# present in the list of strings (e.g. ' foo bar abc ').
kate.wardf51c6162008-06-17 16:38:35 +0000478#
479# Args:
kate.wardda8d2c02011-06-28 13:26:02 +0000480# _flags_str_: string: string to search for in a list of strings
kate.wardf51c6162008-06-17 16:38:35 +0000481# unnamed: list: list of strings
482# Returns:
483# boolean: true if item is in the list
kate.wardda8d2c02011-06-28 13:26:02 +0000484_flags_itemInList() {
kate.wardcc145682008-11-14 00:32:42 +0000485 _flags_str_=$1
kate.wardf51c6162008-06-17 16:38:35 +0000486 shift
487
kate.wardcc145682008-11-14 00:32:42 +0000488 echo " ${*:-} " |grep " ${_flags_str_} " >/dev/null
kate.wardf51c6162008-06-17 16:38:35 +0000489 if [ $? -eq 0 ]; then
490 flags_return=${FLAGS_TRUE}
491 else
492 flags_return=${FLAGS_FALSE}
493 fi
494
kate.wardcc145682008-11-14 00:32:42 +0000495 unset _flags_str_
kate.wardf51c6162008-06-17 16:38:35 +0000496 return ${flags_return}
497}
498
kate.warddadc1642008-11-14 02:00:29 +0000499# Returns the width of the current screen.
kate.ward437639d2008-10-19 17:21:41 +0000500#
kate.ward18d456e2008-11-14 00:46:44 +0000501# Output:
kate.warddadc1642008-11-14 02:00:29 +0000502# integer: width in columns of the current screen.
kate.ward14b33bf2008-11-15 20:03:18 +0000503_flags_columns()
kate.ward437639d2008-10-19 17:21:41 +0000504{
kate.ward14b33bf2008-11-15 20:03:18 +0000505 if [ -z "${__flags_columns}" ]; then
506 # determine the value and store it
507 if eval stty size >/dev/null 2>&1; then
508 # stty size worked :-)
509 set -- `stty size`
510 __flags_columns=$2
511 elif eval tput cols >/dev/null 2>&1; then
512 set -- `tput cols`
513 __flags_columns=$1
514 else
515 __flags_columns=80 # default terminal width
516 fi
kate.warddadc1642008-11-14 02:00:29 +0000517 fi
kate.ward14b33bf2008-11-15 20:03:18 +0000518 echo ${__flags_columns}
kate.ward437639d2008-10-19 17:21:41 +0000519}
520
521# Validate a boolean.
kate.wardf51c6162008-06-17 16:38:35 +0000522#
523# Args:
524# _flags__bool: boolean: value to validate
525# Returns:
526# bool: true if the value is a valid boolean
kate.warde10dd532013-01-04 21:52:23 +0000527_flags_validBool()
kate.wardf51c6162008-06-17 16:38:35 +0000528{
kate.ward60c25982008-11-12 20:12:44 +0000529 _flags_bool_=$1
kate.wardf51c6162008-06-17 16:38:35 +0000530
531 flags_return=${FLAGS_TRUE}
kate.ward60c25982008-11-12 20:12:44 +0000532 case "${_flags_bool_}" in
kate.wardf51c6162008-06-17 16:38:35 +0000533 true|t|0) ;;
534 false|f|1) ;;
535 *) flags_return=${FLAGS_FALSE} ;;
536 esac
537
kate.ward60c25982008-11-12 20:12:44 +0000538 unset _flags_bool_
kate.wardf51c6162008-06-17 16:38:35 +0000539 return ${flags_return}
540}
541
kate.warde10dd532013-01-04 21:52:23 +0000542# Validate a float using built-ins.
543#
544# Args:
545# _flags_float_: float: value to validate
546# _flags_recursed_: boolean: running recursed
547# Returns:
548# bool: true if the value is a valid integer
549_flags_validFloatBuiltin()
550{
551 flags_return=${FLAGS_FALSE}
552 [ -n "$1" ] || return ${flags_return}
553 _flags_float_=$1
554
555 _flags_curr_=${_flags_float_#-} # strip leading negative sign
556
557 # strip leading and trailing integers until only the decimal point remains
558 _flags_last_=${_flags_curr_}
559 _flags_curr_=${_flags_curr_#[0-9]}
560 _flags_curr_=${_flags_curr_%[0-9]}
561 while [ \
562 "${_flags_curr_}" != '.' -a \
563 -n "${_flags_curr_}" -a \
564 "${_flags_curr_}" != "${_flags_last_}" \
565 ]; do
566 _flags_last_=${_flags_curr_}
567 _flags_curr_=${_flags_curr_#[0-9]}
568 _flags_curr_=${_flags_curr_%[0-9]}
569 done
570 [ "${_flags_curr_}" = '.' -o -z "${_flags_curr_}" ] && \
571 flags_return=${FLAGS_TRUE}
572
573 unset _flags_curr_ _flags_float_ _flags_last_
574 return ${flags_return}
575}
576
kate.ward18d456e2008-11-14 00:46:44 +0000577# Validate a float.
kate.wardf51c6162008-06-17 16:38:35 +0000578#
579# Args:
580# _flags__float: float: value to validate
581# Returns:
582# bool: true if the value is a valid float
kate.warde10dd532013-01-04 21:52:23 +0000583_flags_validFloatExpr()
kate.wardf51c6162008-06-17 16:38:35 +0000584{
kate.wardcc145682008-11-14 00:32:42 +0000585 _flags_float_=$1
kate.wardf51c6162008-06-17 16:38:35 +0000586
kate.warde10dd532013-01-04 21:52:23 +0000587 if ${__FLAGS_FX_VALID_INT} ${_flags_float_}; then
kate.wardf51c6162008-06-17 16:38:35 +0000588 flags_return=${FLAGS_TRUE}
589 else
590 flags_return=${FLAGS_TRUE}
kate.wardcc145682008-11-14 00:32:42 +0000591 case ${_flags_float_} in
kate.wardf51c6162008-06-17 16:38:35 +0000592 -*) # negative floats
kate.warda589b9f2012-12-17 23:27:51 +0000593 _flags_test_=`${FLAGS_EXPR_CMD} -- "${_flags_float_}" :\
kate.warde10dd532013-01-04 21:52:23 +0000594 '\(-[0-9]*\.[0-9]*\)'`
kate.wardf51c6162008-06-17 16:38:35 +0000595 ;;
596 *) # positive floats
kate.warda589b9f2012-12-17 23:27:51 +0000597 _flags_test_=`${FLAGS_EXPR_CMD} -- "${_flags_float_}" :\
kate.warde10dd532013-01-04 21:52:23 +0000598 '\([0-9]*\.[0-9]*\)'`
kate.wardf51c6162008-06-17 16:38:35 +0000599 ;;
600 esac
kate.wardcc145682008-11-14 00:32:42 +0000601 [ "${_flags_test_}" != "${_flags_float_}" ] && flags_return=${FLAGS_FALSE}
kate.wardf51c6162008-06-17 16:38:35 +0000602 fi
603
kate.wardcc145682008-11-14 00:32:42 +0000604 unset _flags_float_ _flags_test_
kate.wardf51c6162008-06-17 16:38:35 +0000605 return ${flags_return}
606}
607
kate.warde10dd532013-01-04 21:52:23 +0000608# Validate an integer using built-ins.
kate.wardf51c6162008-06-17 16:38:35 +0000609#
610# Args:
kate.warde10dd532013-01-04 21:52:23 +0000611# _flags_int_: integer: value to validate
612# _flags_recursed_: boolean: running recursed
kate.wardf51c6162008-06-17 16:38:35 +0000613# Returns:
614# bool: true if the value is a valid integer
kate.warde10dd532013-01-04 21:52:23 +0000615_flags_validIntBuiltin()
616{
617 flags_return=${FLAGS_FALSE}
618 [ -n "$1" ] || return ${flags_return}
619 _flags_int_=$1
620
621 _flags_curr_=${_flags_int_#-} # strip leading negative sign
622
623 _flags_last_=${_flags_curr_}
624 _flags_curr_=${_flags_curr_#[0-9]}
625 while [ -n "${_flags_curr_}" -a "${_flags_curr_}" != "${_flags_last_}" ]; do
626 _flags_last_=${_flags_curr_}
627 _flags_curr_=${_flags_curr_#[0-9]}
628 done
629 [ -z "${_flags_curr_}" ] && flags_return=${FLAGS_TRUE}
630
631 unset _flags_curr_ _flags_int_ _flags_last_
632 return ${flags_return}
633}
634
635# Validate an integer using expr.
636#
637# Args:
638# _flags_int_: interger: value to validate
639# Returns:
640# bool: true if the value is a valid integer
641_flags_validIntExpr()
kate.wardf51c6162008-06-17 16:38:35 +0000642{
kate.wardcc145682008-11-14 00:32:42 +0000643 _flags_int_=$1
kate.wardf51c6162008-06-17 16:38:35 +0000644
645 flags_return=${FLAGS_TRUE}
kate.wardcc145682008-11-14 00:32:42 +0000646 case ${_flags_int_} in
kate.wardf51c6162008-06-17 16:38:35 +0000647 -*) # negative ints
kate.warda589b9f2012-12-17 23:27:51 +0000648 _flags_test_=`${FLAGS_EXPR_CMD} -- "${_flags_int_}" : '\(-[0-9][0-9]*\)'`
kate.wardf51c6162008-06-17 16:38:35 +0000649 ;;
650 *) # positive ints
kate.warda589b9f2012-12-17 23:27:51 +0000651 _flags_test_=`${FLAGS_EXPR_CMD} -- "${_flags_int_}" : '\([0-9][0-9]*\)'`
kate.wardf51c6162008-06-17 16:38:35 +0000652 ;;
653 esac
kate.wardcc145682008-11-14 00:32:42 +0000654 [ "${_flags_test_}" != "${_flags_int_}" ] && flags_return=${FLAGS_FALSE}
kate.wardf51c6162008-06-17 16:38:35 +0000655
kate.wardcc145682008-11-14 00:32:42 +0000656 unset _flags_int_ _flags_test_
kate.wardf51c6162008-06-17 16:38:35 +0000657 return ${flags_return}
658}
659
kate.ward18d456e2008-11-14 00:46:44 +0000660# Parse command-line options using the standard getopt.
661#
662# Note: the flag options are passed around in the global __flags_opts so that
663# the formatting is not lost due to shell parsing and such.
664#
665# Args:
666# @: varies: command-line options to parse
667# Returns:
668# integer: a FLAGS success condition
kate.wardc5210682009-03-30 18:54:36 +0000669_flags_getoptStandard()
kate.ward60c25982008-11-12 20:12:44 +0000670{
671 flags_return=${FLAGS_TRUE}
672 _flags_shortOpts_=`_flags_genOptStr ${__FLAGS_OPTSTR_SHORT}`
673
674 # check for spaces in passed options
675 for _flags_opt_ in "$@"; do
kate.ward14b33bf2008-11-15 20:03:18 +0000676 # note: the silliness with the x's is purely for ksh93 on Ubuntu 6.06
677 _flags_match_=`echo "x${_flags_opt_}x" |sed 's/ //g'`
678 if [ "${_flags_match_}" != "x${_flags_opt_}x" ]; then
kate.ward60c25982008-11-12 20:12:44 +0000679 flags_error='the available getopt does not support spaces in options'
680 flags_return=${FLAGS_ERROR}
681 break
682 fi
683 done
684
685 if [ ${flags_return} -eq ${FLAGS_TRUE} ]; then
686 __flags_opts=`getopt ${_flags_shortOpts_} $@ 2>&1`
kate.ward7a3c9c42009-04-24 00:12:35 +0000687 _flags_rtrn_=$?
688 if [ ${_flags_rtrn_} -ne ${FLAGS_TRUE} ]; then
689 _flags_warn "${__flags_opts}"
kate.ward60c25982008-11-12 20:12:44 +0000690 flags_error='unable to parse provided options with getopt.'
691 flags_return=${FLAGS_ERROR}
692 fi
693 fi
694
kate.ward7a3c9c42009-04-24 00:12:35 +0000695 unset _flags_match_ _flags_opt_ _flags_rtrn_ _flags_shortOpts_
kate.ward60c25982008-11-12 20:12:44 +0000696 return ${flags_return}
697}
698
kate.ward18d456e2008-11-14 00:46:44 +0000699# Parse command-line options using the enhanced getopt.
700#
701# Note: the flag options are passed around in the global __flags_opts so that
702# the formatting is not lost due to shell parsing and such.
703#
704# Args:
705# @: varies: command-line options to parse
706# Returns:
707# integer: a FLAGS success condition
kate.wardc5210682009-03-30 18:54:36 +0000708_flags_getoptEnhanced()
kate.ward60c25982008-11-12 20:12:44 +0000709{
710 flags_return=${FLAGS_TRUE}
711 _flags_shortOpts_=`_flags_genOptStr ${__FLAGS_OPTSTR_SHORT}`
712 _flags_boolOpts_=`echo "${__flags_boolNames}" \
kate.wardd9672402008-11-15 16:55:51 +0000713 |sed 's/^ *//;s/ *$//;s/ /,/g'`
kate.ward60c25982008-11-12 20:12:44 +0000714 _flags_longOpts_=`_flags_genOptStr ${__FLAGS_OPTSTR_LONG}`
715
kate.ward1cb79602011-06-10 11:15:49 +0000716 __flags_opts=`${FLAGS_GETOPT_CMD} \
kate.ward60c25982008-11-12 20:12:44 +0000717 -o ${_flags_shortOpts_} \
718 -l "${_flags_longOpts_},${_flags_boolOpts_}" \
719 -- "$@" 2>&1`
kate.ward7a3c9c42009-04-24 00:12:35 +0000720 _flags_rtrn_=$?
721 if [ ${_flags_rtrn_} -ne ${FLAGS_TRUE} ]; then
722 _flags_warn "${__flags_opts}"
kate.ward60c25982008-11-12 20:12:44 +0000723 flags_error='unable to parse provided options with getopt.'
724 flags_return=${FLAGS_ERROR}
725 fi
726
kate.ward7a3c9c42009-04-24 00:12:35 +0000727 unset _flags_boolOpts_ _flags_longOpts_ _flags_rtrn_ _flags_shortOpts_
kate.ward60c25982008-11-12 20:12:44 +0000728 return ${flags_return}
729}
730
kate.ward18d456e2008-11-14 00:46:44 +0000731# Dynamically parse a getopt result and set appropriate variables.
732#
733# This function does the actual conversion of getopt output and runs it through
734# the standard case structure for parsing. The case structure is actually quite
735# dynamic to support any number of flags.
736#
737# Args:
kate.ward9ccba512009-04-01 13:31:18 +0000738# argc: int: original command-line argument count
kate.ward18d456e2008-11-14 00:46:44 +0000739# @: varies: output from getopt parsing
740# Returns:
741# integer: a FLAGS success condition
kate.ward60c25982008-11-12 20:12:44 +0000742_flags_parseGetopt()
743{
kate.ward9ccba512009-04-01 13:31:18 +0000744 _flags_argc_=$1
745 shift
746
kate.ward60c25982008-11-12 20:12:44 +0000747 flags_return=${FLAGS_TRUE}
748
749 if [ ${__FLAGS_GETOPT_VERS} -ne ${__FLAGS_GETOPT_VERS_ENH} ]; then
750 set -- $@
751 else
752 # note the quotes around the `$@' -- they are essential!
753 eval set -- "$@"
754 fi
755
kate.ward4eb31352013-01-01 22:05:32 +0000756 # Provide user with the number of arguments to shift by later.
kate.wardc5210682009-03-30 18:54:36 +0000757 # NOTE: the FLAGS_ARGC variable is obsolete as of 1.0.3 because it does not
758 # properly give user access to non-flag arguments mixed in between flag
759 # arguments. Its usage was replaced by FLAGS_ARGV, and it is being kept only
760 # for backwards compatibility reasons.
kate.wardb0759202013-01-05 13:57:46 +0000761 FLAGS_ARGC=`${__FLAGS_FX_MATH} $# - 1 - ${_flags_argc_}`
kate.wardc5210682009-03-30 18:54:36 +0000762
kate.ward60c25982008-11-12 20:12:44 +0000763 # handle options. note options with values must do an additional shift
764 while true; do
765 _flags_opt_=$1
766 _flags_arg_=${2:-}
767 _flags_type_=${__FLAGS_TYPE_NONE}
kate.ward7a3c9c42009-04-24 00:12:35 +0000768 _flags_name_=''
kate.ward60c25982008-11-12 20:12:44 +0000769
770 # determine long flag name
771 case "${_flags_opt_}" in
772 --) shift; break ;; # discontinue option parsing
773
774 --*) # long option
kate.warda589b9f2012-12-17 23:27:51 +0000775 _flags_opt_=`${FLAGS_EXPR_CMD} -- "${_flags_opt_}" : '--\(.*\)'`
kate.ward60c25982008-11-12 20:12:44 +0000776 _flags_len_=${__FLAGS_LEN_LONG}
777 if _flags_itemInList "${_flags_opt_}" ${__flags_longNames}; then
778 _flags_name_=${_flags_opt_}
779 else
780 # check for negated long boolean version
781 if _flags_itemInList "${_flags_opt_}" ${__flags_boolNames}; then
kate.warda589b9f2012-12-17 23:27:51 +0000782 _flags_name_=`${FLAGS_EXPR_CMD} -- "${_flags_opt_}" : 'no\(.*\)'`
kate.ward60c25982008-11-12 20:12:44 +0000783 _flags_type_=${__FLAGS_TYPE_BOOLEAN}
784 _flags_arg_=${__FLAGS_NULL}
785 fi
786 fi
787 ;;
788
789 -*) # short option
kate.warda589b9f2012-12-17 23:27:51 +0000790 _flags_opt_=`${FLAGS_EXPR_CMD} -- "${_flags_opt_}" : '-\(.*\)'`
kate.ward60c25982008-11-12 20:12:44 +0000791 _flags_len_=${__FLAGS_LEN_SHORT}
792 if _flags_itemInList "${_flags_opt_}" ${__flags_shortNames}; then
793 # yes. match short name to long name. note purposeful off-by-one
794 # (too high) with awk calculations.
795 _flags_pos_=`echo "${__flags_shortNames}" \
796 |awk 'BEGIN{RS=" ";rn=0}$0==e{rn=NR}END{print rn}' \
797 e=${_flags_opt_}`
798 _flags_name_=`echo "${__flags_longNames}" \
799 |awk 'BEGIN{RS=" "}rn==NR{print $0}' rn="${_flags_pos_}"`
800 fi
801 ;;
802 esac
803
804 # die if the flag was unrecognized
805 if [ -z "${_flags_name_}" ]; then
806 flags_error="unrecognized option (${_flags_opt_})"
807 flags_return=${FLAGS_ERROR}
808 break
809 fi
810
kate.ward60c25982008-11-12 20:12:44 +0000811 # set new flag value
kate.wardda8d2c02011-06-28 13:26:02 +0000812 _flags_usName_=`_flags_underscoreName ${_flags_name_}`
kate.ward60c25982008-11-12 20:12:44 +0000813 [ ${_flags_type_} -eq ${__FLAGS_TYPE_NONE} ] && \
kate.wardd9672402008-11-15 16:55:51 +0000814 _flags_type_=`_flags_getFlagInfo \
kate.wardda8d2c02011-06-28 13:26:02 +0000815 "${_flags_usName_}" ${__FLAGS_INFO_TYPE}`
kate.ward60c25982008-11-12 20:12:44 +0000816 case ${_flags_type_} in
817 ${__FLAGS_TYPE_BOOLEAN})
818 if [ ${_flags_len_} -eq ${__FLAGS_LEN_LONG} ]; then
819 if [ "${_flags_arg_}" != "${__FLAGS_NULL}" ]; then
kate.wardda8d2c02011-06-28 13:26:02 +0000820 eval "FLAGS_${_flags_usName_}=${FLAGS_TRUE}"
kate.ward60c25982008-11-12 20:12:44 +0000821 else
kate.wardda8d2c02011-06-28 13:26:02 +0000822 eval "FLAGS_${_flags_usName_}=${FLAGS_FALSE}"
kate.ward60c25982008-11-12 20:12:44 +0000823 fi
824 else
825 _flags_strToEval_="_flags_val_=\
kate.wardda8d2c02011-06-28 13:26:02 +0000826\${__flags_${_flags_usName_}_${__FLAGS_INFO_DEFAULT}}"
kate.ward60c25982008-11-12 20:12:44 +0000827 eval "${_flags_strToEval_}"
828 if [ ${_flags_val_} -eq ${FLAGS_FALSE} ]; then
kate.wardda8d2c02011-06-28 13:26:02 +0000829 eval "FLAGS_${_flags_usName_}=${FLAGS_TRUE}"
kate.ward60c25982008-11-12 20:12:44 +0000830 else
kate.wardda8d2c02011-06-28 13:26:02 +0000831 eval "FLAGS_${_flags_usName_}=${FLAGS_FALSE}"
kate.ward60c25982008-11-12 20:12:44 +0000832 fi
833 fi
834 ;;
835
836 ${__FLAGS_TYPE_FLOAT})
kate.warde10dd532013-01-04 21:52:23 +0000837 if ${__FLAGS_FX_VALID_FLOAT} "${_flags_arg_}"; then
kate.wardda8d2c02011-06-28 13:26:02 +0000838 eval "FLAGS_${_flags_usName_}='${_flags_arg_}'"
kate.ward60c25982008-11-12 20:12:44 +0000839 else
840 flags_error="invalid float value (${_flags_arg_})"
841 flags_return=${FLAGS_ERROR}
842 break
843 fi
844 ;;
845
846 ${__FLAGS_TYPE_INTEGER})
kate.warde10dd532013-01-04 21:52:23 +0000847 if ${__FLAGS_FX_VALID_INT} "${_flags_arg_}"; then
kate.wardda8d2c02011-06-28 13:26:02 +0000848 eval "FLAGS_${_flags_usName_}='${_flags_arg_}'"
kate.ward60c25982008-11-12 20:12:44 +0000849 else
850 flags_error="invalid integer value (${_flags_arg_})"
851 flags_return=${FLAGS_ERROR}
852 break
853 fi
854 ;;
855
856 ${__FLAGS_TYPE_STRING})
kate.wardda8d2c02011-06-28 13:26:02 +0000857 eval "FLAGS_${_flags_usName_}='${_flags_arg_}'"
kate.ward60c25982008-11-12 20:12:44 +0000858 ;;
859 esac
860
kate.ward5b48fc12008-11-13 00:42:43 +0000861 # handle special case help flag
kate.wardda8d2c02011-06-28 13:26:02 +0000862 if [ "${_flags_usName_}" = 'help' ]; then
kate.ward5b48fc12008-11-13 00:42:43 +0000863 if [ ${FLAGS_help} -eq ${FLAGS_TRUE} ]; then
864 flags_help
865 flags_error='help requested'
kate.ward1cb79602011-06-10 11:15:49 +0000866 flags_return=${FLAGS_TRUE}
kate.ward5b48fc12008-11-13 00:42:43 +0000867 break
868 fi
869 fi
870
kate.wardc5210682009-03-30 18:54:36 +0000871 # shift the option and non-boolean arguements out.
kate.ward60c25982008-11-12 20:12:44 +0000872 shift
kate.wardc5210682009-03-30 18:54:36 +0000873 [ ${_flags_type_} != ${__FLAGS_TYPE_BOOLEAN} ] && shift
874 done
kate.ward60c25982008-11-12 20:12:44 +0000875
kate.wardc5210682009-03-30 18:54:36 +0000876 # give user back non-flag arguments
877 FLAGS_ARGV=''
878 while [ $# -gt 0 ]; do
879 FLAGS_ARGV="${FLAGS_ARGV:+${FLAGS_ARGV} }'$1'"
880 shift
kate.ward60c25982008-11-12 20:12:44 +0000881 done
882
883 unset _flags_arg_ _flags_len_ _flags_name_ _flags_opt_ _flags_pos_ \
kate.wardda8d2c02011-06-28 13:26:02 +0000884 _flags_strToEval_ _flags_type_ _flags_usName_ _flags_val_
kate.ward60c25982008-11-12 20:12:44 +0000885 return ${flags_return}
886}
887
kate.wardb0759202013-01-05 13:57:46 +0000888# Perform some path using built-ins.
889#
890# Args:
891# $@: string: math expression to evaluate
892# Returns:
893# string: the result
894_flags_mathBuiltin()
895{
896 # Variable assignment is needed as workaround for Solaris Bourne shell, which
897 # cannot parse a bare $((expression)).
898 _flags_expr_='$(($@))'
899 eval echo ${_flags_expr_}
900}
901
902# Perform some path using expr.
903#
904# Args:
905# $@: string: math expression to evaluate
906# Returns:
907# string: the result
908_flags_mathExpr()
909{
910 eval ${FLAGS_EXPR_CMD} $@
911}
912
kate.wardf51c6162008-06-17 16:38:35 +0000913#------------------------------------------------------------------------------
914# public functions
915#
916
917# A basic boolean flag. Boolean flags do not take any arguments, and their
918# value is either 1 (false) or 0 (true). For long flags, the false value is
919# specified on the command line by prepending the word 'no'. With short flags,
920# the presense of the flag toggles the current value between true and false.
921# Specifying a short boolean flag twice on the command results in returning the
922# value back to the default value.
923#
924# A default value is required for boolean flags.
925#
926# For example, lets say a Boolean flag was created whose long name was 'update'
927# and whose short name was 'x', and the default value was 'false'. This flag
928# could be explicitly set to 'true' with '--update' or by '-x', and it could be
929# explicitly set to 'false' with '--noupdate'.
930DEFINE_boolean() { _flags_define ${__FLAGS_TYPE_BOOLEAN} "$@"; }
931
932# Other basic flags.
933DEFINE_float() { _flags_define ${__FLAGS_TYPE_FLOAT} "$@"; }
934DEFINE_integer() { _flags_define ${__FLAGS_TYPE_INTEGER} "$@"; }
935DEFINE_string() { _flags_define ${__FLAGS_TYPE_STRING} "$@"; }
936
937# Parse the flags.
938#
939# Args:
940# unnamed: list: command-line flags to parse
941# Returns:
942# integer: success of operation, or error
943FLAGS()
944{
kate.wardabae05d2008-11-12 22:37:05 +0000945 # define a standard 'help' flag if one isn't already defined
946 [ -z "${__flags_help_type:-}" ] && \
947 DEFINE_boolean 'help' false 'show this help' 'h'
kate.wardf08c5b62008-07-11 20:32:11 +0000948
kate.wardf51c6162008-06-17 16:38:35 +0000949 # parse options
kate.ward7a3c9c42009-04-24 00:12:35 +0000950 if [ $# -gt 0 ]; then
951 if [ ${__FLAGS_GETOPT_VERS} -ne ${__FLAGS_GETOPT_VERS_ENH} ]; then
952 _flags_getoptStandard "$@"
953 else
954 _flags_getoptEnhanced "$@"
955 fi
956 flags_return=$?
kate.wardf51c6162008-06-17 16:38:35 +0000957 else
kate.ward7a3c9c42009-04-24 00:12:35 +0000958 # nothing passed; won't bother running getopt
959 __flags_opts='--'
960 flags_return=${FLAGS_TRUE}
kate.wardf51c6162008-06-17 16:38:35 +0000961 fi
kate.ward60c25982008-11-12 20:12:44 +0000962
963 if [ ${flags_return} -eq ${FLAGS_TRUE} ]; then
kate.ward7a3c9c42009-04-24 00:12:35 +0000964 _flags_parseGetopt $# "${__flags_opts}"
kate.ward60c25982008-11-12 20:12:44 +0000965 flags_return=$?
kate.wardf51c6162008-06-17 16:38:35 +0000966 fi
967
kate.wardf51c6162008-06-17 16:38:35 +0000968 [ ${flags_return} -eq ${FLAGS_ERROR} ] && _flags_fatal "${flags_error}"
969 return ${flags_return}
970}
971
kate.ward1cb79602011-06-10 11:15:49 +0000972# This is a helper function for determining the 'getopt' version for platforms
kate.wardf51c6162008-06-17 16:38:35 +0000973# where the detection isn't working. It simply outputs debug information that
974# can be included in a bug report.
975#
976# Args:
977# none
kate.ward60c25982008-11-12 20:12:44 +0000978# Output:
979# debug info that can be included in a bug report
kate.wardf51c6162008-06-17 16:38:35 +0000980# Returns:
981# nothing
982flags_getoptInfo()
983{
984 # platform info
985 _flags_debug "uname -a: `uname -a`"
986 _flags_debug "PATH: ${PATH}"
987
988 # shell info
989 if [ -n "${BASH_VERSION:-}" ]; then
990 _flags_debug 'shell: bash'
991 _flags_debug "BASH_VERSION: ${BASH_VERSION}"
992 elif [ -n "${ZSH_VERSION:-}" ]; then
993 _flags_debug 'shell: zsh'
994 _flags_debug "ZSH_VERSION: ${ZSH_VERSION}"
995 fi
996
997 # getopt info
kate.ward1cb79602011-06-10 11:15:49 +0000998 ${FLAGS_GETOPT_CMD} >/dev/null
kate.ward60c25982008-11-12 20:12:44 +0000999 _flags_getoptReturn=$?
1000 _flags_debug "getopt return: ${_flags_getoptReturn}"
kate.ward1cb79602011-06-10 11:15:49 +00001001 _flags_debug "getopt --version: `${FLAGS_GETOPT_CMD} --version 2>&1`"
kate.wardf51c6162008-06-17 16:38:35 +00001002
kate.ward60c25982008-11-12 20:12:44 +00001003 unset _flags_getoptReturn
kate.wardf51c6162008-06-17 16:38:35 +00001004}
1005
kate.ward437639d2008-10-19 17:21:41 +00001006# Returns whether the detected getopt version is the enhanced version.
kate.wardf51c6162008-06-17 16:38:35 +00001007#
1008# Args:
1009# none
kate.ward60c25982008-11-12 20:12:44 +00001010# Output:
1011# none
kate.wardf51c6162008-06-17 16:38:35 +00001012# Returns:
1013# bool: true if getopt is the enhanced version
1014flags_getoptIsEnh()
1015{
kate.wardd9672402008-11-15 16:55:51 +00001016 test ${__FLAGS_GETOPT_VERS} -eq ${__FLAGS_GETOPT_VERS_ENH}
kate.wardf51c6162008-06-17 16:38:35 +00001017}
1018
kate.ward437639d2008-10-19 17:21:41 +00001019# Returns whether the detected getopt version is the standard version.
kate.wardf51c6162008-06-17 16:38:35 +00001020#
1021# Args:
1022# none
1023# Returns:
1024# bool: true if getopt is the standard version
1025flags_getoptIsStd()
1026{
kate.wardd9672402008-11-15 16:55:51 +00001027 test ${__FLAGS_GETOPT_VERS} -eq ${__FLAGS_GETOPT_VERS_STD}
kate.wardf51c6162008-06-17 16:38:35 +00001028}
1029
1030# This is effectively a 'usage()' function. It prints usage information and
1031# exits the program with ${FLAGS_FALSE} if it is ever found in the command line
1032# arguments. Note this function can be overridden so other apps can define
1033# their own --help flag, replacing this one, if they want.
1034#
1035# Args:
1036# none
1037# Returns:
1038# integer: success of operation (always returns true)
1039flags_help()
1040{
1041 if [ -n "${FLAGS_HELP:-}" ]; then
1042 echo "${FLAGS_HELP}" >&2
1043 else
kate.ward4d8e0472008-07-18 11:52:56 +00001044 echo "USAGE: ${FLAGS_PARENT:-$0} [flags] args" >&2
kate.wardf51c6162008-06-17 16:38:35 +00001045 fi
1046 if [ -n "${__flags_longNames}" ]; then
1047 echo 'flags:' >&2
kate.wardd9672402008-11-15 16:55:51 +00001048 for flags_name_ in ${__flags_longNames}; do
1049 flags_flagStr_=''
kate.wardbda39172009-04-01 03:30:22 +00001050 flags_boolStr_=''
kate.wardda8d2c02011-06-28 13:26:02 +00001051 flags_usName_=`_flags_underscoreName ${flags_name_}`
kate.wardf51c6162008-06-17 16:38:35 +00001052
kate.wardd9672402008-11-15 16:55:51 +00001053 flags_default_=`_flags_getFlagInfo \
kate.wardda8d2c02011-06-28 13:26:02 +00001054 "${flags_usName_}" ${__FLAGS_INFO_DEFAULT}`
kate.wardd9672402008-11-15 16:55:51 +00001055 flags_help_=`_flags_getFlagInfo \
kate.wardda8d2c02011-06-28 13:26:02 +00001056 "${flags_usName_}" ${__FLAGS_INFO_HELP}`
kate.wardd9672402008-11-15 16:55:51 +00001057 flags_short_=`_flags_getFlagInfo \
kate.wardda8d2c02011-06-28 13:26:02 +00001058 "${flags_usName_}" ${__FLAGS_INFO_SHORT}`
kate.wardd9672402008-11-15 16:55:51 +00001059 flags_type_=`_flags_getFlagInfo \
kate.wardda8d2c02011-06-28 13:26:02 +00001060 "${flags_usName_}" ${__FLAGS_INFO_TYPE}`
kate.wardf51c6162008-06-17 16:38:35 +00001061
kate.ward1cb79602011-06-10 11:15:49 +00001062 [ "${flags_short_}" != "${__FLAGS_NULL}" ] && \
1063 flags_flagStr_="-${flags_short_}"
kate.ward437639d2008-10-19 17:21:41 +00001064
1065 if [ ${__FLAGS_GETOPT_VERS} -eq ${__FLAGS_GETOPT_VERS_ENH} ]; then
kate.ward1cb79602011-06-10 11:15:49 +00001066 [ "${flags_short_}" != "${__FLAGS_NULL}" ] && \
1067 flags_flagStr_="${flags_flagStr_},"
1068 # add [no] to long boolean flag names, except the 'help' flag
1069 [ ${flags_type_} -eq ${__FLAGS_TYPE_BOOLEAN} \
kate.wardda8d2c02011-06-28 13:26:02 +00001070 -a "${flags_usName_}" != 'help' ] && \
kate.ward1cb79602011-06-10 11:15:49 +00001071 flags_boolStr_='[no]'
kate.wardd9672402008-11-15 16:55:51 +00001072 flags_flagStr_="${flags_flagStr_}--${flags_boolStr_}${flags_name_}:"
kate.ward437639d2008-10-19 17:21:41 +00001073 fi
1074
kate.wardd9672402008-11-15 16:55:51 +00001075 case ${flags_type_} in
kate.ward437639d2008-10-19 17:21:41 +00001076 ${__FLAGS_TYPE_BOOLEAN})
kate.wardd9672402008-11-15 16:55:51 +00001077 if [ ${flags_default_} -eq ${FLAGS_TRUE} ]; then
1078 flags_defaultStr_='true'
kate.ward437639d2008-10-19 17:21:41 +00001079 else
kate.wardd9672402008-11-15 16:55:51 +00001080 flags_defaultStr_='false'
kate.ward437639d2008-10-19 17:21:41 +00001081 fi
1082 ;;
1083 ${__FLAGS_TYPE_FLOAT}|${__FLAGS_TYPE_INTEGER})
kate.wardd9672402008-11-15 16:55:51 +00001084 flags_defaultStr_=${flags_default_} ;;
1085 ${__FLAGS_TYPE_STRING}) flags_defaultStr_="'${flags_default_}'" ;;
kate.ward437639d2008-10-19 17:21:41 +00001086 esac
kate.wardd9672402008-11-15 16:55:51 +00001087 flags_defaultStr_="(default: ${flags_defaultStr_})"
kate.ward437639d2008-10-19 17:21:41 +00001088
kate.wardd9672402008-11-15 16:55:51 +00001089 flags_helpStr_=" ${flags_flagStr_} ${flags_help_} ${flags_defaultStr_}"
kate.warda589b9f2012-12-17 23:27:51 +00001090 flags_helpStrLen_=`${FLAGS_EXPR_CMD} -- "${flags_helpStr_}" : '.*'`
kate.ward14b33bf2008-11-15 20:03:18 +00001091 flags_columns_=`_flags_columns`
1092 if [ ${flags_helpStrLen_} -lt ${flags_columns_} ]; then
kate.wardd9672402008-11-15 16:55:51 +00001093 echo "${flags_helpStr_}" >&2
kate.ward437639d2008-10-19 17:21:41 +00001094 else
kate.ward64099612008-11-17 17:36:14 +00001095 echo " ${flags_flagStr_} ${flags_help_}" >&2
kate.ward87b0c2e2009-04-01 03:48:16 +00001096 # note: the silliness with the x's is purely for ksh93 on Ubuntu 6.06
1097 # because it doesn't like empty strings when used in this manner.
kate.wardbda39172009-04-01 03:30:22 +00001098 flags_emptyStr_="`echo \"x${flags_flagStr_}x\" \
1099 |awk '{printf "%"length($0)-2"s", ""}'`"
kate.ward64099612008-11-17 17:36:14 +00001100 flags_helpStr_=" ${flags_emptyStr_} ${flags_defaultStr_}"
kate.warda589b9f2012-12-17 23:27:51 +00001101 flags_helpStrLen_=`${FLAGS_EXPR_CMD} -- "${flags_helpStr_}" : '.*'`
kate.ward64099612008-11-17 17:36:14 +00001102 if [ ${__FLAGS_GETOPT_VERS} -eq ${__FLAGS_GETOPT_VERS_STD} \
1103 -o ${flags_helpStrLen_} -lt ${flags_columns_} ]; then
1104 # indented to match help string
1105 echo "${flags_helpStr_}" >&2
1106 else
1107 # indented four from left to allow for longer defaults as long flag
1108 # names might be used too, making things too long
1109 echo " ${flags_defaultStr_}" >&2
1110 fi
kate.ward437639d2008-10-19 17:21:41 +00001111 fi
kate.wardf51c6162008-06-17 16:38:35 +00001112 done
1113 fi
1114
kate.wardd9672402008-11-15 16:55:51 +00001115 unset flags_boolStr_ flags_default_ flags_defaultStr_ flags_emptyStr_ \
1116 flags_flagStr_ flags_help_ flags_helpStr flags_helpStrLen flags_name_ \
kate.wardda8d2c02011-06-28 13:26:02 +00001117 flags_columns_ flags_short_ flags_type_ flags_usName_
kate.wardf51c6162008-06-17 16:38:35 +00001118 return ${FLAGS_TRUE}
1119}
1120
kate.wardcecd7bd2009-05-10 17:56:13 +00001121# Reset shflags back to an uninitialized state.
kate.wardf51c6162008-06-17 16:38:35 +00001122#
1123# Args:
1124# none
1125# Returns:
1126# nothing
1127flags_reset()
1128{
kate.wardd9672402008-11-15 16:55:51 +00001129 for flags_name_ in ${__flags_longNames}; do
kate.ward869a71b2011-06-28 13:29:40 +00001130 flags_usName_=`_flags_underscoreName ${flags_name_}`
1131 flags_strToEval_="unset FLAGS_${flags_usName_}"
kate.wardd9672402008-11-15 16:55:51 +00001132 for flags_type_ in \
kate.wardcecd7bd2009-05-10 17:56:13 +00001133 ${__FLAGS_INFO_DEFAULT} \
kate.wardfb0e7182009-05-10 16:59:13 +00001134 ${__FLAGS_INFO_HELP} \
1135 ${__FLAGS_INFO_SHORT} \
1136 ${__FLAGS_INFO_TYPE}
kate.wardf51c6162008-06-17 16:38:35 +00001137 do
kate.wardd9672402008-11-15 16:55:51 +00001138 flags_strToEval_=\
kate.ward869a71b2011-06-28 13:29:40 +00001139"${flags_strToEval_} __flags_${flags_usName_}_${flags_type_}"
kate.wardf51c6162008-06-17 16:38:35 +00001140 done
kate.wardd9672402008-11-15 16:55:51 +00001141 eval ${flags_strToEval_}
kate.wardf51c6162008-06-17 16:38:35 +00001142 done
1143
1144 # reset internal variables
1145 __flags_boolNames=' '
1146 __flags_longNames=' '
1147 __flags_shortNames=' '
kate.wardda8d2c02011-06-28 13:26:02 +00001148 __flags_definedNames=' '
kate.wardf51c6162008-06-17 16:38:35 +00001149
kate.ward869a71b2011-06-28 13:29:40 +00001150 unset flags_name_ flags_type_ flags_strToEval_ flags_usName_
kate.wardf51c6162008-06-17 16:38:35 +00001151}