From 5ede5673a9d86d445a734d0ed0afbf105c5c81b0 Mon Sep 17 00:00:00 2001
From: Dragon Fire <danielbodendahl@gmail.com>
Date: Thu, 10 Dec 2020 08:45:27 -0500
Subject: [PATCH] Fix rarity

---
 structures/pokemon/Pokemon.js | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/structures/pokemon/Pokemon.js b/structures/pokemon/Pokemon.js
index 0689e3b9..e97d0e58 100644
--- a/structures/pokemon/Pokemon.js
+++ b/structures/pokemon/Pokemon.js
@@ -145,7 +145,22 @@ module.exports = class Pokemon {
 				}
 				return false;
 			}))
-			.map(item => ({ url: item.item.url, name: null, rarity: item.rarity }));
+			.map(item => {
+				const inSwordShield = item.version_details
+					.some(version => version.version.name === 'sword' || version.version.name === 'shield');
+				const rarity = item.version_details
+					.find(version => {
+						if (inSwordShield) return true;
+						const sunMoon = version.version.name === 'ultra-sun' || version.version.name === 'ultra-moon';
+						if (!inSwordShield && sunMoon) return true;
+						return false;
+					}).rarity;
+				return {
+					url: item.item.url,
+					name: null,
+					rarity
+				};
+			});
 		await this.fetchHeldItemNames(defaultBody);
 		await this.fetchChain();
 		this.gameDataCached = true;