blob: 7db9573d4b5acc64dc09b8ab78c94a6983e4efec [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#
kate.ward0a4f16f2013-01-07 22:10:49 +000021# DEFINE_boolean: does not take any arguments. Say --myflag to set
22# FLAGS_myflag to true, or --nomyflag to set FLAGS_myflag to false. For short
23# flags, passing the flag on the command-line negates the default value, i.e.
24# if the default is true, passing the flag sets the value to false.
kate.wardf51c6162008-06-17 16:38:35 +000025#
26# DEFINE_float: takes an input and intreprets it as a floating point number. As
27# shell does not support floats per-se, the input is merely validated as
28# being a valid floating point value.
29#
30# DEFINE_integer: takes an input and intreprets it as an integer.
31#
32# SPECIAL FLAGS: There are a few flags that have special meaning:
33# --help (or -?) prints a list of all the flags in a human-readable fashion
kate.ward39913d12009-04-01 14:09:23 +000034# --flagfile=foo read flags from foo. (not implemented yet)
kate.wardf51c6162008-06-17 16:38:35 +000035# -- as in getopt(), terminates flag-processing
36#
37# EXAMPLE USAGE:
38#
kate.wardc5210682009-03-30 18:54:36 +000039# -- begin hello.sh --
kate.wardf51c6162008-06-17 16:38:35 +000040# #! /bin/sh
41# . ./shflags
kate.wardc5210682009-03-30 18:54:36 +000042# DEFINE_string name 'world' "somebody's name" n
43# FLAGS "$@" || exit $?
44# eval set -- "${FLAGS_ARGV}"
kate.wardf51c6162008-06-17 16:38:35 +000045# echo "Hello, ${FLAGS_name}."
kate.wardc5210682009-03-30 18:54:36 +000046# -- end hello.sh --
47#
48# $ ./hello.sh -n Kate
49# Hello, Kate.
kate.ward29ddd4e2008-06-18 19:05:43 +000050#
kate.ward1cb79602011-06-10 11:15:49 +000051# CUSTOMIZABLE BEHAVIOR:
52#
53# A script can override the default 'getopt' command by providing the path to
54# an alternate implementation by defining the FLAGS_GETOPT_CMD variable.
55#
56# NOTES:
57#
58# * Not all systems include a getopt version that supports long flags. On these
59# systems, only short flags are recognized.
kate.wardf51c6162008-06-17 16:38:35 +000060
61#==============================================================================
kate.warda8953952008-06-17 16:48:34 +000062# shFlags
kate.wardf51c6162008-06-17 16:38:35 +000063#
64# Shared attributes:
65# flags_error: last error message
66# flags_return: last return value
67#
68# __flags_longNames: list of long names for all flags
69# __flags_shortNames: list of short names for all flags
70# __flags_boolNames: list of boolean flag names
71#
kate.ward60c25982008-11-12 20:12:44 +000072# __flags_opts: options parsed by getopt
73#
kate.wardf51c6162008-06-17 16:38:35 +000074# Per-flag attributes:
75# FLAGS_<flag_name>: contains value of flag named 'flag_name'
76# __flags_<flag_name>_default: the default flag value
77# __flags_<flag_name>_help: the flag help string
78# __flags_<flag_name>_short: the flag short name
79# __flags_<flag_name>_type: the flag type
80#
81# Notes:
82# - lists of strings are space separated, and a null value is the '~' char.
83
84# return if FLAGS already loaded
kate.wardd3d3d082008-07-10 20:00:42 +000085[ -n "${FLAGS_VERSION:-}" ] && return 0
kate.ward9a183b82010-03-18 00:25:34 +000086FLAGS_VERSION='1.0.4pre'
kate.wardf51c6162008-06-17 16:38:35 +000087
kate.warde10dd532013-01-04 21:52:23 +000088# return values that scripts can use
89FLAGS_TRUE=0
90FLAGS_FALSE=1
91FLAGS_ERROR=2
92
93# determine some reasonable command defaults
kate.ward4ae73d82012-12-17 23:34:29 +000094__FLAGS_UNAME_S=`uname -s`
95case "${__FLAGS_UNAME_S}" in
96 BSD) __FLAGS_EXPR_CMD='gexpr' ;;
97 *) __FLAGS_EXPR_CMD='expr' ;;
kate.warda589b9f2012-12-17 23:27:51 +000098esac
99
kate.warde10dd532013-01-04 21:52:23 +0000100# commands a user can override if needed
kate.ward4ae73d82012-12-17 23:34:29 +0000101FLAGS_EXPR_CMD=${FLAGS_EXPR_CMD:-${__FLAGS_EXPR_CMD}}
kate.ward1cb79602011-06-10 11:15:49 +0000102FLAGS_GETOPT_CMD=${FLAGS_GETOPT_CMD:-getopt}
103
kate.wardf51c6162008-06-17 16:38:35 +0000104# specific shell checks
105if [ -n "${ZSH_VERSION:-}" ]; then
kate.ward1d0ecc42008-07-11 15:33:23 +0000106 setopt |grep "^shwordsplit$" >/dev/null
107 if [ $? -ne ${FLAGS_TRUE} ]; then
108 _flags_fatal 'zsh shwordsplit option is required for proper zsh operation'
kate.wardf51c6162008-06-17 16:38:35 +0000109 fi
110 if [ -z "${FLAGS_PARENT:-}" ]; then
111 _flags_fatal "zsh does not pass \$0 through properly. please declare' \
kate.warda8953952008-06-17 16:48:34 +0000112\"FLAGS_PARENT=\$0\" before calling shFlags"
kate.wardf51c6162008-06-17 16:38:35 +0000113 fi
114fi
115
kate.warde10dd532013-01-04 21:52:23 +0000116# can we use built-ins?
117( echo "${FLAGS_TRUE#0}"; ) >/dev/null 2>&1
118if [ $? -eq ${FLAGS_TRUE} ]; then
kate.wardc66a5fc2013-01-12 23:10:15 +0000119 __FLAGS_USE_BUILTIN=${FLAGS_TRUE}
kate.wardb0759202013-01-05 13:57:46 +0000120 __FLAGS_FX_MATH='_flags_mathBuiltin'
kate.warde10dd532013-01-04 21:52:23 +0000121 __FLAGS_FX_VALID_FLOAT='_flags_validFloatBuiltin'
kate.warde10dd532013-01-04 21:52:23 +0000122else
kate.wardc66a5fc2013-01-12 23:10:15 +0000123 __FLAGS_USE_BUILTIN=${FLAGS_FALSE}
124 __FLAGS_FX_MATH=${FLAGS_EXPR_CMD}
kate.warde10dd532013-01-04 21:52:23 +0000125 __FLAGS_FX_VALID_FLOAT='_flags_validFloatExpr'
kate.warde10dd532013-01-04 21:52:23 +0000126fi
127
kate.wardf51c6162008-06-17 16:38:35 +0000128#
129# constants
130#
131
kate.ward1cb79602011-06-10 11:15:49 +0000132# reserved flag names
133__FLAGS_RESERVED_LIST=' ARGC ARGV ERROR FALSE GETOPT_CMD HELP PARENT TRUE '
134__FLAGS_RESERVED_LIST="${__FLAGS_RESERVED_LIST} VERSION "
135
kate.wardf51c6162008-06-17 16:38:35 +0000136# getopt version
137__FLAGS_GETOPT_VERS_STD=0
138__FLAGS_GETOPT_VERS_ENH=1
139__FLAGS_GETOPT_VERS_BSD=2
140
kate.ward1cb79602011-06-10 11:15:49 +0000141${FLAGS_GETOPT_CMD} >/dev/null 2>&1
kate.wardf51c6162008-06-17 16:38:35 +0000142case $? in
143 0) __FLAGS_GETOPT_VERS=${__FLAGS_GETOPT_VERS_STD} ;; # bsd getopt
144 2)
kate.wardc5210682009-03-30 18:54:36 +0000145 # TODO(kward): look into '-T' option to test the internal getopt() version
kate.ward1cb79602011-06-10 11:15:49 +0000146 if [ "`${FLAGS_GETOPT_CMD} --version`" = '-- ' ]; then
kate.wardf51c6162008-06-17 16:38:35 +0000147 __FLAGS_GETOPT_VERS=${__FLAGS_GETOPT_VERS_STD}
148 else
149 __FLAGS_GETOPT_VERS=${__FLAGS_GETOPT_VERS_ENH}
150 fi
151 ;;
kate.ward1cb79602011-06-10 11:15:49 +0000152 *) _flags_fatal 'unable to determine getopt version' ;;
kate.wardf51c6162008-06-17 16:38:35 +0000153esac
154
155# getopt optstring lengths
156__FLAGS_OPTSTR_SHORT=0
157__FLAGS_OPTSTR_LONG=1
158
159__FLAGS_NULL='~'
160
161# flag info strings
kate.wardfb0e7182009-05-10 16:59:13 +0000162__FLAGS_INFO_DEFAULT='default'
163__FLAGS_INFO_HELP='help'
164__FLAGS_INFO_SHORT='short'
165__FLAGS_INFO_TYPE='type'
kate.wardf51c6162008-06-17 16:38:35 +0000166
167# flag lengths
168__FLAGS_LEN_SHORT=0
169__FLAGS_LEN_LONG=1
170
171# flag types
172__FLAGS_TYPE_NONE=0
173__FLAGS_TYPE_BOOLEAN=1
174__FLAGS_TYPE_FLOAT=2
175__FLAGS_TYPE_INTEGER=3
176__FLAGS_TYPE_STRING=4
177
178# set the constants readonly
kate.wardd9672402008-11-15 16:55:51 +0000179__flags_constants=`set |awk -F= '/^FLAGS_/ || /^__FLAGS_/ {print $1}'`
180for __flags_const in ${__flags_constants}; do
181 # skip certain flags
182 case ${__flags_const} in
kate.ward39913d12009-04-01 14:09:23 +0000183 FLAGS_HELP) continue ;;
kate.wardd9672402008-11-15 16:55:51 +0000184 FLAGS_PARENT) continue ;;
185 esac
186 # set flag readonly
187 if [ -z "${ZSH_VERSION:-}" ]; then
188 readonly ${__flags_const}
189 else # handle zsh
kate.ward1d0ecc42008-07-11 15:33:23 +0000190 case ${ZSH_VERSION} in
kate.wardd9672402008-11-15 16:55:51 +0000191 [123].*) readonly ${__flags_const} ;;
192 *) readonly -g ${__flags_const} ;; # declare readonly constants globally
kate.ward1d0ecc42008-07-11 15:33:23 +0000193 esac
kate.wardf51c6162008-06-17 16:38:35 +0000194 fi
195done
kate.wardd9672402008-11-15 16:55:51 +0000196unset __flags_const __flags_constants
kate.wardf51c6162008-06-17 16:38:35 +0000197
198#
199# internal variables
200#
201
kate.wardda8d2c02011-06-28 13:26:02 +0000202# space separated lists
203__flags_boolNames=' ' # boolean flag names
204__flags_longNames=' ' # long flag names
205__flags_shortNames=' ' # short flag names
206__flags_definedNames=' ' # defined flag names (used for validation)
kate.wardf51c6162008-06-17 16:38:35 +0000207
kate.ward14b33bf2008-11-15 20:03:18 +0000208__flags_columns='' # screen width in columns
kate.wardd9672402008-11-15 16:55:51 +0000209__flags_opts='' # temporary storage for parsed getopt flags
kate.ward60c25982008-11-12 20:12:44 +0000210
kate.wardf51c6162008-06-17 16:38:35 +0000211#------------------------------------------------------------------------------
212# private functions
213#
214
kate.warde10dd532013-01-04 21:52:23 +0000215# logging functions
216_flags_debug() { echo "flags:DEBUG $@" >&2; }
217_flags_warn() { echo "flags:WARN $@" >&2; }
218_flags_error() { echo "flags:ERROR $@" >&2; }
219_flags_fatal() { echo "flags:FATAL $@" >&2; exit ${FLAGS_ERROR}; }
220
kate.wardcc145682008-11-14 00:32:42 +0000221# Define a flag.
222#
223# Calling this function will define the following info variables for the
224# specified flag:
kate.wardf51c6162008-06-17 16:38:35 +0000225# FLAGS_flagname - the name for this flag (based upon the long flag name)
226# __flags_<flag_name>_default - the default value
227# __flags_flagname_help - the help string
228# __flags_flagname_short - the single letter alias
229# __flags_flagname_type - the type of flag (one of __FLAGS_TYPE_*)
230#
231# Args:
232# _flags__type: integer: internal type of flag (__FLAGS_TYPE_*)
233# _flags__name: string: long flag name
234# _flags__default: default flag value
235# _flags__help: string: help string
236# _flags__short: string: (optional) short flag name
237# Returns:
238# integer: success of operation, or error
239_flags_define()
240{
241 if [ $# -lt 4 ]; then
242 flags_error='DEFINE error: too few arguments'
243 flags_return=${FLAGS_ERROR}
244 _flags_error "${flags_error}"
245 return ${flags_return}
246 fi
247
kate.ward60c25982008-11-12 20:12:44 +0000248 _flags_type_=$1
249 _flags_name_=$2
250 _flags_default_=$3
251 _flags_help_=$4
252 _flags_short_=${5:-${__FLAGS_NULL}}
kate.wardf51c6162008-06-17 16:38:35 +0000253
kate.ward60c25982008-11-12 20:12:44 +0000254 _flags_return_=${FLAGS_TRUE}
kate.wardda8d2c02011-06-28 13:26:02 +0000255 _flags_usName_=`_flags_underscoreName ${_flags_name_}`
kate.wardf51c6162008-06-17 16:38:35 +0000256
kate.ward39913d12009-04-01 14:09:23 +0000257 # check whether the flag name is reserved
kate.wardda8d2c02011-06-28 13:26:02 +0000258 _flags_itemInList ${_flags_usName_} "${__FLAGS_RESERVED_LIST}"
kate.ward929261f2010-03-28 23:12:17 +0000259 if [ $? -eq ${FLAGS_TRUE} ]; then
kate.ward39913d12009-04-01 14:09:23 +0000260 flags_error="flag name (${_flags_name_}) is reserved"
261 _flags_return_=${FLAGS_ERROR}
262 fi
263
kate.wardf51c6162008-06-17 16:38:35 +0000264 # require short option for getopt that don't support long options
kate.ward60c25982008-11-12 20:12:44 +0000265 if [ ${_flags_return_} -eq ${FLAGS_TRUE} \
kate.wardf51c6162008-06-17 16:38:35 +0000266 -a ${__FLAGS_GETOPT_VERS} -ne ${__FLAGS_GETOPT_VERS_ENH} \
kate.ward60c25982008-11-12 20:12:44 +0000267 -a "${_flags_short_}" = "${__FLAGS_NULL}" ]
kate.wardf51c6162008-06-17 16:38:35 +0000268 then
kate.ward5b48fc12008-11-13 00:42:43 +0000269 flags_error="short flag required for (${_flags_name_}) on this platform"
kate.ward60c25982008-11-12 20:12:44 +0000270 _flags_return_=${FLAGS_ERROR}
kate.wardf51c6162008-06-17 16:38:35 +0000271 fi
272
273 # check for existing long name definition
kate.ward60c25982008-11-12 20:12:44 +0000274 if [ ${_flags_return_} -eq ${FLAGS_TRUE} ]; then
kate.wardda8d2c02011-06-28 13:26:02 +0000275 if _flags_itemInList ${_flags_usName_} ${__flags_definedNames}; then
276 flags_error="definition for ([no]${_flags_name_}) already exists"
kate.ward20d06782008-10-21 19:57:19 +0000277 _flags_warn "${flags_error}"
kate.ward60c25982008-11-12 20:12:44 +0000278 _flags_return_=${FLAGS_FALSE}
kate.wardf51c6162008-06-17 16:38:35 +0000279 fi
280 fi
281
282 # check for existing short name definition
kate.ward60c25982008-11-12 20:12:44 +0000283 if [ ${_flags_return_} -eq ${FLAGS_TRUE} \
284 -a "${_flags_short_}" != "${__FLAGS_NULL}" ]
kate.wardf51c6162008-06-17 16:38:35 +0000285 then
kate.ward60c25982008-11-12 20:12:44 +0000286 if _flags_itemInList "${_flags_short_}" ${__flags_shortNames}; then
287 flags_error="flag short name (${_flags_short_}) already defined"
kate.ward20d06782008-10-21 19:57:19 +0000288 _flags_warn "${flags_error}"
kate.ward60c25982008-11-12 20:12:44 +0000289 _flags_return_=${FLAGS_FALSE}
kate.wardf51c6162008-06-17 16:38:35 +0000290 fi
291 fi
292
293 # handle default value. note, on several occasions the 'if' portion of an
294 # if/then/else contains just a ':' which does nothing. a binary reversal via
295 # '!' is not done because it does not work on all shells.
kate.ward60c25982008-11-12 20:12:44 +0000296 if [ ${_flags_return_} -eq ${FLAGS_TRUE} ]; then
297 case ${_flags_type_} in
kate.wardf51c6162008-06-17 16:38:35 +0000298 ${__FLAGS_TYPE_BOOLEAN})
kate.warde10dd532013-01-04 21:52:23 +0000299 if _flags_validBool "${_flags_default_}"; then
kate.ward60c25982008-11-12 20:12:44 +0000300 case ${_flags_default_} in
301 true|t|0) _flags_default_=${FLAGS_TRUE} ;;
302 false|f|1) _flags_default_=${FLAGS_FALSE} ;;
kate.wardf51c6162008-06-17 16:38:35 +0000303 esac
304 else
kate.ward60c25982008-11-12 20:12:44 +0000305 flags_error="invalid default flag value '${_flags_default_}'"
306 _flags_return_=${FLAGS_ERROR}
kate.wardf51c6162008-06-17 16:38:35 +0000307 fi
308 ;;
309
310 ${__FLAGS_TYPE_FLOAT})
kate.warde10dd532013-01-04 21:52:23 +0000311 if ${__FLAGS_FX_VALID_FLOAT} "${_flags_default_}"; then
kate.wardf51c6162008-06-17 16:38:35 +0000312 :
313 else
kate.ward60c25982008-11-12 20:12:44 +0000314 flags_error="invalid default flag value '${_flags_default_}'"
315 _flags_return_=${FLAGS_ERROR}
kate.wardf51c6162008-06-17 16:38:35 +0000316 fi
317 ;;
318
319 ${__FLAGS_TYPE_INTEGER})
kate.wardc66a5fc2013-01-12 23:10:15 +0000320 if _flags_validInt "${_flags_default_}"; then
kate.wardf51c6162008-06-17 16:38:35 +0000321 :
322 else
kate.ward60c25982008-11-12 20:12:44 +0000323 flags_error="invalid default flag value '${_flags_default_}'"
324 _flags_return_=${FLAGS_ERROR}
kate.wardf51c6162008-06-17 16:38:35 +0000325 fi
326 ;;
327
328 ${__FLAGS_TYPE_STRING}) ;; # everything in shell is a valid string
329
330 *)
kate.ward60c25982008-11-12 20:12:44 +0000331 flags_error="unrecognized flag type '${_flags_type_}'"
332 _flags_return_=${FLAGS_ERROR}
kate.wardf51c6162008-06-17 16:38:35 +0000333 ;;
334 esac
335 fi
336
kate.ward60c25982008-11-12 20:12:44 +0000337 if [ ${_flags_return_} -eq ${FLAGS_TRUE} ]; then
kate.wardf51c6162008-06-17 16:38:35 +0000338 # store flag information
kate.wardda8d2c02011-06-28 13:26:02 +0000339 eval "FLAGS_${_flags_usName_}='${_flags_default_}'"
340 eval "__flags_${_flags_usName_}_${__FLAGS_INFO_TYPE}=${_flags_type_}"
341 eval "__flags_${_flags_usName_}_${__FLAGS_INFO_DEFAULT}=\
kate.ward60c25982008-11-12 20:12:44 +0000342\"${_flags_default_}\""
kate.wardda8d2c02011-06-28 13:26:02 +0000343 eval "__flags_${_flags_usName_}_${__FLAGS_INFO_HELP}=\"${_flags_help_}\""
344 eval "__flags_${_flags_usName_}_${__FLAGS_INFO_SHORT}='${_flags_short_}'"
kate.wardf51c6162008-06-17 16:38:35 +0000345
kate.wardda8d2c02011-06-28 13:26:02 +0000346 # append flag names to name lists
kate.ward60c25982008-11-12 20:12:44 +0000347 __flags_shortNames="${__flags_shortNames}${_flags_short_} "
kate.wardda8d2c02011-06-28 13:26:02 +0000348 __flags_longNames="${__flags_longNames}${_flags_name_} "
kate.ward60c25982008-11-12 20:12:44 +0000349 [ ${_flags_type_} -eq ${__FLAGS_TYPE_BOOLEAN} ] && \
350 __flags_boolNames="${__flags_boolNames}no${_flags_name_} "
kate.wardda8d2c02011-06-28 13:26:02 +0000351
352 # append flag names to defined names for later validation checks
353 __flags_definedNames="${__flags_definedNames}${_flags_usName_} "
354 [ ${_flags_type_} -eq ${__FLAGS_TYPE_BOOLEAN} ] && \
355 __flags_definedNames="${__flags_definedNames}no${_flags_usName_} "
kate.wardf51c6162008-06-17 16:38:35 +0000356 fi
357
kate.ward60c25982008-11-12 20:12:44 +0000358 flags_return=${_flags_return_}
kate.wardda8d2c02011-06-28 13:26:02 +0000359 unset _flags_default_ _flags_help_ _flags_name_ _flags_return_ \
360 _flags_short_ _flags_type_ _flags_usName_
kate.wardf51c6162008-06-17 16:38:35 +0000361 [ ${flags_return} -eq ${FLAGS_ERROR} ] && _flags_error "${flags_error}"
362 return ${flags_return}
363}
364
kate.wardda8d2c02011-06-28 13:26:02 +0000365# Underscore a flag name by replacing dashes with underscores.
366#
367# Args:
368# unnamed: string: log flag name
369# Output:
370# string: underscored name
371_flags_underscoreName()
372{
373 echo $1 |tr '-' '_'
374}
375
kate.wardcc145682008-11-14 00:32:42 +0000376# Return valid getopt options using currently defined list of long options.
377#
378# This function builds a proper getopt option string for short (and long)
379# options, using the current list of long options for reference.
kate.wardf51c6162008-06-17 16:38:35 +0000380#
381# Args:
kate.wardcc145682008-11-14 00:32:42 +0000382# _flags_optStr: integer: option string type (__FLAGS_OPTSTR_*)
kate.wardf51c6162008-06-17 16:38:35 +0000383# Output:
384# string: generated option string for getopt
385# Returns:
386# boolean: success of operation (always returns True)
387_flags_genOptStr()
388{
kate.wardcc145682008-11-14 00:32:42 +0000389 _flags_optStrType_=$1
kate.wardf51c6162008-06-17 16:38:35 +0000390
kate.wardcc145682008-11-14 00:32:42 +0000391 _flags_opts_=''
kate.wardf51c6162008-06-17 16:38:35 +0000392
kate.wardda8d2c02011-06-28 13:26:02 +0000393 for _flags_name_ in ${__flags_longNames}; do
394 _flags_usName_=`_flags_underscoreName ${_flags_name_}`
395 _flags_type_=`_flags_getFlagInfo ${_flags_usName_} ${__FLAGS_INFO_TYPE}`
kate.ward1cb79602011-06-10 11:15:49 +0000396 [ $? -eq ${FLAGS_TRUE} ] || _flags_fatal 'call to _flags_type_ failed'
kate.wardcc145682008-11-14 00:32:42 +0000397 case ${_flags_optStrType_} in
kate.wardf51c6162008-06-17 16:38:35 +0000398 ${__FLAGS_OPTSTR_SHORT})
kate.wardcc145682008-11-14 00:32:42 +0000399 _flags_shortName_=`_flags_getFlagInfo \
kate.wardda8d2c02011-06-28 13:26:02 +0000400 ${_flags_usName_} ${__FLAGS_INFO_SHORT}`
kate.wardcc145682008-11-14 00:32:42 +0000401 if [ "${_flags_shortName_}" != "${__FLAGS_NULL}" ]; then
402 _flags_opts_="${_flags_opts_}${_flags_shortName_}"
kate.wardc5210682009-03-30 18:54:36 +0000403 # getopt needs a trailing ':' to indicate a required argument
kate.wardcc145682008-11-14 00:32:42 +0000404 [ ${_flags_type_} -ne ${__FLAGS_TYPE_BOOLEAN} ] && \
405 _flags_opts_="${_flags_opts_}:"
kate.wardf51c6162008-06-17 16:38:35 +0000406 fi
407 ;;
408
409 ${__FLAGS_OPTSTR_LONG})
kate.wardda8d2c02011-06-28 13:26:02 +0000410 _flags_opts_="${_flags_opts_:+${_flags_opts_},}${_flags_name_}"
kate.wardc5210682009-03-30 18:54:36 +0000411 # getopt needs a trailing ':' to indicate a required argument
kate.wardcc145682008-11-14 00:32:42 +0000412 [ ${_flags_type_} -ne ${__FLAGS_TYPE_BOOLEAN} ] && \
413 _flags_opts_="${_flags_opts_}:"
kate.wardf51c6162008-06-17 16:38:35 +0000414 ;;
415 esac
416 done
417
kate.wardcc145682008-11-14 00:32:42 +0000418 echo "${_flags_opts_}"
kate.wardda8d2c02011-06-28 13:26:02 +0000419 unset _flags_name_ _flags_opts_ _flags_optStrType_ _flags_shortName_ \
420 _flags_type_ _flags_usName_
kate.wardf51c6162008-06-17 16:38:35 +0000421 return ${FLAGS_TRUE}
422}
423
424# Returns flag details based on a flag name and flag info.
425#
426# Args:
kate.wardda8d2c02011-06-28 13:26:02 +0000427# string: underscored flag name
kate.wardf51c6162008-06-17 16:38:35 +0000428# string: flag info (see the _flags_define function for valid info types)
429# Output:
430# string: value of dereferenced flag variable
431# Returns:
432# integer: one of FLAGS_{TRUE|FALSE|ERROR}
433_flags_getFlagInfo()
434{
kate.wardda8d2c02011-06-28 13:26:02 +0000435 # note: adding gFI to variable names to prevent naming conflicts with calling
436 # functions
437 _flags_gFI_usName_=$1
438 _flags_gFI_info_=$2
kate.wardf51c6162008-06-17 16:38:35 +0000439
kate.wardda8d2c02011-06-28 13:26:02 +0000440 _flags_infoVar_="__flags_${_flags_gFI_usName_}_${_flags_gFI_info_}"
441 _flags_strToEval_="_flags_infoValue_=\"\${${_flags_infoVar_}:-}\""
kate.ward1b600c52008-11-12 21:26:05 +0000442 eval "${_flags_strToEval_}"
kate.wardda8d2c02011-06-28 13:26:02 +0000443 if [ -n "${_flags_infoValue_}" ]; then
kate.wardf51c6162008-06-17 16:38:35 +0000444 flags_return=${FLAGS_TRUE}
445 else
kate.wardda8d2c02011-06-28 13:26:02 +0000446 # see if the _flags_gFI_usName_ variable is a string as strings can be
447 # empty...
kate.ward1b600c52008-11-12 21:26:05 +0000448 # note: the DRY principle would say to have this function call itself for
449 # the next three lines, but doing so results in an infinite loop as an
450 # invalid _flags_name_ will also not have the associated _type variable.
451 # Because it doesn't (it will evaluate to an empty string) the logic will
452 # try to find the _type variable of the _type variable, and so on. Not so
453 # good ;-)
kate.wardda8d2c02011-06-28 13:26:02 +0000454 _flags_typeVar_="__flags_${_flags_gFI_usName_}_${__FLAGS_INFO_TYPE}"
kate.ward929261f2010-03-28 23:12:17 +0000455 _flags_strToEval_="_flags_typeValue_=\"\${${_flags_typeVar_}:-}\""
kate.ward1b600c52008-11-12 21:26:05 +0000456 eval "${_flags_strToEval_}"
kate.ward929261f2010-03-28 23:12:17 +0000457 if [ "${_flags_typeValue_}" = "${__FLAGS_TYPE_STRING}" ]; then
kate.ward437639d2008-10-19 17:21:41 +0000458 flags_return=${FLAGS_TRUE}
459 else
460 flags_return=${FLAGS_ERROR}
kate.wardda8d2c02011-06-28 13:26:02 +0000461 flags_error="missing flag info variable (${_flags_infoVar_})"
kate.ward437639d2008-10-19 17:21:41 +0000462 fi
kate.wardf51c6162008-06-17 16:38:35 +0000463 fi
464
kate.wardda8d2c02011-06-28 13:26:02 +0000465 echo "${_flags_infoValue_}"
466 unset _flags_gFI_usName_ _flags_gfI_info_ _flags_infoValue_ _flags_infoVar_ \
kate.ward929261f2010-03-28 23:12:17 +0000467 _flags_strToEval_ _flags_typeValue_ _flags_typeVar_
kate.wardf51c6162008-06-17 16:38:35 +0000468 [ ${flags_return} -eq ${FLAGS_ERROR} ] && _flags_error "${flags_error}"
469 return ${flags_return}
470}
471
kate.wardda8d2c02011-06-28 13:26:02 +0000472# Check for presense of item in a list.
473#
474# Passed a string (e.g. 'abc'), this function will determine if the string is
475# present in the list of strings (e.g. ' foo bar abc ').
kate.wardf51c6162008-06-17 16:38:35 +0000476#
477# Args:
kate.wardda8d2c02011-06-28 13:26:02 +0000478# _flags_str_: string: string to search for in a list of strings
kate.wardf51c6162008-06-17 16:38:35 +0000479# unnamed: list: list of strings
480# Returns:
481# boolean: true if item is in the list
kate.wardda8d2c02011-06-28 13:26:02 +0000482_flags_itemInList() {
kate.wardcc145682008-11-14 00:32:42 +0000483 _flags_str_=$1
kate.wardf51c6162008-06-17 16:38:35 +0000484 shift
485
kate.wardcc145682008-11-14 00:32:42 +0000486 echo " ${*:-} " |grep " ${_flags_str_} " >/dev/null
kate.wardf51c6162008-06-17 16:38:35 +0000487 if [ $? -eq 0 ]; then
488 flags_return=${FLAGS_TRUE}
489 else
490 flags_return=${FLAGS_FALSE}
491 fi
492
kate.wardcc145682008-11-14 00:32:42 +0000493 unset _flags_str_
kate.wardf51c6162008-06-17 16:38:35 +0000494 return ${flags_return}
495}
496
kate.warddadc1642008-11-14 02:00:29 +0000497# Returns the width of the current screen.
kate.ward437639d2008-10-19 17:21:41 +0000498#
kate.ward18d456e2008-11-14 00:46:44 +0000499# Output:
kate.warddadc1642008-11-14 02:00:29 +0000500# integer: width in columns of the current screen.
kate.ward14b33bf2008-11-15 20:03:18 +0000501_flags_columns()
kate.ward437639d2008-10-19 17:21:41 +0000502{
kate.ward14b33bf2008-11-15 20:03:18 +0000503 if [ -z "${__flags_columns}" ]; then
504 # determine the value and store it
505 if eval stty size >/dev/null 2>&1; then
506 # stty size worked :-)
507 set -- `stty size`
508 __flags_columns=$2
509 elif eval tput cols >/dev/null 2>&1; then
510 set -- `tput cols`
511 __flags_columns=$1
512 else
513 __flags_columns=80 # default terminal width
514 fi
kate.warddadc1642008-11-14 02:00:29 +0000515 fi
kate.ward14b33bf2008-11-15 20:03:18 +0000516 echo ${__flags_columns}
kate.ward437639d2008-10-19 17:21:41 +0000517}
518
519# Validate a boolean.
kate.wardf51c6162008-06-17 16:38:35 +0000520#
521# Args:
522# _flags__bool: boolean: value to validate
523# Returns:
524# bool: true if the value is a valid boolean
kate.warde10dd532013-01-04 21:52:23 +0000525_flags_validBool()
kate.wardf51c6162008-06-17 16:38:35 +0000526{
kate.ward60c25982008-11-12 20:12:44 +0000527 _flags_bool_=$1
kate.wardf51c6162008-06-17 16:38:35 +0000528
529 flags_return=${FLAGS_TRUE}
kate.ward60c25982008-11-12 20:12:44 +0000530 case "${_flags_bool_}" in
kate.wardf51c6162008-06-17 16:38:35 +0000531 true|t|0) ;;
532 false|f|1) ;;
533 *) flags_return=${FLAGS_FALSE} ;;
534 esac
535
kate.ward60c25982008-11-12 20:12:44 +0000536 unset _flags_bool_
kate.wardf51c6162008-06-17 16:38:35 +0000537 return ${flags_return}
538}
539
kate.warde10dd532013-01-04 21:52:23 +0000540# Validate a float using built-ins.
541#
542# Args:
543# _flags_float_: float: value to validate
kate.warde10dd532013-01-04 21:52:23 +0000544# Returns:
545# bool: true if the value is a valid integer
546_flags_validFloatBuiltin()
547{
548 flags_return=${FLAGS_FALSE}
549 [ -n "$1" ] || return ${flags_return}
550 _flags_float_=$1
551
kate.wardc66a5fc2013-01-12 23:10:15 +0000552 if _flags_validInt ${_flags_float_}; then
kate.warde10dd532013-01-04 21:52:23 +0000553 flags_return=${FLAGS_TRUE}
kate.wardc66a5fc2013-01-12 23:10:15 +0000554 else
555 _flags_float_whole_=${_flags_float_%.*}
556 _flags_float_fraction_=${_flags_float_#*.}
557 if _flags_validInt ${_flags_float_whole_:-0} -a \
558 _flags_validInt ${_flags_float_fraction_}; then
559 flags_return=${FLAGS_TRUE}
560 fi
561 fi
kate.warde10dd532013-01-04 21:52:23 +0000562
kate.wardc66a5fc2013-01-12 23:10:15 +0000563 unset _flags_float_ _flags_float_whole_ _flags_float_fraction_
kate.warde10dd532013-01-04 21:52:23 +0000564 return ${flags_return}
565}
566
kate.ward18d456e2008-11-14 00:46:44 +0000567# Validate a float.
kate.wardf51c6162008-06-17 16:38:35 +0000568#
569# Args:
570# _flags__float: float: value to validate
571# Returns:
572# bool: true if the value is a valid float
kate.warde10dd532013-01-04 21:52:23 +0000573_flags_validFloatExpr()
kate.wardf51c6162008-06-17 16:38:35 +0000574{
kate.wardc66a5fc2013-01-12 23:10:15 +0000575 flags_return=${FLAGS_FALSE}
576 [ -n "$1" ] || return ${flags_return}
kate.wardcc145682008-11-14 00:32:42 +0000577 _flags_float_=$1
kate.wardf51c6162008-06-17 16:38:35 +0000578
kate.wardc66a5fc2013-01-12 23:10:15 +0000579 if _flags_validInt "${_flags_float_}"; then
kate.wardf51c6162008-06-17 16:38:35 +0000580 flags_return=${FLAGS_TRUE}
581 else
582 flags_return=${FLAGS_TRUE}
kate.wardcc145682008-11-14 00:32:42 +0000583 case ${_flags_float_} in
kate.wardf51c6162008-06-17 16:38:35 +0000584 -*) # negative floats
kate.warda589b9f2012-12-17 23:27:51 +0000585 _flags_test_=`${FLAGS_EXPR_CMD} -- "${_flags_float_}" :\
kate.warde10dd532013-01-04 21:52:23 +0000586 '\(-[0-9]*\.[0-9]*\)'`
kate.wardf51c6162008-06-17 16:38:35 +0000587 ;;
588 *) # positive floats
kate.warda589b9f2012-12-17 23:27:51 +0000589 _flags_test_=`${FLAGS_EXPR_CMD} -- "${_flags_float_}" :\
kate.warde10dd532013-01-04 21:52:23 +0000590 '\([0-9]*\.[0-9]*\)'`
kate.wardf51c6162008-06-17 16:38:35 +0000591 ;;
592 esac
kate.wardcc145682008-11-14 00:32:42 +0000593 [ "${_flags_test_}" != "${_flags_float_}" ] && flags_return=${FLAGS_FALSE}
kate.wardf51c6162008-06-17 16:38:35 +0000594 fi
595
kate.wardcc145682008-11-14 00:32:42 +0000596 unset _flags_float_ _flags_test_
kate.wardf51c6162008-06-17 16:38:35 +0000597 return ${flags_return}
598}
599
kate.wardc66a5fc2013-01-12 23:10:15 +0000600# Validate an integer.
kate.wardf51c6162008-06-17 16:38:35 +0000601#
602# Args:
kate.warde10dd532013-01-04 21:52:23 +0000603# _flags_int_: integer: value to validate
kate.wardf51c6162008-06-17 16:38:35 +0000604# Returns:
605# bool: true if the value is a valid integer
kate.wardc66a5fc2013-01-12 23:10:15 +0000606_flags_validInt()
kate.warde10dd532013-01-04 21:52:23 +0000607{
608 flags_return=${FLAGS_FALSE}
609 [ -n "$1" ] || return ${flags_return}
610 _flags_int_=$1
611
kate.wardcc145682008-11-14 00:32:42 +0000612 case ${_flags_int_} in
kate.wardc66a5fc2013-01-12 23:10:15 +0000613 -*.*) ;; # ignore negative floats (we'll invalidate them later)
614 -*) # strip possible leading negative sign
615 if _flags_useBuiltin; then
616 _flags_int_=${_flags_int_#-}
617 else
618 _flags_int_=`${FLAGS_EXPR_CMD} -- "${_flags_int_}" : '-\([0-9][0-9]*\)'`
619 fi
kate.wardf51c6162008-06-17 16:38:35 +0000620 ;;
621 esac
kate.wardf51c6162008-06-17 16:38:35 +0000622
kate.wardc66a5fc2013-01-12 23:10:15 +0000623 case "${_flags_int_}" in
624 *[!0-9]*) flags_return=${FLAGS_FALSE} ;;
625 *) flags_return=${FLAGS_TRUE} ;;
626 esac
627
628 unset _flags_int_
kate.wardf51c6162008-06-17 16:38:35 +0000629 return ${flags_return}
630}
631
kate.ward18d456e2008-11-14 00:46:44 +0000632# Parse command-line options using the standard getopt.
633#
634# Note: the flag options are passed around in the global __flags_opts so that
635# the formatting is not lost due to shell parsing and such.
636#
637# Args:
638# @: varies: command-line options to parse
639# Returns:
640# integer: a FLAGS success condition
kate.wardc5210682009-03-30 18:54:36 +0000641_flags_getoptStandard()
kate.ward60c25982008-11-12 20:12:44 +0000642{
643 flags_return=${FLAGS_TRUE}
644 _flags_shortOpts_=`_flags_genOptStr ${__FLAGS_OPTSTR_SHORT}`
645
646 # check for spaces in passed options
647 for _flags_opt_ in "$@"; do
kate.ward14b33bf2008-11-15 20:03:18 +0000648 # note: the silliness with the x's is purely for ksh93 on Ubuntu 6.06
649 _flags_match_=`echo "x${_flags_opt_}x" |sed 's/ //g'`
650 if [ "${_flags_match_}" != "x${_flags_opt_}x" ]; then
kate.ward60c25982008-11-12 20:12:44 +0000651 flags_error='the available getopt does not support spaces in options'
652 flags_return=${FLAGS_ERROR}
653 break
654 fi
655 done
656
657 if [ ${flags_return} -eq ${FLAGS_TRUE} ]; then
658 __flags_opts=`getopt ${_flags_shortOpts_} $@ 2>&1`
kate.ward7a3c9c42009-04-24 00:12:35 +0000659 _flags_rtrn_=$?
660 if [ ${_flags_rtrn_} -ne ${FLAGS_TRUE} ]; then
661 _flags_warn "${__flags_opts}"
kate.ward60c25982008-11-12 20:12:44 +0000662 flags_error='unable to parse provided options with getopt.'
663 flags_return=${FLAGS_ERROR}
664 fi
665 fi
666
kate.ward7a3c9c42009-04-24 00:12:35 +0000667 unset _flags_match_ _flags_opt_ _flags_rtrn_ _flags_shortOpts_
kate.ward60c25982008-11-12 20:12:44 +0000668 return ${flags_return}
669}
670
kate.ward18d456e2008-11-14 00:46:44 +0000671# Parse command-line options using the enhanced getopt.
672#
673# Note: the flag options are passed around in the global __flags_opts so that
674# the formatting is not lost due to shell parsing and such.
675#
676# Args:
677# @: varies: command-line options to parse
678# Returns:
679# integer: a FLAGS success condition
kate.wardc5210682009-03-30 18:54:36 +0000680_flags_getoptEnhanced()
kate.ward60c25982008-11-12 20:12:44 +0000681{
682 flags_return=${FLAGS_TRUE}
683 _flags_shortOpts_=`_flags_genOptStr ${__FLAGS_OPTSTR_SHORT}`
684 _flags_boolOpts_=`echo "${__flags_boolNames}" \
kate.wardd9672402008-11-15 16:55:51 +0000685 |sed 's/^ *//;s/ *$//;s/ /,/g'`
kate.ward60c25982008-11-12 20:12:44 +0000686 _flags_longOpts_=`_flags_genOptStr ${__FLAGS_OPTSTR_LONG}`
687
kate.ward1cb79602011-06-10 11:15:49 +0000688 __flags_opts=`${FLAGS_GETOPT_CMD} \
kate.ward60c25982008-11-12 20:12:44 +0000689 -o ${_flags_shortOpts_} \
690 -l "${_flags_longOpts_},${_flags_boolOpts_}" \
691 -- "$@" 2>&1`
kate.ward7a3c9c42009-04-24 00:12:35 +0000692 _flags_rtrn_=$?
693 if [ ${_flags_rtrn_} -ne ${FLAGS_TRUE} ]; then
694 _flags_warn "${__flags_opts}"
kate.ward60c25982008-11-12 20:12:44 +0000695 flags_error='unable to parse provided options with getopt.'
696 flags_return=${FLAGS_ERROR}
697 fi
698
kate.ward7a3c9c42009-04-24 00:12:35 +0000699 unset _flags_boolOpts_ _flags_longOpts_ _flags_rtrn_ _flags_shortOpts_
kate.ward60c25982008-11-12 20:12:44 +0000700 return ${flags_return}
701}
702
kate.ward18d456e2008-11-14 00:46:44 +0000703# Dynamically parse a getopt result and set appropriate variables.
704#
705# This function does the actual conversion of getopt output and runs it through
706# the standard case structure for parsing. The case structure is actually quite
707# dynamic to support any number of flags.
708#
709# Args:
kate.ward9ccba512009-04-01 13:31:18 +0000710# argc: int: original command-line argument count
kate.ward18d456e2008-11-14 00:46:44 +0000711# @: varies: output from getopt parsing
712# Returns:
713# integer: a FLAGS success condition
kate.ward60c25982008-11-12 20:12:44 +0000714_flags_parseGetopt()
715{
kate.ward9ccba512009-04-01 13:31:18 +0000716 _flags_argc_=$1
717 shift
718
kate.ward60c25982008-11-12 20:12:44 +0000719 flags_return=${FLAGS_TRUE}
720
721 if [ ${__FLAGS_GETOPT_VERS} -ne ${__FLAGS_GETOPT_VERS_ENH} ]; then
722 set -- $@
723 else
724 # note the quotes around the `$@' -- they are essential!
725 eval set -- "$@"
726 fi
727
kate.ward4eb31352013-01-01 22:05:32 +0000728 # Provide user with the number of arguments to shift by later.
kate.wardc5210682009-03-30 18:54:36 +0000729 # NOTE: the FLAGS_ARGC variable is obsolete as of 1.0.3 because it does not
730 # properly give user access to non-flag arguments mixed in between flag
731 # arguments. Its usage was replaced by FLAGS_ARGV, and it is being kept only
732 # for backwards compatibility reasons.
kate.wardb0759202013-01-05 13:57:46 +0000733 FLAGS_ARGC=`${__FLAGS_FX_MATH} $# - 1 - ${_flags_argc_}`
kate.wardc5210682009-03-30 18:54:36 +0000734
kate.ward60c25982008-11-12 20:12:44 +0000735 # handle options. note options with values must do an additional shift
736 while true; do
737 _flags_opt_=$1
738 _flags_arg_=${2:-}
739 _flags_type_=${__FLAGS_TYPE_NONE}
kate.ward7a3c9c42009-04-24 00:12:35 +0000740 _flags_name_=''
kate.ward60c25982008-11-12 20:12:44 +0000741
742 # determine long flag name
743 case "${_flags_opt_}" in
744 --) shift; break ;; # discontinue option parsing
745
746 --*) # long option
kate.warda589b9f2012-12-17 23:27:51 +0000747 _flags_opt_=`${FLAGS_EXPR_CMD} -- "${_flags_opt_}" : '--\(.*\)'`
kate.ward60c25982008-11-12 20:12:44 +0000748 _flags_len_=${__FLAGS_LEN_LONG}
749 if _flags_itemInList "${_flags_opt_}" ${__flags_longNames}; then
750 _flags_name_=${_flags_opt_}
751 else
752 # check for negated long boolean version
753 if _flags_itemInList "${_flags_opt_}" ${__flags_boolNames}; then
kate.warda589b9f2012-12-17 23:27:51 +0000754 _flags_name_=`${FLAGS_EXPR_CMD} -- "${_flags_opt_}" : 'no\(.*\)'`
kate.ward60c25982008-11-12 20:12:44 +0000755 _flags_type_=${__FLAGS_TYPE_BOOLEAN}
756 _flags_arg_=${__FLAGS_NULL}
757 fi
758 fi
759 ;;
760
761 -*) # short option
kate.warda589b9f2012-12-17 23:27:51 +0000762 _flags_opt_=`${FLAGS_EXPR_CMD} -- "${_flags_opt_}" : '-\(.*\)'`
kate.ward60c25982008-11-12 20:12:44 +0000763 _flags_len_=${__FLAGS_LEN_SHORT}
764 if _flags_itemInList "${_flags_opt_}" ${__flags_shortNames}; then
765 # yes. match short name to long name. note purposeful off-by-one
766 # (too high) with awk calculations.
767 _flags_pos_=`echo "${__flags_shortNames}" \
768 |awk 'BEGIN{RS=" ";rn=0}$0==e{rn=NR}END{print rn}' \
769 e=${_flags_opt_}`
770 _flags_name_=`echo "${__flags_longNames}" \
771 |awk 'BEGIN{RS=" "}rn==NR{print $0}' rn="${_flags_pos_}"`
772 fi
773 ;;
774 esac
775
776 # die if the flag was unrecognized
777 if [ -z "${_flags_name_}" ]; then
778 flags_error="unrecognized option (${_flags_opt_})"
779 flags_return=${FLAGS_ERROR}
780 break
781 fi
782
kate.ward60c25982008-11-12 20:12:44 +0000783 # set new flag value
kate.wardda8d2c02011-06-28 13:26:02 +0000784 _flags_usName_=`_flags_underscoreName ${_flags_name_}`
kate.ward60c25982008-11-12 20:12:44 +0000785 [ ${_flags_type_} -eq ${__FLAGS_TYPE_NONE} ] && \
kate.wardd9672402008-11-15 16:55:51 +0000786 _flags_type_=`_flags_getFlagInfo \
kate.wardda8d2c02011-06-28 13:26:02 +0000787 "${_flags_usName_}" ${__FLAGS_INFO_TYPE}`
kate.ward60c25982008-11-12 20:12:44 +0000788 case ${_flags_type_} in
789 ${__FLAGS_TYPE_BOOLEAN})
790 if [ ${_flags_len_} -eq ${__FLAGS_LEN_LONG} ]; then
791 if [ "${_flags_arg_}" != "${__FLAGS_NULL}" ]; then
kate.wardda8d2c02011-06-28 13:26:02 +0000792 eval "FLAGS_${_flags_usName_}=${FLAGS_TRUE}"
kate.ward60c25982008-11-12 20:12:44 +0000793 else
kate.wardda8d2c02011-06-28 13:26:02 +0000794 eval "FLAGS_${_flags_usName_}=${FLAGS_FALSE}"
kate.ward60c25982008-11-12 20:12:44 +0000795 fi
796 else
797 _flags_strToEval_="_flags_val_=\
kate.wardda8d2c02011-06-28 13:26:02 +0000798\${__flags_${_flags_usName_}_${__FLAGS_INFO_DEFAULT}}"
kate.ward60c25982008-11-12 20:12:44 +0000799 eval "${_flags_strToEval_}"
800 if [ ${_flags_val_} -eq ${FLAGS_FALSE} ]; then
kate.wardda8d2c02011-06-28 13:26:02 +0000801 eval "FLAGS_${_flags_usName_}=${FLAGS_TRUE}"
kate.ward60c25982008-11-12 20:12:44 +0000802 else
kate.wardda8d2c02011-06-28 13:26:02 +0000803 eval "FLAGS_${_flags_usName_}=${FLAGS_FALSE}"
kate.ward60c25982008-11-12 20:12:44 +0000804 fi
805 fi
806 ;;
807
808 ${__FLAGS_TYPE_FLOAT})
kate.warde10dd532013-01-04 21:52:23 +0000809 if ${__FLAGS_FX_VALID_FLOAT} "${_flags_arg_}"; then
kate.wardda8d2c02011-06-28 13:26:02 +0000810 eval "FLAGS_${_flags_usName_}='${_flags_arg_}'"
kate.ward60c25982008-11-12 20:12:44 +0000811 else
812 flags_error="invalid float value (${_flags_arg_})"
813 flags_return=${FLAGS_ERROR}
814 break
815 fi
816 ;;
817
818 ${__FLAGS_TYPE_INTEGER})
kate.wardc66a5fc2013-01-12 23:10:15 +0000819 if _flags_validInt "${_flags_arg_}"; then
kate.wardda8d2c02011-06-28 13:26:02 +0000820 eval "FLAGS_${_flags_usName_}='${_flags_arg_}'"
kate.ward60c25982008-11-12 20:12:44 +0000821 else
822 flags_error="invalid integer value (${_flags_arg_})"
823 flags_return=${FLAGS_ERROR}
824 break
825 fi
826 ;;
827
828 ${__FLAGS_TYPE_STRING})
kate.wardda8d2c02011-06-28 13:26:02 +0000829 eval "FLAGS_${_flags_usName_}='${_flags_arg_}'"
kate.ward60c25982008-11-12 20:12:44 +0000830 ;;
831 esac
832
kate.ward5b48fc12008-11-13 00:42:43 +0000833 # handle special case help flag
kate.wardda8d2c02011-06-28 13:26:02 +0000834 if [ "${_flags_usName_}" = 'help' ]; then
kate.ward5b48fc12008-11-13 00:42:43 +0000835 if [ ${FLAGS_help} -eq ${FLAGS_TRUE} ]; then
836 flags_help
837 flags_error='help requested'
kate.ward1cb79602011-06-10 11:15:49 +0000838 flags_return=${FLAGS_TRUE}
kate.ward5b48fc12008-11-13 00:42:43 +0000839 break
840 fi
841 fi
842
kate.wardc5210682009-03-30 18:54:36 +0000843 # shift the option and non-boolean arguements out.
kate.ward60c25982008-11-12 20:12:44 +0000844 shift
kate.wardc5210682009-03-30 18:54:36 +0000845 [ ${_flags_type_} != ${__FLAGS_TYPE_BOOLEAN} ] && shift
846 done
kate.ward60c25982008-11-12 20:12:44 +0000847
kate.wardc5210682009-03-30 18:54:36 +0000848 # give user back non-flag arguments
849 FLAGS_ARGV=''
850 while [ $# -gt 0 ]; do
851 FLAGS_ARGV="${FLAGS_ARGV:+${FLAGS_ARGV} }'$1'"
852 shift
kate.ward60c25982008-11-12 20:12:44 +0000853 done
854
855 unset _flags_arg_ _flags_len_ _flags_name_ _flags_opt_ _flags_pos_ \
kate.wardda8d2c02011-06-28 13:26:02 +0000856 _flags_strToEval_ _flags_type_ _flags_usName_ _flags_val_
kate.ward60c25982008-11-12 20:12:44 +0000857 return ${flags_return}
858}
859
kate.wardb0759202013-01-05 13:57:46 +0000860# Perform some path using built-ins.
861#
862# Args:
863# $@: string: math expression to evaluate
864# Returns:
865# string: the result
866_flags_mathBuiltin()
867{
868 # Variable assignment is needed as workaround for Solaris Bourne shell, which
869 # cannot parse a bare $((expression)).
870 _flags_expr_='$(($@))'
871 eval echo ${_flags_expr_}
872}
873
kate.wardc66a5fc2013-01-12 23:10:15 +0000874# Use built-in helper function to enable unit testing.
kate.wardb0759202013-01-05 13:57:46 +0000875#
876# Args:
kate.wardc66a5fc2013-01-12 23:10:15 +0000877# None
kate.wardb0759202013-01-05 13:57:46 +0000878# Returns:
kate.wardc66a5fc2013-01-12 23:10:15 +0000879# bool: true if built-ins should be used
880_flags_useBuiltin()
kate.wardb0759202013-01-05 13:57:46 +0000881{
kate.wardc66a5fc2013-01-12 23:10:15 +0000882 return ${__FLAGS_USE_BUILTIN}
kate.wardb0759202013-01-05 13:57:46 +0000883}
884
kate.wardf51c6162008-06-17 16:38:35 +0000885#------------------------------------------------------------------------------
886# public functions
kate.wardc66a5fc2013-01-12 23:10:15 +0000887#
kate.wardf51c6162008-06-17 16:38:35 +0000888# A basic boolean flag. Boolean flags do not take any arguments, and their
889# value is either 1 (false) or 0 (true). For long flags, the false value is
890# specified on the command line by prepending the word 'no'. With short flags,
891# the presense of the flag toggles the current value between true and false.
892# Specifying a short boolean flag twice on the command results in returning the
893# value back to the default value.
894#
895# A default value is required for boolean flags.
896#
897# For example, lets say a Boolean flag was created whose long name was 'update'
898# and whose short name was 'x', and the default value was 'false'. This flag
899# could be explicitly set to 'true' with '--update' or by '-x', and it could be
900# explicitly set to 'false' with '--noupdate'.
901DEFINE_boolean() { _flags_define ${__FLAGS_TYPE_BOOLEAN} "$@"; }
902
903# Other basic flags.
904DEFINE_float() { _flags_define ${__FLAGS_TYPE_FLOAT} "$@"; }
905DEFINE_integer() { _flags_define ${__FLAGS_TYPE_INTEGER} "$@"; }
906DEFINE_string() { _flags_define ${__FLAGS_TYPE_STRING} "$@"; }
907
908# Parse the flags.
909#
910# Args:
911# unnamed: list: command-line flags to parse
912# Returns:
913# integer: success of operation, or error
914FLAGS()
915{
kate.wardabae05d2008-11-12 22:37:05 +0000916 # define a standard 'help' flag if one isn't already defined
917 [ -z "${__flags_help_type:-}" ] && \
918 DEFINE_boolean 'help' false 'show this help' 'h'
kate.wardf08c5b62008-07-11 20:32:11 +0000919
kate.wardf51c6162008-06-17 16:38:35 +0000920 # parse options
kate.ward7a3c9c42009-04-24 00:12:35 +0000921 if [ $# -gt 0 ]; then
922 if [ ${__FLAGS_GETOPT_VERS} -ne ${__FLAGS_GETOPT_VERS_ENH} ]; then
923 _flags_getoptStandard "$@"
924 else
925 _flags_getoptEnhanced "$@"
926 fi
927 flags_return=$?
kate.wardf51c6162008-06-17 16:38:35 +0000928 else
kate.ward7a3c9c42009-04-24 00:12:35 +0000929 # nothing passed; won't bother running getopt
930 __flags_opts='--'
931 flags_return=${FLAGS_TRUE}
kate.wardf51c6162008-06-17 16:38:35 +0000932 fi
kate.ward60c25982008-11-12 20:12:44 +0000933
934 if [ ${flags_return} -eq ${FLAGS_TRUE} ]; then
kate.ward7a3c9c42009-04-24 00:12:35 +0000935 _flags_parseGetopt $# "${__flags_opts}"
kate.ward60c25982008-11-12 20:12:44 +0000936 flags_return=$?
kate.wardf51c6162008-06-17 16:38:35 +0000937 fi
938
kate.wardf51c6162008-06-17 16:38:35 +0000939 [ ${flags_return} -eq ${FLAGS_ERROR} ] && _flags_fatal "${flags_error}"
940 return ${flags_return}
941}
942
kate.ward1cb79602011-06-10 11:15:49 +0000943# This is a helper function for determining the 'getopt' version for platforms
kate.wardf51c6162008-06-17 16:38:35 +0000944# where the detection isn't working. It simply outputs debug information that
945# can be included in a bug report.
946#
947# Args:
948# none
kate.ward60c25982008-11-12 20:12:44 +0000949# Output:
950# debug info that can be included in a bug report
kate.wardf51c6162008-06-17 16:38:35 +0000951# Returns:
952# nothing
953flags_getoptInfo()
954{
955 # platform info
956 _flags_debug "uname -a: `uname -a`"
957 _flags_debug "PATH: ${PATH}"
958
959 # shell info
960 if [ -n "${BASH_VERSION:-}" ]; then
961 _flags_debug 'shell: bash'
962 _flags_debug "BASH_VERSION: ${BASH_VERSION}"
963 elif [ -n "${ZSH_VERSION:-}" ]; then
964 _flags_debug 'shell: zsh'
965 _flags_debug "ZSH_VERSION: ${ZSH_VERSION}"
966 fi
967
968 # getopt info
kate.ward1cb79602011-06-10 11:15:49 +0000969 ${FLAGS_GETOPT_CMD} >/dev/null
kate.ward60c25982008-11-12 20:12:44 +0000970 _flags_getoptReturn=$?
971 _flags_debug "getopt return: ${_flags_getoptReturn}"
kate.ward1cb79602011-06-10 11:15:49 +0000972 _flags_debug "getopt --version: `${FLAGS_GETOPT_CMD} --version 2>&1`"
kate.wardf51c6162008-06-17 16:38:35 +0000973
kate.ward60c25982008-11-12 20:12:44 +0000974 unset _flags_getoptReturn
kate.wardf51c6162008-06-17 16:38:35 +0000975}
976
kate.ward437639d2008-10-19 17:21:41 +0000977# Returns whether the detected getopt version is the enhanced version.
kate.wardf51c6162008-06-17 16:38:35 +0000978#
979# Args:
980# none
kate.ward60c25982008-11-12 20:12:44 +0000981# Output:
982# none
kate.wardf51c6162008-06-17 16:38:35 +0000983# Returns:
984# bool: true if getopt is the enhanced version
985flags_getoptIsEnh()
986{
kate.wardd9672402008-11-15 16:55:51 +0000987 test ${__FLAGS_GETOPT_VERS} -eq ${__FLAGS_GETOPT_VERS_ENH}
kate.wardf51c6162008-06-17 16:38:35 +0000988}
989
kate.ward437639d2008-10-19 17:21:41 +0000990# Returns whether the detected getopt version is the standard version.
kate.wardf51c6162008-06-17 16:38:35 +0000991#
992# Args:
993# none
994# Returns:
995# bool: true if getopt is the standard version
996flags_getoptIsStd()
997{
kate.wardd9672402008-11-15 16:55:51 +0000998 test ${__FLAGS_GETOPT_VERS} -eq ${__FLAGS_GETOPT_VERS_STD}
kate.wardf51c6162008-06-17 16:38:35 +0000999}
1000
1001# This is effectively a 'usage()' function. It prints usage information and
1002# exits the program with ${FLAGS_FALSE} if it is ever found in the command line
1003# arguments. Note this function can be overridden so other apps can define
1004# their own --help flag, replacing this one, if they want.
1005#
1006# Args:
1007# none
1008# Returns:
1009# integer: success of operation (always returns true)
1010flags_help()
1011{
1012 if [ -n "${FLAGS_HELP:-}" ]; then
1013 echo "${FLAGS_HELP}" >&2
1014 else
kate.ward4d8e0472008-07-18 11:52:56 +00001015 echo "USAGE: ${FLAGS_PARENT:-$0} [flags] args" >&2
kate.wardf51c6162008-06-17 16:38:35 +00001016 fi
1017 if [ -n "${__flags_longNames}" ]; then
1018 echo 'flags:' >&2
kate.wardd9672402008-11-15 16:55:51 +00001019 for flags_name_ in ${__flags_longNames}; do
1020 flags_flagStr_=''
kate.wardbda39172009-04-01 03:30:22 +00001021 flags_boolStr_=''
kate.wardda8d2c02011-06-28 13:26:02 +00001022 flags_usName_=`_flags_underscoreName ${flags_name_}`
kate.wardf51c6162008-06-17 16:38:35 +00001023
kate.wardd9672402008-11-15 16:55:51 +00001024 flags_default_=`_flags_getFlagInfo \
kate.wardda8d2c02011-06-28 13:26:02 +00001025 "${flags_usName_}" ${__FLAGS_INFO_DEFAULT}`
kate.wardd9672402008-11-15 16:55:51 +00001026 flags_help_=`_flags_getFlagInfo \
kate.wardda8d2c02011-06-28 13:26:02 +00001027 "${flags_usName_}" ${__FLAGS_INFO_HELP}`
kate.wardd9672402008-11-15 16:55:51 +00001028 flags_short_=`_flags_getFlagInfo \
kate.wardda8d2c02011-06-28 13:26:02 +00001029 "${flags_usName_}" ${__FLAGS_INFO_SHORT}`
kate.wardd9672402008-11-15 16:55:51 +00001030 flags_type_=`_flags_getFlagInfo \
kate.wardda8d2c02011-06-28 13:26:02 +00001031 "${flags_usName_}" ${__FLAGS_INFO_TYPE}`
kate.wardf51c6162008-06-17 16:38:35 +00001032
kate.ward1cb79602011-06-10 11:15:49 +00001033 [ "${flags_short_}" != "${__FLAGS_NULL}" ] && \
1034 flags_flagStr_="-${flags_short_}"
kate.ward437639d2008-10-19 17:21:41 +00001035
1036 if [ ${__FLAGS_GETOPT_VERS} -eq ${__FLAGS_GETOPT_VERS_ENH} ]; then
kate.ward1cb79602011-06-10 11:15:49 +00001037 [ "${flags_short_}" != "${__FLAGS_NULL}" ] && \
1038 flags_flagStr_="${flags_flagStr_},"
1039 # add [no] to long boolean flag names, except the 'help' flag
1040 [ ${flags_type_} -eq ${__FLAGS_TYPE_BOOLEAN} \
kate.wardda8d2c02011-06-28 13:26:02 +00001041 -a "${flags_usName_}" != 'help' ] && \
kate.ward1cb79602011-06-10 11:15:49 +00001042 flags_boolStr_='[no]'
kate.wardd9672402008-11-15 16:55:51 +00001043 flags_flagStr_="${flags_flagStr_}--${flags_boolStr_}${flags_name_}:"
kate.ward437639d2008-10-19 17:21:41 +00001044 fi
1045
kate.wardd9672402008-11-15 16:55:51 +00001046 case ${flags_type_} in
kate.ward437639d2008-10-19 17:21:41 +00001047 ${__FLAGS_TYPE_BOOLEAN})
kate.wardd9672402008-11-15 16:55:51 +00001048 if [ ${flags_default_} -eq ${FLAGS_TRUE} ]; then
1049 flags_defaultStr_='true'
kate.ward437639d2008-10-19 17:21:41 +00001050 else
kate.wardd9672402008-11-15 16:55:51 +00001051 flags_defaultStr_='false'
kate.ward437639d2008-10-19 17:21:41 +00001052 fi
1053 ;;
1054 ${__FLAGS_TYPE_FLOAT}|${__FLAGS_TYPE_INTEGER})
kate.wardd9672402008-11-15 16:55:51 +00001055 flags_defaultStr_=${flags_default_} ;;
1056 ${__FLAGS_TYPE_STRING}) flags_defaultStr_="'${flags_default_}'" ;;
kate.ward437639d2008-10-19 17:21:41 +00001057 esac
kate.wardd9672402008-11-15 16:55:51 +00001058 flags_defaultStr_="(default: ${flags_defaultStr_})"
kate.ward437639d2008-10-19 17:21:41 +00001059
kate.wardd9672402008-11-15 16:55:51 +00001060 flags_helpStr_=" ${flags_flagStr_} ${flags_help_} ${flags_defaultStr_}"
kate.warda589b9f2012-12-17 23:27:51 +00001061 flags_helpStrLen_=`${FLAGS_EXPR_CMD} -- "${flags_helpStr_}" : '.*'`
kate.ward14b33bf2008-11-15 20:03:18 +00001062 flags_columns_=`_flags_columns`
1063 if [ ${flags_helpStrLen_} -lt ${flags_columns_} ]; then
kate.wardd9672402008-11-15 16:55:51 +00001064 echo "${flags_helpStr_}" >&2
kate.ward437639d2008-10-19 17:21:41 +00001065 else
kate.ward64099612008-11-17 17:36:14 +00001066 echo " ${flags_flagStr_} ${flags_help_}" >&2
kate.ward87b0c2e2009-04-01 03:48:16 +00001067 # note: the silliness with the x's is purely for ksh93 on Ubuntu 6.06
1068 # because it doesn't like empty strings when used in this manner.
kate.wardbda39172009-04-01 03:30:22 +00001069 flags_emptyStr_="`echo \"x${flags_flagStr_}x\" \
1070 |awk '{printf "%"length($0)-2"s", ""}'`"
kate.ward64099612008-11-17 17:36:14 +00001071 flags_helpStr_=" ${flags_emptyStr_} ${flags_defaultStr_}"
kate.warda589b9f2012-12-17 23:27:51 +00001072 flags_helpStrLen_=`${FLAGS_EXPR_CMD} -- "${flags_helpStr_}" : '.*'`
kate.ward64099612008-11-17 17:36:14 +00001073 if [ ${__FLAGS_GETOPT_VERS} -eq ${__FLAGS_GETOPT_VERS_STD} \
1074 -o ${flags_helpStrLen_} -lt ${flags_columns_} ]; then
1075 # indented to match help string
1076 echo "${flags_helpStr_}" >&2
1077 else
1078 # indented four from left to allow for longer defaults as long flag
1079 # names might be used too, making things too long
1080 echo " ${flags_defaultStr_}" >&2
1081 fi
kate.ward437639d2008-10-19 17:21:41 +00001082 fi
kate.wardf51c6162008-06-17 16:38:35 +00001083 done
1084 fi
1085
kate.wardd9672402008-11-15 16:55:51 +00001086 unset flags_boolStr_ flags_default_ flags_defaultStr_ flags_emptyStr_ \
1087 flags_flagStr_ flags_help_ flags_helpStr flags_helpStrLen flags_name_ \
kate.wardda8d2c02011-06-28 13:26:02 +00001088 flags_columns_ flags_short_ flags_type_ flags_usName_
kate.wardf51c6162008-06-17 16:38:35 +00001089 return ${FLAGS_TRUE}
1090}
1091
kate.wardcecd7bd2009-05-10 17:56:13 +00001092# Reset shflags back to an uninitialized state.
kate.wardf51c6162008-06-17 16:38:35 +00001093#
1094# Args:
1095# none
1096# Returns:
1097# nothing
1098flags_reset()
1099{
kate.wardd9672402008-11-15 16:55:51 +00001100 for flags_name_ in ${__flags_longNames}; do
kate.ward869a71b2011-06-28 13:29:40 +00001101 flags_usName_=`_flags_underscoreName ${flags_name_}`
1102 flags_strToEval_="unset FLAGS_${flags_usName_}"
kate.wardd9672402008-11-15 16:55:51 +00001103 for flags_type_ in \
kate.wardcecd7bd2009-05-10 17:56:13 +00001104 ${__FLAGS_INFO_DEFAULT} \
kate.wardfb0e7182009-05-10 16:59:13 +00001105 ${__FLAGS_INFO_HELP} \
1106 ${__FLAGS_INFO_SHORT} \
1107 ${__FLAGS_INFO_TYPE}
kate.wardf51c6162008-06-17 16:38:35 +00001108 do
kate.wardd9672402008-11-15 16:55:51 +00001109 flags_strToEval_=\
kate.ward869a71b2011-06-28 13:29:40 +00001110"${flags_strToEval_} __flags_${flags_usName_}_${flags_type_}"
kate.wardf51c6162008-06-17 16:38:35 +00001111 done
kate.wardd9672402008-11-15 16:55:51 +00001112 eval ${flags_strToEval_}
kate.wardf51c6162008-06-17 16:38:35 +00001113 done
1114
1115 # reset internal variables
1116 __flags_boolNames=' '
1117 __flags_longNames=' '
1118 __flags_shortNames=' '
kate.wardda8d2c02011-06-28 13:26:02 +00001119 __flags_definedNames=' '
kate.wardf51c6162008-06-17 16:38:35 +00001120
kate.ward869a71b2011-06-28 13:29:40 +00001121 unset flags_name_ flags_type_ flags_strToEval_ flags_usName_
kate.wardf51c6162008-06-17 16:38:35 +00001122}