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
79 lines
3.4 KiB
Python
79 lines
3.4 KiB
Python
"""ToS / Rules acceptance gate commands."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime, timezone
|
|
|
|
import discord
|
|
from redbot.core import commands
|
|
|
|
from ..models.permissions import Role, requires_role
|
|
from ..utils import ok_embed, err_embed, info_embed, COLOUR_INFO
|
|
from ._base import WormholeBase
|
|
|
|
|
|
class ToSCommands(WormholeBase):
|
|
"""Mixin — Terms of Service / rules acceptance gate."""
|
|
|
|
@WormholeBase.wh_rules.command(name="set")
|
|
@requires_role(Role.ADMIN)
|
|
async def wh_rules_set(self, ctx: commands.Context, name: str, *, text: str) -> None:
|
|
"""Set the rules / ToS text."""
|
|
async with self.config.networks() as ns:
|
|
ns[name]["rules_text"] = text
|
|
await ctx.send(embed=ok_embed(f"Rules updated for `{name}`."))
|
|
|
|
@WormholeBase.wh_rules.command(name="require")
|
|
@requires_role(Role.ADMIN)
|
|
async def wh_rules_require(self, ctx: commands.Context, name: str, enabled: bool = True) -> None:
|
|
"""Toggle mandatory rules acceptance before using the network."""
|
|
async with self.config.networks() as ns:
|
|
ns[name]["rules_required"] = enabled
|
|
await ctx.send(embed=ok_embed(
|
|
f"Rules acceptance {'required' if enabled else 'not required'} for `{name}`."
|
|
))
|
|
|
|
@WormholeBase.wh_rules.command(name="view")
|
|
async def wh_rules_view(self, ctx: commands.Context, name: str) -> None:
|
|
"""View the network rules."""
|
|
nd = await self._net(name)
|
|
if not nd:
|
|
return await ctx.send(embed=err_embed(f"Network `{name}` not found."))
|
|
text = nd.get("rules_text") or nd.get("rules") or "No rules set."
|
|
em = discord.Embed(
|
|
title=f"📜 Rules — {name}",
|
|
description=text[:4000],
|
|
colour=COLOUR_INFO,
|
|
)
|
|
accepted = nd.get("rules_accepted", {}).get(str(ctx.author.id))
|
|
if accepted:
|
|
em.set_footer(text=f"You accepted on {accepted[:10]}")
|
|
else:
|
|
em.set_footer(text="⚠️ You haven't accepted these rules yet. Use `wh accept` or `wh agree`.")
|
|
await ctx.send(embed=em)
|
|
|
|
@WormholeBase.wh.command(name="accept")
|
|
async def wh_accept(self, ctx: commands.Context, name: str) -> None:
|
|
"""Accept the network rules / ToS."""
|
|
nd = await self._net(name)
|
|
if not nd:
|
|
return await ctx.send(embed=err_embed(f"Network `{name}` not found."))
|
|
if not nd.get("rules_required"):
|
|
return await ctx.send(embed=info_embed("This network doesn't require rules acceptance."))
|
|
async with self.config.networks() as ns:
|
|
ns[name].setdefault("rules_accepted", {})[str(ctx.author.id)] = datetime.now(timezone.utc).isoformat()
|
|
await ctx.send(embed=ok_embed(f"You've accepted the rules for `{name}`. You can now use the relay."))
|
|
|
|
@WormholeBase.wh.command(name="agree")
|
|
async def wh_agree(self, ctx: commands.Context, name: str) -> None:
|
|
"""Alias for ``wh accept``."""
|
|
await self.wh_accept(ctx, name)
|
|
|
|
@WormholeBase.wh_rules.command(name="reset")
|
|
@requires_role(Role.ADMIN)
|
|
async def wh_rules_reset(self, ctx: commands.Context, name: str) -> None:
|
|
"""Reset all acceptances (force everyone to re-accept)."""
|
|
async with self.config.networks() as ns:
|
|
ns[name]["rules_accepted"] = {}
|
|
await ctx.send(embed=ok_embed(f"All acceptances reset for `{name}`. Users must re-accept."))
|
|
await self._audit(name, "rules_reset", str(ctx.author))
|