bot group - membership - sync child groups

opensourcewebsite-org · Jun 11, 2024 · 577e3a1 · 577e3a1
1 parent 570bd2a
commit 577e3a1
Show file tree

Hide file tree

Showing 10 changed files with 73 additions and 10 deletions.
diff --git a/messages/ru/bot.php b/messages/ru/bot.php
@@ -240,6 +240,8 @@
     'Child groups' => 'Дочерние группы',
     'Child groups associated with the group' => 'Дочерние группы, связанные с этой группой',
     'Membership and verification dates are automatically copied to active child groups' => 'Даты членства и верификации автоматически копируются в активные дочерние группы',
+    'Membership information has been copied to child groups' => 'Информация о членстве скопирована в дочерние группы',
+    'No child groups' => 'Нет дочерних групп',
     'list of premium members' => 'список премиум участников',
     'premium members with intro' => 'премиум участники с интро',
     'Note' => 'Заметка',
@@ -248,6 +250,7 @@
     'Balance, price' => 'Баланс, цена',
     'Tariff, days' => 'Тариф, дни',
     'Balance, days' => 'Баланс, дни',
+    'No premium members' => 'Нет премиум участников',
     'Select a currency in the group settings' => 'Выберите валюту в настройках группы',
     'FAQ' => 'ЧАВО',
     'Help Center' => 'Центр помощи',

diff --git a/modules/bot/components/helpers/Emoji.php b/modules/bot/components/helpers/Emoji.php
@@ -52,4 +52,5 @@ class Emoji
     public const EYE = '👁‍🗨';
     public const THANKS = '🙏';
     public const SWITCH = '🔀';
+    public const SYNC = '🔂';
 }
diff --git a/modules/bot/config/actions.php b/modules/bot/config/actions.php
@@ -128,4 +128,5 @@
     'pass_captcha',
     'child_groups',
     'set_child_group',
+    'sync_child_groups',
 ];
diff --git a/modules/bot/controllers/groups/PremiumMembersController.php b/modules/bot/controllers/groups/PremiumMembersController.php
@@ -32,7 +32,7 @@ public function actionIndex($page = 1)
         $chat = $this->getTelegramChat();
 
         if ($chat->isMembershipOn()) {
-            $query = $chat->getPremiumChatMembers();
+            $query = $chat->getActivePremiumChatMembers();
 
             $pagination = new Pagination([
                 'totalCount' => $query->count(),

diff --git a/modules/bot/controllers/privates/GroupGuestController.php b/modules/bot/controllers/privates/GroupGuestController.php
@@ -299,7 +299,7 @@ public function actionPremiumMembers($id = null, $page = 1)
 
         $this->getState()->clearInputRoute();
 
-        $query = $chat->getPremiumChatMembers();
+        $query = $chat->getActivePremiumChatMembers();
 
         $pagination = new Pagination([
             'totalCount' => $query->count(),

diff --git a/modules/bot/controllers/privates/GroupMembershipController.php b/modules/bot/controllers/privates/GroupMembershipController.php
@@ -194,13 +194,7 @@ public function actionMembers($id = null, $page = 1)
             'id' => $chat->id,
         ]));
 
-        $query = ChatMember::find()
-            ->where([
-                'chat_id' => $chat->id,
-            ])
-            ->andWhere([
-                'not', ['membership_date' => null],
-            ])
+        $query = $chat->getPremiumChatMembers()
             ->orderBy([
                 'membership_date' => SORT_ASC,
                 'user_id' => SORT_ASC,
@@ -1015,7 +1009,7 @@ public function actionSetMemberTariffDaysBalance($id = null, $oid = null)
     }
 
     /**
-    * @param int|null $id Chat->id
+     * @param int|null $id Chat->id
      * @param int $page
      * @return array
      */
@@ -1078,6 +1072,12 @@ public function actionChildGroups($id = null, $page = 1)
                 ]),
                 'text' => Emoji::BACK,
             ],
+            [
+                'callback_data' => self::createRoute('sync-child-groups', [
+                    'id' => $chat->id,
+                ]),
+                'text' => Emoji::SYNC,
+            ],
             [
                 'callback_data' => MenuController::createRoute(),
                 'text' => Emoji::MENU,
@@ -1133,4 +1133,51 @@ public function actionSetChildGroup($id = null, $cid = null)
              'id' => $chat->id,
          ]);
     }
+
+    /**
+     * @param int|null $id Chat->id
+     * @return array
+     */
+    public function actionSyncChildGroups($id = null)
+    {
+        $chat = Yii::$app->cache->get('chat');
+
+        if (!$premiumChatMembers = $chat->premiumChatMembers) {
+            return $this->getResponseBuilder()
+            ->answerCallbackQuery(
+                $this->render('alert-no-premium-members')
+            )
+            ->build();
+        }
+
+        if (!$childChats = $chat->childGroups) {
+            return $this->getResponseBuilder()
+            ->answerCallbackQuery(
+                $this->render('alert-no-child-groups')
+            )
+            ->build();
+        }
+
+        $updatedCount = 0;
+
+        foreach ($childChats as $childChat) {
+            foreach ($premiumChatMembers as $premiumChatMember) {
+                if ($childChatMember = $childChat->getChatMemberByUserId($premiumChatMember->getUserId())) {
+                    $childChatMember->membership_date = $premiumChatMember->membership_date;
+                    $childChatMember->limiter_date = $premiumChatMember->limiter_date;
+                    $childChatMember->save();
+
+                    $updatedCount++;
+                }
+            }
+        }
+
+        return $this->getResponseBuilder()
+        ->answerCallbackQuery(
+            $this->render('alert-sync-child-groups', [
+                'updatedCount' => $updatedCount,
+                ])
+        )
+        ->build();
+    }
 }
diff --git a/modules/bot/models/Chat.php b/modules/bot/models/Chat.php
@@ -380,6 +380,14 @@ public function getChatMemberCreator()
     }
 
     public function getPremiumChatMembers()
+    {
+        return $this->hasMany(ChatMember::class, ['chat_id' => 'id'])
+            ->andWhere([
+                'not', [ChatMember::tableName() . '.membership_date' => null],
+            ]);
+    }
+
+    public function getActivePremiumChatMembers()
     {
         $today = new DateTime('@' . (time() + ($this->timezone * 60)));
 

diff --git a/modules/bot/views/privates/group-membership/alert-no-child-groups.php b/modules/bot/views/privates/group-membership/alert-no-child-groups.php
@@ -0,0 +1 @@
+<?= Yii::t('bot', 'No child groups');
diff --git a/modules/bot/views/privates/group-membership/alert-no-premium-members.php b/modules/bot/views/privates/group-membership/alert-no-premium-members.php
@@ -0,0 +1 @@
+<?= Yii::t('bot', 'No premium members');
diff --git a/modules/bot/views/privates/group-membership/alert-sync-child-groups.php b/modules/bot/views/privates/group-membership/alert-sync-child-groups.php
@@ -0,0 +1 @@
+<?= Yii::t('bot', 'Membership information has been copied to child groups'); ?>. <?= Yii::t('bot', 'Members') ?>: <?= $updatedCount ?><br/>