From c62762fd576a0bef885b8337f230595817f3cb35 Mon Sep 17 00:00:00 2001
From: Waleed Hassan <waleed.hassan.aidid@gmail.com>
Date: Wed, 15 Nov 2023 11:44:31 +0300
Subject: [PATCH] Import: Support Groudon-Primal & Kyogre-Primal

---
 import/src/set-import.ts | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/import/src/set-import.ts b/import/src/set-import.ts
index 34feef312..59bd1d63f 100644
--- a/import/src/set-import.ts
+++ b/import/src/set-import.ts
@@ -296,9 +296,17 @@ function similarFormes(
     }
   }
   if (pset.species === 'Rayquaza-Mega' && format &&
-    (!format.id.includes('balancedhackmons') || format.id.includes('bh'))) {
+    (!format.id.includes('balancedhackmons') && !format.id.includes('bh'))) {
     return ['Rayquaza'] as SpeciesName[];
   }
+  if ((pset.species === 'Groudon' || pset.species === 'Groudon-Primal') &&
+    pset.item === 'Red Orb') {
+    return [pset.species === 'Groudon' ? 'Groudon-Primal' : 'Groudon'] as SpeciesName[];
+  }
+  if ((pset.species === 'Kyogre' || pset.species === 'Kyogre-Primal') &&
+    pset.item === 'Blue Orb') {
+    return [pset.species === 'Kyogre' ? 'Kyogre-Primal' : 'Kyogre'] as SpeciesName[];
+  }
   if (pset.ability === 'Power Construct') {
     return ['Zygarde-Complete'] as SpeciesName[];
   }
@@ -380,7 +388,10 @@ async function importGen(
             if (!statsIgnore[forme]) statsIgnore[forme] = new Set();
             statsIgnore[forme].add(formatID);
             if (!calcSets[forme]) calcSets[forme] = {};
-            if (forme.includes('-Mega')) {
+            // TODO: Make this check smarter - maybe by return a `diffAbility` bool
+            // in `similarFormes`?
+            if (forme.endsWith('-Mega') || forme.startsWith('Groudon') ||
+              forme.startsWith('Kyogre')) {
               const mega = getSpecie(gen, forme);
               calcSets[forme][setName] = {...calcSet, ability: mega.abilities[0]};
             } else {
@@ -428,7 +439,8 @@ async function importGen(
             continue;
           }
           if (!calcSets[forme]) calcSets[forme] = {};
-          if (item.megaEvolves && item.megaStone) {
+          if (forme.endsWith('-Mega') || forme.startsWith('Groudon') ||
+            forme.startsWith('Kyogre')) {
             const mega = getSpecie(gen, forme);
             calcSets[forme][setName] = {...calcSet, ability: mega.abilities[0]};
           } else {