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