first commit

bot/cogs/commands/fastgreet.py

@@ -0,0 +1,78 @@
+import discord
+from discord.ext import commands
+import sqlite3
+import asyncio
+import os
+
+DB_PATH = "./db/fastgreet.db"
+
+class FastGreet(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+        os.makedirs("./db", exist_ok=True)
+        self.init_db()
+
+    def init_db(self):
+        with sqlite3.connect(DB_PATH) as conn:
+            conn.execute("""
+                CREATE TABLE IF NOT EXISTS greet_channels (
+                    guild_id INTEGER,
+                    channel_id INTEGER,
+                    PRIMARY KEY (guild_id, channel_id)
+                )
+            """)
+
+    @commands.command(name="fastgreet_add")
+    @commands.has_permissions(administrator=True)
+    async def add_greet_channel(self, ctx, channel: discord.TextChannel):
+        with sqlite3.connect(DB_PATH) as conn:
+            conn.execute("""
+                INSERT OR IGNORE INTO greet_channels (guild_id, channel_id)
+                VALUES (?, ?)
+            """, (ctx.guild.id, channel.id))
+        await ctx.send(f"✅ {channel.mention} added as a greet channel.")
+
+    @commands.command(name="fastgreet_remove")
+    @commands.has_permissions(administrator=True)
+    async def remove_greet_channel(self, ctx, channel: discord.TextChannel):
+        with sqlite3.connect(DB_PATH) as conn:
+            conn.execute("""
+                DELETE FROM greet_channels WHERE guild_id = ? AND channel_id = ?
+            """, (ctx.guild.id, channel.id))
+        await ctx.send(f"❌ {channel.mention} removed from greet channels.")
+
+    @commands.command(name="fastgreet_list")
+    async def list_greet_channels(self, ctx):
+        with sqlite3.connect(DB_PATH) as conn:
+            cursor = conn.execute("""
+                SELECT channel_id FROM greet_channels WHERE guild_id = ?
+            """, (ctx.guild.id,))
+            rows = cursor.fetchall()
+
+        if not rows:
+            await ctx.send("⚠️ No greet channels configured.")
+            return
+
+        channels = [f"<#{cid[0]}>" for cid in rows]
+        await ctx.send("📋 Greet Channels: " + ", ".join(channels))
+
+    @commands.Cog.listener()
+    async def on_member_join(self, member):
+        with sqlite3.connect(DB_PATH) as conn:
+            cursor = conn.execute("""
+                SELECT channel_id FROM greet_channels WHERE guild_id = ?
+            """, (member.guild.id,))
+            channels = [row[0] for row in cursor.fetchall()]
+
+        for channel_id in channels:
+            channel = self.bot.get_channel(channel_id)
+            if channel:
+                try:
+                    msg = await channel.send(f"{member.mention} Welcome!")
+                    await asyncio.sleep(2)
+                    await msg.delete()
+                except discord.Forbidden:
+                    continue  # Missing permissions
+
+async def setup(bot):
+    await bot.add_cog(FastGreet(bot))