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

<?php

namespace Drupal\if_then_else\core\Nodes\Actions\BanIpAddressAction;

use Drupal\if_then_else\core\Nodes\Actions\Action;
use Drupal\if_then_else\Event\GraphValidationEvent;
use Drupal\if_then_else\Event\NodeSubscriptionEvent;

/**
 * Banned IP address action node class.
 */
class BanIpAddressAction extends Action {

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

  /**
   * {@inheritdoc}
   */
  public function registerNode(NodeSubscriptionEvent $event) {
    $event->nodes[static::getName()] = [
      'label' => t('Ban IP Address'),
      'type' => 'action',
      'class' => 'Drupal\\if_then_else\\core\\Nodes\\Actions\\BanIpAddressAction\\BanIpAddressAction',
      'inputs' => [
        'ip_address' => [
          'label' => t('IP address / List of IP address'),
          'description' => t('Can be a comma-separated string of IP or an array of IP.'),
          'sockets' => ['string', 'array'],
          'required' => TRUE,
        ],
      ],
    ];
  }

  /**
   * {@inheritDoc}.
   */
  public function validateGraph(GraphValidationEvent $event) {

    $nodes = $event->data->nodes;
    foreach ($nodes as $node) {
      if ($node->data->type == 'value' && $node->data->name == 'text_value' && property_exists($node->data, 'value')) {
        $ip_addresses = $node->data->value;
        $banned_ip_addresses = [];
        if (is_string($ip_addresses)) {
          foreach (explode(',', $ip_addresses) as $ip_address) {
            if (!empty($ip_address)) {
              $banned_ip_addresses[] = trim($ip_address);
            }
          }
        }
        elseif (is_array($ip_addresses)) {
          array_merge($banned_ip_addresses, $ip_addresses);
        }
        if (empty($banned_ip_addresses)) {
          $event->errors[] = t('Please enter a valid IP to IfThenElse rule');
        }
        else {
          foreach ($banned_ip_addresses as $ip) {
            $ip = trim($ip);
            if (!filter_var($ip, FILTER_VALIDATE_IP)) {
              $event->errors[] = t('IP @ip is not valid IP. Please enter a valid IP to IfThenElse rule', ['@ip' => $ip]);
            }
          }
        }
      }
    }
  }

  /**
   * {@inheritDoc}.
   */
  public function process() {
    // To check ban module is enable or not.
    if (!\Drupal::moduleHandler()->moduleExists('ban')) {
      \Drupal::logger('if_then_else')->notice(t("Ban module is not enabled. Rule @node_name won't execute.", ['@node_name' => $this->data->name]));
      $this->setSuccess(FALSE);
      return;
    }

    $ip_addresses = $this->inputs['ip_address'];

    $banned_ip_addresses = [];

    if (is_string($ip_addresses)) {
      foreach (explode(',', $ip_addresses) as $ip_address) {
        $banned_ip_addresses[] = trim($ip_address);
      }
    }
    elseif (is_array($ip_addresses)) {
      array_merge($banned_ip_addresses, $ip_addresses);
    }

    if (empty($banned_ip_addresses)) {
      $this->setSuccess(FALSE);
      return;
    }

    $ip_manager = \Drupal::service('ban.ip_manager');
    foreach ($banned_ip_addresses as $ip) {
      $ip = trim($ip);
      if (!$ip_manager->isBanned($ip)) {
        $ip_manager->banIp($ip);
      }
    }
  }

}