79 lines
2.8 KiB
Python
79 lines
2.8 KiB
Python
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))
|