import discord from discord import app_commands from discord.ext import commands import aiohttp import asyncio import json import logging from datetime import datetime, timezone logger = logging.getLogger("bot.sms") SMSPOOL_BASE = "https://api.smspool.net" BLACKLIST_PATH = "data/blacklist.json" DISCORD_SERVICE = "discord" # Full country list for autocomplete COUNTRIES = [ ("United States", "US"), ("United Kingdom", "GB"), ("Canada", "CA"), ("Australia", "AU"), ("Germany", "DE"), ("France", "FR"), ("Netherlands", "NL"), ("Sweden", "SE"), ("Norway", "NO"), ("Denmark", "DK"), ("Finland", "FI"), ("Spain", "ES"), ("Italy", "IT"), ("Poland", "PL"), ("Portugal", "PT"), ("Belgium", "BE"), ("Austria", "AT"), ("Switzerland", "CH"), ("Russia", "RU"), ("Ukraine", "UA"), ("Brazil", "BR"), ("Mexico", "MX"), ("Argentina", "AR"), ("Colombia", "CO"), ("Chile", "CL"), ("India", "IN"), ("Indonesia", "ID"), ("Philippines", "PH"), ("Vietnam", "VN"), ("Thailand", "TH"), ("Malaysia", "MY"), ("Singapore", "SG"), ("Hong Kong", "HK"), ("Japan", "JP"), ("South Korea", "KR"), ("Turkey", "TR"), ("Israel", "IL"), ("Saudi Arabia", "SA"), ("UAE", "AE"), ("Egypt", "EG"), ("Nigeria", "NG"), ("South Africa", "ZA"), ("Kenya", "KE"), ("Ghana", "GH"), ("Romania", "RO"), ("Czech Republic", "CZ"), ("Hungary", "HU"), ("Bulgaria", "BG"), ("Croatia", "HR"), ("Slovakia", "SK"), ("Lithuania", "LT"), ("Latvia", "LV"), ("Estonia", "EE"), ("Serbia", "RS"), ("Moldova", "MD"), ("Georgia", "GE"), ("Armenia", "AM"), ("Azerbaijan", "AZ"), ("Kazakhstan", "KZ"), ("Pakistan", "PK"), ("Bangladesh", "BD"), ("Sri Lanka", "LK"), ("Nepal", "NP"), ("Cambodia", "KH"), ("Myanmar", "MM"), ("New Zealand", "NZ"), ("Peru", "PE"), ("Ecuador", "EC"), ("Bolivia", "BO"), ("Paraguay", "PY"), ("Uruguay", "UY"), ("Guatemala", "GT"), ("Honduras", "HN"), ("El Salvador", "SV"), ("Nicaragua", "NI"), ("Costa Rica", "CR"), ("Panama", "PA"), ("Dominican Republic", "DO"), ("Cuba", "CU"), ("Jamaica", "JM"), ("Haiti", "HT"), ("Trinidad and Tobago", "TT"), ("Barbados", "BB"), ("Morocco", "MA"), ("Algeria", "DZ"), ("Tunisia", "TN"), ("Libya", "LY"), ("Sudan", "SD"), ("Ethiopia", "ET"), ("Tanzania", "TZ"), ("Uganda", "UG"), ("Cameroon", "CM"), ("Ivory Coast", "CI"), ("Senegal", "SN"), ("Zimbabwe", "ZW"), ] def load_blacklist() -> list: try: with open(BLACKLIST_PATH, "r") as f: return json.load(f) except (FileNotFoundError, json.JSONDecodeError): return [] def _load_admins() -> list: try: with open("data/admins.json", "r") as f: return json.load(f) except (FileNotFoundError, json.JSONDecodeError): return [] async def _delete_channel_after(channel: discord.TextChannel, delay: int, reason: str): await asyncio.sleep(delay) try: await channel.delete(reason=reason) except discord.HTTPException: pass class CancelView(discord.ui.View): def __init__(self, cog: "SMSCog", order_id: str, user_id: int, channel: discord.TextChannel): super().__init__(timeout=320) self.cog = cog self.order_id = order_id self.user_id = user_id self.channel = channel self.cancelled = False def _has_permission(self, interaction: discord.Interaction) -> bool: if interaction.user.id == self.user_id: return True if interaction.user.id == interaction.guild.owner_id: return True config = self.cog.bot.config admin_role_id = config.get("admin_role_id") if admin_role_id: role = interaction.guild.get_role(int(admin_role_id)) if role and role in interaction.user.roles: return True return interaction.user.id in _load_admins() @discord.ui.button(label="Cancel Order", style=discord.ButtonStyle.danger, emoji="❌") async def cancel_button(self, interaction: discord.Interaction, button: discord.ui.Button): if not self._has_permission(interaction): await interaction.response.send_message( "❌ Only the order owner or staff can cancel this.", ephemeral=True ) return self.cancelled = True self.stop() await interaction.response.defer() try: await self.cog.cancel_order(self.order_id) except Exception as e: logger.warning(f"Cancel API failed for {self.order_id}: {e}") self.cog.bot.active_orders.pop(self.user_id, None) embed = discord.Embed( title="❌ Order Cancelled", description="Your SMS order was cancelled. This channel closes in **10 seconds**.", color=discord.Color.red(), timestamp=datetime.now(timezone.utc) ) await interaction.followup.send(embed=embed) asyncio.create_task(_delete_channel_after(self.channel, 10, "SMS order cancelled")) class SMSCog(commands.Cog): def __init__(self, bot: commands.Bot): self.bot = bot # ─── SMSPool API ────────────────────────────────────────────────────────── async def order_number(self, country: str) -> dict: key = self.bot.config.get("smspool_api_key", "") async with aiohttp.ClientSession() as session: async with session.post( f"{SMSPOOL_BASE}/purchase/sms", data={"key": key, "country": country, "service": DISCORD_SERVICE} ) as resp: return await resp.json(content_type=None) async def check_sms(self, order_id: str) -> dict: key = self.bot.config.get("smspool_api_key", "") async with aiohttp.ClientSession() as session: async with session.get( f"{SMSPOOL_BASE}/sms/check", params={"key": key, "orderid": order_id} ) as resp: return await resp.json(content_type=None) async def cancel_order(self, order_id: str) -> dict: key = self.bot.config.get("smspool_api_key", "") async with aiohttp.ClientSession() as session: async with session.post( f"{SMSPOOL_BASE}/sms/cancel", data={"key": key, "orderid": order_id} ) as resp: return await resp.json(content_type=None) # ─── Channel + Webhook creation ─────────────────────────────────────────── async def create_sms_channel( self, guild: discord.Guild, user: discord.Member ) -> discord.TextChannel: config = self.bot.config category: discord.CategoryChannel | None = None cat_id = config.get("sms_category_id") if cat_id: category = guild.get_channel(int(cat_id)) overwrites = { guild.default_role: discord.PermissionOverwrite(read_messages=False), user: discord.PermissionOverwrite( read_messages=True, send_messages=False, attach_files=False, ), guild.me: discord.PermissionOverwrite( read_messages=True, send_messages=True, manage_channels=True, manage_webhooks=True, ), } admin_role_id = config.get("admin_role_id") if admin_role_id: role = guild.get_role(int(admin_role_id)) if role: overwrites[role] = discord.PermissionOverwrite( read_messages=True, send_messages=True ) safe_name = "".join( c for c in user.display_name.lower().replace(" ", "-") if c.isalnum() or c == "-" )[:20] channel = await guild.create_text_channel( name=f"sms-{safe_name}", category=category, overwrites=overwrites, topic=f"Discord SMS session for {user} ({user.id})", reason=f"Discord SMS session opened for {user}", ) return channel async def get_or_create_webhook(self, channel: discord.TextChannel) -> discord.Webhook: webhooks = await channel.webhooks() for wh in webhooks: if wh.name == "SMSPool Delivery": return wh return await channel.create_webhook(name="SMSPool Delivery") # ─── Poll for SMS ───────────────────────────────────────────────────────── async def poll_for_sms( self, order_id: str, user_id: int, channel: discord.TextChannel, cancel_view: CancelView, ) -> None: config = self.bot.config interval = int(config.get("poll_interval", 8)) timeout = int(config.get("poll_timeout", 300)) close_delay = int(config.get("channel_close_delay", 10)) webhook = await self.get_or_create_webhook(channel) elapsed = 0 while elapsed < timeout: if cancel_view.cancelled: return await asyncio.sleep(interval) elapsed += interval try: data = await self.check_sms(order_id) except Exception as e: logger.warning(f"Poll error for order {order_id}: {e}") continue status = data.get("status", 0) # SMS received (status == 3 or sms field populated) sms_code = data.get("sms") or data.get("code") if sms_code or status == 3: sms_code = sms_code or "—" self.bot.active_orders.pop(user_id, None) cancel_view.cancelled = True cancel_view.stop() # Send via webhook so it looks clean embed = discord.Embed( title="✅ SMS Received!", color=discord.Color.brand_green(), timestamp=datetime.now(timezone.utc) ) embed.add_field(name="📨 Verification Code", value=f"```{sms_code}```", inline=False) embed.add_field(name="📱 Service", value="Discord", inline=True) embed.add_field(name="Full SMS", value=data.get("full_sms", sms_code), inline=False) embed.set_footer(text=f"Channel deletes in {close_delay}s") try: await webhook.send( content=f"<@{user_id}>", embed=embed, username="SMSPool Delivery", avatar_url="https://smspool.net/favicon.ico", ) except Exception: await channel.send(content=f"<@{user_id}>", embed=embed) await self._log_action( channel.guild, f"✅ SMS received for <@{user_id}> | Order `{order_id}` | Code: `{sms_code}`" ) asyncio.create_task(_delete_channel_after(channel, close_delay, "SMS verified — auto-close")) return # Order expired/failed if status in (2, 4, 5): self.bot.active_orders.pop(user_id, None) cancel_view.cancelled = True cancel_view.stop() embed = discord.Embed( title="⌛ Order Expired", description="No SMS was received and the order has expired.\nThis channel closes in **15 seconds**.", color=discord.Color.orange(), timestamp=datetime.now(timezone.utc) ) try: await webhook.send( content=f"<@{user_id}>", embed=embed, username="SMSPool Delivery", avatar_url="https://smspool.net/favicon.ico", ) except Exception: await channel.send(content=f"<@{user_id}>", embed=embed) asyncio.create_task(_delete_channel_after(channel, 15, "SMS order expired")) return # Timed out on our end if not cancel_view.cancelled: self.bot.active_orders.pop(user_id, None) cancel_view.cancelled = True cancel_view.stop() try: await self.cancel_order(order_id) except Exception: pass embed = discord.Embed( title="⏰ Session Timed Out", description="No SMS was received within the time limit. The order has been cancelled.\nThis channel closes in **15 seconds**.", color=discord.Color.orange(), timestamp=datetime.now(timezone.utc) ) try: await webhook.send( content=f"<@{user_id}>", embed=embed, username="SMSPool Delivery", avatar_url="https://smspool.net/favicon.ico", ) except Exception: await channel.send(content=f"<@{user_id}>", embed=embed) asyncio.create_task(_delete_channel_after(channel, 15, "SMS session timed out")) # ─── Logging helper ─────────────────────────────────────────────────────── async def _log_action(self, guild: discord.Guild, message: str): log_id = self.bot.config.get("log_channel_id") if not log_id: return ch = guild.get_channel(int(log_id)) if ch: try: await ch.send(message) except Exception: pass # ─── Country autocomplete ───────────────────────────────────────────────── async def country_autocomplete( self, interaction: discord.Interaction, current: str ) -> list[app_commands.Choice[str]]: current = current.lower() results = [ app_commands.Choice(name=f"{name} ({code})", value=code) for name, code in COUNTRIES if current in name.lower() or current in code.lower() ] return results[:25] # ─── /sms ───────────────────────────────────────────────────────────────── @app_commands.command( name="sms", description="Generate a Discord phone number to receive a Discord verification SMS" ) @app_commands.describe(country="Country to get a number from (start typing to search)") @app_commands.autocomplete(country=country_autocomplete) async def sms_command( self, interaction: discord.Interaction, country: str = None, ): await interaction.response.defer(ephemeral=True) self.bot.reload_config() config = self.bot.config # Blacklist check if interaction.user.id in load_blacklist(): embed = discord.Embed( title="🚫 Access Denied", description="You are **blacklisted** from using the SMS command.\nContact a staff member if you believe this is an error.", color=discord.Color.red(), ) await interaction.followup.send(embed=embed, ephemeral=True) return # Pause check if config.get("sms_paused"): owner_id = config.get("bot_owner_id") is_owner_bypass = owner_id and interaction.user.id == int(owner_id) if not is_owner_bypass: embed = discord.Embed( title="⏸️ SMS Temporarily Unavailable", description="The SMS generation system is currently **paused** by an administrator.\nPlease try again later.", color=discord.Color.orange(), ) await interaction.followup.send(embed=embed, ephemeral=True) return # Duplicate order check if interaction.user.id in self.bot.active_orders: info = self.bot.active_orders[interaction.user.id] embed = discord.Embed( title="⚠️ Active Order Exists", description=( f"You already have an open SMS session.\n" f"Go to <#{info['channel_id']}> or use `/sms_cancel` to close it first." ), color=discord.Color.yellow(), ) await interaction.followup.send(embed=embed, ephemeral=True) return effective_country = (country or config.get("sms_default_country", "US")).upper() # Create private channel try: channel = await self.create_sms_channel(interaction.guild, interaction.user) except discord.Forbidden: await interaction.followup.send( "❌ I lack permission to create channels. Please contact an admin.", ephemeral=True ) return except Exception as e: logger.error(f"Channel creation failed: {e}") await interaction.followup.send("❌ Failed to create your SMS channel.", ephemeral=True) return # Direct user to channel await interaction.followup.send( embed=discord.Embed( title="📱 SMS Session Opening", description=f"Your private channel is ready: {channel.mention}", color=discord.Color.blurple(), ), ephemeral=True, ) # Order number from SMSPool try: order_data = await self.order_number(effective_country) except Exception as e: logger.error(f"SMSPool order failed: {e}") await channel.send("❌ SMSPool API error. Please try again.") asyncio.create_task(_delete_channel_after(channel, 10, "API error")) return if not order_data.get("success") and "number" not in order_data: err_msg = order_data.get("message", str(order_data)) embed = discord.Embed( title="❌ Order Failed", description=f"SMSPool could not provide a number.\n```{err_msg}```", color=discord.Color.red(), ) await channel.send(embed=embed) asyncio.create_task(_delete_channel_after(channel, 15, "Order failed")) return order_id = str(order_data.get("order_id", order_data.get("orderid", ""))) phone_number = order_data.get("number", "Unknown") # Register active order self.bot.active_orders[interaction.user.id] = { "order_id": order_id, "channel_id": channel.id, "country": effective_country, "started_at": datetime.now(timezone.utc).isoformat(), } # Post order info in the channel via webhook try: webhook = await self.get_or_create_webhook(channel) except Exception: webhook = None # Country display name country_label = next( (name for name, code in COUNTRIES if code == effective_country), effective_country ) order_embed = discord.Embed( title="📱 Discord SMS Number Ready", description=( f"Use the number below to verify your **Discord** account.\n" f"The bot will automatically detect your SMS and post it here." ), color=discord.Color.blurple(), timestamp=datetime.now(timezone.utc), ) order_embed.add_field(name="📞 Phone Number", value=f"```+{phone_number}```", inline=False) order_embed.add_field(name="🌍 Country", value=f"{country_label} (`{effective_country}`)", inline=True) order_embed.add_field(name="🎯 Service", value="Discord", inline=True) order_embed.add_field(name="🔢 Order ID", value=f"`{order_id}`", inline=True) order_embed.add_field( name="⏱️ Timeout", value=f"{int(config.get('poll_timeout', 300)) // 60} minutes", inline=True, ) order_embed.set_footer(text="Channel will auto-delete once SMS is received or timeout is reached.") cancel_view = CancelView(self, order_id, interaction.user.id, channel) try: if webhook: await webhook.send( content=f"<@{interaction.user.id}>", embed=order_embed, view=cancel_view, username="SMSPool Delivery", avatar_url="https://smspool.net/favicon.ico", ) else: await channel.send( content=f"<@{interaction.user.id}>", embed=order_embed, view=cancel_view, ) except Exception as e: logger.error(f"Failed to send order embed: {e}") await self._log_action( interaction.guild, f"📱 SMS order started by <@{interaction.user.id}> | " f"Number: `+{phone_number}` | Country: `{effective_country}` | Order: `{order_id}`" ) # Start polling — runs in background asyncio.create_task( self.poll_for_sms(order_id, interaction.user.id, channel, cancel_view) ) # ─── /sms_cancel ───────────────────────────────────────────────────────── @app_commands.command(name="sms_cancel", description="Cancel your active SMS order and close the session channel") async def sms_cancel(self, interaction: discord.Interaction): await interaction.response.defer(ephemeral=True) order_info = self.bot.active_orders.get(interaction.user.id) # Allow admins to cancel from outside the channel too is_admin_user = interaction.user.id in _load_admins() is_owner_user = interaction.user.id == interaction.guild.owner_id owner_id = self.bot.config.get("bot_owner_id") if owner_id: is_owner_user = is_owner_user or interaction.user.id == int(owner_id) if not order_info and not (is_admin_user or is_owner_user): await interaction.followup.send( embed=discord.Embed( title="⚠️ No Active Order", description="You don't have an active SMS session to cancel.", color=discord.Color.yellow(), ), ephemeral=True, ) return if order_info: try: await self.cancel_order(order_info["order_id"]) except Exception as e: logger.warning(f"Cancel API failed: {e}") self.bot.active_orders.pop(interaction.user.id, None) ch = interaction.guild.get_channel(order_info["channel_id"]) if ch: embed = discord.Embed( title="❌ Session Cancelled", description="Your SMS order was cancelled via `/sms_cancel`. Channel closes in **10 seconds**.", color=discord.Color.red(), timestamp=datetime.now(timezone.utc), ) await ch.send(embed=embed) asyncio.create_task(_delete_channel_after(ch, 10, "Cancelled via /sms_cancel")) await interaction.followup.send( embed=discord.Embed( title="✅ Order Cancelled", description="Your SMS session has been cancelled.", color=discord.Color.green(), ), ephemeral=True, ) async def setup(bot: commands.Bot): await bot.add_cog(SMSCog(bot))