4bc143b26f
Commands now show as 'wh create', 'wh staff add', 'wh set relaymode' instead of 'wh-create', 'wh-staff add', 'wh-set relaymode'. - Added commands/_base.py with root 'wh' group and all 20 subgroup stubs - All 13 command mixins inherit from WormholeBase - Decorators changed: @commands.hybrid_command(name='wh-X') → @WormholeBase.wh.command(name='X') - Subgroup commands: @wh_X.command() → @WormholeBase.wh_X.command() - Removed duplicate 'wh' group from core.py (now in _base.py) - Fixed alias: 'wh-color' → 'color' for colour command
102 lines
4.9 KiB
Python
102 lines
4.9 KiB
Python
"""DM relay commands — subscribe, send, configure DM relay."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import discord
|
|
from redbot.core import commands
|
|
|
|
from ..models.permissions import Role, requires_role
|
|
from ..utils import ok_embed, err_embed, info_embed, dm_embed, truncate
|
|
from ._base import WormholeBase
|
|
|
|
|
|
class DMCommands(WormholeBase):
|
|
"""Mixin — DM relay management."""
|
|
|
|
@WormholeBase.wh_dm.command(name="enable")
|
|
@requires_role(Role.ADMIN)
|
|
async def wh_dm_enable(self, ctx: commands.Context, name: str, enabled: bool = True) -> None:
|
|
"""Enable or disable DM relay for a network."""
|
|
async with self.config.networks() as ns:
|
|
ns[name]["dm_enabled"] = enabled
|
|
await ctx.send(embed=ok_embed(f"DM relay {'enabled' if enabled else 'disabled'} for `{name}`."))
|
|
|
|
@WormholeBase.wh_dm.command(name="subscribe", aliases=["sub"])
|
|
async def wh_dm_sub(self, ctx: commands.Context, name: str) -> None:
|
|
"""Subscribe to receive network messages via DM."""
|
|
nd = await self._net(name)
|
|
if not nd:
|
|
return await ctx.send(embed=err_embed(f"Network `{name}` not found."))
|
|
if not nd.get("dm_enabled"):
|
|
return await ctx.send(embed=err_embed(f"DM relay isn't enabled for `{name}`."))
|
|
async with self.config.networks() as ns:
|
|
subs = ns[name].setdefault("dm_subscribers", [])
|
|
if ctx.author.id in subs:
|
|
return await ctx.send(embed=err_embed("You're already subscribed."))
|
|
subs.append(ctx.author.id)
|
|
await ctx.send(embed=ok_embed(f"Subscribed to DM relay for `{name}`."))
|
|
|
|
@WormholeBase.wh_dm.command(name="unsubscribe", aliases=["unsub"])
|
|
async def wh_dm_unsub(self, ctx: commands.Context, name: str) -> None:
|
|
"""Unsubscribe from DM relay."""
|
|
async with self.config.networks() as ns:
|
|
subs = ns[name].get("dm_subscribers", [])
|
|
if ctx.author.id in subs:
|
|
subs.remove(ctx.author.id)
|
|
await ctx.send(embed=ok_embed(f"Unsubscribed from DM relay for `{name}`."))
|
|
|
|
@WormholeBase.wh_dm.command(name="mode")
|
|
@requires_role(Role.ADMIN)
|
|
async def wh_dm_mode(self, ctx: commands.Context, name: str, mode: str) -> None:
|
|
"""Set DM relay mode: ``embed``, ``compact``, or ``plain``."""
|
|
if mode not in ("embed", "compact", "plain"):
|
|
return await ctx.send(embed=err_embed("Mode must be `embed`, `compact`, or `plain`."))
|
|
async with self.config.networks() as ns:
|
|
ns[name]["dm_relay_mode"] = mode
|
|
await ctx.send(embed=ok_embed(f"DM relay mode set to **{mode}** for `{name}`."))
|
|
|
|
@WormholeBase.wh_dm.command(name="send")
|
|
async def wh_dm_send(self, ctx: commands.Context, name: str, *, message: str) -> None:
|
|
"""Send a message to the network from your DMs."""
|
|
nd = await self._net(name)
|
|
if not nd:
|
|
return await ctx.send(embed=err_embed(f"Network `{name}` not found."))
|
|
if not nd.get("dm_enabled"):
|
|
return await ctx.send(embed=err_embed("DM relay isn't enabled for this network."))
|
|
if ctx.author.id in nd.get("banned_users", []):
|
|
return await ctx.send(embed=err_embed("You're banned from this network."))
|
|
if ctx.author.id in nd.get("muted_users", []):
|
|
return await ctx.send(embed=err_embed("You're muted in this network."))
|
|
|
|
from ..utils import build_dm_relay_embed
|
|
em = build_dm_relay_embed(ctx.author, message, name)
|
|
for ch_id in nd.get("channels", []):
|
|
ch = self.bot.get_channel(ch_id)
|
|
if ch:
|
|
try:
|
|
await ch.send(embed=em)
|
|
except Exception:
|
|
pass
|
|
await ctx.send(embed=ok_embed("Message sent to the network."))
|
|
|
|
@WormholeBase.wh_dm.command(name="ignore")
|
|
async def wh_dm_ignore(self, ctx: commands.Context, name: str, user: discord.User) -> None:
|
|
"""Ignore a user in DM relay (you won't see their messages)."""
|
|
async with self.config.networks() as ns:
|
|
ignores = ns[name].setdefault("user_ignores", {})
|
|
uid_str = str(ctx.author.id)
|
|
user_ignores = ignores.setdefault(uid_str, [])
|
|
if user.id not in user_ignores:
|
|
user_ignores.append(user.id)
|
|
await ctx.send(embed=ok_embed(f"{user.display_name} ignored in DM relay for `{name}`."))
|
|
|
|
@WormholeBase.wh_dm.command(name="unignore")
|
|
async def wh_dm_unignore(self, ctx: commands.Context, name: str, user: discord.User) -> None:
|
|
"""Stop ignoring a user in DM relay."""
|
|
async with self.config.networks() as ns:
|
|
ignores = ns[name].get("user_ignores", {})
|
|
uid_str = str(ctx.author.id)
|
|
user_ignores = ignores.get(uid_str, [])
|
|
if user.id in user_ignores:
|
|
user_ignores.remove(user.id)
|
|
await ctx.send(embed=ok_embed(f"{user.display_name} unignored in DM relay for `{name}`."))
|