暫無描述

HelperSet.php 2.4KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109
  1. <?php
  2. /*
  3. * This file is part of the Symfony package.
  4. *
  5. * (c) Fabien Potencier <fabien@symfony.com>
  6. *
  7. * For the full copyright and license information, please view the LICENSE
  8. * file that was distributed with this source code.
  9. */
  10. namespace Symfony\Component\Console\Helper;
  11. use Symfony\Component\Console\Command\Command;
  12. use Symfony\Component\Console\Exception\InvalidArgumentException;
  13. /**
  14. * HelperSet represents a set of helpers to be used with a command.
  15. *
  16. * @author Fabien Potencier <fabien@symfony.com>
  17. */
  18. class HelperSet implements \IteratorAggregate
  19. {
  20. /**
  21. * @var Helper[]
  22. */
  23. private $helpers = [];
  24. private $command;
  25. /**
  26. * @param Helper[] $helpers An array of helper
  27. */
  28. public function __construct(array $helpers = [])
  29. {
  30. foreach ($helpers as $alias => $helper) {
  31. $this->set($helper, \is_int($alias) ? null : $alias);
  32. }
  33. }
  34. /**
  35. * Sets a helper.
  36. *
  37. * @param string $alias An alias
  38. */
  39. public function set(HelperInterface $helper, $alias = null)
  40. {
  41. $this->helpers[$helper->getName()] = $helper;
  42. if (null !== $alias) {
  43. $this->helpers[$alias] = $helper;
  44. }
  45. $helper->setHelperSet($this);
  46. }
  47. /**
  48. * Returns true if the helper if defined.
  49. *
  50. * @param string $name The helper name
  51. *
  52. * @return bool true if the helper is defined, false otherwise
  53. */
  54. public function has($name)
  55. {
  56. return isset($this->helpers[$name]);
  57. }
  58. /**
  59. * Gets a helper value.
  60. *
  61. * @param string $name The helper name
  62. *
  63. * @return HelperInterface The helper instance
  64. *
  65. * @throws InvalidArgumentException if the helper is not defined
  66. */
  67. public function get($name)
  68. {
  69. if (!$this->has($name)) {
  70. throw new InvalidArgumentException(sprintf('The helper "%s" is not defined.', $name));
  71. }
  72. return $this->helpers[$name];
  73. }
  74. public function setCommand(Command $command = null)
  75. {
  76. $this->command = $command;
  77. }
  78. /**
  79. * Gets the command associated with this helper set.
  80. *
  81. * @return Command A Command instance
  82. */
  83. public function getCommand()
  84. {
  85. return $this->command;
  86. }
  87. /**
  88. * @return \Traversable<Helper>
  89. */
  90. #[\ReturnTypeWillChange]
  91. public function getIterator()
  92. {
  93. return new \ArrayIterator($this->helpers);
  94. }
  95. }