import logging import os import re import sqlite3 from dataclasses import dataclass from datetime import datetime, timedelta, timezone from typing import Optional import discord from discord import app_commands from discord.ext import commands from dotenv import load_dotenv load_dotenv() TOKEN = os.getenv("DISCORD_TOKEN") TEST_GUILD_ID = os.getenv("GUILD_ID") DB_PATH = os.getenv("DATABASE_PATH", "clans_v100000000.db") CLAN_CATEGORY_ID = int(os.getenv("CLAN_CATEGORY_ID", "1352322719588417607")) INVITE_EXPIRY_MINUTES = int(os.getenv("INVITE_EXPIRY_MINUTES", "5")) DEFAULT_CLAN_COLOR = os.getenv("DEFAULT_CLAN_COLOR", "#8B5CF6") LOG_LEVEL = os.getenv("LOG_LEVEL", "INFO").upper() logging.basicConfig( level=getattr(logging, LOG_LEVEL, logging.INFO), format="[%(asctime)s] [%(levelname)s] %(name)s: %(message)s", ) logger = logging.getLogger("rust-clan-bot-v100000000") HEX_RE = re.compile(r"^#?(?:[0-9a-fA-F]{6})$") URL_RE = re.compile(r"^https?://", re.IGNORECASE) CHANNEL_SAFE_RE = re.compile(r"[^a-z0-9-]") @dataclass(slots=True) class ClanRecord: id: int guild_id: int name: str color_hex: str role_id: Optional[int] leader_id: int channel_id: Optional[int] home_message_id: Optional[int] banner_url: Optional[str] tag: Optional[str] bio: Optional[str] rust_server: Optional[str] invite_url: Optional[str] region: Optional[str] playstyle: Optional[str] focus: Optional[str] recruitment_open: int requirements: Optional[str] created_at: str class ClanDatabase: def __init__(self, path: str): self.conn = sqlite3.connect(path) self.conn.row_factory = sqlite3.Row self.conn.execute("PRAGMA foreign_keys = ON") self.create_tables() @staticmethod def now_iso() -> str: return datetime.now(timezone.utc).isoformat() def close(self) -> None: self.conn.close() def create_tables(self) -> None: with self.conn: self.conn.execute( """ CREATE TABLE IF NOT EXISTS clans ( id INTEGER PRIMARY KEY AUTOINCREMENT, guild_id INTEGER NOT NULL, name TEXT NOT NULL, color_hex TEXT NOT NULL, role_id INTEGER, leader_id INTEGER NOT NULL, channel_id INTEGER, home_message_id INTEGER, banner_url TEXT, tag TEXT, bio TEXT, rust_server TEXT, invite_url TEXT, region TEXT, playstyle TEXT, focus TEXT, recruitment_open INTEGER NOT NULL DEFAULT 0, requirements TEXT, created_at TEXT NOT NULL, UNIQUE(guild_id, name) ) """ ) self.conn.execute( """ CREATE TABLE IF NOT EXISTS clan_members ( guild_id INTEGER NOT NULL, user_id INTEGER NOT NULL, clan_id INTEGER NOT NULL, role TEXT NOT NULL, specialty TEXT DEFAULT '', note TEXT DEFAULT '', joined_at TEXT NOT NULL, PRIMARY KEY (guild_id, user_id), FOREIGN KEY (clan_id) REFERENCES clans(id) ON DELETE CASCADE ) """ ) self.conn.execute( """ CREATE TABLE IF NOT EXISTS clan_invites ( guild_id INTEGER NOT NULL, user_id INTEGER NOT NULL, clan_id INTEGER NOT NULL, inviter_id INTEGER NOT NULL, expires_at TEXT NOT NULL, PRIMARY KEY (guild_id, user_id), FOREIGN KEY (clan_id) REFERENCES clans(id) ON DELETE CASCADE ) """ ) self.conn.execute( """ CREATE TABLE IF NOT EXISTS clan_applications ( guild_id INTEGER NOT NULL, user_id INTEGER NOT NULL, clan_id INTEGER NOT NULL, message_id INTEGER, status TEXT NOT NULL DEFAULT 'pending', created_at TEXT NOT NULL, PRIMARY KEY (guild_id, user_id), FOREIGN KEY (clan_id) REFERENCES clans(id) ON DELETE CASCADE ) """ ) self.conn.execute( """ CREATE TABLE IF NOT EXISTS clan_activity ( id INTEGER PRIMARY KEY AUTOINCREMENT, clan_id INTEGER NOT NULL, actor_id INTEGER, action TEXT NOT NULL, details TEXT, created_at TEXT NOT NULL, FOREIGN KEY (clan_id) REFERENCES clans(id) ON DELETE CASCADE ) """ ) def _clan_from_row(self, row: Optional[sqlite3.Row]) -> Optional[ClanRecord]: return ClanRecord(**dict(row)) if row else None def cleanup_expired_invites(self) -> None: with self.conn: self.conn.execute("DELETE FROM clan_invites WHERE expires_at <= ?", (self.now_iso(),)) def log(self, clan_id: int, action: str, details: str = "", actor_id: Optional[int] = None) -> None: with self.conn: self.conn.execute( "INSERT INTO clan_activity (clan_id, actor_id, action, details, created_at) VALUES (?, ?, ?, ?, ?)", (clan_id, actor_id, action, details, self.now_iso()), ) def get_activity(self, clan_id: int, limit: int = 12) -> list[sqlite3.Row]: return self.conn.execute( "SELECT * FROM clan_activity WHERE clan_id = ? ORDER BY id DESC LIMIT ?", (clan_id, limit), ).fetchall() def get_clan_by_id(self, clan_id: int) -> Optional[ClanRecord]: return self._clan_from_row(self.conn.execute("SELECT * FROM clans WHERE id = ?", (clan_id,)).fetchone()) def get_clan_by_name(self, guild_id: int, name: str) -> Optional[ClanRecord]: return self._clan_from_row( self.conn.execute("SELECT * FROM clans WHERE guild_id = ? AND lower(name) = lower(?)", (guild_id, name)).fetchone() ) def get_clan_by_channel(self, guild_id: int, channel_id: int) -> Optional[ClanRecord]: return self._clan_from_row( self.conn.execute("SELECT * FROM clans WHERE guild_id = ? AND channel_id = ?", (guild_id, channel_id)).fetchone() ) def get_user_membership(self, guild_id: int, user_id: int) -> Optional[sqlite3.Row]: return self.conn.execute( """ SELECT cm.*, c.name AS clan_name, c.color_hex, c.role_id, c.leader_id, c.channel_id FROM clan_members cm JOIN clans c ON c.id = cm.clan_id WHERE cm.guild_id = ? AND cm.user_id = ? """, (guild_id, user_id), ).fetchone() def get_members(self, clan_id: int) -> list[sqlite3.Row]: return self.conn.execute( """ SELECT * FROM clan_members WHERE clan_id = ? ORDER BY CASE role WHEN 'leader' THEN 3 WHEN 'co_leader' THEN 2 ELSE 1 END DESC, joined_at ASC """, (clan_id,), ).fetchall() def create_clan(self, guild_id: int, leader_id: int, name: str, color_hex: str, role_id: int, channel_id: int) -> int: created_at = self.now_iso() with self.conn: cur = self.conn.execute( """ INSERT INTO clans ( guild_id, name, color_hex, role_id, leader_id, channel_id, home_message_id, banner_url, tag, bio, rust_server, invite_url, region, playstyle, focus, recruitment_open, requirements, created_at ) VALUES (?, ?, ?, ?, ?, ?, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, 0, NULL, ?) """, (guild_id, name, color_hex, role_id, leader_id, channel_id, created_at), ) clan_id = int(cur.lastrowid) self.conn.execute( "INSERT INTO clan_members (guild_id, user_id, clan_id, role, specialty, note, joined_at) VALUES (?, ?, ?, 'leader', '', '', ?)", (guild_id, leader_id, clan_id, created_at), ) return clan_id def update_clan(self, clan_id: int, **fields: object) -> None: allowed = { "name", "color_hex", "role_id", "channel_id", "home_message_id", "banner_url", "tag", "bio", "rust_server", "invite_url", "region", "playstyle", "focus", "recruitment_open", "requirements" } parts = [] values = [] for key, value in fields.items(): if key in allowed: parts.append(f"{key} = ?") values.append(value) if not parts: return values.append(clan_id) with self.conn: self.conn.execute(f"UPDATE clans SET {', '.join(parts)} WHERE id = ?", tuple(values)) def add_member(self, guild_id: int, user_id: int, clan_id: int, role: str = "member") -> None: with self.conn: self.conn.execute( "INSERT INTO clan_members (guild_id, user_id, clan_id, role, specialty, note, joined_at) VALUES (?, ?, ?, ?, '', '', ?)", (guild_id, user_id, clan_id, role, self.now_iso()), ) def remove_member(self, guild_id: int, user_id: int) -> None: with self.conn: self.conn.execute("DELETE FROM clan_members WHERE guild_id = ? AND user_id = ?", (guild_id, user_id)) self.conn.execute("DELETE FROM clan_invites WHERE guild_id = ? AND user_id = ?", (guild_id, user_id)) self.conn.execute("DELETE FROM clan_applications WHERE guild_id = ? AND user_id = ?", (guild_id, user_id)) def set_member_role(self, guild_id: int, user_id: int, role: str) -> None: with self.conn: self.conn.execute("UPDATE clan_members SET role = ? WHERE guild_id = ? AND user_id = ?", (role, guild_id, user_id)) def set_member_specialty(self, guild_id: int, user_id: int, specialty: str) -> None: with self.conn: self.conn.execute("UPDATE clan_members SET specialty = ? WHERE guild_id = ? AND user_id = ?", (specialty, guild_id, user_id)) def set_member_note(self, guild_id: int, user_id: int, note: str) -> None: with self.conn: self.conn.execute("UPDATE clan_members SET note = ? WHERE guild_id = ? AND user_id = ?", (note, guild_id, user_id)) def transfer_leadership(self, guild_id: int, clan_id: int, old_leader_id: int, new_leader_id: int) -> None: with self.conn: self.conn.execute("UPDATE clan_members SET role = 'co_leader' WHERE guild_id = ? AND user_id = ?", (guild_id, old_leader_id)) self.conn.execute("UPDATE clan_members SET role = 'leader' WHERE guild_id = ? AND user_id = ?", (guild_id, new_leader_id)) self.conn.execute("UPDATE clans SET leader_id = ? WHERE id = ?", (new_leader_id, clan_id)) def set_invite(self, guild_id: int, user_id: int, clan_id: int, inviter_id: int, expires_at: str) -> None: with self.conn: self.conn.execute( """ INSERT INTO clan_invites (guild_id, user_id, clan_id, inviter_id, expires_at) VALUES (?, ?, ?, ?, ?) ON CONFLICT(guild_id, user_id) DO UPDATE SET clan_id = excluded.clan_id, inviter_id = excluded.inviter_id, expires_at = excluded.expires_at """, (guild_id, user_id, clan_id, inviter_id, expires_at), ) def get_invite(self, guild_id: int, user_id: int) -> Optional[sqlite3.Row]: self.cleanup_expired_invites() return self.conn.execute("SELECT * FROM clan_invites WHERE guild_id = ? AND user_id = ?", (guild_id, user_id)).fetchone() def delete_invite(self, guild_id: int, user_id: int) -> None: with self.conn: self.conn.execute("DELETE FROM clan_invites WHERE guild_id = ? AND user_id = ?", (guild_id, user_id)) def set_application(self, guild_id: int, user_id: int, clan_id: int, message_id: int) -> None: with self.conn: self.conn.execute( """ INSERT INTO clan_applications (guild_id, user_id, clan_id, message_id, status, created_at) VALUES (?, ?, ?, ?, 'pending', ?) ON CONFLICT(guild_id, user_id) DO UPDATE SET clan_id = excluded.clan_id, message_id = excluded.message_id, status = 'pending', created_at = excluded.created_at """, (guild_id, user_id, clan_id, message_id, self.now_iso()), ) def get_application_by_message(self, message_id: int) -> Optional[sqlite3.Row]: return self.conn.execute("SELECT * FROM clan_applications WHERE message_id = ?", (message_id,)).fetchone() def get_pending_application(self, guild_id: int, user_id: int) -> Optional[sqlite3.Row]: return self.conn.execute( "SELECT * FROM clan_applications WHERE guild_id = ? AND user_id = ? AND status = 'pending'", (guild_id, user_id), ).fetchone() def set_application_status(self, guild_id: int, user_id: int, status: str) -> None: with self.conn: self.conn.execute("UPDATE clan_applications SET status = ? WHERE guild_id = ? AND user_id = ?", (status, guild_id, user_id)) def disband_clan(self, clan_id: int) -> None: with self.conn: self.conn.execute("DELETE FROM clan_applications WHERE clan_id = ?", (clan_id,)) self.conn.execute("DELETE FROM clan_invites WHERE clan_id = ?", (clan_id,)) self.conn.execute("DELETE FROM clan_members WHERE clan_id = ?", (clan_id,)) self.conn.execute("DELETE FROM clan_activity WHERE clan_id = ?", (clan_id,)) self.conn.execute("DELETE FROM clans WHERE id = ?", (clan_id,)) class ReadyCheckView(discord.ui.View): def __init__(self, title: str): super().__init__(timeout=6 * 60 * 60) self.title = title self.ready_ids: set[int] = set() self.maybe_ids: set[int] = set() self.no_ids: set[int] = set() self.message: Optional[discord.Message] = None def _set_bucket(self, user_id: int, bucket: str) -> None: self.ready_ids.discard(user_id) self.maybe_ids.discard(user_id) self.no_ids.discard(user_id) getattr(self, f"{bucket}_ids").add(user_id) async def _sync(self) -> None: if self.message is None: return embed = self.message.embeds[0] if self.message.embeds else discord.Embed(title=self.title) embed.description = ( f"✅ Ready: **{len(self.ready_ids)}**\n" f"❔ Maybe: **{len(self.maybe_ids)}**\n" f"❌ Not on: **{len(self.no_ids)}**" ) await self.message.edit(embed=embed, view=self) @discord.ui.button(label="Ready", style=discord.ButtonStyle.success) async def ready(self, interaction: discord.Interaction, button: discord.ui.Button) -> None: self._set_bucket(interaction.user.id, "ready") await self._sync() await interaction.response.send_message("Marked as ready.", ephemeral=True) @discord.ui.button(label="Maybe", style=discord.ButtonStyle.secondary) async def maybe(self, interaction: discord.Interaction, button: discord.ui.Button) -> None: self._set_bucket(interaction.user.id, "maybe") await self._sync() await interaction.response.send_message("Marked as maybe.", ephemeral=True) @discord.ui.button(label="Not On", style=discord.ButtonStyle.danger) async def not_on(self, interaction: discord.Interaction, button: discord.ui.Button) -> None: self._set_bucket(interaction.user.id, "no") await self._sync() await interaction.response.send_message("Marked as not on.", ephemeral=True) class InviteView(discord.ui.View): def __init__(self, cog: "ClanCog", clan_id: int, user_id: int): super().__init__(timeout=INVITE_EXPIRY_MINUTES * 60) self.cog = cog self.clan_id = clan_id self.user_id = user_id self.message: Optional[discord.Message] = None async def interaction_check(self, interaction: discord.Interaction) -> bool: if interaction.user.id != self.user_id: await interaction.response.send_message("That invite is not for you.", ephemeral=True) return False return True async def disable_all(self) -> None: for child in self.children: if isinstance(child, discord.ui.Button): child.disabled = True if self.message: try: await self.message.edit(view=self) except discord.HTTPException: pass async def on_timeout(self) -> None: await self.disable_all() @discord.ui.button(label="Accept", style=discord.ButtonStyle.success) async def accept(self, interaction: discord.Interaction, button: discord.ui.Button) -> None: await self.cog.handle_invite_response(interaction, True, self) @discord.ui.button(label="Decline", style=discord.ButtonStyle.danger) async def decline(self, interaction: discord.Interaction, button: discord.ui.Button) -> None: await self.cog.handle_invite_response(interaction, False, self) class HomeProfileModal(discord.ui.Modal): tag = discord.ui.TextInput(label="Clan tag", required=False, max_length=10) bio = discord.ui.TextInput(label="Clan bio", required=False, style=discord.TextStyle.paragraph, max_length=300) rust_server = discord.ui.TextInput(label="Rust server", required=False, max_length=100) invite_url = discord.ui.TextInput(label="Clan Discord link", required=False, max_length=200) region = discord.ui.TextInput(label="Region", required=False, max_length=32) def __init__(self, cog: "ClanCog", clan: ClanRecord): super().__init__(title="Edit Clan Profile") self.cog = cog self.clan = clan self.tag.default = clan.tag or "" self.bio.default = clan.bio or "" self.rust_server.default = clan.rust_server or "" self.invite_url.default = clan.invite_url or "" self.region.default = clan.region or "" async def on_submit(self, interaction: discord.Interaction) -> None: invite_value = str(self.invite_url).strip() if invite_value and not URL_RE.match(invite_value): await interaction.response.send_message("Discord link must start with http:// or https://", ephemeral=True) return self.cog.db.update_clan( self.clan.id, tag=str(self.tag).strip().upper(), bio=str(self.bio).strip(), rust_server=str(self.rust_server).strip(), invite_url=invite_value, region=str(self.region).strip(), ) self.cog.db.log(self.clan.id, "profile_updated", "Updated profile", interaction.user.id) await self.cog.refresh_home(interaction.guild, self.clan.id, page="overview") await interaction.response.send_message("Clan profile updated.", ephemeral=True) class HomeSettingsModal(discord.ui.Modal): playstyle = discord.ui.TextInput(label="Playstyle", required=False, max_length=40) focus = discord.ui.TextInput(label="Current focus", required=False, max_length=60) recruitment = discord.ui.TextInput(label="Recruitment (open/closed)", required=False, max_length=10) requirements = discord.ui.TextInput(label="Requirements", required=False, style=discord.TextStyle.paragraph, max_length=300) def __init__(self, cog: "ClanCog", clan: ClanRecord): super().__init__(title="Edit Clan Settings") self.cog = cog self.clan = clan self.playstyle.default = clan.playstyle or "" self.focus.default = clan.focus or "" self.recruitment.default = "open" if clan.recruitment_open else "closed" self.requirements.default = clan.requirements or "" async def on_submit(self, interaction: discord.Interaction) -> None: raw = str(self.recruitment).strip().lower() recruitment_open = 1 if raw in {"open", "yes", "1", "true"} else 0 self.cog.db.update_clan( self.clan.id, playstyle=str(self.playstyle).strip(), focus=str(self.focus).strip(), recruitment_open=recruitment_open, requirements=str(self.requirements).strip(), ) self.cog.db.log(self.clan.id, "settings_updated", f"Recruitment {'open' if recruitment_open else 'closed'}", interaction.user.id) await self.cog.refresh_home(interaction.guild, self.clan.id, page="recruitment") await interaction.response.send_message("Clan settings updated.", ephemeral=True) class HomeBannerModal(discord.ui.Modal): banner_url = discord.ui.TextInput(label="Banner image URL", required=False, max_length=400) def __init__(self, cog: "ClanCog", clan: ClanRecord): super().__init__(title="Set Clan Banner") self.cog = cog self.clan = clan self.banner_url.default = clan.banner_url or "" async def on_submit(self, interaction: discord.Interaction) -> None: value = str(self.banner_url).strip() if value and not URL_RE.match(value): await interaction.response.send_message("Banner URL must start with http:// or https://", ephemeral=True) return self.cog.db.update_clan(self.clan.id, banner_url=value) self.cog.db.log(self.clan.id, "banner_updated", "Updated banner", interaction.user.id) await self.cog.refresh_home(interaction.guild, self.clan.id, page="overview") await interaction.response.send_message("Clan banner updated.", ephemeral=True) class QuickAnnouncementModal(discord.ui.Modal): title_input = discord.ui.TextInput(label="Announcement title", max_length=80) body_input = discord.ui.TextInput(label="Announcement body", style=discord.TextStyle.paragraph, max_length=500) def __init__(self, cog: "ClanCog"): super().__init__(title="Post Clan Announcement") self.cog = cog async def on_submit(self, interaction: discord.Interaction) -> None: clan, membership = await self.cog.get_channel_clan_and_membership(interaction) if not clan or not membership or membership["role"] not in {"leader", "co_leader"}: await interaction.response.send_message("Only leaders and admins can post announcements here.", ephemeral=True) return channel = interaction.channel if not isinstance(channel, discord.TextChannel): await interaction.response.send_message("This only works in the clan channel.", ephemeral=True) return embed = discord.Embed( title=f"📢 {self.title_input}", description=str(self.body_input), colour=self.cog.to_colour(clan.color_hex), timestamp=datetime.now(timezone.utc), ) embed.set_footer(text=f"Posted by {interaction.user.display_name}") await channel.send(embed=embed) self.cog.db.log(clan.id, "announcement", str(self.title_input), interaction.user.id) await self.cog.refresh_home(interaction.guild, clan.id, page="activity") await interaction.response.send_message("Announcement posted.", ephemeral=True) class ApplicationDecisionView(discord.ui.View): def __init__(self, cog: "ClanCog"): super().__init__(timeout=None) self.cog = cog async def interaction_check(self, interaction: discord.Interaction) -> bool: if interaction.guild is None or interaction.message is None: await interaction.response.send_message("This only works in a server.", ephemeral=True) return False app_row = self.cog.db.get_application_by_message(interaction.message.id) if not app_row or app_row["status"] != "pending": await interaction.response.send_message("This application is no longer active.", ephemeral=True) return False membership = self.cog.db.get_user_membership(interaction.guild.id, interaction.user.id) if not membership or membership["clan_id"] != app_row["clan_id"] or membership["role"] not in {"leader", "co_leader"}: await interaction.response.send_message("Only that clan's leaders/admins can decide this.", ephemeral=True) return False return True @discord.ui.button(label="Accept Application", style=discord.ButtonStyle.success, custom_id="clanapp:accept") async def accept(self, interaction: discord.Interaction, button: discord.ui.Button) -> None: await self.cog.handle_application_decision(interaction, True) @discord.ui.button(label="Deny Application", style=discord.ButtonStyle.danger, custom_id="clanapp:deny") async def deny(self, interaction: discord.Interaction, button: discord.ui.Button) -> None: await self.cog.handle_application_decision(interaction, False) class ClanHomeView(discord.ui.View): def __init__(self, cog: "ClanCog"): super().__init__(timeout=None) self.cog = cog async def _clan(self, interaction: discord.Interaction) -> Optional[ClanRecord]: if interaction.guild is None or interaction.channel is None: return None return self.cog.db.get_clan_by_channel(interaction.guild.id, interaction.channel.id) async def _membership(self, interaction: discord.Interaction) -> Optional[sqlite3.Row]: if interaction.guild is None: return None return self.cog.db.get_user_membership(interaction.guild.id, interaction.user.id) @discord.ui.button(label="Overview", style=discord.ButtonStyle.primary, custom_id="home:overview", row=0) async def overview(self, interaction: discord.Interaction, button: discord.ui.Button) -> None: clan = await self._clan(interaction) if clan is None: await interaction.response.send_message("This isn't a clan channel.", ephemeral=True) return await interaction.response.defer() await self.cog.refresh_home(interaction.guild, clan.id, page="overview") @discord.ui.button(label="Members", style=discord.ButtonStyle.primary, custom_id="home:members", row=0) async def members(self, interaction: discord.Interaction, button: discord.ui.Button) -> None: clan = await self._clan(interaction) if clan is None: await interaction.response.send_message("This isn't a clan channel.", ephemeral=True) return await interaction.response.defer() await self.cog.refresh_home(interaction.guild, clan.id, page="members") @discord.ui.button(label="Recruitment", style=discord.ButtonStyle.primary, custom_id="home:recruitment", row=0) async def recruitment(self, interaction: discord.Interaction, button: discord.ui.Button) -> None: clan = await self._clan(interaction) if clan is None: await interaction.response.send_message("This isn't a clan channel.", ephemeral=True) return await interaction.response.defer() await self.cog.refresh_home(interaction.guild, clan.id, page="recruitment") @discord.ui.button(label="Activity", style=discord.ButtonStyle.primary, custom_id="home:activity", row=0) async def activity(self, interaction: discord.Interaction, button: discord.ui.Button) -> None: clan = await self._clan(interaction) if clan is None: await interaction.response.send_message("This isn't a clan channel.", ephemeral=True) return await interaction.response.defer() await self.cog.refresh_home(interaction.guild, clan.id, page="activity") @discord.ui.button(label="Edit Profile", style=discord.ButtonStyle.secondary, custom_id="home:profile", row=1) async def edit_profile(self, interaction: discord.Interaction, button: discord.ui.Button) -> None: clan = await self._clan(interaction) membership = await self._membership(interaction) if not clan or not membership or membership["clan_id"] != clan.id or membership["role"] not in {"leader", "co_leader"}: await interaction.response.send_message("Only leaders and admins can edit this.", ephemeral=True) return await interaction.response.send_modal(HomeProfileModal(self.cog, clan)) @discord.ui.button(label="Settings", style=discord.ButtonStyle.secondary, custom_id="home:settings", row=1) async def edit_settings(self, interaction: discord.Interaction, button: discord.ui.Button) -> None: clan = await self._clan(interaction) membership = await self._membership(interaction) if not clan or not membership or membership["clan_id"] != clan.id or membership["role"] not in {"leader", "co_leader"}: await interaction.response.send_message("Only leaders and admins can edit this.", ephemeral=True) return await interaction.response.send_modal(HomeSettingsModal(self.cog, clan)) @discord.ui.button(label="Banner", style=discord.ButtonStyle.secondary, custom_id="home:banner", row=1) async def banner(self, interaction: discord.Interaction, button: discord.ui.Button) -> None: clan = await self._clan(interaction) membership = await self._membership(interaction) if not clan or not membership or membership["clan_id"] != clan.id or membership["role"] not in {"leader", "co_leader"}: await interaction.response.send_message("Only leaders and admins can edit the banner.", ephemeral=True) return await interaction.response.send_modal(HomeBannerModal(self.cog, clan)) @discord.ui.button(label="Ready Check", style=discord.ButtonStyle.success, custom_id="home:readycheck", row=2) async def ready_check(self, interaction: discord.Interaction, button: discord.ui.Button) -> None: clan, membership = await self.cog.get_channel_clan_and_membership(interaction) if not clan or not membership or membership["role"] not in {"leader", "co_leader"}: await interaction.response.send_message("Only leaders and admins can start ready checks.", ephemeral=True) return channel = interaction.channel if not isinstance(channel, discord.TextChannel): await interaction.response.send_message("This only works in the clan channel.", ephemeral=True) return embed = discord.Embed( title="⚔️ Ready Check", description="✅ Ready: **0**\n❔ Maybe: **0**\n❌ Not on: **0**", colour=self.cog.to_colour(clan.color_hex), timestamp=datetime.now(timezone.utc), ) embed.set_footer(text=f"Started by {interaction.user.display_name}") view = ReadyCheckView("⚔️ Ready Check") msg = await channel.send(embed=embed, view=view) view.message = msg self.cog.db.log(clan.id, "ready_check", "Started from home panel", interaction.user.id) await self.cog.refresh_home(interaction.guild, clan.id, page="activity") await interaction.response.send_message("Ready check posted.", ephemeral=True) @discord.ui.button(label="Announcement", style=discord.ButtonStyle.success, custom_id="home:announce", row=2) async def announce(self, interaction: discord.Interaction, button: discord.ui.Button) -> None: clan, membership = await self.cog.get_channel_clan_and_membership(interaction) if not clan or not membership or membership["role"] not in {"leader", "co_leader"}: await interaction.response.send_message("Only leaders and admins can post announcements.", ephemeral=True) return await interaction.response.send_modal(QuickAnnouncementModal(self.cog)) @discord.ui.button(label="Refresh", style=discord.ButtonStyle.secondary, custom_id="home:refresh", row=2) async def refresh(self, interaction: discord.Interaction, button: discord.ui.Button) -> None: clan = await self._clan(interaction) if clan is None: await interaction.response.send_message("This isn't a clan channel.", ephemeral=True) return await interaction.response.defer() await self.cog.refresh_home(interaction.guild, clan.id, page="overview") class ClanCog(commands.Cog): clan = app_commands.Group(name="clan", description="Rust clan system") def __init__(self, bot: commands.Bot, db: ClanDatabase): self.bot = bot self.db = db self.home_view = ClanHomeView(self) self.application_view = ApplicationDecisionView(self) @staticmethod def normalize_hex(value: str) -> str: if not HEX_RE.fullmatch(value): raise ValueError("Color must be a 6-digit hex like #8B5CF6") return f"#{value.lstrip('#').upper()}" @staticmethod def to_colour(hex_color: str) -> discord.Colour: return discord.Colour(int(hex_color.lstrip("#"), 16)) @staticmethod def validate_name(name: str) -> str: name = name.strip() if len(name) < 3 or len(name) > 24: raise ValueError("Clan names must be 3 to 24 characters long.") return name @staticmethod def channel_name(name: str) -> str: slug = name.lower().replace(" ", "-") slug = CHANNEL_SAFE_RE.sub("", slug) slug = re.sub(r"-+", "-", slug).strip("-") return f"clan-{slug or 'base'}" def clan_title(self, clan: ClanRecord) -> str: return f"[{clan.tag}] {clan.name}" if clan.tag else clan.name async def ensure_guild(self, interaction: discord.Interaction) -> discord.Guild: if interaction.guild is None: raise RuntimeError("This command only works in a server.") return interaction.guild async def get_member_obj(self, guild: discord.Guild, user_id: int) -> Optional[discord.Member]: member = guild.get_member(user_id) if member: return member try: return await guild.fetch_member(user_id) except (discord.NotFound, discord.Forbidden, discord.HTTPException): return None async def mention(self, guild: discord.Guild, user_id: int) -> str: member = await self.get_member_obj(guild, user_id) return member.mention if member else f"<@{user_id}>" async def get_or_create_role(self, guild: discord.Guild, clan: ClanRecord) -> discord.Role: role = guild.get_role(clan.role_id) if clan.role_id else None if role: return role role = await guild.create_role(name=clan.name, colour=self.to_colour(clan.color_hex), reason="Recreated clan role") self.db.update_clan(clan.id, role_id=role.id) return role async def sync_member_role(self, guild: discord.Guild, user_id: int, clan: Optional[ClanRecord]) -> None: member = await self.get_member_obj(guild, user_id) if member is None: return role_ids = [row[0] for row in self.db.conn.execute("SELECT role_id FROM clans WHERE guild_id = ? AND role_id IS NOT NULL", (guild.id,)).fetchall()] remove = [guild.get_role(role_id) for role_id in role_ids if guild.get_role(role_id) in member.roles] if remove: try: await member.remove_roles(*remove, reason="Clan role sync cleanup") except discord.HTTPException: logger.exception("Failed removing clan roles from %s", user_id) if clan is not None: role = await self.get_or_create_role(guild, clan) if role not in member.roles: try: await member.add_roles(role, reason="Clan role sync add") except discord.HTTPException: logger.exception("Failed adding clan role to %s", user_id) async def ensure_clan_channel(self, guild: discord.Guild, clan: ClanRecord, role: discord.Role) -> discord.TextChannel: channel = guild.get_channel(clan.channel_id) if clan.channel_id else None me = guild.me or (await self.get_member_obj(guild, self.bot.user.id)) if self.bot.user else None overwrites = { guild.default_role: discord.PermissionOverwrite(view_channel=False), role: discord.PermissionOverwrite(view_channel=True, send_messages=True, read_message_history=True, attach_files=True, embed_links=True), } if me: overwrites[me] = discord.PermissionOverwrite(view_channel=True, send_messages=True, manage_messages=True, manage_channels=True, read_message_history=True) if isinstance(channel, discord.TextChannel): await channel.edit(name=self.channel_name(clan.name), overwrites=overwrites, reason="Clan channel sync") return channel category = guild.get_channel(CLAN_CATEGORY_ID) if not isinstance(category, discord.CategoryChannel): raise RuntimeError(f"Category {CLAN_CATEGORY_ID} was not found.") channel = await category.create_text_channel(name=self.channel_name(clan.name), overwrites=overwrites, reason=f"Clan channel for {clan.name}") self.db.update_clan(clan.id, channel_id=channel.id) return channel async def get_channel_clan_and_membership(self, interaction: discord.Interaction) -> tuple[Optional[ClanRecord], Optional[sqlite3.Row]]: if interaction.guild is None or interaction.channel is None: return None, None clan = self.db.get_clan_by_channel(interaction.guild.id, interaction.channel.id) membership = self.db.get_user_membership(interaction.guild.id, interaction.user.id) return clan, membership async def get_clan_and_membership(self, guild_id: int, user_id: int) -> tuple[Optional[ClanRecord], Optional[sqlite3.Row]]: membership = self.db.get_user_membership(guild_id, user_id) if not membership: return None, None return self.db.get_clan_by_id(membership["clan_id"]), membership async def build_home_embed(self, guild: discord.Guild, clan: ClanRecord, page: str) -> discord.Embed: members = self.db.get_members(clan.id) embed = discord.Embed( title={ "overview": f"🛡️ Command Center • {self.clan_title(clan)}", "members": f"👥 Roster Board • {self.clan_title(clan)}", "recruitment": f"📡 Recruitment • {self.clan_title(clan)}", "activity": f"📜 Activity Feed • {self.clan_title(clan)}", }.get(page, f"🛡️ Command Center • {self.clan_title(clan)}"), colour=self.to_colour(clan.color_hex), timestamp=datetime.now(timezone.utc), ) embed.set_footer(text="Rust Clan Bot v100000000") if page == "overview": admins = [] specialties = [] for row in members: if row["role"] == "co_leader": admins.append(await self.mention(guild, row["user_id"])) if row["specialty"]: specialties.append(f"{await self.mention(guild, row['user_id'])} — *{row['specialty']}*") embed.description = clan.bio or "No clan bio yet. Use the control buttons below to customize this home panel." embed.add_field(name="Leader", value=await self.mention(guild, clan.leader_id), inline=True) embed.add_field(name="Admins", value="\n".join(admins) if admins else "None", inline=True) embed.add_field(name="Members", value=str(len(members)), inline=True) embed.add_field(name="Rust Server", value=clan.rust_server or "Not set", inline=True) embed.add_field(name="Region", value=clan.region or "Not set", inline=True) embed.add_field(name="Playstyle", value=clan.playstyle or "Not set", inline=True) embed.add_field(name="Current Focus", value=clan.focus or "Not set", inline=True) embed.add_field(name="Recruitment", value="Open" if clan.recruitment_open else "Closed", inline=True) embed.add_field(name="Clan Discord", value=clan.invite_url or "Not set", inline=True) if specialties: embed.add_field(name="Specialties", value="\n".join(specialties[:8]), inline=False) if clan.banner_url: embed.set_image(url=clan.banner_url) elif page == "members": leaders, admins, regulars = [], [], [] for row in members: line = f"{await self.mention(guild, row['user_id'])}" if row["specialty"]: line += f" — *{row['specialty']}*" if row["role"] == "leader": leaders.append(line) elif row["role"] == "co_leader": admins.append(line) else: regulars.append(line) embed.description = "Leadership, specialties, and full roster in one place." embed.add_field(name="Leader", value="\n".join(leaders) if leaders else "None", inline=False) embed.add_field(name="Admins", value="\n".join(admins) if admins else "None", inline=False) embed.add_field(name="Members", value="\n".join(regulars[:25]) if regulars else "None", inline=False) elif page == "recruitment": embed.description = "Public-facing recruitment info and application details." embed.add_field(name="Status", value="🟢 Open" if clan.recruitment_open else "🔴 Closed", inline=True) embed.add_field(name="Region", value=clan.region or "Not set", inline=True) embed.add_field(name="Playstyle", value=clan.playstyle or "Not set", inline=True) embed.add_field(name="Current Focus", value=clan.focus or "Not set", inline=False) embed.add_field(name="Requirements", value=clan.requirements or "No requirements set.", inline=False) embed.add_field(name="How to Apply", value=f"Use `/clan apply {clan.name}` while recruitment is open.", inline=False) if clan.banner_url: embed.set_image(url=clan.banner_url) else: logs = self.db.get_activity(clan.id) lines = [] for row in logs: stamp = datetime.fromisoformat(row["created_at"]).strftime("%m/%d %H:%M") actor = f"<@{row['actor_id']}>" if row["actor_id"] else "System" details = f" — {row['details']}" if row["details"] else "" lines.append(f"`{stamp}` **{row['action'].replace('_', ' ').title()}** by {actor}{details}") embed.description = "\n".join(lines) if lines else "No activity yet." return embed async def ensure_home_message(self, guild: discord.Guild, clan: ClanRecord) -> discord.Message: role = await self.get_or_create_role(guild, clan) channel = await self.ensure_clan_channel(guild, clan, role) message = None if clan.home_message_id: try: message = await channel.fetch_message(clan.home_message_id) except (discord.NotFound, discord.HTTPException): message = None if message is None: embed = await self.build_home_embed(guild, self.db.get_clan_by_id(clan.id), "overview") message = await channel.send(content="## Rust Clan Command Center", embed=embed, view=self.home_view) try: await message.pin(reason="Clan command center") except discord.HTTPException: pass self.db.update_clan(clan.id, home_message_id=message.id) return message async def refresh_home(self, guild: Optional[discord.Guild], clan_id: int, page: str = "overview") -> None: if guild is None: return clan = self.db.get_clan_by_id(clan_id) if clan is None: return message = await self.ensure_home_message(guild, clan) clan = self.db.get_clan_by_id(clan.id) embed = await self.build_home_embed(guild, clan, page) await message.edit(content="## Rust Clan Command Center", embed=embed, view=self.home_view) async def handle_invite_response(self, interaction: discord.Interaction, accept: bool, view: InviteView) -> None: guild = interaction.guild if guild is None: await interaction.response.send_message("This only works in a server.", ephemeral=True) return invite = self.db.get_invite(guild.id, interaction.user.id) if not invite or invite["clan_id"] != view.clan_id: await interaction.response.send_message("That invite is no longer valid.", ephemeral=True) await view.disable_all() return clan = self.db.get_clan_by_id(invite["clan_id"]) if clan is None: self.db.delete_invite(guild.id, interaction.user.id) await interaction.response.send_message("That clan no longer exists.", ephemeral=True) await view.disable_all() return if self.db.get_user_membership(guild.id, interaction.user.id): self.db.delete_invite(guild.id, interaction.user.id) await interaction.response.send_message("You are already in a clan.", ephemeral=True) await view.disable_all() return if accept: self.db.add_member(guild.id, interaction.user.id, clan.id, "member") self.db.delete_invite(guild.id, interaction.user.id) await self.sync_member_role(guild, interaction.user.id, clan) self.db.log(clan.id, "member_joined", f"{interaction.user.display_name} joined from invite", interaction.user.id) await self.refresh_home(guild, clan.id, page="members") await interaction.response.send_message(f"You joined **{clan.name}**.", ephemeral=True) else: self.db.delete_invite(guild.id, interaction.user.id) self.db.log(clan.id, "invite_declined", "Invite declined", interaction.user.id) await self.refresh_home(guild, clan.id, page="activity") await interaction.response.send_message(f"You declined **{clan.name}**.", ephemeral=True) await view.disable_all() async def handle_application_decision(self, interaction: discord.Interaction, accept: bool) -> None: if interaction.guild is None or interaction.message is None: await interaction.response.send_message("This only works in a server.", ephemeral=True) return app_row = self.db.get_application_by_message(interaction.message.id) if not app_row or app_row["status"] != "pending": await interaction.response.send_message("This application is no longer active.", ephemeral=True) return clan = self.db.get_clan_by_id(app_row["clan_id"]) if clan is None: await interaction.response.send_message("That clan no longer exists.", ephemeral=True) return applicant_id = app_row["user_id"] if self.db.get_user_membership(interaction.guild.id, applicant_id): self.db.set_application_status(interaction.guild.id, applicant_id, "closed") await interaction.response.send_message("That user is already in a clan.", ephemeral=True) return for child in self.application_view.children: if isinstance(child, discord.ui.Button): child.disabled = True try: await interaction.message.edit(view=self.application_view) except discord.HTTPException: pass if accept: self.db.add_member(interaction.guild.id, applicant_id, clan.id, "member") self.db.set_application_status(interaction.guild.id, applicant_id, "accepted") await self.sync_member_role(interaction.guild, applicant_id, clan) self.db.log(clan.id, "application_accepted", f"Accepted <@{applicant_id}>", interaction.user.id) await self.refresh_home(interaction.guild, clan.id, page="members") await interaction.response.send_message(f"Accepted <@{applicant_id}> into **{clan.name}**.") else: self.db.set_application_status(interaction.guild.id, applicant_id, "denied") self.db.log(clan.id, "application_denied", f"Denied <@{applicant_id}>", interaction.user.id) await self.refresh_home(interaction.guild, clan.id, page="activity") await interaction.response.send_message(f"Denied <@{applicant_id}>.") @clan.command(name="create", description="Create a clan") async def create(self, interaction: discord.Interaction, name: str, color: Optional[str] = None) -> None: guild = await self.ensure_guild(interaction) try: name = self.validate_name(name) color_hex = self.normalize_hex(color or DEFAULT_CLAN_COLOR) except ValueError as exc: await interaction.response.send_message(str(exc), ephemeral=True) return if self.db.get_user_membership(guild.id, interaction.user.id): await interaction.response.send_message("You are already in a clan.", ephemeral=True) return if self.db.get_clan_by_name(guild.id, name): await interaction.response.send_message("A clan with that name already exists.", ephemeral=True) return category = guild.get_channel(CLAN_CATEGORY_ID) if not isinstance(category, discord.CategoryChannel): await interaction.response.send_message(f"Category `{CLAN_CATEGORY_ID}` was not found.", ephemeral=True) return await interaction.response.defer(ephemeral=True, thinking=True) role = await guild.create_role(name=name, colour=self.to_colour(color_hex), reason=f"Clan created by {interaction.user}") channel = await category.create_text_channel(name=self.channel_name(name), overwrites={guild.default_role: discord.PermissionOverwrite(view_channel=False), role: discord.PermissionOverwrite(view_channel=True, send_messages=True, read_message_history=True, attach_files=True, embed_links=True)}, reason=f"Clan channel for {name}") clan_id = self.db.create_clan(guild.id, interaction.user.id, name, color_hex, role.id, channel.id) clan = self.db.get_clan_by_id(clan_id) await self.sync_member_role(guild, interaction.user.id, clan) self.db.log(clan.id, "clan_created", f"Created {name}", interaction.user.id) await self.refresh_home(guild, clan.id, page="overview") await interaction.followup.send(f"Created **{name}**. Your clan channel is {channel.mention}.", ephemeral=True) @clan.command(name="info", description="Show clan info") async def info(self, interaction: discord.Interaction, name: Optional[str] = None) -> None: guild = await self.ensure_guild(interaction) clan = self.db.get_clan_by_name(guild.id, name) if name else None if clan is None and not name: membership = self.db.get_user_membership(guild.id, interaction.user.id) if membership: clan = self.db.get_clan_by_id(membership["clan_id"]) if clan is None: await interaction.response.send_message("Clan not found.", ephemeral=True) return await interaction.response.send_message(embed=await self.build_home_embed(guild, clan, "overview")) @clan.command(name="invite", description="Invite a user to your clan") async def invite(self, interaction: discord.Interaction, user: discord.Member) -> None: guild = await self.ensure_guild(interaction) clan, membership = await self.get_clan_and_membership(guild.id, interaction.user.id) if not clan or not membership: await interaction.response.send_message("You are not in a clan.", ephemeral=True) return if membership["role"] not in {"leader", "co_leader"}: await interaction.response.send_message("Only leaders and admins can invite users.", ephemeral=True) return if user.bot or user.id == interaction.user.id: await interaction.response.send_message("Pick a real user that isn't yourself.", ephemeral=True) return if self.db.get_user_membership(guild.id, user.id): await interaction.response.send_message("That user is already in a clan.", ephemeral=True) return expires_at = (datetime.now(timezone.utc) + timedelta(minutes=INVITE_EXPIRY_MINUTES)).isoformat() self.db.set_invite(guild.id, user.id, clan.id, interaction.user.id, expires_at) view = InviteView(self, clan.id, user.id) embed = discord.Embed( title="📨 Clan Invite", description=f"{user.mention}, {interaction.user.mention} invited you to **{clan.name}**.\nInvite expires in **{INVITE_EXPIRY_MINUTES} minutes**.", colour=self.to_colour(clan.color_hex), ) await interaction.response.send_message(embed=embed, view=view) try: view.message = await interaction.original_response() except discord.HTTPException: pass self.db.log(clan.id, "invite_sent", f"Invited {user.display_name}", interaction.user.id) await self.refresh_home(guild, clan.id, page="activity") @clan.command(name="apply", description="Apply to an open clan") async def apply(self, interaction: discord.Interaction, clan_name: str) -> None: guild = await self.ensure_guild(interaction) clan = self.db.get_clan_by_name(guild.id, clan_name) if clan is None: await interaction.response.send_message("Clan not found.", ephemeral=True) return if not clan.recruitment_open: await interaction.response.send_message("That clan is not recruiting right now.", ephemeral=True) return if self.db.get_user_membership(guild.id, interaction.user.id): await interaction.response.send_message("You are already in a clan.", ephemeral=True) return if self.db.get_pending_application(guild.id, interaction.user.id): await interaction.response.send_message("You already have a pending application.", ephemeral=True) return channel = guild.get_channel(clan.channel_id) if clan.channel_id else None if not isinstance(channel, discord.TextChannel): await interaction.response.send_message("That clan channel is missing.", ephemeral=True) return embed = discord.Embed( title="📝 Clan Application", description=f"Applicant: {interaction.user.mention}\nClan: **{clan.name}**", colour=self.to_colour(clan.color_hex), timestamp=datetime.now(timezone.utc), ) embed.add_field(name="Region", value=clan.region or "Not set", inline=True) embed.add_field(name="Playstyle", value=clan.playstyle or "Not set", inline=True) embed.add_field(name="Requirements", value=clan.requirements or "None", inline=False) message = await channel.send(embed=embed, view=self.application_view) self.db.set_application(guild.id, interaction.user.id, clan.id, message.id) self.db.log(clan.id, "application_submitted", f"{interaction.user.display_name} applied", interaction.user.id) await self.refresh_home(guild, clan.id, page="activity") await interaction.response.send_message(f"Application sent to **{clan.name}**.", ephemeral=True) @clan.command(name="promote", description="Promote a member to admin") async def promote(self, interaction: discord.Interaction, user: discord.Member) -> None: guild = await self.ensure_guild(interaction) clan, actor = await self.get_clan_and_membership(guild.id, interaction.user.id) target = self.db.get_user_membership(guild.id, user.id) if not clan or not actor: await interaction.response.send_message("You are not in a clan.", ephemeral=True) return if actor["role"] != "leader": await interaction.response.send_message("Only the leader can promote admins.", ephemeral=True) return if not target or target["clan_id"] != clan.id or target["role"] != "member": await interaction.response.send_message("You can only promote members from your clan.", ephemeral=True) return self.db.set_member_role(guild.id, user.id, "co_leader") self.db.log(clan.id, "promoted", f"Promoted {user.display_name} to admin", interaction.user.id) await self.refresh_home(guild, clan.id, page="members") await interaction.response.send_message(f"{user.mention} is now a clan **Admin**.") @clan.command(name="demote", description="Demote an admin to member") async def demote(self, interaction: discord.Interaction, user: discord.Member) -> None: guild = await self.ensure_guild(interaction) clan, actor = await self.get_clan_and_membership(guild.id, interaction.user.id) target = self.db.get_user_membership(guild.id, user.id) if not clan or not actor: await interaction.response.send_message("You are not in a clan.", ephemeral=True) return if actor["role"] != "leader": await interaction.response.send_message("Only the leader can demote admins.", ephemeral=True) return if not target or target["clan_id"] != clan.id or target["role"] != "co_leader": await interaction.response.send_message("You can only demote admins from your clan.", ephemeral=True) return self.db.set_member_role(guild.id, user.id, "member") self.db.log(clan.id, "demoted", f"Demoted {user.display_name} to member", interaction.user.id) await self.refresh_home(guild, clan.id, page="members") await interaction.response.send_message(f"{user.mention} is now a **Member**.") @clan.command(name="transfer", description="Transfer clan ownership") async def transfer(self, interaction: discord.Interaction, user: discord.Member) -> None: guild = await self.ensure_guild(interaction) clan, actor = await self.get_clan_and_membership(guild.id, interaction.user.id) target = self.db.get_user_membership(guild.id, user.id) if not clan or not actor: await interaction.response.send_message("You are not in a clan.", ephemeral=True) return if actor["role"] != "leader": await interaction.response.send_message("Only the leader can transfer ownership.", ephemeral=True) return if not target or target["clan_id"] != clan.id or user.id == interaction.user.id: await interaction.response.send_message("Choose another member in your clan.", ephemeral=True) return self.db.transfer_leadership(guild.id, clan.id, interaction.user.id, user.id) self.db.log(clan.id, "ownership_transferred", f"Transferred ownership to {user.display_name}", interaction.user.id) await self.refresh_home(guild, clan.id, page="members") await interaction.response.send_message(f"Transferred **{clan.name}** to {user.mention}.") @clan.command(name="kick", description="Kick a member from the clan") async def kick(self, interaction: discord.Interaction, user: discord.Member) -> None: guild = await self.ensure_guild(interaction) clan, actor = await self.get_clan_and_membership(guild.id, interaction.user.id) target = self.db.get_user_membership(guild.id, user.id) if not clan or not actor: await interaction.response.send_message("You are not in a clan.", ephemeral=True) return if not target or target["clan_id"] != clan.id: await interaction.response.send_message("That user is not in your clan.", ephemeral=True) return if user.id == interaction.user.id: await interaction.response.send_message("Use `/clan leave` instead.", ephemeral=True) return if target["role"] == "leader": await interaction.response.send_message("You cannot kick the leader.", ephemeral=True) return if actor["role"] == "member" or (actor["role"] == "co_leader" and target["role"] != "member"): await interaction.response.send_message("You do not have permission to kick that user.", ephemeral=True) return self.db.remove_member(guild.id, user.id) await self.sync_member_role(guild, user.id, None) self.db.log(clan.id, "member_kicked", f"Kicked {user.display_name}", interaction.user.id) await self.refresh_home(guild, clan.id, page="members") await interaction.response.send_message(f"{user.mention} was kicked from **{clan.name}**.") @clan.command(name="leave", description="Leave your clan") async def leave(self, interaction: discord.Interaction) -> None: guild = await self.ensure_guild(interaction) clan, membership = await self.get_clan_and_membership(guild.id, interaction.user.id) if not clan or not membership: await interaction.response.send_message("You are not in a clan.", ephemeral=True) return members = self.db.get_members(clan.id) if membership["role"] == "leader" and len(members) > 1: await interaction.response.send_message("Transfer ownership or disband the clan before leaving.", ephemeral=True) return if membership["role"] == "leader" and len(members) == 1: role = guild.get_role(clan.role_id) if clan.role_id else None channel = guild.get_channel(clan.channel_id) if clan.channel_id else None self.db.disband_clan(clan.id) if isinstance(channel, discord.TextChannel): try: await channel.delete(reason="Solo leader left and disbanded clan") except discord.HTTPException: pass if role: try: await role.delete(reason="Clan disbanded") except discord.HTTPException: pass await interaction.response.send_message(f"You left and disbanded **{clan.name}**.") return self.db.remove_member(guild.id, interaction.user.id) await self.sync_member_role(guild, interaction.user.id, None) self.db.log(clan.id, "member_left", f"{interaction.user.display_name} left", interaction.user.id) await self.refresh_home(guild, clan.id, page="members") await interaction.response.send_message(f"You left **{clan.name}**.", ephemeral=True) @clan.command(name="disband", description="Disband your clan") async def disband(self, interaction: discord.Interaction) -> None: guild = await self.ensure_guild(interaction) clan, membership = await self.get_clan_and_membership(guild.id, interaction.user.id) if not clan or not membership: await interaction.response.send_message("You are not in a clan.", ephemeral=True) return if membership["role"] != "leader": await interaction.response.send_message("Only the leader can disband the clan.", ephemeral=True) return role = guild.get_role(clan.role_id) if clan.role_id else None channel = guild.get_channel(clan.channel_id) if clan.channel_id else None members = self.db.get_members(clan.id) self.db.disband_clan(clan.id) for row in members: await self.sync_member_role(guild, row["user_id"], None) if isinstance(channel, discord.TextChannel): try: await channel.delete(reason="Clan disbanded") except discord.HTTPException: pass if role: try: await role.delete(reason="Clan disbanded") except discord.HTTPException: pass await interaction.response.send_message(f"**{clan.name}** has been disbanded.") @clan.command(name="rename", description="Rename the clan") async def rename(self, interaction: discord.Interaction, new_name: str) -> None: guild = await self.ensure_guild(interaction) clan, membership = await self.get_clan_and_membership(guild.id, interaction.user.id) if not clan or not membership: await interaction.response.send_message("You are not in a clan.", ephemeral=True) return if membership["role"] != "leader": await interaction.response.send_message("Only the leader can rename the clan.", ephemeral=True) return try: new_name = self.validate_name(new_name) except ValueError as exc: await interaction.response.send_message(str(exc), ephemeral=True) return existing = self.db.get_clan_by_name(guild.id, new_name) if existing and existing.id != clan.id: await interaction.response.send_message("Another clan already uses that name.", ephemeral=True) return role = await self.get_or_create_role(guild, clan) await role.edit(name=new_name, reason="Clan renamed") channel = guild.get_channel(clan.channel_id) if clan.channel_id else None if isinstance(channel, discord.TextChannel): try: await channel.edit(name=self.channel_name(new_name), reason="Clan renamed") except discord.HTTPException: pass self.db.update_clan(clan.id, name=new_name) self.db.log(clan.id, "renamed", f"Renamed clan to {new_name}", interaction.user.id) await self.refresh_home(guild, clan.id, page="overview") await interaction.response.send_message(f"Clan renamed to **{new_name}**.") @clan.command(name="setcolor", description="Change your clan color") async def setcolor(self, interaction: discord.Interaction, hex_color: str) -> None: guild = await self.ensure_guild(interaction) clan, membership = await self.get_clan_and_membership(guild.id, interaction.user.id) if not clan or not membership: await interaction.response.send_message("You are not in a clan.", ephemeral=True) return if membership["role"] != "leader": await interaction.response.send_message("Only the leader can change the clan color.", ephemeral=True) return try: hex_color = self.normalize_hex(hex_color) except ValueError as exc: await interaction.response.send_message(str(exc), ephemeral=True) return role = await self.get_or_create_role(guild, clan) await role.edit(colour=self.to_colour(hex_color), reason="Clan color updated") self.db.update_clan(clan.id, color_hex=hex_color) self.db.log(clan.id, "color_updated", hex_color, interaction.user.id) await self.refresh_home(guild, clan.id, page="overview") await interaction.response.send_message(f"Clan color updated to **{hex_color}**.") @clan.command(name="specialty", description="Set your clan specialty") async def specialty(self, interaction: discord.Interaction, specialty: str) -> None: guild = await self.ensure_guild(interaction) clan, membership = await self.get_clan_and_membership(guild.id, interaction.user.id) if not clan or not membership: await interaction.response.send_message("You are not in a clan.", ephemeral=True) return specialty = specialty.strip()[:40] self.db.set_member_specialty(guild.id, interaction.user.id, specialty) self.db.log(clan.id, "specialty_updated", specialty, interaction.user.id) await self.refresh_home(guild, clan.id, page="members") await interaction.response.send_message(f"Your specialty is now **{specialty}**.", ephemeral=True) @clan.command(name="note", description="Set a private admin note for a member") async def note(self, interaction: discord.Interaction, user: discord.Member, note: str) -> None: guild = await self.ensure_guild(interaction) clan, membership = await self.get_clan_and_membership(guild.id, interaction.user.id) target = self.db.get_user_membership(guild.id, user.id) if not clan or not membership: await interaction.response.send_message("You are not in a clan.", ephemeral=True) return if membership["role"] not in {"leader", "co_leader"}: await interaction.response.send_message("Only leaders and admins can set notes.", ephemeral=True) return if not target or target["clan_id"] != clan.id: await interaction.response.send_message("That user is not in your clan.", ephemeral=True) return note = note.strip()[:120] self.db.set_member_note(guild.id, user.id, note) self.db.log(clan.id, "note_updated", f"Updated note for {user.display_name}", interaction.user.id) await self.refresh_home(guild, clan.id, page="activity") await interaction.response.send_message(f"Saved a private note for {user.mention}.", ephemeral=True) @app_commands.choices(kind=[ app_commands.Choice(name="Raid", value="raid"), app_commands.Choice(name="Defend", value="defend"), app_commands.Choice(name="Roam", value="roam"), app_commands.Choice(name="Farm", value="farm"), app_commands.Choice(name="Wipe", value="wipe"), ]) @clan.command(name="announce", description="Post a preset clan announcement") async def announce(self, interaction: discord.Interaction, kind: app_commands.Choice[str], message: str) -> None: guild = await self.ensure_guild(interaction) clan, membership = await self.get_clan_and_membership(guild.id, interaction.user.id) if not clan or not membership: await interaction.response.send_message("You are not in a clan.", ephemeral=True) return if membership["role"] not in {"leader", "co_leader"}: await interaction.response.send_message("Only leaders and admins can announce.", ephemeral=True) return channel = guild.get_channel(clan.channel_id) if clan.channel_id else None if not isinstance(channel, discord.TextChannel): await interaction.response.send_message("Your clan channel is missing.", ephemeral=True) return emojis = {"raid": "💥", "defend": "🛡️", "roam": "🚗", "farm": "🌾", "wipe": "🧨"} embed = discord.Embed( title=f"{emojis[kind.value]} {kind.name} Alert", description=message, colour=self.to_colour(clan.color_hex), timestamp=datetime.now(timezone.utc), ) embed.set_footer(text=f"Posted by {interaction.user.display_name}") await channel.send(embed=embed) self.db.log(clan.id, "announcement", kind.value, interaction.user.id) await self.refresh_home(guild, clan.id, page="activity") await interaction.response.send_message("Announcement posted.", ephemeral=True) @clan.command(name="readycheck", description="Post a ready check") async def readycheck(self, interaction: discord.Interaction, title: Optional[str] = None) -> None: guild = await self.ensure_guild(interaction) clan, membership = await self.get_clan_and_membership(guild.id, interaction.user.id) if not clan or not membership: await interaction.response.send_message("You are not in a clan.", ephemeral=True) return if membership["role"] not in {"leader", "co_leader"}: await interaction.response.send_message("Only leaders and admins can start ready checks.", ephemeral=True) return channel = guild.get_channel(clan.channel_id) if clan.channel_id else None if not isinstance(channel, discord.TextChannel): await interaction.response.send_message("Your clan channel is missing.", ephemeral=True) return title = title or "⚔️ Ready Check" embed = discord.Embed(title=title, description="✅ Ready: **0**\n❔ Maybe: **0**\n❌ Not on: **0**", colour=self.to_colour(clan.color_hex), timestamp=datetime.now(timezone.utc)) embed.set_footer(text=f"Started by {interaction.user.display_name}") view = ReadyCheckView(title) msg = await channel.send(embed=embed, view=view) view.message = msg self.db.log(clan.id, "ready_check", title, interaction.user.id) await self.refresh_home(guild, clan.id, page="activity") await interaction.response.send_message("Ready check posted.", ephemeral=True) @clan.command(name="repair", description="Repair clan role, channel, and home message") async def repair(self, interaction: discord.Interaction) -> None: guild = await self.ensure_guild(interaction) clan, membership = await self.get_clan_and_membership(guild.id, interaction.user.id) if not clan or not membership: await interaction.response.send_message("You are not in a clan.", ephemeral=True) return if membership["role"] not in {"leader", "co_leader"}: await interaction.response.send_message("Only leaders and admins can run repairs.", ephemeral=True) return role = await self.get_or_create_role(guild, clan) await role.edit(name=clan.name, colour=self.to_colour(clan.color_hex), reason="Clan repair sync") channel = await self.ensure_clan_channel(guild, self.db.get_clan_by_id(clan.id), role) members = self.db.get_members(clan.id) for row in members: await self.sync_member_role(guild, row["user_id"], clan) await self.refresh_home(guild, clan.id, page="overview") self.db.log(clan.id, "repair", "Repaired role/channel/home", interaction.user.id) await interaction.response.send_message(f"Repair complete. Clan channel: {channel.mention}", ephemeral=True) @clan.command(name="help", description="Show clan help") async def help(self, interaction: discord.Interaction) -> None: embed = discord.Embed(title="Rust Clan Commands", colour=discord.Colour.blurple()) items = [ ("/clan create [color]", "Create your clan, role, private channel, and command center."), ("/clan invite ", "Invite someone to your clan."), ("/clan apply ", "Apply to a clan if recruitment is open."), ("/clan promote / demote / transfer", "Manage leadership."), ("/clan specialty ", "Set your role specialty like PvP, Builder, Farmer, Pilot."), ("/clan announce", "Post raid/defend/roam/farm/wipe alerts."), ("/clan readycheck", "Start a ready check with buttons."), ("/clan repair", "Recreate or sync the clan role/channel/home message."), ("Clan channel buttons", "Overview, roster, recruitment, activity, edit profile, banner, settings, announcement, ready check."), ] for name, value in items: embed.add_field(name=name, value=value, inline=False) await interaction.response.send_message(embed=embed, ephemeral=True) class ClanBot(commands.Bot): def __init__(self, db: ClanDatabase): intents = discord.Intents.default() super().__init__(command_prefix=commands.when_mentioned, intents=intents) self.db = db self.clan_cog: Optional[ClanCog] = None async def setup_hook(self) -> None: self.clan_cog = ClanCog(self, self.db) await self.add_cog(self.clan_cog) self.add_view(self.clan_cog.home_view) self.add_view(self.clan_cog.application_view) if TEST_GUILD_ID: guild = discord.Object(id=int(TEST_GUILD_ID)) self.tree.copy_global_to(guild=guild) synced = await self.tree.sync(guild=guild) logger.info("Synced %s commands to test guild %s", len(synced), TEST_GUILD_ID) else: synced = await self.tree.sync() logger.info("Synced %s global commands", len(synced)) async def close(self) -> None: try: if self.db: self.db.close() finally: await super().close() async def on_ready(self) -> None: logger.info("Logged in as %s (%s)", self.user, self.user.id if self.user else "unknown") def main() -> None: if not TOKEN or TOKEN == "your_bot_token_here": raise RuntimeError("DISCORD_TOKEN is missing or still set to a placeholder value in .env") db = ClanDatabase(DB_PATH) bot = ClanBot(db) try: bot.run(TOKEN) except discord.LoginFailure: logger.error("Discord rejected the token. Check DISCORD_TOKEN in your .env file.") raise if __name__ == "__main__": main()