/shell/src/main/java/org/jboss/forge/shell/command/parser/OrderedValueVarargsOptionParser.java

https://github.com/includeeasy/core · Java · 75 lines · 45 code · 5 blank · 25 comment · 4 complexity · b5ae420bfc4349a5a15db90afa1b632c MD5 · raw file

  1. /*
  2. * JBoss, by Red Hat.
  3. * Copyright 2010, Red Hat, Inc., and individual contributors
  4. * by the @authors tag. See the copyright.txt in the distribution for a
  5. * full listing of individual contributors.
  6. *
  7. * This is free software; you can redistribute it and/or modify it
  8. * under the terms of the GNU Lesser General Public License as
  9. * published by the Free Software Foundation; either version 2.1 of
  10. * the License, or (at your option) any later version.
  11. *
  12. * This software is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15. * Lesser General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU Lesser General Public
  18. * License along with this software; if not, write to the Free
  19. * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
  20. * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
  21. */
  22. package org.jboss.forge.shell.command.parser;
  23. import java.util.ArrayList;
  24. import java.util.List;
  25. import java.util.Queue;
  26. import org.jboss.forge.parser.java.util.Strings;
  27. import org.jboss.forge.shell.command.CommandMetadata;
  28. import org.jboss.forge.shell.command.OptionMetadata;
  29. /**
  30. * @author <a href="mailto:lincolnbaxter@gmail.com">Lincoln Baxter, III</a>
  31. */
  32. public class OrderedValueVarargsOptionParser implements CommandParser
  33. {
  34. @Override
  35. public CommandParserContext parse(final CommandMetadata command, final Queue<String> tokens,
  36. final CommandParserContext ctx)
  37. {
  38. try
  39. {
  40. OptionMetadata option = command.getOrderedOptionByIndex(ctx.getOrderedParamCount());
  41. if (option.isVarargs())
  42. {
  43. List<String> args = new ArrayList<String>();
  44. String lastToken = null;
  45. // gobble unless we hit a named token
  46. while (!tokens.isEmpty())
  47. {
  48. lastToken = tokens.peek();
  49. if (lastToken.startsWith("-") && command.hasOption(lastToken.replaceAll("^--?", "")))
  50. {
  51. break;
  52. }
  53. lastToken = tokens.remove();
  54. lastToken = Strings.stripQuotes(lastToken);
  55. args.add(lastToken);
  56. }
  57. ctx.put(option, args.toArray(new String[args.size()]), Strings.stripQuotes(lastToken));
  58. ctx.incrementParmCount();
  59. }
  60. }
  61. catch (IllegalArgumentException e)
  62. {
  63. ctx.addWarning("The command [" + command + "] takes ["
  64. + command.getNumOrderedOptions() + "] unnamed argument(s), but found ["
  65. + (ctx.getOrderedParamCount() + 1)
  66. + "].");
  67. }
  68. return ctx;
  69. }
  70. }