101 lines
4.7 KiB
Python
101 lines
4.7 KiB
Python
# ╔══════════════════════════════════════════════════════════════════╗
|
|
# ║ ║
|
|
# ║ +-+-+-+-+-+-+-+-+ ║
|
|
# ║ |H|e|x|a|H|o|s|t| ║
|
|
# ║ +-+-+-+-+-+-+-+-+ ║
|
|
# ║ ║
|
|
# ║ © 2026 HexaHost — All Rights Reserved ║
|
|
# ║ ║
|
|
# ║ discord ── https://discord.gg/hexahost ║
|
|
# ║ github ── https://github.com/theoneandonlymace ║
|
|
# ║ ║
|
|
# ╚══════════════════════════════════════════════════════════════════╝
|
|
from utils.db_paths import jsondb_path
|
|
|
|
import discord
|
|
from utils.emoji import BLOBPART
|
|
from discord.ext import commands
|
|
import json
|
|
from utils.cv2 import CV2
|
|
|
|
class joindm(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.joindm_messages = {}
|
|
self.load_joindm_messages()
|
|
|
|
def load_joindm_messages(self):
|
|
# Load the join DM messages from file
|
|
try:
|
|
with open(jsondb_path('joindm_messages.json'), 'r') as f:
|
|
self.joindm_messages = json.load(f)
|
|
except FileNotFoundError:
|
|
self.joindm_messages = {}
|
|
|
|
def save_joindm_messages(self):
|
|
# Save the join DM messages to file
|
|
with open(jsondb_path('joindm_messages.json'), 'w') as f:
|
|
json.dump(self.joindm_messages, f)
|
|
|
|
@commands.hybrid_group(invoke_without_command=True)
|
|
@commands.has_permissions(administrator=True)
|
|
async def joindm(self, ctx):
|
|
# Display the current join DM message
|
|
guild_id = str(ctx.guild.id)
|
|
if guild_id in self.joindm_messages:
|
|
await ctx.send(view=CV2("✅ Join DM Status", f"The current join DM message is:\n`{self.joindm_messages[guild_id]}`"))
|
|
else:
|
|
await ctx.send(view=CV2("❌ Error", "No custom join DM message has been set for this server."))
|
|
|
|
@joindm.command()
|
|
@commands.has_permissions(administrator=True)
|
|
async def message(self, ctx, *, message=None):
|
|
# Set the custom join DM message
|
|
if message is None:
|
|
await ctx.send(view=CV2("❌ Error", "Please provide a custom join DM message."))
|
|
else:
|
|
self.joindm_messages[str(ctx.guild.id)] = message
|
|
self.save_joindm_messages()
|
|
await ctx.send(view=CV2("✅ Success", "Custom join DM message set successfully."))
|
|
|
|
@joindm.command()
|
|
@commands.has_permissions(administrator=True)
|
|
async def enable(self, ctx):
|
|
# Enable the join DM module
|
|
self.bot.add_listener(self.on_member_join, 'on_member_join')
|
|
await ctx.send(view=CV2("✅ Success", "Join DM module enabled. Custom DM will be sent to new members."))
|
|
|
|
@joindm.command()
|
|
@commands.has_permissions(administrator=True)
|
|
async def disable(self, ctx):
|
|
# Disable the join DM module
|
|
self.bot.remove_listener(self.on_member_join)
|
|
await ctx.send(view=CV2("✅ Success", "Join DM module disabled. Custom DM will not be sent to new members."))
|
|
|
|
@joindm.command()
|
|
async def test(self, ctx):
|
|
# Send a test join DM to the author of the command
|
|
guild_id = str(ctx.guild.id)
|
|
if guild_id in self.joindm_messages:
|
|
message = self.joindm_messages[guild_id]
|
|
server_name = ctx.guild.name
|
|
join_dm_message = f"{message}\n\n ``Sent from {server_name} `` "
|
|
await ctx.send(view=CV2("✅ Test Sent", "Test Join DM Sent To Your Dm"))
|
|
await ctx.message.add_reaction(BLOBPART)
|
|
await ctx.author.send(view=CV2("👋 Welcome", join_dm_message))
|
|
else:
|
|
await ctx.send(view=CV2("❌ Error", "No custom join DM message has been set for this server."))
|
|
|
|
async def on_member_join(self, member):
|
|
|
|
guild_id = str(member.guild.id)
|
|
if guild_id in self.joindm_messages:
|
|
message = self.joindm_messages[guild_id]
|
|
dm_channel = await member.create_dm()
|
|
server_name = member.guild.name
|
|
join_dm_message = f"{message}\n\n``Sent from {server_name} ``"
|
|
await dm_channel.send(view=CV2("👋 Welcome", join_dm_message))
|
|
|
|
async def setup(bot):
|
|
await bot.add_cog(joindm(bot))
|