Feat/giveaway blacklist

bot/cogs/giveaway.py

@@ -0,0 +1,176 @@
+import logging
+from datetime import datetime, timedelta, timezone
+from typing import Optional
+
+import discord
+from discord import app_commands, Interaction
+from discord.ext import commands, tasks
+
+from bot.bot import WarnetBot
+from bot.config import BLACKLIST_GA_ROLE_ID, GUILD_ID
+
+logger = logging.getLogger(__name__)
+
+
+@commands.guild_only()
+class Giveaway(commands.GroupCog, group_name='warnet-ga'):
+    def __init__(self, bot: WarnetBot) -> None:
+        self.bot = bot
+        self.db_pool = bot.get_db_pool()
+
+    @commands.Cog.listener()
+    async def on_connect(self) -> None:
+        self._check_blacklist_ga.start()
+
+    @app_commands.command(name='blacklist', description='Blacklist a user from giveaway')
+    @app_commands.describe(
+        big='Yes if the giveaway is large than Rp50.000',
+        winners='Winner of the giveaway. Example: 1234567890,0987654321',
+        ghosts='Member who do not claim the giveaway. Example: 1234567890,0987654321',
+    )
+    async def add_giveaway_blacklist(
+        self,
+        interaction: Interaction,
+        big: bool,
+        winners: str,
+        ghosts: Optional[str] = None,
+    ) -> None:
+        await interaction.response.defer()
+        if not interaction.user.guild_permissions.administrator:
+            return await interaction.followup.send('You are not an admin', ephemeral=True)
+
+        winner_list: list[discord.Member] = []
+        winners = winners.split(',')
+        for winner in winners:
+            if (user := interaction.guild.get_member(int(winner))) is None:
+                return await interaction.followup.send(f'User {winner} not found', ephemeral=True)
+            winner_list.append(user)
+
+        ghost_list: list[discord.Member] = []
+        if ghosts:
+            ghosts = ghosts.split(',')
+            for ghost in ghosts:
+                if (user := interaction.guild.get_member(int(ghost))) is None:
+                    return await interaction.followup.send(
+                        f'User {ghost} not found', ephemeral=True
+                    )
+                ghost_list.append(user)
+
+        if big:
+            winner_day = 30
+            ghosting_day = 15
+        else:
+            winner_day = 15
+            ghosting_day = 7
+
+        now = datetime.now(timezone.utc)
+        end_time_no_streak = now + timedelta(days=winner_day)
+        end_time_streak = now + timedelta(days=winner_day * 2)
+        end_time_ghosting = now + timedelta(days=ghosting_day)
+        blacklist_role = interaction.guild.get_role(BLACKLIST_GA_ROLE_ID)
+
+        async with self.db_pool.acquire() as conn:
+            streak_user_records = await conn.fetch(
+                'SELECT user_id FROM blacklist_ga WHERE status_user = 1',
+            )
+            streak_user = {record['user_id'] for record in streak_user_records}
+
+            for winner in winner_list:
+                await winner.add_roles(blacklist_role)
+                if winner.id in streak_user:
+                    await conn.execute(
+                        '''
+                        UPDATE blacklist_ga 
+                        SET end_time = $2, status_user = $3, cooldown_time = $4
+                        WHERE user_id = $1
+                        ''',
+                        winner.id,
+                        end_time_streak,
+                        0,
+                        end_time_streak,
+                    )
+                else:
+                    await conn.execute(
+                        'INSERT INTO blacklist_ga (user_id, end_time, status_user, cooldown_time) VALUES ($1, $2, $3, $4)',
+                        winner.id,
+                        end_time_no_streak,
+                        1,
+                        end_time_streak,
+                    )
+                logger.info(
+                    f'Added role {blacklist_role.id} to user {winner.id} for {winner_day} days'
+                )
+
+            for ghost in ghost_list:
+                await ghost.add_roles(blacklist_role)
+                if ghost.id in streak_user:
+                    await conn.execute(
+                        'UPDATE blacklist_ga SET end_time = $2 WHERE user_id = $1',
+                        ghost.id,
+                        end_time_ghosting,
+                    )
+                else:
+                    await conn.execute(
+                        'INSERT INTO blacklist_ga (user_id, end_time, status_user) VALUES ($1, $2, $3)',
+                        ghost.id,
+                        end_time_ghosting,
+                        0,
+                    )
+                logger.info(
+                    f'Added role {blacklist_role.id} to user {ghost.id} for {ghosting_day} days'
+                )
+
+        embed = discord.Embed(
+            title="Role Added",
+            description=f"Successfully added the role {blacklist_role.mention} to the {len(winner_list)} winners and {len(ghost_list)} ghosts",
+            color=discord.Color.green(),
+        )
+        embed.add_field(
+            name="List of Winners",
+            value=f"{', '.join([winner.mention for winner in winner_list])}",
+        )
+        embed.add_field(
+            name="List of Ghosts",
+            value=f"{', '.join([ghost.mention for ghost in ghost_list])}",
+        )
+        await interaction.followup.send(embed=embed)
+
+    @tasks.loop(seconds=60)
+    async def _check_blacklist_ga(self) -> None:
+        guild = self.bot.get_guild(GUILD_ID)
+        blacklist_role = guild.get_role(BLACKLIST_GA_ROLE_ID)
+        now = datetime.now(timezone.utc)
+
+        async with self.db_pool.acquire() as conn:
+            user_want_remove = await conn.fetch(
+                'SELECT user_id FROM blacklist_ga WHERE end_time <= $1 AND has_role = TRUE',
+                now,
+            )
+            for user in user_want_remove:
+                if user := guild.get_member(int(user['user_id'])):
+                    await user.remove_roles(blacklist_role)
+                    await conn.execute(
+                        'UPDATE blacklist_ga SET has_role = FALSE WHERE user_id = $1',
+                        user.id,
+                    )
+                    logger.info(f'Removed role {blacklist_role.id} from user {user.id} (rm role)')
+
+            user_want_delete = await conn.fetch(
+                'SELECT user_id FROM blacklist_ga WHERE has_role = FALSE AND status_user = 0 OR cooldown_time <= $1',
+                now,
+            )
+            for user in user_want_delete:
+                if user:
+                    await conn.execute(
+                        'DELETE FROM blacklist_ga WHERE user_id = $1',
+                        int(user['user_id']),
+                    )
+                    logger.info(f'Deleted user {user} (rm row table)')
+
+    @_check_blacklist_ga.before_loop
+    async def _before_check_blacklist_ga(self):
+        await self.bot.wait_until_ready()
+
+
+async def setup(bot: WarnetBot) -> None:
+    await bot.add_cog(Giveaway(bot))

