import discord from discord import app_commands from discord.ext import commands import random from datetime import datetime, timedelta from database import Database class Economy(commands.Cog): def __init__(self, bot): self.bot = bot self.db = Database() self.currency_symbol = "💰" self.xp_symbol = "⭐" @app_commands.command(name='balance', description='Check your or another user\'s balance') async def balance(self, interaction: discord.Interaction, member: discord.Member = None): member = member or interaction.user user_data = self.db.get_user(member.id) embed = discord.Embed( title=f"{member.display_name}'s Balance", color=discord.Color.gold() ) embed.set_thumbnail(url=member.display_avatar.url) embed.add_field(name=f"{self.currency_symbol} Credits", value=f"**{user_data['balance']:,}**", inline=True) embed.add_field(name=f"{self.xp_symbol} Level", value=f"**{user_data['level']}**", inline=True) embed.add_field(name="XP", value=f"**{user_data['xp']:,}** / **{self.xp_for_level(user_data['level'] + 1):,}**", inline=True) await interaction.response.send_message(embed=embed) def xp_for_level(self, level): return 100 * (level ** 2) @app_commands.command(name='daily', description='Claim your daily reward') async def daily(self, interaction: discord.Interaction): user_data = self.db.get_user(interaction.user.id) now = datetime.utcnow() last_daily = datetime.fromisoformat(user_data['last_daily']) if user_data['last_daily'] else None if last_daily and (now - last_daily).total_seconds() < 86400: time_left = 86400 - (now - last_daily).total_seconds() hours = int(time_left // 3600) minutes = int((time_left % 3600) // 60) embed = discord.Embed( title="⏰ Daily Reward", description=f"You've already claimed your daily reward!\nCome back in **{hours}h {minutes}m**", color=discord.Color.red() ) await interaction.response.send_message(embed=embed, ephemeral=True) return streak = user_data['daily_streak'] if last_daily and (now - last_daily).total_seconds() < 172800: streak += 1 else: streak = 1 base_reward = 500 streak_bonus = min(streak * 50, 1000) total_reward = base_reward + streak_bonus self.db.update_balance(interaction.user.id, total_reward) self.db.update_daily(interaction.user.id, now.isoformat(), streak) embed = discord.Embed( title="🎁 Daily Reward Claimed!", description=f"You received **{total_reward:,}** {self.currency_symbol}", color=discord.Color.green() ) embed.add_field(name="Daily Streak", value=f"🔥 **{streak}** days", inline=True) embed.add_field(name="Streak Bonus", value=f"+**{streak_bonus:,}** {self.currency_symbol}", inline=True) await interaction.response.send_message(embed=embed) @app_commands.command(name='work', description='Work to earn credits') async def work(self, interaction: discord.Interaction): user_data = self.db.get_user(interaction.user.id) now = datetime.utcnow() last_work = datetime.fromisoformat(user_data['last_work']) if user_data['last_work'] else None if last_work and (now - last_work).total_seconds() < 3600: time_left = 3600 - (now - last_work).total_seconds() minutes = int(time_left // 60) seconds = int(time_left % 60) embed = discord.Embed( title="⏰ Work Cooldown", description=f"You're tired! Rest for **{minutes}m {seconds}s**", color=discord.Color.red() ) await interaction.response.send_message(embed=embed, ephemeral=True) return jobs = [ ("worked as a developer", 200, 400), ("delivered packages", 150, 350), ("streamed on Twitch", 180, 420), ("created content", 160, 380), ("tutored students", 140, 320), ("designed graphics", 190, 410), ("wrote articles", 170, 390), ("managed social media", 160, 360), ] job, min_earn, max_earn = random.choice(jobs) earned = random.randint(min_earn, max_earn) self.db.update_balance(interaction.user.id, earned) self.db.update_work(interaction.user.id, now.isoformat()) embed = discord.Embed( title="💼 Work Complete!", description=f"You {job} and earned **{earned:,}** {self.currency_symbol}", color=discord.Color.green() ) await interaction.response.send_message(embed=embed) @app_commands.command(name='crime', description='Commit a crime for high risk, high reward') async def crime(self, interaction: discord.Interaction): user_data = self.db.get_user(interaction.user.id) now = datetime.utcnow() last_crime = datetime.fromisoformat(user_data['last_crime']) if user_data['last_crime'] else None if last_crime and (now - last_crime).total_seconds() < 7200: time_left = 7200 - (now - last_crime).total_seconds() hours = int(time_left // 3600) minutes = int((time_left % 3600) // 60) embed = discord.Embed( title="⏰ Crime Cooldown", description=f"Lay low for **{hours}h {minutes}m**", color=discord.Color.red() ) await interaction.response.send_message(embed=embed, ephemeral=True) return crimes = [ ("robbed a bank", 800, 1500), ("hacked a corporation", 900, 1600), ("stole a luxury car", 700, 1400), ("smuggled rare items", 850, 1550), ("pulled off a heist", 1000, 1800), ] success_rate = 0.55 if random.random() < success_rate: crime, min_earn, max_earn = random.choice(crimes) earned = random.randint(min_earn, max_earn) self.db.update_balance(interaction.user.id, earned) self.db.update_crime(interaction.user.id, now.isoformat()) embed = discord.Embed( title="🎭 Crime Success!", description=f"You {crime} and got away with **{earned:,}** {self.currency_symbol}", color=discord.Color.green() ) else: fine = random.randint(300, 800) fine = min(fine, user_data['balance']) self.db.update_balance(interaction.user.id, -fine) self.db.update_crime(interaction.user.id, now.isoformat()) embed = discord.Embed( title="🚔 Crime Failed!", description=f"You got caught and paid a fine of **{fine:,}** {self.currency_symbol}", color=discord.Color.red() ) await interaction.response.send_message(embed=embed) @app_commands.command(name='gamble', description='Gamble your credits') async def gamble(self, interaction: discord.Interaction, amount: int): user_data = self.db.get_user(interaction.user.id) if amount < 100: await interaction.response.send_message("Minimum bet is **100** credits!", ephemeral=True) return if amount > user_data['balance']: await interaction.response.send_message("You don't have enough credits!", ephemeral=True) return if amount > 10000: await interaction.response.send_message("Maximum bet is **10,000** credits!", ephemeral=True) return slots = ['🍒', '🍋', '🍊', '🍇', '💎', '7️⃣'] result = [random.choice(slots) for _ in range(3)] multiplier = 0 if result[0] == result[1] == result[2]: if result[0] == '💎': multiplier = 10 elif result[0] == '7️⃣': multiplier = 5 else: multiplier = 3 elif result[0] == result[1] or result[1] == result[2]: multiplier = 1.5 if multiplier > 0: winnings = int(amount * multiplier) self.db.update_balance(interaction.user.id, winnings - amount) embed = discord.Embed( title="🎰 Slot Machine", description=f"{result[0]} | {result[1]} | {result[2]}\n\n**YOU WIN!**\nWon **{winnings:,}** {self.currency_symbol} (x{multiplier})", color=discord.Color.green() ) else: self.db.update_balance(interaction.user.id, -amount) embed = discord.Embed( title="🎰 Slot Machine", description=f"{result[0]} | {result[1]} | {result[2]}\n\n**YOU LOSE!**\nLost **{amount:,}** {self.currency_symbol}", color=discord.Color.red() ) await interaction.response.send_message(embed=embed) @app_commands.command(name='coinflip', description='Flip a coin and bet on the outcome') async def coinflip(self, interaction: discord.Interaction, amount: int, choice: str): user_data = self.db.get_user(interaction.user.id) if choice.lower() not in ['heads', 'tails']: await interaction.response.send_message("Choose **heads** or **tails**!", ephemeral=True) return if amount < 50: await interaction.response.send_message("Minimum bet is **50** credits!", ephemeral=True) return if amount > user_data['balance']: await interaction.response.send_message("You don't have enough credits!", ephemeral=True) return result = random.choice(['heads', 'tails']) if result == choice.lower(): winnings = amount * 2 self.db.update_balance(interaction.user.id, amount) embed = discord.Embed( title="🪙 Coinflip", description=f"The coin landed on **{result}**!\n\n**YOU WIN!**\nWon **{winnings:,}** {self.currency_symbol}", color=discord.Color.green() ) else: self.db.update_balance(interaction.user.id, -amount) embed = discord.Embed( title="🪙 Coinflip", description=f"The coin landed on **{result}**!\n\n**YOU LOSE!**\nLost **{amount:,}** {self.currency_symbol}", color=discord.Color.red() ) await interaction.response.send_message(embed=embed) @app_commands.command(name='shop', description='View the item shop') async def shop(self, interaction: discord.Interaction): items = self.db.get_shop_items() embed = discord.Embed( title="🛒 Item Shop", description="Purchase items with your credits!", color=discord.Color.blue() ) for item in items: embed.add_field( name=f"{item['emoji']} {item['name']} - {item['price']:,} {self.currency_symbol}", value=item['description'], inline=False ) embed.set_footer(text="Use /buy to purchase an item") await interaction.response.send_message(embed=embed) @app_commands.command(name='buy', description='Buy an item from the shop') async def buy(self, interaction: discord.Interaction, item_name: str): user_data = self.db.get_user(interaction.user.id) item = self.db.get_shop_item(item_name) if not item: await interaction.response.send_message("Item not found!", ephemeral=True) return if user_data['balance'] < item['price']: await interaction.response.send_message(f"You need **{item['price']:,}** {self.currency_symbol} to buy this item!", ephemeral=True) return self.db.update_balance(interaction.user.id, -item['price']) self.db.add_inventory_item(interaction.user.id, item['name']) embed = discord.Embed( title="✅ Purchase Successful!", description=f"You bought **{item['emoji']} {item['name']}** for **{item['price']:,}** {self.currency_symbol}", color=discord.Color.green() ) await interaction.response.send_message(embed=embed) @app_commands.command(name='inventory', description='View your inventory') async def inventory(self, interaction: discord.Interaction, member: discord.Member = None): member = member or interaction.user items = self.db.get_inventory(member.id) embed = discord.Embed( title=f"{member.display_name}'s Inventory", color=discord.Color.blue() ) embed.set_thumbnail(url=member.display_avatar.url) if not items: embed.description = "Inventory is empty!" else: item_counts = {} for item in items: item_counts[item] = item_counts.get(item, 0) + 1 shop_items = {i['name']: i for i in self.db.get_shop_items()} for item_name, count in item_counts.items(): shop_item = shop_items.get(item_name, {}) emoji = shop_item.get('emoji', '📦') embed.add_field( name=f"{emoji} {item_name}", value=f"Quantity: **{count}**", inline=True ) await interaction.response.send_message(embed=embed) @app_commands.command(name='give', description='Give credits to another user') async def give(self, interaction: discord.Interaction, member: discord.Member, amount: int): if member.id == interaction.user.id: await interaction.response.send_message("You can't give credits to yourself!", ephemeral=True) return if member.bot: await interaction.response.send_message("You can't give credits to bots!", ephemeral=True) return if amount < 1: await interaction.response.send_message("Amount must be positive!", ephemeral=True) return user_data = self.db.get_user(interaction.user.id) if amount > user_data['balance']: await interaction.response.send_message("You don't have enough credits!", ephemeral=True) return self.db.update_balance(interaction.user.id, -amount) self.db.update_balance(member.id, amount) embed = discord.Embed( title="💸 Transfer Complete", description=f"{interaction.user.mention} gave **{amount:,}** {self.currency_symbol} to {member.mention}", color=discord.Color.green() ) await interaction.response.send_message(embed=embed) @app_commands.command(name='leaderboard', description='View the richest users') async def leaderboard(self, interaction: discord.Interaction): top_users = self.db.get_leaderboard(10) embed = discord.Embed( title="💰 Leaderboard - Top 10 Richest Users", color=discord.Color.gold() ) medals = ['🥇', '🥈', '🥉'] description = "" for idx, user_data in enumerate(top_users): user = self.bot.get_user(user_data['user_id']) if user: medal = medals[idx] if idx < 3 else f"**{idx + 1}.**" description += f"{medal} {user.mention} - **{user_data['balance']:,}** {self.currency_symbol}\n" embed.description = description if description else "No users found!" await interaction.response.send_message(embed=embed) @app_commands.command(name='levelboard', description='View the highest level users') async def levelboard(self, interaction: discord.Interaction): top_users = self.db.get_level_leaderboard(10) embed = discord.Embed( title="⭐ Leaderboard - Top 10 Highest Levels", color=discord.Color.gold() ) medals = ['🥇', '🥈', '🥉'] description = "" for idx, user_data in enumerate(top_users): user = self.bot.get_user(user_data['user_id']) if user: medal = medals[idx] if idx < 3 else f"**{idx + 1}.**" description += f"{medal} {user.mention} - Level **{user_data['level']}** ({user_data['xp']:,} XP)\n" embed.description = description if description else "No users found!" await interaction.response.send_message(embed=embed) @app_commands.command(name='rob', description='Attempt to rob another user') async def rob(self, interaction: discord.Interaction, member: discord.Member): if member.id == interaction.user.id: await interaction.response.send_message("You can't rob yourself!", ephemeral=True) return if member.bot: await interaction.response.send_message("You can't rob bots!", ephemeral=True) return user_data = self.db.get_user(interaction.user.id) target_data = self.db.get_user(member.id) now = datetime.utcnow() last_rob = datetime.fromisoformat(user_data['last_rob']) if user_data['last_rob'] else None if last_rob and (now - last_rob).total_seconds() < 14400: time_left = 14400 - (now - last_rob).total_seconds() hours = int(time_left // 3600) minutes = int((time_left % 3600) // 60) embed = discord.Embed( title="⏰ Rob Cooldown", description=f"Wait **{hours}h {minutes}m** before robbing again", color=discord.Color.red() ) await interaction.response.send_message(embed=embed, ephemeral=True) return if target_data['balance'] < 500: await interaction.response.send_message(f"{member.mention} doesn't have enough credits to rob!", ephemeral=True) return success_rate = 0.45 if random.random() < success_rate: stolen = random.randint(100, min(1000, target_data['balance'] // 2)) self.db.update_balance(interaction.user.id, stolen) self.db.update_balance(member.id, -stolen) self.db.update_rob(interaction.user.id, now.isoformat()) embed = discord.Embed( title="💰 Robbery Success!", description=f"You robbed **{stolen:,}** {self.currency_symbol} from {member.mention}", color=discord.Color.green() ) else: fine = random.randint(200, 600) fine = min(fine, user_data['balance']) self.db.update_balance(interaction.user.id, -fine) self.db.update_rob(interaction.user.id, now.isoformat()) embed = discord.Embed( title="🚔 Robbery Failed!", description=f"You got caught and paid a fine of **{fine:,}** {self.currency_symbol}", color=discord.Color.red() ) await interaction.response.send_message(embed=embed) async def setup(bot): await bot.add_cog(Economy(bot))