Merge 3e44516095 into a3634c19cf

VKAPI/Handlers/Wall.php

@@ -620,10 +620,6 @@ final class Wall extends VKAPIRequestHandler
             return (object) ["post_id" => $post->getVirtualId()];
         }
 
-        if (\openvk\Web\Util\EventRateLimiter::i()->tryToLimit($this->getUser(), "wall.post", false)) {
-            $this->failTooOften();
-        }
-
         $anon = OPENVK_ROOT_CONF["openvk"]["preferences"]["wall"]["anonymousPosting"]["enable"];
         if ($wallOwner instanceof Club && $from_group == 1 && $signed != 1 && $anon) {
             $manager = $wallOwner->getManager($this->getUser());
@@ -717,6 +713,10 @@ final class Wall extends VKAPIRequestHandler
                 $post->setSuggested(1);
             }
 
+            if (\openvk\Web\Util\EventRateLimiter::i()->tryToLimit($this->getUser(), "wall.post")) {
+                $this->failTooOften();
+            }
+
             $post->save();
         } catch (\LogicException $ex) {
             $this->fail(100, "One of the parameters specified was missing or invalid");
@@ -730,8 +730,6 @@ final class Wall extends VKAPIRequestHandler
             (new WallPostNotification($wallOwner, $post, $this->getUser()))->emit();
         }
 
-        \openvk\Web\Util\EventRateLimiter::i()->writeEvent("wall.post", $this->getUser(), $wallOwner);
-
         return (object) ["post_id" => $post->getVirtualId()];
     }
 

Web/Models/Entities/Traits/TSubscribable.php

@@ -34,7 +34,6 @@ trait TSubscribable
             "target"   => $this->getId(),
         ];
         $sub  = $ctx->table("subscriptions")->where($data);
-
         if (!($sub->fetch())) {
             $ctx->table("subscriptions")->insert($data);
 

Web/Models/Entities/User.php

@@ -114,7 +114,6 @@ class User extends RowModel
 
     public function getChandlerUser(): ChandlerUser
     {
-        # TODO cache this function
         return new ChandlerUser($this->getRecord()->ref("ChandlerUsers", "user"));
     }
 
@@ -1740,41 +1739,51 @@ class User extends RowModel
         return DatabaseConnection::i()->getContext()->table("blacklist_relations")->where("author", $this->getId())->count();
     }
 
-    public function recieveEventsData(array $list): array
+    public function getEventCounters(array $list): array
     {
+        $count_of_keys = sizeof(array_keys($list));
         $ev_str = $this->getRecord()->events_counters;
-        $values = [];
+        $counters = [];
 
         if (!$ev_str) {
             for ($i = 0; $i < sizeof(array_keys($list)); $i++) {
-                $values[] = 0;
+                $counters[] = 0;
             }
         } else {
-            $keys = array_keys($list);
+            $counters = unpack("S" . $count_of_keys, base64_decode($ev_str, true));
-            $values = unpack("S*", base64_decode($ev_str));
         }
 
         return [
-            'counters' => $values,
+            'counters' => array_combine(array_keys($list), $counters),
             'refresh_time' => $this->getRecord()->events_refresh_time,
         ];
     }
 
-    public function stateEvents(array $list): void
+    public function stateEvents(array $state_list): void
     {
-        $this->stateChanges("events_counters", base64_encode(pack("S*", array_values($list))));
+        $pack_str = "";
+
+        foreach ($state_list as $item => $id) {
+            $pack_str .= "S";
+        }
+
+        $this->stateChanges("events_counters", base64_encode(pack($pack_str, ...array_values($state_list))));
+
+        if (!$this->getRecord()->events_refresh_time) {
+            $this->stateChanges("events_refresh_time", time());
+        }
     }
 
-    public function resetEvents(array $list, int $restriction_length)
+    public function resetEvents(array $list): void
     {
         $values = [];
 
-        for ($i = 0; $i < sizeof(array_keys($list)); $i++) {
+        foreach ($list as $key => $val) {
-            $values[] = 0;
+            $values[$key] = 0;
         }
 
         $this->stateEvents($values);
-        $this->stateChanges("events_refresh_time", $restriction_length + time());
+        $this->stateChanges("events_refresh_time", time());
         $this->save();
     }
 }

