|
56 | 56 | ## |
57 | 57 |
|
58 | 58 | show_error() { |
59 | | - echo "Error: $1." >&2 |
60 | | - echo "See --help for usage and options." >&2 |
| 59 | + echo "Error: $1." >&2 |
| 60 | + echo "See --help for usage and options." >&2 |
61 | 61 | } |
62 | 62 |
|
63 | 63 | parse_documentation() { |
64 | | - documentation="$(grep "^##" "$(which "$0")")(no-trim)" |
65 | | - documentation=$(echo "$documentation" | sed -r "s/## ?//" | sed -r "s/@script.name/$(basename "$0")/g" | sed "s/@#/@/g") |
66 | | - documentation=${documentation%(no-trim)} |
| 64 | + documentation="$(grep "^##" "$0")" |
| 65 | + documentation=$(echo "$documentation" | sed -r "s/## ?//" | sed -r "s/@script.name/$(basename "$0")/g" | sed "s/@#/@/g") |
| 66 | + documentation=${documentation%(no-trim)} |
67 | 67 | } |
68 | 68 |
|
69 | 69 | parse_options() { |
70 | 70 |
|
71 | | - local short_option_vars |
72 | | - local short_options |
73 | | - local documentation |
74 | | - local next_is_value |
75 | | - local argument |
76 | | - |
77 | | - local option |
78 | | - local option_name |
79 | | - local option_value |
80 | | - local option_var |
81 | | - |
82 | | - local known_option |
83 | | - local known_option_name |
84 | | - local known_option_var |
85 | | - |
86 | | - # Parse known options from documentation |
87 | | - if [[ -z ${options+defined} ]]; then |
88 | | - parse_documentation |
89 | | - while read -r line; do |
90 | | - case "$line" in |
91 | | - "-h, --help"*) continue ;; |
92 | | - "--help, -h"*) continue ;; |
93 | | - -*," "--*) option=$(echo "$line" | awk -F'(^-|, --| )' '{ print $2"="$3 }') ;; |
94 | | - --*," "-*) option=$(echo "$line" | awk -F'(--|, -| )' '{ print $3"="$2 }') ;; |
95 | | - --*=*) option=$(echo "$line" | awk -F'(--|=| )' '{ print $2"=?" }') ;; |
96 | | - --*" "*) option=$(echo "$line" | awk -F'(--| )' '{ print $2 }') ;; |
97 | | - *) continue ;; |
98 | | - esac |
99 | | - options+=("$option") |
100 | | - done <<< "$documentation" |
| 71 | + local short_option_vars |
| 72 | + local short_options |
| 73 | + local documentation |
| 74 | + local next_is_value |
| 75 | + local argument |
| 76 | + |
| 77 | + local option |
| 78 | + local option_name |
| 79 | + local option_value |
| 80 | + local option_var |
| 81 | + |
| 82 | + local known_option |
| 83 | + local known_option_name |
| 84 | + local known_option_var |
| 85 | + |
| 86 | + # Parse known options from documentation |
| 87 | + if [[ -z ${options+defined} ]]; then |
| 88 | + parse_documentation |
| 89 | + while read -r line; do |
| 90 | + case "$line" in |
| 91 | + "-h, --help"*) continue ;; |
| 92 | + "--help, -h"*) continue ;; |
| 93 | + -*," "--*) option=$(echo "$line" | awk -F'(^-|, --| )' '{ print $2"="$3 }') ;; |
| 94 | + --*," "-*) option=$(echo "$line" | awk -F'(--|, -| )' '{ print $3"="$2 }') ;; |
| 95 | + --*=*) option=$(echo "$line" | awk -F'(--|=| )' '{ print $2"=?" }') ;; |
| 96 | + --*" "*) option=$(echo "$line" | awk -F'(--| )' '{ print $2 }') ;; |
| 97 | + *) continue ;; |
| 98 | + esac |
| 99 | + options+=("$option") |
| 100 | + done <<<"$documentation" |
| 101 | + fi |
| 102 | + |
| 103 | + options+=(h=help) |
| 104 | + arguments=() |
| 105 | + |
| 106 | + # Prepare known options |
| 107 | + for option in "${options[@]}"; do |
| 108 | + option_var=${option#*=} |
| 109 | + option_name=${option%="$option_var"} |
| 110 | + if [[ "${#option_name}" = "1" ]]; then |
| 111 | + short_options="${short_options}${option_name}" |
| 112 | + if [[ "${#option_var}" -gt "1" ]]; then |
| 113 | + short_option_vars+=("$option_var") |
| 114 | + fi |
101 | 115 | fi |
102 | | - |
103 | | - options+=(h=help) |
104 | | - arguments=() |
105 | | - |
106 | | - # Prepare known options |
107 | | - for option in "${options[@]}"; do |
108 | | - option_var=${option#*=} |
109 | | - option_name=${option%=$option_var} |
110 | | - if [[ "${#option_name}" = "1" ]]; then |
111 | | - short_options="${short_options}${option_name}" |
112 | | - if [[ "${#option_var}" > "1" ]]; then |
113 | | - short_option_vars+=("$option_var") |
114 | | - fi |
| 116 | + done |
| 117 | + |
| 118 | + # Extract regular arguments |
| 119 | + index=1 |
| 120 | + parameters=() |
| 121 | + for argument in "$@"; do |
| 122 | + if [[ "$argument" = -* ]]; then |
| 123 | + parameters+=("$argument") |
| 124 | + for known_option in "${options[@]}"; do |
| 125 | + known_option_var=${known_option#*=} |
| 126 | + known_option_name=${known_option%="$known_option_var"} |
| 127 | + if [[ "$known_option_var" = "?" && "$argument" = --$known_option_name ]]; then |
| 128 | + next_is_value="yes" |
| 129 | + break |
115 | 130 | fi |
116 | | - done |
117 | | - |
118 | | - # Extract regular arguments |
119 | | - index=1 |
120 | | - parameters=() |
121 | | - for argument in "$@"; do |
122 | | - if [[ "$argument" = -* ]]; then |
123 | | - parameters+=("$argument") |
124 | | - for known_option in "${options[@]}"; do |
125 | | - known_option_var=${known_option#*=} |
126 | | - known_option_name=${known_option%=$known_option_var} |
127 | | - if [[ "$known_option_var" = "?" && "$argument" = --$known_option_name ]]; then |
128 | | - next_is_value="yes" |
129 | | - break |
130 | | - fi |
131 | | - done |
132 | | - else |
133 | | - if [[ -z "$next_is_value" ]]; then |
134 | | - arguments+=("${!index}") |
135 | | - else |
136 | | - parameters+=("$argument") |
137 | | - fi |
138 | | - next_is_value="" |
| 131 | + done |
| 132 | + else |
| 133 | + if [[ -z "$next_is_value" ]]; then |
| 134 | + arguments+=("${!index}") |
| 135 | + else |
| 136 | + parameters+=("$argument") |
| 137 | + fi |
| 138 | + next_is_value="" |
| 139 | + fi |
| 140 | + index=$((index + 1)) |
| 141 | + done |
| 142 | + set -- "${parameters[@]}" |
| 143 | + |
| 144 | + # Parse the provided options |
| 145 | + while getopts ":${short_options}-:" option; do |
| 146 | + option="${option}${OPTARG}" |
| 147 | + option_value="" |
| 148 | + |
| 149 | + # Set the corresponding variable for known options |
| 150 | + for known_option in "${options[@]}" "${short_option_vars[@]}"; do |
| 151 | + known_option_var=${known_option#*=} |
| 152 | + known_option_name=${known_option%="$known_option_var"} |
| 153 | + |
| 154 | + # Short option |
| 155 | + if [[ "$option" = "$known_option_name" ]]; then |
| 156 | + option_value="yes" |
| 157 | + known_option_var=$(echo "$known_option_var" | tr "-" "_") |
| 158 | + eval "$known_option_var=\"$option_value\"" |
| 159 | + break |
| 160 | + |
| 161 | + # Long option |
| 162 | + elif [[ "$option" = -$known_option_name && "$known_option_var" != "?" ]]; then |
| 163 | + option_value="yes" |
| 164 | + known_option_var=$(echo "$known_option_var" | tr "-" "_") |
| 165 | + eval "$known_option_var=\"$option_value\"" |
| 166 | + break |
| 167 | + |
| 168 | + # Long option with value in next parameter |
| 169 | + elif [[ "$option" = -$known_option_name && "$known_option_var" = "?" ]]; then |
| 170 | + eval option_value="\$$OPTIND" |
| 171 | + if [[ -z "$option_value" || "$option_value" = -* ]]; then |
| 172 | + show_error "you must specify a value for --$known_option_name" |
| 173 | + exit 1 |
139 | 174 | fi |
140 | | - index=$((index + 1)) |
| 175 | + OPTIND=$((OPTIND + 1)) |
| 176 | + known_option_var=$(echo "$known_option_name" | tr "-" "_") |
| 177 | + eval "$known_option_var=\"$option_value\"" |
| 178 | + break |
| 179 | + |
| 180 | + # Long option with value after equal sign |
| 181 | + elif [[ "$option" = -$known_option_name=* && "$known_option_var" = "?" ]]; then |
| 182 | + option_value=${option#*=} |
| 183 | + known_option_var=$(echo "$known_option_name" | tr "-" "_") |
| 184 | + eval "$known_option_var=\"$option_value\"" |
| 185 | + break |
| 186 | + |
| 187 | + # Long option with unnecessary value |
| 188 | + elif [[ "$option" = -$known_option_name=* && "$known_option_var" != "?" ]]; then |
| 189 | + option_value=${option#*=} |
| 190 | + show_error "--$known_option_name does not accept a value, you specified \"$option_value\"" |
| 191 | + exit 1 |
| 192 | + fi |
141 | 193 | done |
142 | | - set -- "${parameters[@]}" |
143 | | - |
144 | | - # Parse the provided options |
145 | | - while getopts ":${short_options}-:" option; do |
146 | | - option="${option}${OPTARG}" |
147 | | - option_value="" |
148 | | - |
149 | | - # Set the corresponding variable for known options |
150 | | - for known_option in "${options[@]}" "${short_option_vars[@]}"; do |
151 | | - known_option_var=${known_option#*=} |
152 | | - known_option_name=${known_option%=$known_option_var} |
153 | | - |
154 | | - # Short option |
155 | | - if [[ "$option" = "$known_option_name" ]]; then |
156 | | - option_value="yes" |
157 | | - known_option_var=$(echo "$known_option_var" | tr "-" "_") |
158 | | - eval "$known_option_var=\"$option_value\"" |
159 | | - break |
160 | | - |
161 | | - # Long option |
162 | | - elif [[ "$option" = -$known_option_name && "$known_option_var" != "?" ]]; then |
163 | | - option_value="yes" |
164 | | - known_option_var=$(echo "$known_option_var" | tr "-" "_") |
165 | | - eval "$known_option_var=\"$option_value\"" |
166 | | - break |
167 | | - |
168 | | - # Long option with value in next parameter |
169 | | - elif [[ "$option" = -$known_option_name && "$known_option_var" = "?" ]]; then |
170 | | - eval option_value="\$$OPTIND" |
171 | | - if [[ -z "$option_value" || "$option_value" = -* ]]; then |
172 | | - show_error "you must specify a value for --$known_option_name" |
173 | | - exit 1 |
174 | | - fi |
175 | | - OPTIND=$((OPTIND + 1)) |
176 | | - known_option_var=$(echo "$known_option_name" | tr "-" "_") |
177 | | - eval "$known_option_var=\"$option_value\"" |
178 | | - break |
179 | | - |
180 | | - # Long option with value after equal sign |
181 | | - elif [[ "$option" = -$known_option_name=* && "$known_option_var" = "?" ]]; then |
182 | | - option_value=${option#*=} |
183 | | - known_option_var=$(echo "$known_option_name" | tr "-" "_") |
184 | | - eval "$known_option_var=\"$option_value\"" |
185 | | - break |
186 | | - |
187 | | - # Long option with unnecessary value |
188 | | - elif [[ "$option" = -$known_option_name=* && "$known_option_var" != "?" ]]; then |
189 | | - option_value=${option#*=} |
190 | | - show_error "--$known_option_name does not accept a value, you specified \"$option_value\"" |
191 | | - exit 1 |
192 | | - fi |
193 | | - done |
194 | | - |
195 | | - # Unknown option |
196 | | - if [[ -z "$option_value" ]]; then |
197 | | - option=${option%%=*} |
198 | | - [[ "$option" = \?* ]] && option=${option#*\?} |
199 | | - show_error "unrecognized option -$option" |
200 | | - exit 1 |
201 | | - fi |
202 | 194 |
|
203 | | - # Help option |
204 | | - if [[ -n "$help" ]]; then |
205 | | - [[ -z "$documentation" ]] && parse_documentation |
206 | | - echo "$documentation" |
207 | | - exit |
208 | | - fi |
209 | | - done |
| 195 | + # Unknown option |
| 196 | + if [[ -z "$option_value" ]]; then |
| 197 | + option=${option%%=*} |
| 198 | + [[ "$option" = \?* ]] && option=${option#*\?} |
| 199 | + show_error "unrecognized option -$option" |
| 200 | + exit 1 |
| 201 | + fi |
| 202 | + |
| 203 | + # Help option |
| 204 | + if [[ -n "$help" ]]; then |
| 205 | + [[ -z "$documentation" ]] && parse_documentation |
| 206 | + echo "$documentation" |
| 207 | + exit |
| 208 | + fi |
| 209 | + done |
210 | 210 | } |
211 | 211 |
|
212 | 212 | parse_options "$@" |
0 commit comments