From 3c18d552c2c9eb15834b33687915a7b16c98883b Mon Sep 17 00:00:00 2001
From: Parker <contact@pkrm.dev>
Date: Sat, 25 Jan 2025 22:58:41 -0600
Subject: reupload

---
 code/cogs/gambling_helpers.py | 139 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 139 insertions(+)
 create mode 100644 code/cogs/gambling_helpers.py

(limited to 'code/cogs/gambling_helpers.py')

diff --git a/code/cogs/gambling_helpers.py b/code/cogs/gambling_helpers.py
new file mode 100644
index 0000000..5579df3
--- /dev/null
+++ b/code/cogs/gambling_helpers.py
@@ -0,0 +1,139 @@
+import discord
+from discord.ext import commands
+from discord import app_commands
+import random
+
+from bot import InsufficientFundsException
+from global_variables import BOT_COLOR, BONUS_COOLDOWN
+from database import Database
+
+
+class GamblingHelpers(commands.Cog):
+    def __init__(self, bot: commands.Bot) -> None:
+        self.bot = bot
+        self.economy = Database(bot)
+
+    async def check_bet(
+        self,
+        interaction: discord.Interaction,
+        bet
+    ):
+        bet = int(bet)
+        if bet <= 0:
+            raise commands.errors.BadArgument()
+        current = (await self.economy.get_entry(interaction.user.id))[1]
+        if bet > current:
+            raise InsufficientFundsException()
+
+
+    # This is only here temporarily, because I accidentally deleted the database
+    @commands.command()
+    @commands.is_owner()
+    @commands.dm_only()
+    async def refund(
+        self,
+        ctx: commands.Context,
+        id,
+        amount: int
+    ):
+        user = await self.bot.fetch_user(id)
+        if user:
+            await self.economy.add_money(id, amount)
+            await ctx.send(f"Refunded {user} ${amount:,}")
+        else:
+            await ctx.send("User not found")
+
+
+    @app_commands.command()
+    @app_commands.checks.cooldown(1, BONUS_COOLDOWN*3600)
+    async def add(
+        self,
+        interaction: discord.Interaction
+    ):
+        f"Add $10,000 to your balance every {BONUS_COOLDOWN} hours"
+        amount = 10000
+        await self.economy.add_money(interaction.user.id, amount)
+        embed = discord.Embed(
+            title="I've added $10,000 to you balance",
+            description=f"Come back again in {BONUS_COOLDOWN} hours.",
+            color=BOT_COLOR
+        )
+        await interaction.response.send_message(embed=embed, ephemeral=True)
+
+
+    @app_commands.command()
+    @app_commands.checks.cooldown(1, 120)
+    async def work(
+        self,
+        interaction: discord.Interaction
+    ):
+        "Work for a randomized amount of money every 2 minutes"
+        a = random.randint(500, 2500)
+        b = random.randint(500, 2500)
+        if a == b:
+            num = 50000000
+        else:
+            num = a + b
+
+        await self.economy.add_money(interaction.user.id, num)
+        embed = discord.Embed(
+            title=f"You worked and earned ${num:,}",
+            description="Come back again in 2 minutes.",
+            color=BOT_COLOR
+        )
+        await interaction.response.send_message(embed=embed, ephemeral=True)
+
+
+    @app_commands.command()
+    async def leaderboard(
+        self,
+        interaction: discord.Interaction
+    ):
+        "Show the global currency leaderboard"
+        entries = await self.economy.top_entries(5)
+        embed = discord.Embed(
+            title="Global Economy Leaderboard:",
+            description="",
+            color=discord.Color.gold()
+        )
+
+        for i, entry in enumerate(entries):
+            id = entry[0]
+            name = f"<@{id}>"
+            embed.description += f"**`{i+1}.` {name}**\n${entry[1]:,}\n\n"
+
+        await interaction.response.send_message(embed=embed)
+
+
+    @app_commands.command()
+    @app_commands.describe(user="The user to give money to")
+    @app_commands.describe(amount="The amount of money to give the user")
+    async def give(
+        self,
+        interaction: discord.Interaction,
+        user: discord.Member,
+        amount: app_commands.Range[int, 1, None]
+    ):
+        "Give money to another user"
+        if user == interaction.user:
+            embed = discord.Embed(
+                title="Self Error",
+                description="You cannot give money to yourself, please try again with a different user.",
+                color=BOT_COLOR
+            )
+            return await interaction.response.send_message(embed=embed, ephemeral=True)
+
+        else:
+            await self.check_bet(interaction, amount)
+            await self.economy.add_money(user.id, amount)
+            await self.economy.add_money(interaction.user.id, amount*-1)
+            embed = discord.Embed(
+                title="Gift Success",
+                description=f"You have successfully given {user.mention} ${amount:,}!",
+                color=discord.Color.green()
+            )
+            return await interaction.response.send_message(embed=embed)
+
+
+async def setup(bot: commands.Bot):
+    await bot.add_cog(GamblingHelpers(bot))
\ No newline at end of file
-- 
cgit v1.2.3-70-g09d2