blob: 0d0f9de7d1ea933b2eaa0bad2e5fb3c085b985ab [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#
53# NOTE: Not all systems include a getopt version that supports long flags. On
54# these systems, only short flags are recognized.
kate.wardf51c6162008-06-17 16:38:35 +000055
56#==============================================================================
kate.warda8953952008-06-17 16:48:34 +000057# shFlags
kate.wardf51c6162008-06-17 16:38:35 +000058#
59# Shared attributes:
60# flags_error: last error message
61# flags_return: last return value
62#
63# __flags_longNames: list of long names for all flags
64# __flags_shortNames: list of short names for all flags
65# __flags_boolNames: list of boolean flag names
66#
kate.ward60c25982008-11-12 20:12:44 +000067# __flags_opts: options parsed by getopt
68#
kate.wardf51c6162008-06-17 16:38:35 +000069# Per-flag attributes:
70# FLAGS_<flag_name>: contains value of flag named 'flag_name'
71# __flags_<flag_name>_default: the default flag value
72# __flags_<flag_name>_help: the flag help string
73# __flags_<flag_name>_short: the flag short name
74# __flags_<flag_name>_type: the flag type
75#
76# Notes:
77# - lists of strings are space separated, and a null value is the '~' char.
78
79# return if FLAGS already loaded
kate.wardd3d3d082008-07-10 20:00:42 +000080[ -n "${FLAGS_VERSION:-}" ] && return 0
kate.ward9a183b82010-03-18 00:25:34 +000081FLAGS_VERSION='1.0.4pre'
kate.wardf51c6162008-06-17 16:38:35 +000082
kate.ward39913d12009-04-01 14:09:23 +000083# return values
kate.wardd2258ed2008-11-12 23:21:48 +000084FLAGS_TRUE=0
85FLAGS_FALSE=1
86FLAGS_ERROR=2
87
kate.ward39913d12009-04-01 14:09:23 +000088# reserved flag names
kate.ward929261f2010-03-28 23:12:17 +000089FLAGS_RESERVED_LIST=' ARGC ARGV ERROR FALSE HELP PARENT RESERVED TRUE VERSION '
kate.ward39913d12009-04-01 14:09:23 +000090
kate.wardf51c6162008-06-17 16:38:35 +000091_flags_debug() { echo "flags:DEBUG $@" >&2; }
kate.wardb7fa8d62008-08-28 21:18:21 +000092_flags_warn() { echo "flags:WARN $@" >&2; }
kate.wardf51c6162008-06-17 16:38:35 +000093_flags_error() { echo "flags:ERROR $@" >&2; }
94_flags_fatal() { echo "flags:FATAL $@" >&2; }
95
kate.wardf51c6162008-06-17 16:38:35 +000096# specific shell checks
97if [ -n "${ZSH_VERSION:-}" ]; then
kate.ward1d0ecc42008-07-11 15:33:23 +000098 setopt |grep "^shwordsplit$" >/dev/null
99 if [ $? -ne ${FLAGS_TRUE} ]; then
100 _flags_fatal 'zsh shwordsplit option is required for proper zsh operation'
kate.wardf51c6162008-06-17 16:38:35 +0000101 exit ${FLAGS_ERROR}
102 fi
103 if [ -z "${FLAGS_PARENT:-}" ]; then
104 _flags_fatal "zsh does not pass \$0 through properly. please declare' \
kate.warda8953952008-06-17 16:48:34 +0000105\"FLAGS_PARENT=\$0\" before calling shFlags"
kate.wardf51c6162008-06-17 16:38:35 +0000106 exit ${FLAGS_ERROR}
107 fi
108fi
109
kate.wardf51c6162008-06-17 16:38:35 +0000110#
111# constants
112#
113
kate.wardf51c6162008-06-17 16:38:35 +0000114# getopt version
115__FLAGS_GETOPT_VERS_STD=0
116__FLAGS_GETOPT_VERS_ENH=1
117__FLAGS_GETOPT_VERS_BSD=2
118
119getopt >/dev/null 2>&1
120case $? in
121 0) __FLAGS_GETOPT_VERS=${__FLAGS_GETOPT_VERS_STD} ;; # bsd getopt
122 2)
kate.wardc5210682009-03-30 18:54:36 +0000123 # TODO(kward): look into '-T' option to test the internal getopt() version
kate.wardf51c6162008-06-17 16:38:35 +0000124 if [ "`getopt --version`" = '-- ' ]; then
125 __FLAGS_GETOPT_VERS=${__FLAGS_GETOPT_VERS_STD}
126 else
127 __FLAGS_GETOPT_VERS=${__FLAGS_GETOPT_VERS_ENH}
128 fi
129 ;;
130 *)
131 _flags_fatal 'unable to determine getopt version'
132 exit ${FLAGS_ERROR}
133 ;;
134esac
135
136# getopt optstring lengths
137__FLAGS_OPTSTR_SHORT=0
138__FLAGS_OPTSTR_LONG=1
139
140__FLAGS_NULL='~'
141
142# flag info strings
kate.wardfb0e7182009-05-10 16:59:13 +0000143__FLAGS_INFO_DEFAULT='default'
144__FLAGS_INFO_HELP='help'
145__FLAGS_INFO_SHORT='short'
146__FLAGS_INFO_TYPE='type'
kate.wardf51c6162008-06-17 16:38:35 +0000147
148# flag lengths
149__FLAGS_LEN_SHORT=0
150__FLAGS_LEN_LONG=1
151
152# flag types
153__FLAGS_TYPE_NONE=0
154__FLAGS_TYPE_BOOLEAN=1
155__FLAGS_TYPE_FLOAT=2
156__FLAGS_TYPE_INTEGER=3
157__FLAGS_TYPE_STRING=4
158
159# set the constants readonly
kate.wardd9672402008-11-15 16:55:51 +0000160__flags_constants=`set |awk -F= '/^FLAGS_/ || /^__FLAGS_/ {print $1}'`
161for __flags_const in ${__flags_constants}; do
162 # skip certain flags
163 case ${__flags_const} in
kate.ward39913d12009-04-01 14:09:23 +0000164 FLAGS_HELP) continue ;;
kate.wardd9672402008-11-15 16:55:51 +0000165 FLAGS_PARENT) continue ;;
166 esac
167 # set flag readonly
168 if [ -z "${ZSH_VERSION:-}" ]; then
169 readonly ${__flags_const}
170 else # handle zsh
kate.ward1d0ecc42008-07-11 15:33:23 +0000171 case ${ZSH_VERSION} in
kate.wardd9672402008-11-15 16:55:51 +0000172 [123].*) readonly ${__flags_const} ;;
173 *) readonly -g ${__flags_const} ;; # declare readonly constants globally
kate.ward1d0ecc42008-07-11 15:33:23 +0000174 esac
kate.wardf51c6162008-06-17 16:38:35 +0000175 fi
176done
kate.wardd9672402008-11-15 16:55:51 +0000177unset __flags_const __flags_constants
kate.wardf51c6162008-06-17 16:38:35 +0000178
179#
180# internal variables
181#
182
kate.wardd9672402008-11-15 16:55:51 +0000183__flags_boolNames=' ' # space separated list of boolean flag names
184__flags_longNames=' ' # space separated list of long flag names
185__flags_shortNames=' ' # space separated list of short flag names
kate.wardf51c6162008-06-17 16:38:35 +0000186
kate.ward14b33bf2008-11-15 20:03:18 +0000187__flags_columns='' # screen width in columns
kate.wardd9672402008-11-15 16:55:51 +0000188__flags_opts='' # temporary storage for parsed getopt flags
kate.ward60c25982008-11-12 20:12:44 +0000189
kate.wardf51c6162008-06-17 16:38:35 +0000190#------------------------------------------------------------------------------
191# private functions
192#
193
kate.wardcc145682008-11-14 00:32:42 +0000194# Define a flag.
195#
196# Calling this function will define the following info variables for the
197# specified flag:
kate.wardf51c6162008-06-17 16:38:35 +0000198# FLAGS_flagname - the name for this flag (based upon the long flag name)
199# __flags_<flag_name>_default - the default value
200# __flags_flagname_help - the help string
201# __flags_flagname_short - the single letter alias
202# __flags_flagname_type - the type of flag (one of __FLAGS_TYPE_*)
203#
204# Args:
205# _flags__type: integer: internal type of flag (__FLAGS_TYPE_*)
206# _flags__name: string: long flag name
207# _flags__default: default flag value
208# _flags__help: string: help string
209# _flags__short: string: (optional) short flag name
210# Returns:
211# integer: success of operation, or error
212_flags_define()
213{
214 if [ $# -lt 4 ]; then
215 flags_error='DEFINE error: too few arguments'
216 flags_return=${FLAGS_ERROR}
217 _flags_error "${flags_error}"
218 return ${flags_return}
219 fi
220
kate.ward60c25982008-11-12 20:12:44 +0000221 _flags_type_=$1
222 _flags_name_=$2
223 _flags_default_=$3
224 _flags_help_=$4
225 _flags_short_=${5:-${__FLAGS_NULL}}
kate.wardf51c6162008-06-17 16:38:35 +0000226
kate.ward60c25982008-11-12 20:12:44 +0000227 _flags_return_=${FLAGS_TRUE}
kate.wardf51c6162008-06-17 16:38:35 +0000228
229 # TODO(kward): check for validity of the flag name (e.g. dashes)
kate.wardf51c6162008-06-17 16:38:35 +0000230
kate.ward39913d12009-04-01 14:09:23 +0000231 # check whether the flag name is reserved
kate.ward929261f2010-03-28 23:12:17 +0000232 _flags_itemInList ${_flags_name_} "${FLAGS_RESERVED_LIST}"
233 if [ $? -eq ${FLAGS_TRUE} ]; then
kate.ward39913d12009-04-01 14:09:23 +0000234 flags_error="flag name (${_flags_name_}) is reserved"
235 _flags_return_=${FLAGS_ERROR}
236 fi
237
kate.wardf51c6162008-06-17 16:38:35 +0000238 # require short option for getopt that don't support long options
kate.ward60c25982008-11-12 20:12:44 +0000239 if [ ${_flags_return_} -eq ${FLAGS_TRUE} \
kate.wardf51c6162008-06-17 16:38:35 +0000240 -a ${__FLAGS_GETOPT_VERS} -ne ${__FLAGS_GETOPT_VERS_ENH} \
kate.ward60c25982008-11-12 20:12:44 +0000241 -a "${_flags_short_}" = "${__FLAGS_NULL}" ]
kate.wardf51c6162008-06-17 16:38:35 +0000242 then
kate.ward5b48fc12008-11-13 00:42:43 +0000243 flags_error="short flag required for (${_flags_name_}) on this platform"
kate.ward60c25982008-11-12 20:12:44 +0000244 _flags_return_=${FLAGS_ERROR}
kate.wardf51c6162008-06-17 16:38:35 +0000245 fi
246
247 # check for existing long name definition
kate.ward60c25982008-11-12 20:12:44 +0000248 if [ ${_flags_return_} -eq ${FLAGS_TRUE} ]; then
249 if _flags_itemInList "${_flags_name_}" \
kate.wardf51c6162008-06-17 16:38:35 +0000250 ${__flags_longNames} ${__flags_boolNames}
251 then
kate.ward60c25982008-11-12 20:12:44 +0000252 flags_error="flag name ([no]${_flags_name_}) already defined"
kate.ward20d06782008-10-21 19:57:19 +0000253 _flags_warn "${flags_error}"
kate.ward60c25982008-11-12 20:12:44 +0000254 _flags_return_=${FLAGS_FALSE}
kate.wardf51c6162008-06-17 16:38:35 +0000255 fi
256 fi
257
258 # check for existing short name definition
kate.ward60c25982008-11-12 20:12:44 +0000259 if [ ${_flags_return_} -eq ${FLAGS_TRUE} \
260 -a "${_flags_short_}" != "${__FLAGS_NULL}" ]
kate.wardf51c6162008-06-17 16:38:35 +0000261 then
kate.ward60c25982008-11-12 20:12:44 +0000262 if _flags_itemInList "${_flags_short_}" ${__flags_shortNames}; then
263 flags_error="flag short name (${_flags_short_}) already defined"
kate.ward20d06782008-10-21 19:57:19 +0000264 _flags_warn "${flags_error}"
kate.ward60c25982008-11-12 20:12:44 +0000265 _flags_return_=${FLAGS_FALSE}
kate.wardf51c6162008-06-17 16:38:35 +0000266 fi
267 fi
268
269 # handle default value. note, on several occasions the 'if' portion of an
270 # if/then/else contains just a ':' which does nothing. a binary reversal via
271 # '!' is not done because it does not work on all shells.
kate.ward60c25982008-11-12 20:12:44 +0000272 if [ ${_flags_return_} -eq ${FLAGS_TRUE} ]; then
273 case ${_flags_type_} in
kate.wardf51c6162008-06-17 16:38:35 +0000274 ${__FLAGS_TYPE_BOOLEAN})
kate.ward60c25982008-11-12 20:12:44 +0000275 if _flags_validateBoolean "${_flags_default_}"; then
276 case ${_flags_default_} in
277 true|t|0) _flags_default_=${FLAGS_TRUE} ;;
278 false|f|1) _flags_default_=${FLAGS_FALSE} ;;
kate.wardf51c6162008-06-17 16:38:35 +0000279 esac
280 else
kate.ward60c25982008-11-12 20:12:44 +0000281 flags_error="invalid default flag value '${_flags_default_}'"
282 _flags_return_=${FLAGS_ERROR}
kate.wardf51c6162008-06-17 16:38:35 +0000283 fi
284 ;;
285
286 ${__FLAGS_TYPE_FLOAT})
kate.ward60c25982008-11-12 20:12:44 +0000287 if _flags_validateFloat "${_flags_default_}"; then
kate.wardf51c6162008-06-17 16:38:35 +0000288 :
289 else
kate.ward60c25982008-11-12 20:12:44 +0000290 flags_error="invalid default flag value '${_flags_default_}'"
291 _flags_return_=${FLAGS_ERROR}
kate.wardf51c6162008-06-17 16:38:35 +0000292 fi
293 ;;
294
295 ${__FLAGS_TYPE_INTEGER})
kate.ward60c25982008-11-12 20:12:44 +0000296 if _flags_validateInteger "${_flags_default_}"; then
kate.wardf51c6162008-06-17 16:38:35 +0000297 :
298 else
kate.ward60c25982008-11-12 20:12:44 +0000299 flags_error="invalid default flag value '${_flags_default_}'"
300 _flags_return_=${FLAGS_ERROR}
kate.wardf51c6162008-06-17 16:38:35 +0000301 fi
302 ;;
303
304 ${__FLAGS_TYPE_STRING}) ;; # everything in shell is a valid string
305
306 *)
kate.ward60c25982008-11-12 20:12:44 +0000307 flags_error="unrecognized flag type '${_flags_type_}'"
308 _flags_return_=${FLAGS_ERROR}
kate.wardf51c6162008-06-17 16:38:35 +0000309 ;;
310 esac
311 fi
312
kate.ward60c25982008-11-12 20:12:44 +0000313 if [ ${_flags_return_} -eq ${FLAGS_TRUE} ]; then
kate.wardf51c6162008-06-17 16:38:35 +0000314 # store flag information
kate.ward60c25982008-11-12 20:12:44 +0000315 eval "FLAGS_${_flags_name_}='${_flags_default_}'"
kate.wardfb0e7182009-05-10 16:59:13 +0000316 eval "__flags_${_flags_name_}_${__FLAGS_INFO_TYPE}=${_flags_type_}"
317 eval "__flags_${_flags_name_}_${__FLAGS_INFO_DEFAULT}=\
kate.ward60c25982008-11-12 20:12:44 +0000318\"${_flags_default_}\""
kate.wardfb0e7182009-05-10 16:59:13 +0000319 eval "__flags_${_flags_name_}_${__FLAGS_INFO_HELP}=\"${_flags_help_}\""
320 eval "__flags_${_flags_name_}_${__FLAGS_INFO_SHORT}='${_flags_short_}'"
kate.wardf51c6162008-06-17 16:38:35 +0000321
322 # append flag name(s) to list of names
kate.ward60c25982008-11-12 20:12:44 +0000323 __flags_longNames="${__flags_longNames}${_flags_name_} "
324 __flags_shortNames="${__flags_shortNames}${_flags_short_} "
325 [ ${_flags_type_} -eq ${__FLAGS_TYPE_BOOLEAN} ] && \
326 __flags_boolNames="${__flags_boolNames}no${_flags_name_} "
kate.wardf51c6162008-06-17 16:38:35 +0000327 fi
328
kate.ward60c25982008-11-12 20:12:44 +0000329 flags_return=${_flags_return_}
330 unset _flags_default_ _flags_help_ _flags_name_ _flags_return_ _flags_short_ \
331 _flags_type_
kate.wardf51c6162008-06-17 16:38:35 +0000332 [ ${flags_return} -eq ${FLAGS_ERROR} ] && _flags_error "${flags_error}"
333 return ${flags_return}
334}
335
kate.wardcc145682008-11-14 00:32:42 +0000336# Return valid getopt options using currently defined list of long options.
337#
338# This function builds a proper getopt option string for short (and long)
339# options, using the current list of long options for reference.
kate.wardf51c6162008-06-17 16:38:35 +0000340#
341# Args:
kate.wardcc145682008-11-14 00:32:42 +0000342# _flags_optStr: integer: option string type (__FLAGS_OPTSTR_*)
kate.wardf51c6162008-06-17 16:38:35 +0000343# Output:
344# string: generated option string for getopt
345# Returns:
346# boolean: success of operation (always returns True)
347_flags_genOptStr()
348{
kate.wardcc145682008-11-14 00:32:42 +0000349 _flags_optStrType_=$1
kate.wardf51c6162008-06-17 16:38:35 +0000350
kate.wardcc145682008-11-14 00:32:42 +0000351 _flags_opts_=''
kate.wardf51c6162008-06-17 16:38:35 +0000352
kate.wardcc145682008-11-14 00:32:42 +0000353 for _flags_flag_ in ${__flags_longNames}; do
kate.wardfb0e7182009-05-10 16:59:13 +0000354 _flags_type_=`_flags_getFlagInfo ${_flags_flag_} ${__FLAGS_INFO_TYPE}`
kate.ward929261f2010-03-28 23:12:17 +0000355 [ $? -eq ${FLAGS_TRUE} ] || \
356 ( _flags_fatal 'call to _flags_type_ failed'; return ${FLAGS_ERROR} )
kate.wardcc145682008-11-14 00:32:42 +0000357 case ${_flags_optStrType_} in
kate.wardf51c6162008-06-17 16:38:35 +0000358 ${__FLAGS_OPTSTR_SHORT})
kate.wardcc145682008-11-14 00:32:42 +0000359 _flags_shortName_=`_flags_getFlagInfo \
kate.wardfb0e7182009-05-10 16:59:13 +0000360 ${_flags_flag_} ${__FLAGS_INFO_SHORT}`
kate.wardcc145682008-11-14 00:32:42 +0000361 if [ "${_flags_shortName_}" != "${__FLAGS_NULL}" ]; then
362 _flags_opts_="${_flags_opts_}${_flags_shortName_}"
kate.wardc5210682009-03-30 18:54:36 +0000363 # getopt needs a trailing ':' to indicate a required argument
kate.wardcc145682008-11-14 00:32:42 +0000364 [ ${_flags_type_} -ne ${__FLAGS_TYPE_BOOLEAN} ] && \
365 _flags_opts_="${_flags_opts_}:"
kate.wardf51c6162008-06-17 16:38:35 +0000366 fi
367 ;;
368
369 ${__FLAGS_OPTSTR_LONG})
kate.wardcc145682008-11-14 00:32:42 +0000370 _flags_opts_="${_flags_opts_:+${_flags_opts_},}${_flags_flag_}"
kate.wardc5210682009-03-30 18:54:36 +0000371 # getopt needs a trailing ':' to indicate a required argument
kate.wardcc145682008-11-14 00:32:42 +0000372 [ ${_flags_type_} -ne ${__FLAGS_TYPE_BOOLEAN} ] && \
373 _flags_opts_="${_flags_opts_}:"
kate.wardf51c6162008-06-17 16:38:35 +0000374 ;;
375 esac
376 done
377
kate.wardcc145682008-11-14 00:32:42 +0000378 echo "${_flags_opts_}"
379 unset _flags_flag_ _flags_opts_ _flags_optStrType_ _flags_shortName_ \
380 _flags_type_
kate.wardf51c6162008-06-17 16:38:35 +0000381 return ${FLAGS_TRUE}
382}
383
384# Returns flag details based on a flag name and flag info.
385#
386# Args:
387# string: long flag name
388# string: flag info (see the _flags_define function for valid info types)
389# Output:
390# string: value of dereferenced flag variable
391# Returns:
392# integer: one of FLAGS_{TRUE|FALSE|ERROR}
393_flags_getFlagInfo()
394{
kate.ward1b600c52008-11-12 21:26:05 +0000395 _flags_name_=$1
396 _flags_info_=$2
kate.wardf51c6162008-06-17 16:38:35 +0000397
kate.ward1b600c52008-11-12 21:26:05 +0000398 _flags_nameVar_="__flags_${_flags_name_}_${_flags_info_}"
kate.ward929261f2010-03-28 23:12:17 +0000399 _flags_strToEval_="_flags_nameValue_=\"\${${_flags_nameVar_}:-}\""
kate.ward1b600c52008-11-12 21:26:05 +0000400 eval "${_flags_strToEval_}"
kate.ward929261f2010-03-28 23:12:17 +0000401 if [ -n "${_flags_nameValue_}" ]; then
kate.wardf51c6162008-06-17 16:38:35 +0000402 flags_return=${FLAGS_TRUE}
403 else
kate.ward1b600c52008-11-12 21:26:05 +0000404 # see if the _flags_name_ variable is a string as strings can be empty...
405 # note: the DRY principle would say to have this function call itself for
406 # the next three lines, but doing so results in an infinite loop as an
407 # invalid _flags_name_ will also not have the associated _type variable.
408 # Because it doesn't (it will evaluate to an empty string) the logic will
409 # try to find the _type variable of the _type variable, and so on. Not so
410 # good ;-)
kate.wardfb0e7182009-05-10 16:59:13 +0000411 _flags_typeVar_="__flags_${_flags_name_}_${__FLAGS_INFO_TYPE}"
kate.ward929261f2010-03-28 23:12:17 +0000412 _flags_strToEval_="_flags_typeValue_=\"\${${_flags_typeVar_}:-}\""
kate.ward1b600c52008-11-12 21:26:05 +0000413 eval "${_flags_strToEval_}"
kate.ward929261f2010-03-28 23:12:17 +0000414 if [ "${_flags_typeValue_}" = "${__FLAGS_TYPE_STRING}" ]; then
kate.ward437639d2008-10-19 17:21:41 +0000415 flags_return=${FLAGS_TRUE}
416 else
417 flags_return=${FLAGS_ERROR}
kate.ward1b600c52008-11-12 21:26:05 +0000418 flags_error="invalid flag name (${_flags_nameVar_})"
kate.ward437639d2008-10-19 17:21:41 +0000419 fi
kate.wardf51c6162008-06-17 16:38:35 +0000420 fi
421
kate.ward929261f2010-03-28 23:12:17 +0000422 echo "${_flags_nameValue_}"
423 unset _flags_info_ _flags_name_ _flags_nameValue_ _flags_nameVar_ \
424 _flags_strToEval_ _flags_typeValue_ _flags_typeVar_
kate.wardf51c6162008-06-17 16:38:35 +0000425 [ ${flags_return} -eq ${FLAGS_ERROR} ] && _flags_error "${flags_error}"
426 return ${flags_return}
427}
428
429# check for presense of item in a list. passed a string (e.g. 'abc'), this
430# function will determine if the string is present in the list of strings (e.g.
431# ' foo bar abc ').
432#
433# Args:
434# _flags__str: string: string to search for in a list of strings
435# unnamed: list: list of strings
436# Returns:
437# boolean: true if item is in the list
kate.wardcc145682008-11-14 00:32:42 +0000438_flags_itemInList()
439{
440 _flags_str_=$1
kate.wardf51c6162008-06-17 16:38:35 +0000441 shift
442
kate.wardcc145682008-11-14 00:32:42 +0000443 echo " ${*:-} " |grep " ${_flags_str_} " >/dev/null
kate.wardf51c6162008-06-17 16:38:35 +0000444 if [ $? -eq 0 ]; then
445 flags_return=${FLAGS_TRUE}
446 else
447 flags_return=${FLAGS_FALSE}
448 fi
449
kate.wardcc145682008-11-14 00:32:42 +0000450 unset _flags_str_
kate.wardf51c6162008-06-17 16:38:35 +0000451 return ${flags_return}
452}
453
kate.warddadc1642008-11-14 02:00:29 +0000454# Returns the width of the current screen.
kate.ward437639d2008-10-19 17:21:41 +0000455#
kate.ward18d456e2008-11-14 00:46:44 +0000456# Output:
kate.warddadc1642008-11-14 02:00:29 +0000457# integer: width in columns of the current screen.
kate.ward14b33bf2008-11-15 20:03:18 +0000458_flags_columns()
kate.ward437639d2008-10-19 17:21:41 +0000459{
kate.ward14b33bf2008-11-15 20:03:18 +0000460 if [ -z "${__flags_columns}" ]; then
461 # determine the value and store it
462 if eval stty size >/dev/null 2>&1; then
463 # stty size worked :-)
464 set -- `stty size`
465 __flags_columns=$2
466 elif eval tput cols >/dev/null 2>&1; then
467 set -- `tput cols`
468 __flags_columns=$1
469 else
470 __flags_columns=80 # default terminal width
471 fi
kate.warddadc1642008-11-14 02:00:29 +0000472 fi
kate.ward14b33bf2008-11-15 20:03:18 +0000473 echo ${__flags_columns}
kate.ward437639d2008-10-19 17:21:41 +0000474}
475
476# Validate a boolean.
kate.wardf51c6162008-06-17 16:38:35 +0000477#
478# Args:
479# _flags__bool: boolean: value to validate
480# Returns:
481# bool: true if the value is a valid boolean
482_flags_validateBoolean()
483{
kate.ward60c25982008-11-12 20:12:44 +0000484 _flags_bool_=$1
kate.wardf51c6162008-06-17 16:38:35 +0000485
486 flags_return=${FLAGS_TRUE}
kate.ward60c25982008-11-12 20:12:44 +0000487 case "${_flags_bool_}" in
kate.wardf51c6162008-06-17 16:38:35 +0000488 true|t|0) ;;
489 false|f|1) ;;
490 *) flags_return=${FLAGS_FALSE} ;;
491 esac
492
kate.ward60c25982008-11-12 20:12:44 +0000493 unset _flags_bool_
kate.wardf51c6162008-06-17 16:38:35 +0000494 return ${flags_return}
495}
496
kate.ward18d456e2008-11-14 00:46:44 +0000497# Validate a float.
kate.wardf51c6162008-06-17 16:38:35 +0000498#
499# Args:
500# _flags__float: float: value to validate
501# Returns:
502# bool: true if the value is a valid float
503_flags_validateFloat()
504{
kate.wardcc145682008-11-14 00:32:42 +0000505 _flags_float_=$1
kate.wardf51c6162008-06-17 16:38:35 +0000506
kate.wardcc145682008-11-14 00:32:42 +0000507 if _flags_validateInteger ${_flags_float_}; then
kate.wardf51c6162008-06-17 16:38:35 +0000508 flags_return=${FLAGS_TRUE}
509 else
510 flags_return=${FLAGS_TRUE}
kate.wardcc145682008-11-14 00:32:42 +0000511 case ${_flags_float_} in
kate.wardf51c6162008-06-17 16:38:35 +0000512 -*) # negative floats
kate.ward9a183b82010-03-18 00:25:34 +0000513 _flags_test_=`expr -- "${_flags_float_}" :\
514 '\(-[0-9][0-9]*\.[0-9][0-9]*\)'`
kate.wardf51c6162008-06-17 16:38:35 +0000515 ;;
516 *) # positive floats
kate.ward9a183b82010-03-18 00:25:34 +0000517 _flags_test_=`expr -- "${_flags_float_}" :\
518 '\([0-9][0-9]*\.[0-9][0-9]*\)'`
kate.wardf51c6162008-06-17 16:38:35 +0000519 ;;
520 esac
kate.wardcc145682008-11-14 00:32:42 +0000521 [ "${_flags_test_}" != "${_flags_float_}" ] && flags_return=${FLAGS_FALSE}
kate.wardf51c6162008-06-17 16:38:35 +0000522 fi
523
kate.wardcc145682008-11-14 00:32:42 +0000524 unset _flags_float_ _flags_test_
kate.wardf51c6162008-06-17 16:38:35 +0000525 return ${flags_return}
526}
527
kate.ward18d456e2008-11-14 00:46:44 +0000528# Validate an integer.
kate.wardf51c6162008-06-17 16:38:35 +0000529#
530# Args:
531# _flags__integer: interger: value to validate
532# Returns:
533# bool: true if the value is a valid integer
534_flags_validateInteger()
535{
kate.wardcc145682008-11-14 00:32:42 +0000536 _flags_int_=$1
kate.wardf51c6162008-06-17 16:38:35 +0000537
538 flags_return=${FLAGS_TRUE}
kate.wardcc145682008-11-14 00:32:42 +0000539 case ${_flags_int_} in
kate.wardf51c6162008-06-17 16:38:35 +0000540 -*) # negative ints
kate.ward9a183b82010-03-18 00:25:34 +0000541 _flags_test_=`expr -- "${_flags_int_}" : '\(-[0-9][0-9]*\)'`
kate.wardf51c6162008-06-17 16:38:35 +0000542 ;;
543 *) # positive ints
kate.ward9a183b82010-03-18 00:25:34 +0000544 _flags_test_=`expr -- "${_flags_int_}" : '\([0-9][0-9]*\)'`
kate.wardf51c6162008-06-17 16:38:35 +0000545 ;;
546 esac
kate.wardcc145682008-11-14 00:32:42 +0000547 [ "${_flags_test_}" != "${_flags_int_}" ] && flags_return=${FLAGS_FALSE}
kate.wardf51c6162008-06-17 16:38:35 +0000548
kate.wardcc145682008-11-14 00:32:42 +0000549 unset _flags_int_ _flags_test_
kate.wardf51c6162008-06-17 16:38:35 +0000550 return ${flags_return}
551}
552
kate.ward18d456e2008-11-14 00:46:44 +0000553# Parse command-line options using the standard getopt.
554#
555# Note: the flag options are passed around in the global __flags_opts so that
556# the formatting is not lost due to shell parsing and such.
557#
558# Args:
559# @: varies: command-line options to parse
560# Returns:
561# integer: a FLAGS success condition
kate.wardc5210682009-03-30 18:54:36 +0000562_flags_getoptStandard()
kate.ward60c25982008-11-12 20:12:44 +0000563{
564 flags_return=${FLAGS_TRUE}
565 _flags_shortOpts_=`_flags_genOptStr ${__FLAGS_OPTSTR_SHORT}`
566
567 # check for spaces in passed options
568 for _flags_opt_ in "$@"; do
kate.ward14b33bf2008-11-15 20:03:18 +0000569 # note: the silliness with the x's is purely for ksh93 on Ubuntu 6.06
570 _flags_match_=`echo "x${_flags_opt_}x" |sed 's/ //g'`
571 if [ "${_flags_match_}" != "x${_flags_opt_}x" ]; then
kate.ward60c25982008-11-12 20:12:44 +0000572 flags_error='the available getopt does not support spaces in options'
573 flags_return=${FLAGS_ERROR}
574 break
575 fi
576 done
577
578 if [ ${flags_return} -eq ${FLAGS_TRUE} ]; then
579 __flags_opts=`getopt ${_flags_shortOpts_} $@ 2>&1`
kate.ward7a3c9c42009-04-24 00:12:35 +0000580 _flags_rtrn_=$?
581 if [ ${_flags_rtrn_} -ne ${FLAGS_TRUE} ]; then
582 _flags_warn "${__flags_opts}"
kate.ward60c25982008-11-12 20:12:44 +0000583 flags_error='unable to parse provided options with getopt.'
584 flags_return=${FLAGS_ERROR}
585 fi
586 fi
587
kate.ward7a3c9c42009-04-24 00:12:35 +0000588 unset _flags_match_ _flags_opt_ _flags_rtrn_ _flags_shortOpts_
kate.ward60c25982008-11-12 20:12:44 +0000589 return ${flags_return}
590}
591
kate.ward18d456e2008-11-14 00:46:44 +0000592# Parse command-line options using the enhanced getopt.
593#
594# Note: the flag options are passed around in the global __flags_opts so that
595# the formatting is not lost due to shell parsing and such.
596#
597# Args:
598# @: varies: command-line options to parse
599# Returns:
600# integer: a FLAGS success condition
kate.wardc5210682009-03-30 18:54:36 +0000601_flags_getoptEnhanced()
kate.ward60c25982008-11-12 20:12:44 +0000602{
603 flags_return=${FLAGS_TRUE}
604 _flags_shortOpts_=`_flags_genOptStr ${__FLAGS_OPTSTR_SHORT}`
605 _flags_boolOpts_=`echo "${__flags_boolNames}" \
kate.wardd9672402008-11-15 16:55:51 +0000606 |sed 's/^ *//;s/ *$//;s/ /,/g'`
kate.ward60c25982008-11-12 20:12:44 +0000607 _flags_longOpts_=`_flags_genOptStr ${__FLAGS_OPTSTR_LONG}`
608
609 __flags_opts=`getopt \
610 -o ${_flags_shortOpts_} \
611 -l "${_flags_longOpts_},${_flags_boolOpts_}" \
612 -- "$@" 2>&1`
kate.ward7a3c9c42009-04-24 00:12:35 +0000613 _flags_rtrn_=$?
614 if [ ${_flags_rtrn_} -ne ${FLAGS_TRUE} ]; then
615 _flags_warn "${__flags_opts}"
kate.ward60c25982008-11-12 20:12:44 +0000616 flags_error='unable to parse provided options with getopt.'
617 flags_return=${FLAGS_ERROR}
618 fi
619
kate.ward7a3c9c42009-04-24 00:12:35 +0000620 unset _flags_boolOpts_ _flags_longOpts_ _flags_rtrn_ _flags_shortOpts_
kate.ward60c25982008-11-12 20:12:44 +0000621 return ${flags_return}
622}
623
kate.ward18d456e2008-11-14 00:46:44 +0000624# Dynamically parse a getopt result and set appropriate variables.
625#
626# This function does the actual conversion of getopt output and runs it through
627# the standard case structure for parsing. The case structure is actually quite
628# dynamic to support any number of flags.
629#
630# Args:
kate.ward9ccba512009-04-01 13:31:18 +0000631# argc: int: original command-line argument count
kate.ward18d456e2008-11-14 00:46:44 +0000632# @: varies: output from getopt parsing
633# Returns:
634# integer: a FLAGS success condition
kate.ward60c25982008-11-12 20:12:44 +0000635_flags_parseGetopt()
636{
kate.ward9ccba512009-04-01 13:31:18 +0000637 _flags_argc_=$1
638 shift
639
kate.ward60c25982008-11-12 20:12:44 +0000640 flags_return=${FLAGS_TRUE}
641
642 if [ ${__FLAGS_GETOPT_VERS} -ne ${__FLAGS_GETOPT_VERS_ENH} ]; then
643 set -- $@
644 else
645 # note the quotes around the `$@' -- they are essential!
646 eval set -- "$@"
647 fi
648
kate.wardc5210682009-03-30 18:54:36 +0000649 # provide user with number of arguments to shift by later
650 # NOTE: the FLAGS_ARGC variable is obsolete as of 1.0.3 because it does not
651 # properly give user access to non-flag arguments mixed in between flag
652 # arguments. Its usage was replaced by FLAGS_ARGV, and it is being kept only
653 # for backwards compatibility reasons.
kate.ward9ccba512009-04-01 13:31:18 +0000654 FLAGS_ARGC=`expr $# - 1 - ${_flags_argc_}`
kate.wardc5210682009-03-30 18:54:36 +0000655
kate.ward60c25982008-11-12 20:12:44 +0000656 # handle options. note options with values must do an additional shift
657 while true; do
658 _flags_opt_=$1
659 _flags_arg_=${2:-}
660 _flags_type_=${__FLAGS_TYPE_NONE}
kate.ward7a3c9c42009-04-24 00:12:35 +0000661 _flags_name_=''
kate.ward60c25982008-11-12 20:12:44 +0000662
663 # determine long flag name
664 case "${_flags_opt_}" in
665 --) shift; break ;; # discontinue option parsing
666
667 --*) # long option
kate.ward9a183b82010-03-18 00:25:34 +0000668 _flags_opt_=`expr -- "${_flags_opt_}" : '--\(.*\)'`
kate.ward60c25982008-11-12 20:12:44 +0000669 _flags_len_=${__FLAGS_LEN_LONG}
670 if _flags_itemInList "${_flags_opt_}" ${__flags_longNames}; then
671 _flags_name_=${_flags_opt_}
672 else
673 # check for negated long boolean version
674 if _flags_itemInList "${_flags_opt_}" ${__flags_boolNames}; then
kate.ward9a183b82010-03-18 00:25:34 +0000675 _flags_name_=`expr -- "${_flags_opt_}" : 'no\(.*\)'`
kate.ward60c25982008-11-12 20:12:44 +0000676 _flags_type_=${__FLAGS_TYPE_BOOLEAN}
677 _flags_arg_=${__FLAGS_NULL}
678 fi
679 fi
680 ;;
681
682 -*) # short option
kate.ward9a183b82010-03-18 00:25:34 +0000683 _flags_opt_=`expr -- "${_flags_opt_}" : '-\(.*\)'`
kate.ward60c25982008-11-12 20:12:44 +0000684 _flags_len_=${__FLAGS_LEN_SHORT}
685 if _flags_itemInList "${_flags_opt_}" ${__flags_shortNames}; then
686 # yes. match short name to long name. note purposeful off-by-one
687 # (too high) with awk calculations.
688 _flags_pos_=`echo "${__flags_shortNames}" \
689 |awk 'BEGIN{RS=" ";rn=0}$0==e{rn=NR}END{print rn}' \
690 e=${_flags_opt_}`
691 _flags_name_=`echo "${__flags_longNames}" \
692 |awk 'BEGIN{RS=" "}rn==NR{print $0}' rn="${_flags_pos_}"`
693 fi
694 ;;
695 esac
696
697 # die if the flag was unrecognized
698 if [ -z "${_flags_name_}" ]; then
699 flags_error="unrecognized option (${_flags_opt_})"
700 flags_return=${FLAGS_ERROR}
701 break
702 fi
703
kate.ward60c25982008-11-12 20:12:44 +0000704 # set new flag value
705 [ ${_flags_type_} -eq ${__FLAGS_TYPE_NONE} ] && \
kate.wardd9672402008-11-15 16:55:51 +0000706 _flags_type_=`_flags_getFlagInfo \
kate.wardfb0e7182009-05-10 16:59:13 +0000707 "${_flags_name_}" ${__FLAGS_INFO_TYPE}`
kate.ward60c25982008-11-12 20:12:44 +0000708 case ${_flags_type_} in
709 ${__FLAGS_TYPE_BOOLEAN})
710 if [ ${_flags_len_} -eq ${__FLAGS_LEN_LONG} ]; then
711 if [ "${_flags_arg_}" != "${__FLAGS_NULL}" ]; then
712 eval "FLAGS_${_flags_name_}=${FLAGS_TRUE}"
713 else
714 eval "FLAGS_${_flags_name_}=${FLAGS_FALSE}"
715 fi
716 else
717 _flags_strToEval_="_flags_val_=\
kate.wardfb0e7182009-05-10 16:59:13 +0000718\${__flags_${_flags_name_}_${__FLAGS_INFO_DEFAULT}}"
kate.ward60c25982008-11-12 20:12:44 +0000719 eval "${_flags_strToEval_}"
720 if [ ${_flags_val_} -eq ${FLAGS_FALSE} ]; then
721 eval "FLAGS_${_flags_name_}=${FLAGS_TRUE}"
722 else
723 eval "FLAGS_${_flags_name_}=${FLAGS_FALSE}"
724 fi
725 fi
726 ;;
727
728 ${__FLAGS_TYPE_FLOAT})
729 if _flags_validateFloat "${_flags_arg_}"; then
730 eval "FLAGS_${_flags_name_}='${_flags_arg_}'"
731 else
732 flags_error="invalid float value (${_flags_arg_})"
733 flags_return=${FLAGS_ERROR}
734 break
735 fi
736 ;;
737
738 ${__FLAGS_TYPE_INTEGER})
739 if _flags_validateInteger "${_flags_arg_}"; then
740 eval "FLAGS_${_flags_name_}='${_flags_arg_}'"
741 else
742 flags_error="invalid integer value (${_flags_arg_})"
743 flags_return=${FLAGS_ERROR}
744 break
745 fi
746 ;;
747
748 ${__FLAGS_TYPE_STRING})
749 eval "FLAGS_${_flags_name_}='${_flags_arg_}'"
750 ;;
751 esac
752
kate.ward5b48fc12008-11-13 00:42:43 +0000753 # handle special case help flag
754 if [ "${_flags_name_}" = 'help' ]; then
755 if [ ${FLAGS_help} -eq ${FLAGS_TRUE} ]; then
756 flags_help
757 flags_error='help requested'
758 flags_return=${FLAGS_FALSE}
759 break
760 fi
761 fi
762
kate.wardc5210682009-03-30 18:54:36 +0000763 # shift the option and non-boolean arguements out.
kate.ward60c25982008-11-12 20:12:44 +0000764 shift
kate.wardc5210682009-03-30 18:54:36 +0000765 [ ${_flags_type_} != ${__FLAGS_TYPE_BOOLEAN} ] && shift
766 done
kate.ward60c25982008-11-12 20:12:44 +0000767
kate.wardc5210682009-03-30 18:54:36 +0000768 # give user back non-flag arguments
769 FLAGS_ARGV=''
770 while [ $# -gt 0 ]; do
771 FLAGS_ARGV="${FLAGS_ARGV:+${FLAGS_ARGV} }'$1'"
772 shift
kate.ward60c25982008-11-12 20:12:44 +0000773 done
774
775 unset _flags_arg_ _flags_len_ _flags_name_ _flags_opt_ _flags_pos_ \
776 _flags_strToEval_ _flags_type_ _flags_val_
777 return ${flags_return}
778}
779
kate.wardf51c6162008-06-17 16:38:35 +0000780#------------------------------------------------------------------------------
781# public functions
782#
783
784# A basic boolean flag. Boolean flags do not take any arguments, and their
785# value is either 1 (false) or 0 (true). For long flags, the false value is
786# specified on the command line by prepending the word 'no'. With short flags,
787# the presense of the flag toggles the current value between true and false.
788# Specifying a short boolean flag twice on the command results in returning the
789# value back to the default value.
790#
791# A default value is required for boolean flags.
792#
793# For example, lets say a Boolean flag was created whose long name was 'update'
794# and whose short name was 'x', and the default value was 'false'. This flag
795# could be explicitly set to 'true' with '--update' or by '-x', and it could be
796# explicitly set to 'false' with '--noupdate'.
797DEFINE_boolean() { _flags_define ${__FLAGS_TYPE_BOOLEAN} "$@"; }
798
799# Other basic flags.
800DEFINE_float() { _flags_define ${__FLAGS_TYPE_FLOAT} "$@"; }
801DEFINE_integer() { _flags_define ${__FLAGS_TYPE_INTEGER} "$@"; }
802DEFINE_string() { _flags_define ${__FLAGS_TYPE_STRING} "$@"; }
803
804# Parse the flags.
805#
806# Args:
807# unnamed: list: command-line flags to parse
808# Returns:
809# integer: success of operation, or error
810FLAGS()
811{
kate.wardabae05d2008-11-12 22:37:05 +0000812 # define a standard 'help' flag if one isn't already defined
813 [ -z "${__flags_help_type:-}" ] && \
814 DEFINE_boolean 'help' false 'show this help' 'h'
kate.wardf08c5b62008-07-11 20:32:11 +0000815
kate.wardf51c6162008-06-17 16:38:35 +0000816 # parse options
kate.ward7a3c9c42009-04-24 00:12:35 +0000817 if [ $# -gt 0 ]; then
818 if [ ${__FLAGS_GETOPT_VERS} -ne ${__FLAGS_GETOPT_VERS_ENH} ]; then
819 _flags_getoptStandard "$@"
820 else
821 _flags_getoptEnhanced "$@"
822 fi
823 flags_return=$?
kate.wardf51c6162008-06-17 16:38:35 +0000824 else
kate.ward7a3c9c42009-04-24 00:12:35 +0000825 # nothing passed; won't bother running getopt
826 __flags_opts='--'
827 flags_return=${FLAGS_TRUE}
kate.wardf51c6162008-06-17 16:38:35 +0000828 fi
kate.ward60c25982008-11-12 20:12:44 +0000829
830 if [ ${flags_return} -eq ${FLAGS_TRUE} ]; then
kate.ward7a3c9c42009-04-24 00:12:35 +0000831 _flags_parseGetopt $# "${__flags_opts}"
kate.ward60c25982008-11-12 20:12:44 +0000832 flags_return=$?
kate.wardf51c6162008-06-17 16:38:35 +0000833 fi
834
kate.wardf51c6162008-06-17 16:38:35 +0000835 [ ${flags_return} -eq ${FLAGS_ERROR} ] && _flags_fatal "${flags_error}"
836 return ${flags_return}
837}
838
839# This is a helper function for determining the `getopt` version for platforms
840# where the detection isn't working. It simply outputs debug information that
841# can be included in a bug report.
842#
843# Args:
844# none
kate.ward60c25982008-11-12 20:12:44 +0000845# Output:
846# debug info that can be included in a bug report
kate.wardf51c6162008-06-17 16:38:35 +0000847# Returns:
848# nothing
849flags_getoptInfo()
850{
851 # platform info
852 _flags_debug "uname -a: `uname -a`"
853 _flags_debug "PATH: ${PATH}"
854
855 # shell info
856 if [ -n "${BASH_VERSION:-}" ]; then
857 _flags_debug 'shell: bash'
858 _flags_debug "BASH_VERSION: ${BASH_VERSION}"
859 elif [ -n "${ZSH_VERSION:-}" ]; then
860 _flags_debug 'shell: zsh'
861 _flags_debug "ZSH_VERSION: ${ZSH_VERSION}"
862 fi
863
864 # getopt info
865 getopt >/dev/null
kate.ward60c25982008-11-12 20:12:44 +0000866 _flags_getoptReturn=$?
867 _flags_debug "getopt return: ${_flags_getoptReturn}"
kate.wardf51c6162008-06-17 16:38:35 +0000868 _flags_debug "getopt --version: `getopt --version 2>&1`"
869
kate.ward60c25982008-11-12 20:12:44 +0000870 unset _flags_getoptReturn
kate.wardf51c6162008-06-17 16:38:35 +0000871}
872
kate.ward437639d2008-10-19 17:21:41 +0000873# Returns whether the detected getopt version is the enhanced version.
kate.wardf51c6162008-06-17 16:38:35 +0000874#
875# Args:
876# none
kate.ward60c25982008-11-12 20:12:44 +0000877# Output:
878# none
kate.wardf51c6162008-06-17 16:38:35 +0000879# Returns:
880# bool: true if getopt is the enhanced version
881flags_getoptIsEnh()
882{
kate.wardd9672402008-11-15 16:55:51 +0000883 test ${__FLAGS_GETOPT_VERS} -eq ${__FLAGS_GETOPT_VERS_ENH}
kate.wardf51c6162008-06-17 16:38:35 +0000884}
885
kate.ward437639d2008-10-19 17:21:41 +0000886# Returns whether the detected getopt version is the standard version.
kate.wardf51c6162008-06-17 16:38:35 +0000887#
888# Args:
889# none
890# Returns:
891# bool: true if getopt is the standard version
892flags_getoptIsStd()
893{
kate.wardd9672402008-11-15 16:55:51 +0000894 test ${__FLAGS_GETOPT_VERS} -eq ${__FLAGS_GETOPT_VERS_STD}
kate.wardf51c6162008-06-17 16:38:35 +0000895}
896
897# This is effectively a 'usage()' function. It prints usage information and
898# exits the program with ${FLAGS_FALSE} if it is ever found in the command line
899# arguments. Note this function can be overridden so other apps can define
900# their own --help flag, replacing this one, if they want.
901#
902# Args:
903# none
904# Returns:
905# integer: success of operation (always returns true)
906flags_help()
907{
908 if [ -n "${FLAGS_HELP:-}" ]; then
909 echo "${FLAGS_HELP}" >&2
910 else
kate.ward4d8e0472008-07-18 11:52:56 +0000911 echo "USAGE: ${FLAGS_PARENT:-$0} [flags] args" >&2
kate.wardf51c6162008-06-17 16:38:35 +0000912 fi
913 if [ -n "${__flags_longNames}" ]; then
914 echo 'flags:' >&2
kate.wardd9672402008-11-15 16:55:51 +0000915 for flags_name_ in ${__flags_longNames}; do
916 flags_flagStr_=''
kate.wardbda39172009-04-01 03:30:22 +0000917 flags_boolStr_=''
kate.wardf51c6162008-06-17 16:38:35 +0000918
kate.wardd9672402008-11-15 16:55:51 +0000919 flags_default_=`_flags_getFlagInfo \
kate.wardfb0e7182009-05-10 16:59:13 +0000920 "${flags_name_}" ${__FLAGS_INFO_DEFAULT}`
kate.wardd9672402008-11-15 16:55:51 +0000921 flags_help_=`_flags_getFlagInfo \
kate.wardfb0e7182009-05-10 16:59:13 +0000922 "${flags_name_}" ${__FLAGS_INFO_HELP}`
kate.wardd9672402008-11-15 16:55:51 +0000923 flags_short_=`_flags_getFlagInfo \
kate.wardfb0e7182009-05-10 16:59:13 +0000924 "${flags_name_}" ${__FLAGS_INFO_SHORT}`
kate.wardd9672402008-11-15 16:55:51 +0000925 flags_type_=`_flags_getFlagInfo \
kate.wardfb0e7182009-05-10 16:59:13 +0000926 "${flags_name_}" ${__FLAGS_INFO_TYPE}`
kate.wardf51c6162008-06-17 16:38:35 +0000927
kate.wardbda39172009-04-01 03:30:22 +0000928 [ "${flags_short_}" != "${__FLAGS_NULL}" ] \
929 && flags_flagStr_="-${flags_short_}"
kate.ward437639d2008-10-19 17:21:41 +0000930
931 if [ ${__FLAGS_GETOPT_VERS} -eq ${__FLAGS_GETOPT_VERS_ENH} ]; then
kate.wardbda39172009-04-01 03:30:22 +0000932 [ "${flags_short_}" != "${__FLAGS_NULL}" ] \
933 && flags_flagStr_="${flags_flagStr_},"
934 [ ${flags_type_} -eq ${__FLAGS_TYPE_BOOLEAN} ] \
935 && flags_boolStr_='[no]'
kate.wardd9672402008-11-15 16:55:51 +0000936 flags_flagStr_="${flags_flagStr_}--${flags_boolStr_}${flags_name_}:"
kate.ward437639d2008-10-19 17:21:41 +0000937 fi
938
kate.wardd9672402008-11-15 16:55:51 +0000939 case ${flags_type_} in
kate.ward437639d2008-10-19 17:21:41 +0000940 ${__FLAGS_TYPE_BOOLEAN})
kate.wardd9672402008-11-15 16:55:51 +0000941 if [ ${flags_default_} -eq ${FLAGS_TRUE} ]; then
942 flags_defaultStr_='true'
kate.ward437639d2008-10-19 17:21:41 +0000943 else
kate.wardd9672402008-11-15 16:55:51 +0000944 flags_defaultStr_='false'
kate.ward437639d2008-10-19 17:21:41 +0000945 fi
946 ;;
947 ${__FLAGS_TYPE_FLOAT}|${__FLAGS_TYPE_INTEGER})
kate.wardd9672402008-11-15 16:55:51 +0000948 flags_defaultStr_=${flags_default_} ;;
949 ${__FLAGS_TYPE_STRING}) flags_defaultStr_="'${flags_default_}'" ;;
kate.ward437639d2008-10-19 17:21:41 +0000950 esac
kate.wardd9672402008-11-15 16:55:51 +0000951 flags_defaultStr_="(default: ${flags_defaultStr_})"
kate.ward437639d2008-10-19 17:21:41 +0000952
kate.wardd9672402008-11-15 16:55:51 +0000953 flags_helpStr_=" ${flags_flagStr_} ${flags_help_} ${flags_defaultStr_}"
kate.ward9a183b82010-03-18 00:25:34 +0000954 flags_helpStrLen_=`expr -- "${flags_helpStr_}" : '.*'`
kate.ward14b33bf2008-11-15 20:03:18 +0000955 flags_columns_=`_flags_columns`
956 if [ ${flags_helpStrLen_} -lt ${flags_columns_} ]; then
kate.wardd9672402008-11-15 16:55:51 +0000957 echo "${flags_helpStr_}" >&2
kate.ward437639d2008-10-19 17:21:41 +0000958 else
kate.ward64099612008-11-17 17:36:14 +0000959 echo " ${flags_flagStr_} ${flags_help_}" >&2
kate.ward87b0c2e2009-04-01 03:48:16 +0000960 # note: the silliness with the x's is purely for ksh93 on Ubuntu 6.06
961 # because it doesn't like empty strings when used in this manner.
kate.wardbda39172009-04-01 03:30:22 +0000962 flags_emptyStr_="`echo \"x${flags_flagStr_}x\" \
963 |awk '{printf "%"length($0)-2"s", ""}'`"
kate.ward64099612008-11-17 17:36:14 +0000964 flags_helpStr_=" ${flags_emptyStr_} ${flags_defaultStr_}"
kate.ward9a183b82010-03-18 00:25:34 +0000965 flags_helpStrLen_=`expr -- "${flags_helpStr_}" : '.*'`
kate.ward64099612008-11-17 17:36:14 +0000966 if [ ${__FLAGS_GETOPT_VERS} -eq ${__FLAGS_GETOPT_VERS_STD} \
967 -o ${flags_helpStrLen_} -lt ${flags_columns_} ]; then
968 # indented to match help string
969 echo "${flags_helpStr_}" >&2
970 else
971 # indented four from left to allow for longer defaults as long flag
972 # names might be used too, making things too long
973 echo " ${flags_defaultStr_}" >&2
974 fi
kate.ward437639d2008-10-19 17:21:41 +0000975 fi
kate.wardf51c6162008-06-17 16:38:35 +0000976 done
977 fi
978
kate.wardd9672402008-11-15 16:55:51 +0000979 unset flags_boolStr_ flags_default_ flags_defaultStr_ flags_emptyStr_ \
980 flags_flagStr_ flags_help_ flags_helpStr flags_helpStrLen flags_name_ \
kate.ward14b33bf2008-11-15 20:03:18 +0000981 flags_columns_ flags_short_ flags_type_
kate.wardf51c6162008-06-17 16:38:35 +0000982 return ${FLAGS_TRUE}
983}
984
kate.wardcecd7bd2009-05-10 17:56:13 +0000985# Reset shflags back to an uninitialized state.
kate.wardf51c6162008-06-17 16:38:35 +0000986#
987# Args:
988# none
989# Returns:
990# nothing
991flags_reset()
992{
kate.wardd9672402008-11-15 16:55:51 +0000993 for flags_name_ in ${__flags_longNames}; do
994 flags_strToEval_="unset FLAGS_${flags_name_}"
995 for flags_type_ in \
kate.wardcecd7bd2009-05-10 17:56:13 +0000996 ${__FLAGS_INFO_DEFAULT} \
kate.wardfb0e7182009-05-10 16:59:13 +0000997 ${__FLAGS_INFO_HELP} \
998 ${__FLAGS_INFO_SHORT} \
999 ${__FLAGS_INFO_TYPE}
kate.wardf51c6162008-06-17 16:38:35 +00001000 do
kate.wardd9672402008-11-15 16:55:51 +00001001 flags_strToEval_=\
1002"${flags_strToEval_} __flags_${flags_name_}_${flags_type_}"
kate.wardf51c6162008-06-17 16:38:35 +00001003 done
kate.wardd9672402008-11-15 16:55:51 +00001004 eval ${flags_strToEval_}
kate.wardf51c6162008-06-17 16:38:35 +00001005 done
1006
1007 # reset internal variables
1008 __flags_boolNames=' '
1009 __flags_longNames=' '
1010 __flags_shortNames=' '
1011
kate.wardd9672402008-11-15 16:55:51 +00001012 unset flags_name_ flags_type_ flags_strToEval_
kate.wardf51c6162008-06-17 16:38:35 +00001013}