from __future__ import annotations import discord from discord import app_commands from discord.ext import commands from datetime import datetime, timedelta import random import utils.embeds as emb from typing import Optional async def _ensure(bot, guild_id, user_id): await bot.db.execute("INSERT OR IGNORE INTO economy(guild_id,user_id) VALUES(?,?)", (guild_id, user_id)) await bot.db.commit() async def _get(bot, guild_id, user_id) -> tuple: await _ensure(bot, guild_id, user_id) async with bot.db.execute("SELECT balance, bank, last_daily, last_weekly, last_work, last_crime FROM economy WHERE guild_id=? AND user_id=?", (guild_id, user_id)) as cur: return await cur.fetchone() async def _set(bot, guild_id, user_id, **kwargs): await _ensure(bot, guild_id, user_id) if not kwargs: return sets = ", ".join(f"{k}=?" for k in kwargs) vals = list(kwargs.values()) + [guild_id, user_id] await bot.db.execute(f"UPDATE economy SET {sets} WHERE guild_id=? AND user_id=?", vals) await bot.db.commit() class Economy(commands.Cog): def __init__(self, bot): self.bot = bot @property def _cfg(self): return self.bot.config.get("economy", {}) @property def _sym(self) -> str: return self._cfg.get("currency_emoji", "🪙") @property def _name(self) -> str: return self._cfg.get("currency_name", "coins") @app_commands.command(name="balance", description="Check your coin balance.") @app_commands.describe(member="Member to check (default: you)") async def balance(self, interaction: discord.Interaction, member: Optional[discord.Member] = None): await interaction.response.defer() t = member or interaction.user row = await _get(self.bot, interaction.guild_id, t.id) bal, bank = row[0], row[1] e = discord.Embed(title=f"{self._sym}\u2002{t.display_name}'s Balance", color=emb.GOLD) e.add_field(name="👛 Wallet", value=f"`{bal:,}` {self._name}", inline=True) e.add_field(name="🏦 Bank", value=f"`{bank:,}` {self._name}", inline=True) e.add_field(name="💰 Net Worth", value=f"`{bal+bank:,}` {self._name}", inline=True) e.set_thumbnail(url=t.display_avatar.url) e.timestamp = datetime.utcnow() await interaction.followup.send(embed=e) @app_commands.command(name="deposit", description="Deposit coins into your bank.") @app_commands.describe(amount="Amount or 'all'") async def deposit(self, interaction: discord.Interaction, amount: str): await interaction.response.defer() row = await _get(self.bot, interaction.guild_id, interaction.user.id) bal, bank = row[0], row[1] amt = bal if amount.lower() == "all" else (int(amount) if amount.isdigit() else -1) if amt <= 0 or amt > bal: return await interaction.followup.send(embed=emb.error("Invalid amount or insufficient wallet balance.")) await _set(self.bot, interaction.guild_id, interaction.user.id, balance=bal - amt, bank=bank + amt) await interaction.followup.send(embed=emb.success(f"Deposited **{amt:,}** {self._sym} into your bank.")) @app_commands.command(name="withdraw", description="Withdraw coins from your bank.") @app_commands.describe(amount="Amount or 'all'") async def withdraw(self, interaction: discord.Interaction, amount: str): await interaction.response.defer() row = await _get(self.bot, interaction.guild_id, interaction.user.id) bal, bank = row[0], row[1] amt = bank if amount.lower() == "all" else (int(amount) if amount.isdigit() else -1) if amt <= 0 or amt > bank: return await interaction.followup.send(embed=emb.error("Invalid amount or insufficient bank balance.")) await _set(self.bot, interaction.guild_id, interaction.user.id, balance=bal + amt, bank=bank - amt) await interaction.followup.send(embed=emb.success(f"Withdrew **{amt:,}** {self._sym} from your bank.")) @app_commands.command(name="daily", description="Claim your daily coin reward.") async def daily(self, interaction: discord.Interaction): await interaction.response.defer() row = await _get(self.bot, interaction.guild_id, interaction.user.id) bal, _, last_daily = row[0], row[1], row[2] if last_daily: since = (datetime.utcnow() - datetime.fromisoformat(last_daily)) if since < timedelta(hours=20): rem = timedelta(hours=20) - since h, m = divmod(int(rem.total_seconds()), 3600) m //= 60 return await interaction.followup.send(embed=emb.error(f"Daily already claimed! Come back in **{h}h {m}m**.")) amount = self._cfg.get("daily_amount", 200) await _set(self.bot, interaction.guild_id, interaction.user.id, balance=bal + amount, last_daily=datetime.utcnow().isoformat()) await interaction.followup.send(embed=emb._e("📅\u2002Daily Reward", f"You claimed **{amount:,}** {self._sym}!\nNew balance: **{bal+amount:,}** {self._sym}", emb.SUCCESS)) @app_commands.command(name="weekly", description="Claim your weekly coin reward.") async def weekly(self, interaction: discord.Interaction): await interaction.response.defer() row = await _get(self.bot, interaction.guild_id, interaction.user.id) bal, _, _d, last_weekly = row[0], row[1], row[2], row[3] if last_weekly: since = datetime.utcnow() - datetime.fromisoformat(last_weekly) if since < timedelta(days=7): rem = timedelta(days=7) - since d = rem.days h = rem.seconds // 3600 return await interaction.followup.send(embed=emb.error(f"Weekly already claimed! Come back in **{d}d {h}h**.")) amount = self._cfg.get("weekly_amount", 1000) await _set(self.bot, interaction.guild_id, interaction.user.id, balance=bal + amount, last_weekly=datetime.utcnow().isoformat()) await interaction.followup.send(embed=emb._e("📆\u2002Weekly Reward", f"You claimed **{amount:,}** {self._sym}!\nNew balance: **{bal+amount:,}** {self._sym}", emb.GOLD)) @app_commands.command(name="work", description="Work to earn coins (30-minute cooldown).") async def work(self, interaction: discord.Interaction): await interaction.response.defer() row = await _get(self.bot, interaction.guild_id, interaction.user.id) bal, last_work = row[0], row[4] if last_work and (datetime.utcnow() - datetime.fromisoformat(last_work)) < timedelta(minutes=30): rem = timedelta(minutes=30) - (datetime.utcnow() - datetime.fromisoformat(last_work)) m = int(rem.total_seconds() / 60) return await interaction.followup.send(embed=emb.error(f"You're tired! Rest for **{m}m**.")) jobs = [ ("coded a new feature", "developer"), ("designed a UI mockup", "designer"), ("tested the product", "QA engineer"), ("wrote documentation", "writer"), ("fixed a critical bug", "engineer"), ("deployed to production", "DevOps"), ("reviewed pull requests", "senior dev"), ("managed a client", "project manager"), ] task, role = random.choice(jobs) amount = random.randint(self._cfg.get("work_min", 50), self._cfg.get("work_max", 300)) await _set(self.bot, interaction.guild_id, interaction.user.id, balance=bal + amount, last_work=datetime.utcnow().isoformat()) await interaction.followup.send(embed=emb._e("💼\u2002Work", f"You {task} as a **{role}** and earned **{amount:,}** {self._sym}!\nBalance: **{bal+amount:,}** {self._sym}", emb.SUCCESS)) @app_commands.command(name="crime", description="Commit a crime for big rewards — but risk a fine.") async def crime(self, interaction: discord.Interaction): await interaction.response.defer() row = await _get(self.bot, interaction.guild_id, interaction.user.id) bal, last_crime = row[0], row[5] if last_crime and (datetime.utcnow() - datetime.fromisoformat(last_crime)) < timedelta(hours=1): rem = timedelta(hours=1) - (datetime.utcnow() - datetime.fromisoformat(last_crime)) m = int(rem.total_seconds() / 60) return await interaction.followup.send(embed=emb.error(f"Lay low for **{m}m** before your next crime.")) crimes = ["robbed a bank", "hacked a server", "sold stolen goods", "scammed an investor", "stole a wallet"] crime_done = random.choice(crimes) fail_chance = self._cfg.get("crime_fail_chance", 30) if random.randint(1, 100) <= fail_chance: fine = random.randint(50, min(bal, 500)) if bal > 0 else 0 await _set(self.bot, interaction.guild_id, interaction.user.id, balance=max(0, bal - fine), last_crime=datetime.utcnow().isoformat()) e = emb._e("🚔\u2002Arrested!", f"You tried to {crime_done} but got caught!\nFine: **{fine:,}** {self._sym}", emb.ERROR) else: amount = random.randint(self._cfg.get("crime_min", 100), self._cfg.get("crime_max", 500)) await _set(self.bot, interaction.guild_id, interaction.user.id, balance=bal + amount, last_crime=datetime.utcnow().isoformat()) e = emb._e("🦹\u2002Crime Successful!", f"You {crime_done} and got away with **{amount:,}** {self._sym}!", emb.SUCCESS) await interaction.followup.send(embed=e) @app_commands.command(name="rob", description="Try to rob another user's wallet.") @app_commands.describe(member="Member to rob") async def rob(self, interaction: discord.Interaction, member: discord.Member): await interaction.response.defer() if member.id == interaction.user.id or member.bot: return await interaction.followup.send(embed=emb.error("Invalid target.")) rob_row = await _get(self.bot, interaction.guild_id, interaction.user.id) tgt_row = await _get(self.bot, interaction.guild_id, member.id) rob_bal, tgt_bal = rob_row[0], tgt_row[0] if tgt_bal < 100: return await interaction.followup.send(embed=emb.error(f"**{member.display_name}** is too broke to rob (< 100 {self._name}).")) if random.randint(1, 100) <= 40: fine = random.randint(50, min(rob_bal, 300)) if rob_bal > 0 else 0 await _set(self.bot, interaction.guild_id, interaction.user.id, balance=max(0, rob_bal - fine)) e = emb._e("🚔\u2002Caught!", f"You tried to rob **{member.display_name}** but got caught!\nFine: **{fine:,}** {self._sym}", emb.ERROR) else: stolen = random.randint(50, min(tgt_bal // 2, 500)) await _set(self.bot, interaction.guild_id, interaction.user.id, balance=rob_bal + stolen) await _set(self.bot, interaction.guild_id, member.id, balance=tgt_bal - stolen) e = emb._e("💸\u2002Robbery!", f"You robbed **{member.display_name}** and stole **{stolen:,}** {self._sym}!", emb.SUCCESS) await interaction.followup.send(embed=e) @app_commands.command(name="pay", description="Pay another user coins from your wallet.") @app_commands.describe(member="Who to pay", amount="How much") async def pay(self, interaction: discord.Interaction, member: discord.Member, amount: int): await interaction.response.defer() if member.id == interaction.user.id or member.bot or amount <= 0: return await interaction.followup.send(embed=emb.error("Invalid target or amount.")) row = await _get(self.bot, interaction.guild_id, interaction.user.id) bal = row[0] if amount > bal: return await interaction.followup.send(embed=emb.error("Insufficient wallet balance.")) trow = await _get(self.bot, interaction.guild_id, member.id) await _set(self.bot, interaction.guild_id, interaction.user.id, balance=bal - amount) await _set(self.bot, interaction.guild_id, member.id, balance=trow[0] + amount) await interaction.followup.send(embed=emb.success(f"Sent **{amount:,}** {self._sym} to {member.mention}.")) @app_commands.command(name="gamble", description="Bet coins on a 50/50 flip.") @app_commands.describe(amount="Amount to bet") async def gamble(self, interaction: discord.Interaction, amount: int): await interaction.response.defer() if amount <= 0: return await interaction.followup.send(embed=emb.error("Bet must be positive.")) row = await _get(self.bot, interaction.guild_id, interaction.user.id) bal = row[0] if amount > bal: return await interaction.followup.send(embed=emb.error("Insufficient wallet balance.")) win = random.random() < 0.5 new_bal = bal + amount if win else bal - amount await _set(self.bot, interaction.guild_id, interaction.user.id, balance=new_bal) e = emb._e( "🎉\u2002You Won!" if win else "💀\u2002You Lost!", f"Bet: **{amount:,}** {self._sym}\n{'Won' if win else 'Lost'}: **{amount:,}** {self._sym}\nNew balance: **{new_bal:,}** {self._sym}", emb.SUCCESS if win else emb.ERROR, ) await interaction.followup.send(embed=e) @app_commands.command(name="slots", description="Play the slot machine!") @app_commands.describe(amount="Amount to bet") async def slots(self, interaction: discord.Interaction, amount: int): await interaction.response.defer() if amount <= 0: return await interaction.followup.send(embed=emb.error("Bet must be positive.")) row = await _get(self.bot, interaction.guild_id, interaction.user.id) bal = row[0] if amount > bal: return await interaction.followup.send(embed=emb.error("Insufficient wallet balance.")) symbols = ["🍒", "🍋", "🍊", "🍇", "⭐", "💎", "7️⃣"] weights = [30, 25, 20, 15, 5, 3, 2 ] reels = random.choices(symbols, weights=weights, k=3) display = " | ".join(reels) if reels[0] == reels[1] == reels[2]: mults = {"💎": 50, "7️⃣": 30, "⭐": 15, "🍇": 8, "🍊": 5, "🍋": 3, "🍒": 2} mult = mults.get(reels[0], 2) winnings = amount * mult await _set(self.bot, interaction.guild_id, interaction.user.id, balance=bal - amount + winnings) e = emb._e("🎰\u2002JACKPOT!", f"**{display}**\n\n{reels[0]} **{mult}x multiplier!** +**{winnings:,}** {self._sym}\nBalance: **{bal-amount+winnings:,}** {self._sym}", emb.GOLD) elif reels[0] == reels[1] or reels[1] == reels[2] or reels[0] == reels[2]: await _set(self.bot, interaction.guild_id, interaction.user.id, balance=bal + amount) e = emb._e("🎰\u2002Two of a Kind!", f"**{display}**\n\n+**{amount:,}** {self._sym}\nBalance: **{bal+amount:,}** {self._sym}", emb.SUCCESS) else: await _set(self.bot, interaction.guild_id, interaction.user.id, balance=bal - amount) e = emb._e("🎰\u2002No luck...", f"**{display}**\n\n-**{amount:,}** {self._sym}\nBalance: **{bal-amount:,}** {self._sym}", emb.ERROR) await interaction.followup.send(embed=e) @app_commands.command(name="richest", description="View the richest members.") async def richest(self, interaction: discord.Interaction): await interaction.response.defer() async with self.bot.db.execute("SELECT user_id, balance+bank as net FROM economy WHERE guild_id=? ORDER BY net DESC LIMIT 10", (interaction.guild_id,)) as cur: rows = await cur.fetchall() medals = ["🥇", "🥈", "🥉"] + ["🏅"] * 7 e = discord.Embed(title=f"{self._sym}\u2002Richest Members", color=emb.GOLD) lines = [] for i, (uid, net) in enumerate(rows): m = interaction.guild.get_member(uid) name = m.display_name if m else f"<@{uid}>" lines.append(f"{medals[i]} **{name}** — `{net:,}` {self._name}") e.description = "\n".join(lines) if lines else "No data yet." e.timestamp = datetime.utcnow() await interaction.followup.send(embed=e) async def setup(bot): await bot.add_cog(Economy(bot))