Chris@0
|
1 <?php
|
Chris@0
|
2
|
Chris@0
|
3 /*
|
Chris@0
|
4 * This file is part of the Symfony package.
|
Chris@0
|
5 *
|
Chris@0
|
6 * (c) Fabien Potencier <fabien@symfony.com>
|
Chris@0
|
7 *
|
Chris@0
|
8 * For the full copyright and license information, please view the LICENSE
|
Chris@0
|
9 * file that was distributed with this source code.
|
Chris@0
|
10 */
|
Chris@0
|
11
|
Chris@0
|
12 namespace Symfony\Component\Console\Input;
|
Chris@0
|
13
|
Chris@0
|
14 use Symfony\Component\Console\Exception\RuntimeException;
|
Chris@0
|
15
|
Chris@0
|
16 /**
|
Chris@0
|
17 * ArgvInput represents an input coming from the CLI arguments.
|
Chris@0
|
18 *
|
Chris@0
|
19 * Usage:
|
Chris@0
|
20 *
|
Chris@0
|
21 * $input = new ArgvInput();
|
Chris@0
|
22 *
|
Chris@0
|
23 * By default, the `$_SERVER['argv']` array is used for the input values.
|
Chris@0
|
24 *
|
Chris@0
|
25 * This can be overridden by explicitly passing the input values in the constructor:
|
Chris@0
|
26 *
|
Chris@0
|
27 * $input = new ArgvInput($_SERVER['argv']);
|
Chris@0
|
28 *
|
Chris@0
|
29 * If you pass it yourself, don't forget that the first element of the array
|
Chris@0
|
30 * is the name of the running application.
|
Chris@0
|
31 *
|
Chris@0
|
32 * When passing an argument to the constructor, be sure that it respects
|
Chris@0
|
33 * the same rules as the argv one. It's almost always better to use the
|
Chris@0
|
34 * `StringInput` when you want to provide your own input.
|
Chris@0
|
35 *
|
Chris@0
|
36 * @author Fabien Potencier <fabien@symfony.com>
|
Chris@0
|
37 *
|
Chris@0
|
38 * @see http://www.gnu.org/software/libc/manual/html_node/Argument-Syntax.html
|
Chris@0
|
39 * @see http://www.opengroup.org/onlinepubs/009695399/basedefs/xbd_chap12.html#tag_12_02
|
Chris@0
|
40 */
|
Chris@0
|
41 class ArgvInput extends Input
|
Chris@0
|
42 {
|
Chris@0
|
43 private $tokens;
|
Chris@0
|
44 private $parsed;
|
Chris@0
|
45
|
Chris@0
|
46 /**
|
Chris@0
|
47 * Constructor.
|
Chris@0
|
48 *
|
Chris@0
|
49 * @param array|null $argv An array of parameters from the CLI (in the argv format)
|
Chris@0
|
50 * @param InputDefinition|null $definition A InputDefinition instance
|
Chris@0
|
51 */
|
Chris@0
|
52 public function __construct(array $argv = null, InputDefinition $definition = null)
|
Chris@0
|
53 {
|
Chris@0
|
54 if (null === $argv) {
|
Chris@0
|
55 $argv = $_SERVER['argv'];
|
Chris@0
|
56 }
|
Chris@0
|
57
|
Chris@0
|
58 // strip the application name
|
Chris@0
|
59 array_shift($argv);
|
Chris@0
|
60
|
Chris@0
|
61 $this->tokens = $argv;
|
Chris@0
|
62
|
Chris@0
|
63 parent::__construct($definition);
|
Chris@0
|
64 }
|
Chris@0
|
65
|
Chris@0
|
66 protected function setTokens(array $tokens)
|
Chris@0
|
67 {
|
Chris@0
|
68 $this->tokens = $tokens;
|
Chris@0
|
69 }
|
Chris@0
|
70
|
Chris@0
|
71 /**
|
Chris@0
|
72 * {@inheritdoc}
|
Chris@0
|
73 */
|
Chris@0
|
74 protected function parse()
|
Chris@0
|
75 {
|
Chris@0
|
76 $parseOptions = true;
|
Chris@0
|
77 $this->parsed = $this->tokens;
|
Chris@0
|
78 while (null !== $token = array_shift($this->parsed)) {
|
Chris@0
|
79 if ($parseOptions && '' == $token) {
|
Chris@0
|
80 $this->parseArgument($token);
|
Chris@0
|
81 } elseif ($parseOptions && '--' == $token) {
|
Chris@0
|
82 $parseOptions = false;
|
Chris@0
|
83 } elseif ($parseOptions && 0 === strpos($token, '--')) {
|
Chris@0
|
84 $this->parseLongOption($token);
|
Chris@0
|
85 } elseif ($parseOptions && '-' === $token[0] && '-' !== $token) {
|
Chris@0
|
86 $this->parseShortOption($token);
|
Chris@0
|
87 } else {
|
Chris@0
|
88 $this->parseArgument($token);
|
Chris@0
|
89 }
|
Chris@0
|
90 }
|
Chris@0
|
91 }
|
Chris@0
|
92
|
Chris@0
|
93 /**
|
Chris@0
|
94 * Parses a short option.
|
Chris@0
|
95 *
|
Chris@0
|
96 * @param string $token The current token
|
Chris@0
|
97 */
|
Chris@0
|
98 private function parseShortOption($token)
|
Chris@0
|
99 {
|
Chris@0
|
100 $name = substr($token, 1);
|
Chris@0
|
101
|
Chris@0
|
102 if (strlen($name) > 1) {
|
Chris@0
|
103 if ($this->definition->hasShortcut($name[0]) && $this->definition->getOptionForShortcut($name[0])->acceptValue()) {
|
Chris@0
|
104 // an option with a value (with no space)
|
Chris@0
|
105 $this->addShortOption($name[0], substr($name, 1));
|
Chris@0
|
106 } else {
|
Chris@0
|
107 $this->parseShortOptionSet($name);
|
Chris@0
|
108 }
|
Chris@0
|
109 } else {
|
Chris@0
|
110 $this->addShortOption($name, null);
|
Chris@0
|
111 }
|
Chris@0
|
112 }
|
Chris@0
|
113
|
Chris@0
|
114 /**
|
Chris@0
|
115 * Parses a short option set.
|
Chris@0
|
116 *
|
Chris@0
|
117 * @param string $name The current token
|
Chris@0
|
118 *
|
Chris@0
|
119 * @throws RuntimeException When option given doesn't exist
|
Chris@0
|
120 */
|
Chris@0
|
121 private function parseShortOptionSet($name)
|
Chris@0
|
122 {
|
Chris@0
|
123 $len = strlen($name);
|
Chris@0
|
124 for ($i = 0; $i < $len; ++$i) {
|
Chris@0
|
125 if (!$this->definition->hasShortcut($name[$i])) {
|
Chris@0
|
126 throw new RuntimeException(sprintf('The "-%s" option does not exist.', $name[$i]));
|
Chris@0
|
127 }
|
Chris@0
|
128
|
Chris@0
|
129 $option = $this->definition->getOptionForShortcut($name[$i]);
|
Chris@0
|
130 if ($option->acceptValue()) {
|
Chris@0
|
131 $this->addLongOption($option->getName(), $i === $len - 1 ? null : substr($name, $i + 1));
|
Chris@0
|
132
|
Chris@0
|
133 break;
|
Chris@0
|
134 } else {
|
Chris@0
|
135 $this->addLongOption($option->getName(), null);
|
Chris@0
|
136 }
|
Chris@0
|
137 }
|
Chris@0
|
138 }
|
Chris@0
|
139
|
Chris@0
|
140 /**
|
Chris@0
|
141 * Parses a long option.
|
Chris@0
|
142 *
|
Chris@0
|
143 * @param string $token The current token
|
Chris@0
|
144 */
|
Chris@0
|
145 private function parseLongOption($token)
|
Chris@0
|
146 {
|
Chris@0
|
147 $name = substr($token, 2);
|
Chris@0
|
148
|
Chris@0
|
149 if (false !== $pos = strpos($name, '=')) {
|
Chris@0
|
150 if (0 === strlen($value = substr($name, $pos + 1))) {
|
Chris@0
|
151 array_unshift($this->parsed, null);
|
Chris@0
|
152 }
|
Chris@0
|
153 $this->addLongOption(substr($name, 0, $pos), $value);
|
Chris@0
|
154 } else {
|
Chris@0
|
155 $this->addLongOption($name, null);
|
Chris@0
|
156 }
|
Chris@0
|
157 }
|
Chris@0
|
158
|
Chris@0
|
159 /**
|
Chris@0
|
160 * Parses an argument.
|
Chris@0
|
161 *
|
Chris@0
|
162 * @param string $token The current token
|
Chris@0
|
163 *
|
Chris@0
|
164 * @throws RuntimeException When too many arguments are given
|
Chris@0
|
165 */
|
Chris@0
|
166 private function parseArgument($token)
|
Chris@0
|
167 {
|
Chris@0
|
168 $c = count($this->arguments);
|
Chris@0
|
169
|
Chris@0
|
170 // if input is expecting another argument, add it
|
Chris@0
|
171 if ($this->definition->hasArgument($c)) {
|
Chris@0
|
172 $arg = $this->definition->getArgument($c);
|
Chris@0
|
173 $this->arguments[$arg->getName()] = $arg->isArray() ? array($token) : $token;
|
Chris@0
|
174
|
Chris@0
|
175 // if last argument isArray(), append token to last argument
|
Chris@0
|
176 } elseif ($this->definition->hasArgument($c - 1) && $this->definition->getArgument($c - 1)->isArray()) {
|
Chris@0
|
177 $arg = $this->definition->getArgument($c - 1);
|
Chris@0
|
178 $this->arguments[$arg->getName()][] = $token;
|
Chris@0
|
179
|
Chris@0
|
180 // unexpected argument
|
Chris@0
|
181 } else {
|
Chris@0
|
182 $all = $this->definition->getArguments();
|
Chris@0
|
183 if (count($all)) {
|
Chris@0
|
184 throw new RuntimeException(sprintf('Too many arguments, expected arguments "%s".', implode('" "', array_keys($all))));
|
Chris@0
|
185 }
|
Chris@0
|
186
|
Chris@0
|
187 throw new RuntimeException(sprintf('No arguments expected, got "%s".', $token));
|
Chris@0
|
188 }
|
Chris@0
|
189 }
|
Chris@0
|
190
|
Chris@0
|
191 /**
|
Chris@0
|
192 * Adds a short option value.
|
Chris@0
|
193 *
|
Chris@0
|
194 * @param string $shortcut The short option key
|
Chris@0
|
195 * @param mixed $value The value for the option
|
Chris@0
|
196 *
|
Chris@0
|
197 * @throws RuntimeException When option given doesn't exist
|
Chris@0
|
198 */
|
Chris@0
|
199 private function addShortOption($shortcut, $value)
|
Chris@0
|
200 {
|
Chris@0
|
201 if (!$this->definition->hasShortcut($shortcut)) {
|
Chris@0
|
202 throw new RuntimeException(sprintf('The "-%s" option does not exist.', $shortcut));
|
Chris@0
|
203 }
|
Chris@0
|
204
|
Chris@0
|
205 $this->addLongOption($this->definition->getOptionForShortcut($shortcut)->getName(), $value);
|
Chris@0
|
206 }
|
Chris@0
|
207
|
Chris@0
|
208 /**
|
Chris@0
|
209 * Adds a long option value.
|
Chris@0
|
210 *
|
Chris@0
|
211 * @param string $name The long option key
|
Chris@0
|
212 * @param mixed $value The value for the option
|
Chris@0
|
213 *
|
Chris@0
|
214 * @throws RuntimeException When option given doesn't exist
|
Chris@0
|
215 */
|
Chris@0
|
216 private function addLongOption($name, $value)
|
Chris@0
|
217 {
|
Chris@0
|
218 if (!$this->definition->hasOption($name)) {
|
Chris@0
|
219 throw new RuntimeException(sprintf('The "--%s" option does not exist.', $name));
|
Chris@0
|
220 }
|
Chris@0
|
221
|
Chris@0
|
222 $option = $this->definition->getOption($name);
|
Chris@0
|
223
|
Chris@0
|
224 // Convert empty values to null
|
Chris@0
|
225 if (!isset($value[0])) {
|
Chris@0
|
226 $value = null;
|
Chris@0
|
227 }
|
Chris@0
|
228
|
Chris@0
|
229 if (null !== $value && !$option->acceptValue()) {
|
Chris@0
|
230 throw new RuntimeException(sprintf('The "--%s" option does not accept a value.', $name));
|
Chris@0
|
231 }
|
Chris@0
|
232
|
Chris@0
|
233 if (null === $value && $option->acceptValue() && count($this->parsed)) {
|
Chris@0
|
234 // if option accepts an optional or mandatory argument
|
Chris@0
|
235 // let's see if there is one provided
|
Chris@0
|
236 $next = array_shift($this->parsed);
|
Chris@0
|
237 if (isset($next[0]) && '-' !== $next[0]) {
|
Chris@0
|
238 $value = $next;
|
Chris@0
|
239 } elseif (empty($next)) {
|
Chris@0
|
240 $value = null;
|
Chris@0
|
241 } else {
|
Chris@0
|
242 array_unshift($this->parsed, $next);
|
Chris@0
|
243 }
|
Chris@0
|
244 }
|
Chris@0
|
245
|
Chris@0
|
246 if (null === $value) {
|
Chris@0
|
247 if ($option->isValueRequired()) {
|
Chris@0
|
248 throw new RuntimeException(sprintf('The "--%s" option requires a value.', $name));
|
Chris@0
|
249 }
|
Chris@0
|
250
|
Chris@0
|
251 if (!$option->isArray()) {
|
Chris@0
|
252 $value = $option->isValueOptional() ? $option->getDefault() : true;
|
Chris@0
|
253 }
|
Chris@0
|
254 }
|
Chris@0
|
255
|
Chris@0
|
256 if ($option->isArray()) {
|
Chris@0
|
257 $this->options[$name][] = $value;
|
Chris@0
|
258 } else {
|
Chris@0
|
259 $this->options[$name] = $value;
|
Chris@0
|
260 }
|
Chris@0
|
261 }
|
Chris@0
|
262
|
Chris@0
|
263 /**
|
Chris@0
|
264 * {@inheritdoc}
|
Chris@0
|
265 */
|
Chris@0
|
266 public function getFirstArgument()
|
Chris@0
|
267 {
|
Chris@0
|
268 foreach ($this->tokens as $token) {
|
Chris@0
|
269 if ($token && '-' === $token[0]) {
|
Chris@0
|
270 continue;
|
Chris@0
|
271 }
|
Chris@0
|
272
|
Chris@0
|
273 return $token;
|
Chris@0
|
274 }
|
Chris@0
|
275 }
|
Chris@0
|
276
|
Chris@0
|
277 /**
|
Chris@0
|
278 * {@inheritdoc}
|
Chris@0
|
279 */
|
Chris@0
|
280 public function hasParameterOption($values, $onlyParams = false)
|
Chris@0
|
281 {
|
Chris@0
|
282 $values = (array) $values;
|
Chris@0
|
283
|
Chris@0
|
284 foreach ($this->tokens as $token) {
|
Chris@0
|
285 if ($onlyParams && $token === '--') {
|
Chris@0
|
286 return false;
|
Chris@0
|
287 }
|
Chris@0
|
288 foreach ($values as $value) {
|
Chris@0
|
289 if ($token === $value || 0 === strpos($token, $value.'=')) {
|
Chris@0
|
290 return true;
|
Chris@0
|
291 }
|
Chris@0
|
292 }
|
Chris@0
|
293 }
|
Chris@0
|
294
|
Chris@0
|
295 return false;
|
Chris@0
|
296 }
|
Chris@0
|
297
|
Chris@0
|
298 /**
|
Chris@0
|
299 * {@inheritdoc}
|
Chris@0
|
300 */
|
Chris@0
|
301 public function getParameterOption($values, $default = false, $onlyParams = false)
|
Chris@0
|
302 {
|
Chris@0
|
303 $values = (array) $values;
|
Chris@0
|
304 $tokens = $this->tokens;
|
Chris@0
|
305
|
Chris@0
|
306 while (0 < count($tokens)) {
|
Chris@0
|
307 $token = array_shift($tokens);
|
Chris@0
|
308 if ($onlyParams && $token === '--') {
|
Chris@0
|
309 return false;
|
Chris@0
|
310 }
|
Chris@0
|
311
|
Chris@0
|
312 foreach ($values as $value) {
|
Chris@0
|
313 if ($token === $value || 0 === strpos($token, $value.'=')) {
|
Chris@0
|
314 if (false !== $pos = strpos($token, '=')) {
|
Chris@0
|
315 return substr($token, $pos + 1);
|
Chris@0
|
316 }
|
Chris@0
|
317
|
Chris@0
|
318 return array_shift($tokens);
|
Chris@0
|
319 }
|
Chris@0
|
320 }
|
Chris@0
|
321 }
|
Chris@0
|
322
|
Chris@0
|
323 return $default;
|
Chris@0
|
324 }
|
Chris@0
|
325
|
Chris@0
|
326 /**
|
Chris@0
|
327 * Returns a stringified representation of the args passed to the command.
|
Chris@0
|
328 *
|
Chris@0
|
329 * @return string
|
Chris@0
|
330 */
|
Chris@0
|
331 public function __toString()
|
Chris@0
|
332 {
|
Chris@0
|
333 $tokens = array_map(function ($token) {
|
Chris@0
|
334 if (preg_match('{^(-[^=]+=)(.+)}', $token, $match)) {
|
Chris@0
|
335 return $match[1].$this->escapeToken($match[2]);
|
Chris@0
|
336 }
|
Chris@0
|
337
|
Chris@0
|
338 if ($token && $token[0] !== '-') {
|
Chris@0
|
339 return $this->escapeToken($token);
|
Chris@0
|
340 }
|
Chris@0
|
341
|
Chris@0
|
342 return $token;
|
Chris@0
|
343 }, $this->tokens);
|
Chris@0
|
344
|
Chris@0
|
345 return implode(' ', $tokens);
|
Chris@0
|
346 }
|
Chris@0
|
347 }
|