rules-8.x-3.x-dev/src/Context/GlobalContextRepositoryTrait.php

<?php

namespace Drupal\rules\Context;

use Drupal\Core\Plugin\Context\ContextRepositoryInterface;

/**
 * Trait for classes that depend on the global context repository.
 */
trait GlobalContextRepositoryTrait {

  /**
   * The global context repository.
   *
   * @var \Drupal\Core\Plugin\Context\ContextRepositoryInterface
   */
  protected $contextRepository;

  /**
   * Sets the global context repository.
   *
   * @param \Drupal\Core\Plugin\Context\ContextRepositoryInterface $context_repository
   *   The global context repository.
   *
   * @return $this
   */
  public function setGlobalContextRepository(ContextRepositoryInterface $context_repository) {
    $this->contextRepository = $context_repository;
    return $this;
  }

  /**
   * Gets the global context repository.
   *
   * @return \Drupal\Core\Plugin\Context\ContextRepositoryInterface
   *   The context repository.
   */
  public function getGlobalContextRepository() {
    if (empty($this->contextRepository)) {
      $this->contextRepository = \Drupal::service('context.repository');
    }
    return $this->contextRepository;
  }

}