Chris@17: <?php
Chris@17: 
Chris@17: namespace Drupal\Component\Plugin\Definition;
Chris@17: 
Chris@17: use Drupal\Component\Plugin\Context\ContextDefinitionInterface;
Chris@17: use Drupal\Component\Plugin\Exception\ContextException;
Chris@17: 
Chris@17: /**
Chris@17:  * Provides a trait for context-aware object-based plugin definitions.
Chris@17:  */
Chris@17: trait ContextAwarePluginDefinitionTrait {
Chris@17: 
Chris@17:   /**
Chris@17:    * The context definitions for this plugin definition.
Chris@17:    *
Chris@17:    * @var \Drupal\Component\Plugin\Context\ContextDefinitionInterface[]
Chris@17:    */
Chris@17:   protected $contextDefinitions = [];
Chris@17: 
Chris@17:   /**
Chris@17:    * Implements \Drupal\Component\Plugin\Definition\ContextAwarePluginDefinitionInterface::hasContextDefinition().
Chris@17:    */
Chris@17:   public function hasContextDefinition($name) {
Chris@17:     return array_key_exists($name, $this->contextDefinitions);
Chris@17:   }
Chris@17: 
Chris@17:   /**
Chris@17:    * Implements \Drupal\Component\Plugin\Definition\ContextAwarePluginDefinitionInterface::getContextDefinitions().
Chris@17:    */
Chris@17:   public function getContextDefinitions() {
Chris@17:     return $this->contextDefinitions;
Chris@17:   }
Chris@17: 
Chris@17:   /**
Chris@17:    * Implements \Drupal\Component\Plugin\Definition\ContextAwarePluginDefinitionInterface::getContextDefinition().
Chris@17:    */
Chris@17:   public function getContextDefinition($name) {
Chris@17:     if ($this->hasContextDefinition($name)) {
Chris@17:       return $this->contextDefinitions[$name];
Chris@17:     }
Chris@17:     throw new ContextException($this->id() . " does not define a '$name' context");
Chris@17:   }
Chris@17: 
Chris@17:   /**
Chris@17:    * Implements \Drupal\Component\Plugin\Definition\ContextAwarePluginDefinitionInterface::addContextDefinition().
Chris@17:    */
Chris@17:   public function addContextDefinition($name, ContextDefinitionInterface $definition) {
Chris@17:     $this->contextDefinitions[$name] = $definition;
Chris@17:     return $this;
Chris@17:   }
Chris@17: 
Chris@17:   /**
Chris@17:    * Implements \Drupal\Component\Plugin\Definition\ContextAwarePluginDefinitionInterface::removeContextDefinition().
Chris@17:    */
Chris@17:   public function removeContextDefinition($name) {
Chris@17:     unset($this->contextDefinitions[$name]);
Chris@17:     return $this;
Chris@17:   }
Chris@17: 
Chris@17: }