import logging import os import random import secrets import string import discord from discord.ext import commands import pterodactyl as ptero from database import Database log = logging.getLogger("cogs.create") def is_admin_user(user_id: int) -> bool: """Check if user is owner, CEO, or admin from .env""" owner_ids = os.getenv("OWNER_IDS", "").split(",") ceo_ids = os.getenv("CEO_IDS", "").split(",") admin_ids = os.getenv("ADMIN_IDS", "").split(",") all_admin_ids = owner_ids + ceo_ids + admin_ids return str(user_id) in [aid.strip() for aid in all_admin_ids if aid.strip()] def _random_password(length: int = 16) -> str: chars = string.ascii_letters + string.digits + "!@#$%^&*" return "".join(secrets.choice(chars) for _ in range(length)) def _random_username(base: str) -> str: suffix = random.randint(1000, 9999) clean = "".join(c for c in base if c.isalnum())[:12] return f"{clean}{suffix}".lower() _DEFAULT_EGG_ID = int(os.getenv("PTERO_DEFAULT_EGG_ID", "1")) _DEFAULT_NEST_ID = int(os.getenv("PTERO_DEFAULT_NEST_ID", "1")) _FALLBACK_STARTUP = "java -Xms128M -Xmx{{SERVER_MEMORY}}M -jar {{SERVER_JARFILE}}" _FALLBACK_DOCKER = "ghcr.io/pterodactyl/yolks:java_17" _FALLBACK_ENV = {"SERVER_JARFILE": "server.jar", "VANILLA_VERSION": "latest"} class CreateCog(commands.Cog): def __init__(self, bot: commands.Bot): self.bot = bot self.db: Database = bot.db async def _create_server_for_user( self, ctx: commands.Context, email: str, password: str, ram: int, cpu: int, disk: int, node: int, plan_type: str ): """Internal method to create server""" if not is_admin_user(ctx.author.id): embed = discord.Embed( title="❌ Permission Denied", description="Only bot owners, CEOs, and admins can use this command.", color=discord.Color.red() ) await ctx.send(embed=embed) return await ctx.typing() username = _random_username(email.split("@")[0]) try: ptero_user = await ptero.app_create_user( email=email, username=username, first_name="Server", last_name="User", password=password, ) except ptero.PterodactylError as e: embed = discord.Embed( title="❌ User Creation Failed", description=f"Could not create Pterodactyl user:\n```{e}```", color=discord.Color.red(), ) await ctx.send(embed=embed) return ptero_id = int(ptero_user.get("id", 0)) try: allocation_id = await ptero.app_get_free_allocation(node) except ptero.PterodactylError as e: embed = discord.Embed( title="❌ No Free Ports", description=f"Could not find free port on node {node}:\n```{e}```", color=discord.Color.red(), ) await ctx.send(embed=embed) return try: egg_data = await ptero.app_get_egg(_DEFAULT_NEST_ID, _DEFAULT_EGG_ID) startup = egg_data.get("startup") or _FALLBACK_STARTUP docker_image = egg_data.get("docker_image") or _FALLBACK_DOCKER variables = egg_data.get("relationships", {}).get("variables", {}).get("data", []) environment = {} for var in variables: attrs = var.get("attributes", {}) key = attrs.get("env_variable", "") val = attrs.get("default_value", "") if key: environment[key] = val if not environment: environment = dict(_FALLBACK_ENV) except ptero.PterodactylError: startup = _FALLBACK_STARTUP docker_image = _FALLBACK_DOCKER environment = dict(_FALLBACK_ENV) server_name = f"{plan_type}-{username}" try: server = await ptero.app_create_server( name=server_name, user_id=ptero_id, egg_id=_DEFAULT_EGG_ID, nest_id=_DEFAULT_NEST_ID, allocation_id=allocation_id, ram_mb=ram, cpu=cpu, disk_mb=disk, startup=startup, docker_image=docker_image, environment=environment, ) except ptero.PterodactylError as e: embed = discord.Embed( title="❌ Server Creation Failed", description=f"Could not create server:\n```{e}```", color=discord.Color.red(), ) await ctx.send(embed=embed) return server_id = server.get("identifier") or str(server.get("id", "unknown")) embed = discord.Embed( title="✅ Server Created Successfully!", description=f"**{plan_type.upper()}** server has been provisioned!", color=discord.Color.green() ) embed.add_field(name="🆔 Server ID", value=f"`{server_id}`", inline=True) embed.add_field(name="📧 Email", value=f"`{email}`", inline=True) embed.add_field(name="🔑 Password", value=f"`{password}`", inline=True) embed.add_field(name="🧠 RAM", value=f"`{ram}MB`", inline=True) embed.add_field(name="⚡ CPU", value=f"`{cpu}%`", inline=True) embed.add_field(name="💾 Disk", value=f"`{disk}MB`", inline=True) embed.add_field(name="🖥️ Node", value=f"`{node}`", inline=True) embed.add_field(name="🔌 Port", value=f"`{allocation_id}`", inline=True) embed.add_field(name="📦 Plan Type", value=f"`{plan_type.upper()}`", inline=True) embed.set_footer(text="TheRynzo PteroControl • Server Management") await ctx.send(embed=embed) log.info(f"Admin {ctx.author} created {plan_type} server: {server_id} for {email}") @commands.group(name="create", invoke_without_command=True) async def create(self, ctx: commands.Context): """Create server commands""" if ctx.invoked_subcommand is None: embed = discord.Embed( title="📝 Create Server Commands", description="Use one of the following subcommands:", color=discord.Color.blue() ) embed.add_field( name="`.create paid `", value="Create a paid server (Plan: Paid 1)", inline=False ) embed.add_field( name="`.create free `", value="Create a free server (Plan: Free 1)", inline=False ) embed.add_field( name="Example", value="`.create paid user@example.com MyPass123 1024 100 10240 1`", inline=False ) await ctx.send(embed=embed) @create.command(name="paid") async def create_paid( self, ctx: commands.Context, email: str, password: str, ram: int, cpu: int, disk: int, node: int ): """Create a paid server - Admin only""" await self._create_server_for_user(ctx, email, password, ram, cpu, disk, node, "Paid 1") @create.command(name="free") async def create_free( self, ctx: commands.Context, email: str, password: str, ram: int, cpu: int, disk: int, node: int ): """Create a free server - Admin only""" await self._create_server_for_user(ctx, email, password, ram, cpu, disk, node, "Free 1") async def setup(bot: commands.Bot): await bot.add_cog(CreateCog(bot))