init

Author: Hanson

.gitignore

@@ -1,5 +1,6 @@
 composer.phar
 /vendor/
+/.idea/
 
 # Commit your application's lock file http://getcomposer.org/doc/01-basic-usage.md#composer-lock-the-lock-file
 # You may choose to ignore a library lock file http://getcomposer.org/doc/02-libraries.md#lock-file

composer.json

@@ -0,0 +1,18 @@
+{
+  "name": "vbot/blacklist",
+  "license": "MIT",
+  "authors": [
+    {
+      "name": "HanSon",
+      "email": "[email protected]"
+    }
+  ],
+  "require-dev": {
+    "hanson/vbot": "^2.0.4"
+  },
+  "autoload": {
+    "psr-4": {
+      "Vbot\\Blacklist\\": "src/"
+    }
+  }
+}

src/Blacklist.php

@@ -0,0 +1,95 @@
+<?php
+
+
+namespace Vbot\Blacklist;
+
+
+use Hanson\Vbot\Extension\AbstractMessageHandler;
+use Illuminate\Support\Collection;
+use Predis\Client;
+use Redis;
+
+class Blacklist extends AbstractMessageHandler
+{
+    public $author = 'HanSon';
+
+    public $version = '1.0';
+
+    public $name = 'blacklist';
+
+    public $zhName = '黑名单';
+
+    /** @var  Redis */
+    private $cache;
+
+    private $frequencyPrefix = 'vbot.blacklist.frequency.';
+
+    private $prefix = 'vbot.blacklist.';
+
+    public $status = true;
+
+    private $configs;
+
+    public function handler(Collection $message)
+    {
+        if (!in_array($message['type'], $this->configs['type'])) {
+            return false;
+        }
+
+        if ($this->cache->get($this->prefix.$message['type'].$message['username'])) {
+            return true;
+        }
+
+        $key = $this->frequencyPrefix.$message['type'].$message['username'];
+        $count = $this->cache->get($key) + 1;
+        $this->cache->set($key,$count);
+        $this->cache->expire($key, 10);
+
+        if ($count == 4) {
+            $this->warn($message);
+            return true;
+        }
+
+        if ($count >= 7) {
+            $this->block($message);
+            return true;
+        }
+
+    }
+
+    private function warn($message)
+    {
+        if (is_callable($callable = $this->configs['warn'])) {
+            call_user_func_array($callable, [$message]);
+        }
+    }
+
+    private function block($message)
+    {
+
+        if ($this->cache->get($this->prefix.$message['type'].$message['username'])) {
+            return true;
+        }
+
+        $this->cache->set($this->prefix.$message['type'].$message['username'], 1);
+
+        if (is_callable($callable = $this->configs['block'])) {
+            call_user_func_array($callable, [$message]);
+        }
+    }
+
+
+    /**
+     * 注册拓展时的操作.
+     */
+    public function register()
+    {
+        $this->configs = vbot('config')->get('extension.'.$this->name);
+        $this->cache = new Client([
+            'host' => vbot('config')->get('database.redis.default.host'),
+            'port' => vbot('config')->get('database.redis.default.port'),
+            'password' => vbot('config')->get('database.redis.default.password'),
+            'database' => vbot('config')->get('database.redis.default.database'),
+        ]);
+    }
+}