import random import asyncio from datetime import datetime, timedelta from typing import Optional import discord from discord.ext import commands from discord import app_commands from sqlalchemy import select, func, desc from bot.models.sql_models import EconomyAccount, ShopItem from bot.utils.embeds import EmbedBuilder from bot.utils.pagination import Paginator DAILY_BASE = 500 DAILY_STREAK_BONUS = 50 WORK_MIN = 100 WORK_MAX = 350 CRIME_MIN = 200 CRIME_MAX = 800 CRIME_FAIL_CHANCE = 0.35 CRIME_FINE_MIN = 100 CRIME_FINE_MAX = 400 WORK_MESSAGES = [ "You worked as a software engineer and earned {amount} coins! 💻", "You drove a taxi and earned {amount} coins! 🚕", "You cooked food at a restaurant and earned {amount} coins! 🍳", "You delivered packages and earned {amount} coins! 📦", "You streamed on Twitch and earned {amount} coins! 🎮", "You wrote code as a freelancer and earned {amount} coins! ⌨️", ] CRIME_SUCCESS = [ "You hacked a database and stole {amount} coins! 💻", "You robbed a bank vault and grabbed {amount} coins! 🏦", "You sold stolen NFTs for {amount} coins! 🎨", ] CRIME_FAIL = [ "The police caught you! You were fined {amount} coins! 🚔", "Your plan failed miserably! You lost {amount} coins in legal fees! ⚖️", ] class Economy(commands.Cog): def __init__(self, bot): self.bot = bot async def _get_or_create_account(self, guild_id: str, user_id: str) -> EconomyAccount: async with self.bot.db.session() as session: result = await session.execute( select(EconomyAccount).where( EconomyAccount.guild_id == guild_id, EconomyAccount.user_id == user_id ) ) account = result.scalar_one_or_none() if not account: async with session.begin(): account = EconomyAccount(guild_id=guild_id, user_id=user_id) session.add(account) async with session.begin(): result = await session.execute( select(EconomyAccount).where( EconomyAccount.guild_id == guild_id, EconomyAccount.user_id == user_id ) ) account = result.scalar_one() return account async def _update_account(self, guild_id: str, user_id: str, **kwargs): async with self.bot.db.session() as session: async with session.begin(): result = await session.execute( select(EconomyAccount).where( EconomyAccount.guild_id == guild_id, EconomyAccount.user_id == user_id ) ) account = result.scalar_one() for key, value in kwargs.items(): setattr(account, key, value) account.net_worth = account.wallet + account.bank @app_commands.command(name="balance", description="Check your or another user's balance") @app_commands.describe(user="User to check balance (defaults to yourself)") async def balance(self, interaction: discord.Interaction, user: Optional[discord.Member] = None): target = user or interaction.user account = await self._get_or_create_account(str(interaction.guild.id), str(target.id)) embed = discord.Embed( title=f"💰 {target.display_name}'s Balance", color=0xFFD700, timestamp=datetime.utcnow() ) embed.set_thumbnail(url=target.display_avatar.url) embed.add_field(name="👛 Wallet", value=f"**{account.wallet:,}** coins", inline=True) embed.add_field(name="🏦 Bank", value=f"**{account.bank:,}** / {account.bank_capacity:,} coins", inline=True) embed.add_field(name="💎 Net Worth", value=f"**{account.net_worth:,}** coins", inline=True) embed.add_field(name="🏆 Prestige", value=f"Level **{account.prestige}**", inline=True) embed.add_field(name="🔥 Streak", value=f"**{account.streak}** days", inline=True) await interaction.response.send_message(embed=embed) @app_commands.command(name="daily", description="Claim your daily coins") async def daily(self, interaction: discord.Interaction): account = await self._get_or_create_account(str(interaction.guild.id), str(interaction.user.id)) now = datetime.utcnow() if account.last_daily and (now - account.last_daily).total_seconds() < 86400: remaining = 86400 - (now - account.last_daily).total_seconds() h, m = divmod(int(remaining // 60), 60) await interaction.response.send_message( f"⏰ Daily already claimed! Next claim in **{h}h {m}m**.", ephemeral=True ) return streak = account.streak + 1 if account.last_daily and (now - account.last_daily).total_seconds() < 172800 else 1 amount = DAILY_BASE + (min(streak, 30) * DAILY_STREAK_BONUS) new_wallet = account.wallet + amount await self._update_account( str(interaction.guild.id), str(interaction.user.id), wallet=new_wallet, last_daily=now, streak=streak, total_earned=account.total_earned + amount ) embed = discord.Embed( title="🎁 Daily Reward Claimed!", description=f"You received **{amount:,}** coins!\n🔥 Current Streak: **{streak}** days", color=0x00FF88, timestamp=now ) if streak >= 7: embed.add_field(name="🎉 Streak Bonus", value=f"+{min(streak, 30) * DAILY_STREAK_BONUS:,} coins", inline=True) embed.set_footer(text=f"💰 New Balance: {new_wallet:,} coins") await interaction.response.send_message(embed=embed) @app_commands.command(name="work", description="Work to earn coins") async def work(self, interaction: discord.Interaction): account = await self._get_or_create_account(str(interaction.guild.id), str(interaction.user.id)) now = datetime.utcnow() cooldown = 3600 if account.last_work and (now - account.last_work).total_seconds() < cooldown: remaining = cooldown - (now - account.last_work).total_seconds() m = int(remaining // 60) await interaction.response.send_message(f"⏰ You must rest before working again! **{m}m** remaining.", ephemeral=True) return amount = random.randint(WORK_MIN, WORK_MAX) msg = random.choice(WORK_MESSAGES).format(amount=f"{amount:,}") new_wallet = account.wallet + amount await self._update_account( str(interaction.guild.id), str(interaction.user.id), wallet=new_wallet, last_work=now, total_earned=account.total_earned + amount ) embed = discord.Embed(description=f"💼 {msg}", color=0x00BFFF, timestamp=now) embed.set_footer(text=f"💰 Balance: {new_wallet:,} coins") await interaction.response.send_message(embed=embed) self.bot.metrics.record_economy_transaction("work") @app_commands.command(name="crime", description="Commit a crime for big rewards (risky!)") async def crime(self, interaction: discord.Interaction): account = await self._get_or_create_account(str(interaction.guild.id), str(interaction.user.id)) now = datetime.utcnow() cooldown = 7200 if account.last_crime and (now - account.last_crime).total_seconds() < cooldown: remaining = cooldown - (now - account.last_crime).total_seconds() h, m = divmod(int(remaining // 60), 60) await interaction.response.send_message(f"⏰ Lay low! Next crime in **{h}h {m}m**.", ephemeral=True) return failed = random.random() < CRIME_FAIL_CHANCE if failed: fine = random.randint(CRIME_FINE_MIN, CRIME_FINE_MAX) new_wallet = max(0, account.wallet - fine) msg = random.choice(CRIME_FAIL).format(amount=f"{fine:,}") await self._update_account( str(interaction.guild.id), str(interaction.user.id), wallet=new_wallet, last_crime=now ) embed = discord.Embed(description=f"🚔 {msg}", color=0xFF4444, timestamp=now) embed.set_footer(text=f"💰 Balance: {new_wallet:,} coins") await interaction.response.send_message(embed=embed) else: amount = random.randint(CRIME_MIN, CRIME_MAX) msg = random.choice(CRIME_SUCCESS).format(amount=f"{amount:,}") new_wallet = account.wallet + amount await self._update_account( str(interaction.guild.id), str(interaction.user.id), wallet=new_wallet, last_crime=now, total_earned=account.total_earned + amount ) embed = discord.Embed(description=f"😈 {msg}", color=0xAA00FF, timestamp=now) embed.set_footer(text=f"💰 Balance: {new_wallet:,} coins") await interaction.response.send_message(embed=embed) @app_commands.command(name="deposit", description="Deposit coins into your bank") @app_commands.describe(amount="Amount to deposit (or 'all')") async def deposit(self, interaction: discord.Interaction, amount: str): account = await self._get_or_create_account(str(interaction.guild.id), str(interaction.user.id)) amt = account.wallet if amount.lower() == "all" else int(amount) if amt <= 0: await interaction.response.send_message("❌ Invalid amount.", ephemeral=True) return if amt > account.wallet: await interaction.response.send_message(f"❌ You only have **{account.wallet:,}** coins in your wallet.", ephemeral=True) return space = account.bank_capacity - account.bank actual = min(amt, space) if actual <= 0: await interaction.response.send_message("❌ Your bank is full!", ephemeral=True) return await self._update_account( str(interaction.guild.id), str(interaction.user.id), wallet=account.wallet - actual, bank=account.bank + actual ) embed = EmbedBuilder.success( "🏦 Deposit Successful", f"Deposited **{actual:,}** coins.\n💰 Wallet: **{account.wallet - actual:,}** | 🏦 Bank: **{account.bank + actual:,}**" ) await interaction.response.send_message(embed=embed) @app_commands.command(name="withdraw", description="Withdraw coins from your bank") @app_commands.describe(amount="Amount to withdraw (or 'all')") async def withdraw(self, interaction: discord.Interaction, amount: str): account = await self._get_or_create_account(str(interaction.guild.id), str(interaction.user.id)) amt = account.bank if amount.lower() == "all" else int(amount) if amt <= 0 or amt > account.bank: await interaction.response.send_message(f"❌ Invalid amount. Bank balance: **{account.bank:,}**", ephemeral=True) return await self._update_account( str(interaction.guild.id), str(interaction.user.id), wallet=account.wallet + amt, bank=account.bank - amt ) embed = EmbedBuilder.success( "💵 Withdrawal Successful", f"Withdrew **{amt:,}** coins.\n💰 Wallet: **{account.wallet + amt:,}** | 🏦 Bank: **{account.bank - amt:,}**" ) await interaction.response.send_message(embed=embed) @app_commands.command(name="pay", description="Transfer coins to another user") @app_commands.describe(user="User to pay", amount="Amount to transfer") async def pay(self, interaction: discord.Interaction, user: discord.Member, amount: int): if user.bot or user == interaction.user: await interaction.response.send_message("❌ Invalid transaction target.", ephemeral=True) return if amount <= 0: await interaction.response.send_message("❌ Amount must be positive.", ephemeral=True) return account = await self._get_or_create_account(str(interaction.guild.id), str(interaction.user.id)) if amount > account.wallet: await interaction.response.send_message(f"❌ Insufficient funds. Wallet: **{account.wallet:,}**", ephemeral=True) return target_account = await self._get_or_create_account(str(interaction.guild.id), str(user.id)) await self._update_account(str(interaction.guild.id), str(interaction.user.id), wallet=account.wallet - amount) await self._update_account(str(interaction.guild.id), str(user.id), wallet=target_account.wallet + amount) embed = EmbedBuilder.success( "💸 Transfer Complete", f"Sent **{amount:,}** coins to {user.mention}" ) await interaction.response.send_message(embed=embed) @app_commands.command(name="leaderboard", description="View the richest members") async def leaderboard(self, interaction: discord.Interaction): await interaction.response.defer() async with self.bot.db.session() as session: result = await session.execute( select(EconomyAccount).where( EconomyAccount.guild_id == str(interaction.guild.id) ).order_by(desc(EconomyAccount.net_worth)).limit(10) ) accounts = result.scalars().all() if not accounts: await interaction.followup.send("No economy data found.", ephemeral=True) return embed = discord.Embed(title="🏆 Richest Members", color=0xFFD700, timestamp=datetime.utcnow()) medals = ["🥇", "🥈", "🥉"] for i, acc in enumerate(accounts): member = interaction.guild.get_member(int(acc.user_id)) name = member.display_name if member else f"User {acc.user_id}" medal = medals[i] if i < 3 else f"**#{i+1}**" embed.add_field( name=f"{medal} {name}", value=f"💎 Net Worth: **{acc.net_worth:,}** | 👛 Wallet: **{acc.wallet:,}**", inline=False ) await interaction.followup.send(embed=embed) @app_commands.command(name="gamble", description="Gamble your coins") @app_commands.describe(amount="Amount to gamble") async def gamble(self, interaction: discord.Interaction, amount: str): account = await self._get_or_create_account(str(interaction.guild.id), str(interaction.user.id)) amt = account.wallet if amount.lower() == "all" else int(amount) if amt <= 0 or amt > account.wallet: await interaction.response.send_message(f"❌ Invalid amount. Wallet: **{account.wallet:,}**", ephemeral=True) return if amt > 100000: await interaction.response.send_message("❌ Maximum gambling amount is 100,000 coins.", ephemeral=True) return won = random.random() < 0.475 delta = amt if won else -amt new_wallet = account.wallet + delta await self._update_account(str(interaction.guild.id), str(interaction.user.id), wallet=max(0, new_wallet)) color = 0x00FF88 if won else 0xFF4444 result_msg = f"🎉 You **WON** **{amt:,}** coins!" if won else f"💸 You **LOST** **{amt:,}** coins!" embed = discord.Embed( title="🎰 Gamble Results", description=result_msg, color=color, timestamp=datetime.utcnow() ) embed.set_footer(text=f"💰 New Balance: {max(0, new_wallet):,} coins") await interaction.response.send_message(embed=embed) @app_commands.command(name="shop", description="Browse the server shop") async def shop(self, interaction: discord.Interaction): await interaction.response.defer() async with self.bot.db.session() as session: result = await session.execute( select(ShopItem).where( ShopItem.guild_id == str(interaction.guild.id), ShopItem.is_active == True ).order_by(ShopItem.price) ) items = result.scalars().all() if not items: await interaction.followup.send("🛒 The shop is empty. Ask an admin to add items!", ephemeral=True) return embed = discord.Embed(title="🛍️ Server Shop", color=0x5865F2, timestamp=datetime.utcnow()) for item in items[:15]: stock_str = "∞" if item.stock == -1 else str(item.stock) embed.add_field( name=f"{item.emoji} {item.name} — {item.price:,} coins", value=f"{item.description}\n📦 Stock: {stock_str}", inline=False ) embed.set_footer(text=f"Use /buy to purchase") await interaction.followup.send(embed=embed) async def setup(bot): await bot.add_cog(Economy(bot))