Chris@0: <?php
Chris@0: 
Chris@0: namespace Drupal\migrate\Event;
Chris@0: 
Chris@0: use Drupal\migrate\Plugin\MigrationInterface;
Chris@0: use Drupal\migrate\MigrateMessageInterface;
Chris@0: use Symfony\Component\EventDispatcher\Event as SymfonyEvent;
Chris@0: 
Chris@0: class EventBase extends SymfonyEvent {
Chris@0: 
Chris@0:   /**
Chris@0:    * The migration.
Chris@0:    *
Chris@0:    * @var \Drupal\migrate\Plugin\MigrationInterface
Chris@0:    */
Chris@0:   protected $migration;
Chris@0: 
Chris@0:   /**
Chris@0:    * The current message service.
Chris@0:    *
Chris@0:    * @var \Drupal\migrate\MigrateMessageInterface
Chris@0:    */
Chris@0:   protected $message;
Chris@0: 
Chris@0:   /**
Chris@0:    * Constructs a Migrate event object.
Chris@0:    *
Chris@0:    * @param \Drupal\migrate\Plugin\MigrationInterface $migration
Chris@0:    *   The migration being run.
Chris@0:    * @param \Drupal\migrate\MigrateMessageInterface $message
Chris@0:    *   The Migrate message service.
Chris@0:    */
Chris@0:   public function __construct(MigrationInterface $migration, MigrateMessageInterface $message) {
Chris@0:     $this->migration = $migration;
Chris@0:     $this->message = $message;
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * Gets the migration.
Chris@0:    *
Chris@0:    * @return \Drupal\migrate\Plugin\MigrationInterface
Chris@0:    *   The migration being run.
Chris@0:    */
Chris@0:   public function getMigration() {
Chris@0:     return $this->migration;
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * Logs a message using the Migrate message service.
Chris@0:    *
Chris@0:    * @param string $message
Chris@0:    *   The message to log.
Chris@0:    * @param string $type
Chris@0:    *   The type of message, for example: status or warning.
Chris@0:    */
Chris@0:   public function logMessage($message, $type = 'status') {
Chris@0:     $this->message->display($message, $type);
Chris@0:   }
Chris@0: 
Chris@0: }