From fe78df580fb94095b788728e2964b087a62fee7a Mon Sep 17 00:00:00 2001
From: Luke Robles <lukelrobles@gmail.com>
Date: Thu, 25 Jan 2024 15:59:19 -0800
Subject: [PATCH] rewrite of the paldeck command to make a nice embed

---
 app/cogs/palworld.py | 98 ++++++++++++++++++++++++++++++++++++++------
 1 file changed, 85 insertions(+), 13 deletions(-)

diff --git a/app/cogs/palworld.py b/app/cogs/palworld.py
index 8709f030..ab9da472 100755
--- a/app/cogs/palworld.py
+++ b/app/cogs/palworld.py
@@ -1,6 +1,8 @@
 from discord import option
 from discord.ext import commands, tasks
 import discord
+import requests
+from bs4 import BeautifulSoup
 
 
 class PalWorld(commands.Cog):
@@ -16,22 +18,19 @@ class PalWorld(commands.Cog):
         import requests
         from bs4 import BeautifulSoup
 
-        url = "https://game8.co/games/Palworld/archives/439556"
+        all_pals = []
+
+        url = "https://paldex.io/palworld/pals/"
         response = requests.get(url)
 
         soup = BeautifulSoup(response.text, "html.parser")
 
-        all_pals = []
-
-        table = soup.find("table", class_="a-table a-table a-table flexible-cell")
-        if table:
-            rows = table.find_all("tr")
-            for row in rows:
-                first_a = row.find("a")
-                if first_a:
-                    all_pals.append(first_a.text.strip())
-                else:
-                    pass
+        x = soup.find_all(
+            "p",
+            class_="text-base lg:text-lg group-hover:underline underline-offset-4 text-center",
+        )
+        for pal in x:
+            all_pals.append(pal.text)
 
         return all_pals
 
@@ -49,7 +48,80 @@ class PalWorld(commands.Cog):
             description="Pal to look up",
         ),
     ):
-        await ctx.respond("https://palpedia.net/pals/%s" % pal.replace(" ", "+"))
+        pal_url = "https://paldex.io/palworld/pals/%s/" % pal.lower()
+        response = requests.get(pal_url)
+        soup = BeautifulSoup(response.text, "html.parser")
+
+        color_lookup = {"Fire": discord.Color.red()}
+
+        pal_element_badge = soup.find("img", class_="rounded-full w-12 px-2 py-2")
+        element_text = pal_element_badge.find_next("p").text
+        element_icon = "https://paldex.io" + pal_element_badge["src"]
+
+        embed = discord.Embed(
+            description="-------",
+            color=discord.Color.blue(),
+            type="rich",
+            title=pal,
+        )
+
+        embed.set_author(
+            name=element_text,
+            icon_url=element_icon,
+            url=pal_url,
+        )
+
+        embed.set_thumbnail(
+            url=soup.find(
+                "img", class_="rounded-full w-32 sm:w-auto border-2 border-amber-500"
+            )["src"]
+        )
+
+        pal_description = soup.find("h3", class_="text-lg mt-2 text-slate-300").text
+        embed.add_field(name="**Description**", value=pal_description, inline=False)
+
+        embed.add_field(name="Work Skills", value="-----", inline=False)
+
+        # work skills
+        skills_emojis = {
+            "Kindling": "🔥",
+            "Watering": "💦",
+            "Planting": "🌱",
+            "Generating Electricity": "⚡️",
+            "Handwork": "🤚",
+            "Gathering": "🍃",
+            "Lumbering": "🪵",
+            "Mining": "⛏️",
+            "Medicine Production": "💉",
+            "Cooling": "❄️",
+            "Transporting": "📦",
+            "Farming": "🚜",
+        }
+
+        specialties = soup.find("div", class_="grid grid-cols-2 grid-rows-6 gap-2")
+        for work_skill in specialties.find_all(
+            "div",
+            class_="inline-flex flex-row gap-1 items-center bg-black-300 border border-amber-300 break-all",
+        ):
+            skill_name = work_skill.text.split("Lv")[0]
+            skill_level = work_skill.text.split("Lv")[1]
+
+            embed.add_field(
+                name=skills_emojis[skill_name] + " " + skill_name,
+                value="%s" % (int(skill_level) * "⭐️"),
+                inline=True,
+            )
+
+        # Drops
+        embed.add_field(name="Drops", value="-----", inline=False)
+        for div in soup.find("div", class_="flex flex-col gap-2"):
+            embed.add_field(
+                name=div.text.split("(")[0],
+                value="",
+                inline=False,
+            )
+        await ctx.defer()
+        await ctx.send_followup(embed=embed)
 
     @palworld.command(
         guild_ids=None,