ifthenelse-8.x-1.x-dev-no-core/modules/contrib/if_then_else/src/core/Nodes/Conditions/EntityHasBundleCondition/EntityHasBundleCondition.php

<?php

namespace Drupal\if_then_else\core\Nodes\Conditions\EntityHasBundleCondition;

use Drupal\if_then_else\core\Nodes\Conditions\Condition;
use Drupal\if_then_else\Event\NodeSubscriptionEvent;
use Drupal\if_then_else\Event\NodeValidationEvent;

/**
 * Entity Has Bundle condition class.
 */
class EntityHasBundleCondition extends Condition {

  /**
   * {@inheritdoc}
   */
  public static function getName() {
    return 'entity_has_bundle_condition';
  }

  /**
   * {@inheritdoc}
   */
  public function registerNode(NodeSubscriptionEvent $event) {
    // Calling custom service for if then else utilities. To
    // fetch values of entities and bundles.
    $if_then_else_utilities = \Drupal::service('ifthenelse.utilities');
    $form_entity_info = $if_then_else_utilities->getContentEntitiesAndBundles();

    $event->nodes[static::getName()] = [
      'label' => t('Entity Bundle'),
      'type' => 'condition',
      'class' => 'Drupal\\if_then_else\\core\\Nodes\\Conditions\\EntityHasBundleCondition\\EntityHasBundleCondition',
      'library' => 'if_then_else/EntityHasBundleCondition',
      'control_class_name' => 'EntityHasBundleConditionControl',
      'entity_info' => $form_entity_info,
      'inputs' => [
        'entity' => [
          'label' => t('Entity'),
          'description' => t('The entity to check the bundle and type of.'),
          'sockets' => ['object.entity'],
          'required' => TRUE,
        ],
      ],
      'outputs' => [
        'success' => [
          'label' => t('Success'),
          'description' => t('TRUE if the provided entity is of the provided type and bundle.'),
          'socket' => 'bool',
        ],
      ],
    ];
  }

  /**
   * {@inheritdoc}
   */
  public function validateNode(NodeValidationEvent $event) {
    $data = $event->node->data;

    if (!property_exists($data, 'selected_entity') || !property_exists($data, 'selected_bundle')) {
      // Make sure that both selected_entity and selected_bundle are set.
      $event->errors[] = t('Select both entity type and bundle in "@node_name".', ['@node_name' => $event->node->name]);
    }
  }

  /**
   * {@inheritdoc}
   */
  public function process() {
    /** @var \Drupal\Core\Entity\EntityBase $entity */
    $entity = $this->inputs['entity'];

    if (!$entity) {
      \Drupal::logger('if_then_else')->notice(t("Rule @node_name did not run as the instance of the entity could not be found", ['@node_name' => $this->data->name]));
      $this->setSuccess(FALSE);
      return;
    }
    $type = $this->data->selected_entity->value;
    $bundle = $this->data->selected_bundle->value;

    $entity_type = $entity->getEntityTypeId();
    $entity_bundle = $entity->bundle();

    $output = FALSE;
    // To match the entity's bundle and type with the specified values.
    if (($entity_bundle == $bundle) && ($entity_type == $type)) {
      $output = TRUE;
    }

    $this->outputs['success'] = $output;

  }

}