bot/cogs/temporary.py

@@ -7,7 +7,7 @@
 
 from bot.bot import WarnetBot
 from bot.cogs.ext.temprole.time import parse_time_string
-from bot.config import GiveawayConfig, GUILD_ID
+from bot.config import BLACKLIST_GA_ROLE_ID, GUILD_ID
 
 logger = logging.getLogger(__name__)
 
@@ -36,7 +36,7 @@ async def add_temporary_role(
         role: discord.Role,
     ) -> None:
         await interaction.response.defer()
-        if role.id == GiveawayConfig.BLACKLIST_ROLE_ID:
+        if role.id == BLACKLIST_GA_ROLE_ID:
             return await interaction.followup.send(
                 'Cannot add blacklist giveaway role!!', ephemeral=True
             )

bot/config/__init__.py

@@ -33,6 +33,7 @@
 ADMINISTRATOR_ROLE_ID = {'admin': '761650159833841674', 'mod': '761662280541798421'}
 NON_ADMINISTRATOR_ROLE_ID = {'staff': '951170972671701063'}
 BOOSTER_ROLE_ID = 768874803712753727
+BLACKLIST_GA_ROLE_ID = 944559967191576636
 
 ANNOUNCEMENT_CHANNEL_ID = 761680720245686313
 ADMIN_CHANNEL_ID = 761684443915485184
@@ -72,8 +73,3 @@ class TCGConfig:
         TCG_TITLE_ROLE_ID[2]: '<:MasterDuelist:1052440400822018078>',
         TCG_TITLE_ROLE_ID[3]: '<:ImmortalDuelist:1052440404135518228>',
     }
-
-
-class GiveawayConfig:
-    BLACKLIST_ROLE_ID = 944559967191576636
-    BIG_GIVEAWAY = 50000

bot/data/db.sql

@@ -73,4 +73,17 @@ CREATE TABLE temp_role (
     PRIMARY KEY(id),
     UNIQUE(user_id, role_id)
 );
-CREATE INDEX IF NOT EXISTS temp_role_user_role_idx ON temp_role (user_id, role_id);
+CREATE INDEX IF NOT EXISTS temp_role_user_role_idx ON temp_role (user_id, role_id);
+
+--- GIVEAWAY BLACKLIST FEATURE ----
+-----------------------------------
+CREATE TABLE blacklist_ga(
+    user_id BIGINT NOT NULL,
+    end_time TIMESTAMP WITH TIME ZONE NOT NULL,
+	cooldown_time TIMESTAMP WITH TIME ZONE,
+	has_role BOOLEAN DEFAULT TRUE NOT NULL,
+	status_user INT NOT NULL, -- 1: in streak phase, 0: not in streak phase 
+	PRIMARY KEY(user_id),
+	UNIQUE(user_id)
+);
+CREATE INDEX IF NOT EXISTS blacklist_ga_id_idx ON blacklist_ga (user_id);