Web/Presenters/GiftsPresenter.php

@@ -106,7 +106,7 @@ final class GiftsPresenter extends OpenVKPresenter
             return;
         }
 
-        if (\openvk\Web\Util\EventRateLimiter::i()->tryToLimit($this->user->identity, "gifts.send", false)) {
+        if (\openvk\Web\Util\EventRateLimiter::i()->tryToLimit($this->user->identity, "gifts.send")) {
             $this->flashFail("err", tr("error"), tr("limit_exceed_exception"));
         }
 

Web/Presenters/GroupPresenter.php

@@ -63,15 +63,15 @@ final class GroupPresenter extends OpenVKPresenter
 
         if ($_SERVER["REQUEST_METHOD"] === "POST") {
             if (!empty($this->postParam("name")) && mb_strlen(trim($this->postParam("name"))) > 0) {
-                if (\openvk\Web\Util\EventRateLimiter::i()->tryToLimit($this->user->identity, "groups.create")) {
-                    $this->flashFail("err", tr("error"), tr("limit_exceed_exception"));
-                }
-
                 $club = new Club();
                 $club->setName($this->postParam("name"));
                 $club->setAbout(empty($this->postParam("about")) ? null : $this->postParam("about"));
                 $club->setOwner($this->user->id);
 
+                if (\openvk\Web\Util\EventRateLimiter::i()->tryToLimit($this->user->identity, "groups.create")) {
+                    $this->flashFail("err", tr("error"), tr("limit_exceed_exception"));
+                }
+
                 try {
                     $club->save();
                 } catch (\PDOException $ex) {
@@ -108,6 +108,12 @@ final class GroupPresenter extends OpenVKPresenter
             $this->flashFail("err", tr("error"), tr("forbidden"));
         }
 
+        if (!$club->getSubscriptionStatus($this->user->identity)) {
+            if (\openvk\Web\Util\EventRateLimiter::i()->tryToLimit($this->user->identity, "groups.sub")) {
+                $this->flashFail("err", tr("error"), tr("limit_exceed_exception"));
+            }
+        }
+
         $club->toggleSubscription($this->user->identity);
 
         $this->redirect($club->getURL());

Web/Presenters/UserPresenter.php

@@ -418,6 +418,12 @@ final class UserPresenter extends OpenVKPresenter
         if ($this->postParam("act") == "rej") {
             $user->changeFlags($this->user->identity, 0b10000000, true);
         } else {
+            if ($user->getSubscriptionStatus($this->user->identity) == \openvk\Web\Models\Entities\User::SUBSCRIPTION_ABSENT) {
+                if (\openvk\Web\Util\EventRateLimiter::i()->tryToLimit($this->user->identity, "friends.outgoing_sub")) {
+                    $this->flashFail("err", tr("error"), tr("limit_exceed_exception"));
+                }
+            }
+
             $user->toggleSubscription($this->user->identity);
         }
 

Web/Presenters/WallPresenter.php

@@ -356,6 +356,10 @@ final class WallPresenter extends OpenVKPresenter
             $this->flashFail("err", tr("failed_to_publish_post"), tr("post_is_empty_or_too_big"));
         }
 
+        if (\openvk\Web\Util\EventRateLimiter::i()->tryToLimit($this->user->identity, "wall.post")) {
+            $this->flashFail("err", tr("error"), tr("limit_exceed_exception"));
+        }
+
         $should_be_suggested = $wall < 0 && !$wallOwner->canBeModifiedBy($this->user->identity) && $wallOwner->getWallType() == 2;
         try {
             $post = new Post();

Web/Util/EventRateLimiter.php

@@ -13,28 +13,29 @@ class EventRateLimiter
     use TSimpleSingleton;
 
     private $config;
-    private $availableFields;
 
     public function __construct()
     {
         $this->config = OPENVK_ROOT_CONF["openvk"]["preferences"]["security"]["rateLimits"]["eventsLimit"];
-
-        $this->availableFields = array_keys($this->config['list']);
     }
 
-    /* 
+    public function tryToLimit(?User $user, string $event_type, bool $is_update = true): bool
-    Checks count of actions for last hours
-
-    Uses config path OPENVK_ROOT_CONF["openvk"]["preferences"]["security"]["rateLimits"]["eventsLimit"]
-
-    Returns:
-
-    true — limit has exceed and the action must be restricted
-
-    false — the action can be performed
-    */
-    public function tryToLimit(?User $user, string $event_type, bool $distinct = true): bool
     {
+
+        /* 
+        Checks count of actions for last x seconds.
+
+        Uses OPENVK_ROOT_CONF["openvk"]["preferences"]["security"]["rateLimits"]["eventsLimit"]
+
+        This check should be peformed only after checking other conditions cuz by default it increments counter
+
+        Returns:
+
+        true — limit has exceed and the action must be restricted
+
+        false — the action can be performed
+        */
+
         $isEnabled = $this->config['enable'];
         $isIgnoreForAdmins = $this->config['ignoreForAdmins'];
         $restrictionTime = $this->config['restrictionTime'];
@@ -48,75 +49,45 @@ class EventRateLimiter
             return false;
         }
 
+        $eventsStats = $user->getEventCounters($eventsList);
         $limitForThatEvent = $eventsList[$event_type];
-        $stat = $this->getEvent($event_type, $user);
-        bdump($stat);
 
-        $is_restrict_over = $stat["refresh_time"] < time() - $restrictionTime;
+        $counters = $eventsStats["counters"];
+        $refresh_time = $eventsStats["refresh_time"];
+        $is_restrict_over = $refresh_time < (time() - $restrictionTime);
+        $event_counter = $counters[$event_type];
 
-        if ($is_restrict_over) {
+        if ($refresh_time && $is_restrict_over) {
-            $user->resetEvents($eventsList, $restrictionTime);
+            $user->resetEvents($eventsList);
 
             return false;
         }
 
-        $is = $stat["compared"] > $limitForThatEvent;
+        $is_limit_exceed = $event_counter >= $limitForThatEvent;
 
-        if ($is === false) {
+        if (!$is_limit_exceed && $is_update) {
-            $this->incrementEvent($event_type, $user);
+            $this->incrementEvent($counters, $event_type, $user);
         }
 
-        return $is;
+        return $is_limit_exceed;
     }
 
-    public function getEvent(string $event_type, User $by_user): array
+    public function incrementEvent(array $old_values, string $event_type, User $initiator): bool
-    {
-        $ev_data = $by_user->recieveEventsData($this->config['list']);
-        $values  = $ev_data['counters'];
-        $i = 0;
-
-        $compared = [];
-        bdump($values);
-
-        foreach ($this->config['list'] as $name => $value) {
-            bdump($value);
-            $compared[$name] = $values[$i];
-            $i += 1;
-        }
-
-        return [
-            "compared" => $compared,
-            "refresh_time" => $ev_data["refresh_time"]
-        ];
-    }
-
-    /*
-    Updates counter for user
-    */
-    public function incrementEvent(string $event_type, User $initiator): bool
     {
+        /*
+        Updates counter for user
+        */
         $isEnabled = $this->config['enable'];
-        $eventsList = OPENVK_ROOT_CONF["openvk"]["preferences"]["security"]["rateLimits"]["eventsLimit"];
+        $eventsList = $this->config['list'];
 
         if (!$isEnabled) {
             return false;
         }
 
-        $ev_data = $initiator->recieveEventsData($eventsList);
+        $old_values[$event_type] += 1;
-        $values  = $ev_data['counters'];
-        $i = 0;
 
-        $compared = [];
+        $initiator->stateEvents($old_values);
-
+        $initiator->save();
-        foreach ($eventsList as $name => $value) {
-            $compared[$name] = $values[$i];
-            $i += 1;
-        }
-
-        $compared[$event_type] += 1;
-
-        bdump($compared);
-        $initiator->stateEvents($compared);
 
         return true;
     }

openvk-example.yml

@@ -50,7 +50,7 @@ openvk:
                         groups.sub: 50
                         friends.outgoing_sub: 25
                         wall.post: 5000
-                        gifts.send: 20
+                        gifts.send: 30
         blacklists:
             limit: 100
             applyToAdmins: true