Update petsimgo.py

petsimgo.py

@@ -1,11 +1,17 @@
 import discord
 from discord import app_commands
 import aiohttp
+import random
+import time
 
-@app_commands.command(name="petsimgo", description="get info on pet on petsgo u")
-async def petsimgo(interaction: discord.Interaction, petname: str):
-    await interaction.response.defer()
+from cash import user_cash
+
+luck_multipliers = {}
+luck_expiration = {}
+luck_opportunities = {}
+used_luck_opportunities = set()
 
+async def perform_roll(interaction: discord.Interaction):
     async def fetch_data(url):
         async with aiohttp.ClientSession() as session:
             async with session.get(url) as response:
@@ -13,26 +19,34 @@ async def petsimgo(interaction: discord.Interaction, petname: str):
                     return await response.json()
                 return None
 
-    exists_data = await fetch_data("https://petsgo.biggamesapi.io/api/exists")
     rap_data = await fetch_data("https://petsgo.biggamesapi.io/api/Rap")
     collection_data = await fetch_data("https://petsgo.biggamesapi.io/api/collection/Pets")
 
-    if not exists_data or not rap_data or not collection_data:
-        await interaction.followup.send("error")
-        return
+    if not rap_data or not collection_data:
+        return None
+
+    pets = [pet for pet in collection_data['data'] if pet['configName'] in [p['configData']['id'] for p in rap_data['data']]]
+    
+    if not pets:
+        return None
 
-    pet_exists = next((pet for pet in exists_data['data'] if pet['configData']['id'].lower() == petname.lower()), None)
-    pet_rap = next((pet for pet in rap_data['data'] if pet['configData']['id'].lower() == petname.lower()), None)
-    pet_info = next((pet for pet in collection_data['data'] if pet['configName'].lower() == petname.lower()), None)
+    user_id = interaction.user.id
+    luck_multiplier = luck_multipliers.get(user_id, 1)
+    
+    sorted_pets = sorted(pets, key=lambda x: x['configData']['difficulty'])
+    
+    max_index = len(sorted_pets) - 1
+    index = int(max_index * (luck_multiplier - 1) / 9)
+    
+    rolled_pet = random.choice(sorted_pets[:index+1])
 
-    if not pet_exists or not pet_rap or not pet_info:
-        await interaction.followup.send(f"Pet '{petname}' not found.")
-        return
+    pet_rap = next((pet for pet in rap_data['data'] if pet['configData']['id'] == rolled_pet['configName']), None)
+
+    if not pet_rap:
+        return None
 
-    exists_value = pet_exists['value']
     rap_value = pet_rap['value']
-    thumbnail_id = pet_info['configData']['thumbnail'].split('://')[1]
-    
+    thumbnail_id = rolled_pet['configData']['thumbnail'].split('://')[1]
     thumbnail_url = f"https://api.rbxgleaks1.workers.dev/asset/{thumbnail_id}"
 
     def format_difficulty(difficulty):
@@ -45,12 +59,102 @@ async def petsimgo(interaction: discord.Interaction, petname: str):
         else:
             return f"{difficulty} ({difficulty:,})"
 
-    embed = discord.Embed(title=f"PetsGo: {pet_info['configData']['name']}", color=0x787878)
-    embed.add_field(name="value", value=f"{rap_value:,} diamonds", inline=True)
-    embed.add_field(name="existing", value=f"{exists_value:,}", inline=True)
-    embed.add_field(name="difficulty", value=format_difficulty(pet_info['configData']['difficulty']), inline=True)
-    embed.add_field(name="category", value=pet_info['category'], inline=True)
+    embed = discord.Embed(title=f"{interaction.user.name} rolled: {rolled_pet['configData']['name']}", color=0x787878)
+    embed.add_field(name="Value", value=f"{rap_value:,} diamonds", inline=True)
+    embed.add_field(name="Difficulty", value=format_difficulty(rolled_pet['configData']['difficulty']), inline=True)
+    embed.add_field(name="Category", value=rolled_pet['category'], inline=True)
     embed.set_thumbnail(url=thumbnail_url)
