BanIpManager.php

<?php

namespace Drupal\ban;

use Drupal\Core\Database\Connection;

/**
 * Ban IP manager.
 */
class BanIpManager implements BanIpManagerInterface {
    
    /**
     * The database connection used to check the IP against.
     *
     * @var \Drupal\Core\Database\Connection
     */
    protected $connection;
    
    /**
     * Constructs a BanIpManager object.
     *
     * @param \Drupal\Core\Database\Connection $connection
     *   The database connection which will be used to check the IP against.
     */
    public function __construct(Connection $connection) {
        $this->connection = $connection;
    }
    
    /**
     * {@inheritdoc}
     */
    public function isBanned($ip) {
        return (bool) $this->connection
            ->query("SELECT * FROM {ban_ip} WHERE [ip] = :ip", [
            ':ip' => $ip,
        ])
            ->fetchField();
    }
    
    /**
     * {@inheritdoc}
     */
    public function findAll() {
        return $this->connection
            ->query('SELECT * FROM {ban_ip}');
    }
    
    /**
     * {@inheritdoc}
     */
    public function banIp($ip) {
        $this->connection
            ->merge('ban_ip')
            ->key([
            'ip' => $ip,
        ])
            ->fields([
            'ip' => $ip,
        ])
            ->execute();
    }
    
    /**
     * {@inheritdoc}
     */
    public function unbanIp($id) {
        $this->connection
            ->delete('ban_ip')
            ->condition('ip', $id)
            ->execute();
    }
    
    /**
     * {@inheritdoc}
     */
    public function findById($ban_id) {
        return $this->connection
            ->query("SELECT [ip] FROM {ban_ip} WHERE [iid] = :iid", [
            ':iid' => $ban_id,
        ])
            ->fetchField();
    }

}