-    embed.set_footer(text="hello everyone can i please get a burrito now")
+    
+    luck_text = ""
+    if user_id in luck_expiration:
+        remaining_time = int(luck_expiration[user_id] - time.time())
+        if remaining_time > 0:
+            luck_percentage = (luck_multiplier - 1) * 100
+            luck_text = f"\nYou have {remaining_time // 60} minutes and {remaining_time % 60} seconds of luck left! ({luck_percentage}% luck)"
+        else:
+            del luck_multipliers[user_id]
+            del luck_expiration[user_id]
+    
+    embed.set_footer(text=f"Click 'Roll Again' to roll again!{luck_text}")
+
+    roll_again_button = discord.ui.Button(style=discord.ButtonStyle.primary, label="Roll Again", custom_id="roll_again")
+    
+    async def roll_again_callback(interaction: discord.Interaction):
+        await interaction.response.defer()
+        result = await perform_roll(interaction)
+        if result:
+            await interaction.followup.send(embed=result[0], view=result[1])
+        else:
+            await interaction.followup.send("An error occurred.")
+
+    roll_again_button.callback = roll_again_callback
+
+    view = discord.ui.View()
+    view.add_item(roll_again_button)
+
+    sell_button = discord.ui.Button(style=discord.ButtonStyle.success, label=f"Sell Pet for ${rap_value}", custom_id="sell_pet")
+
+    async def sell_pet_callback(interaction: discord.Interaction):
+        if interaction.user.id != user_id:
+            await interaction.response.send_message("You cannot sell this pet.", ephemeral=True)
+            return
+
+        sell_value = rap_value
+        user_cash[user_id] = user_cash.get(user_id, 0) + sell_value
+        await interaction.response.send_message(f"You sold the pet for ${sell_value}. Your new balance is ${user_cash[user_id]}.", ephemeral=True)
+        for item in view.children:
+            if item.custom_id == "sell_pet":
+                view.remove_item(item)
+                break
+        await interaction.message.edit(view=view)
+
+    sell_button.callback = sell_pet_callback
+    view.add_item(sell_button)
+
+    if random.random() < 0.6 and luck_opportunities.get(user_id, 0) < 3:
+        luck_opportunities[user_id] = luck_opportunities.get(user_id, 0) + 1
+        increase_luck_button = discord.ui.Button(style=discord.ButtonStyle.success, label="Increase Luck", custom_id=f"increase_luck_{luck_opportunities[user_id]}")
+        
+        async def increase_luck_callback(interaction: discord.Interaction):
+            if interaction.user.id != user_id:
+                await interaction.response.send_message("You cannot use this button.", ephemeral=True)
+                return
+            
+            if user_id in used_luck_opportunities:
+                await interaction.response.send_message("You have already used your luck opportunity.", ephemeral=True)
+                return
+
+            current_luck = luck_multipliers.get(user_id, 1)
+            new_luck = min(current_luck + 1, 10)
+            luck_multipliers[user_id] = new_luck
+            luck_expiration[user_id] = time.time() + 1800
+            used_luck_opportunities.add(user_id)
+            
+            luck_percentage = (new_luck - 1) * 100
+            await interaction.response.send_message(f"Your luck has been increased to {luck_percentage}% for 30 minutes!", ephemeral=True)
+            
+            for item in view.children:
+                if item.custom_id == interaction.custom_id:
+                    view.remove_item(item)
+                    break
+            await interaction.message.edit(view=view)
+        
+        increase_luck_button.callback = increase_luck_callback
+        view.add_item(increase_luck_button)
+
+    return embed, view
+
+@app_commands.command(name="petroll", description="Roll for a random pet")
+async def petroll(interaction: discord.Interaction):
+    await interaction.response.defer()
+    result = await perform_roll(interaction)
+    if result:
+        await interaction.followup.send(embed=result[0], view=result[1])
+    else:
+        await interaction.followup.send("An error occurred.")
 
-    await interaction.followup.send(embed=embed)
+@app_commands.command(name="balance", description="Check your current balance")
+async def balance(interaction: discord.Interaction):
+    user_id = interaction.user.id
+    current_balance = user_cash.get(user_id, 0)
+    await interaction.response.send_message(f"Your current balance is ${current_balance}.", ephemeral=True)