From 178140f63fca71959d94e4408374aeb51a5a07ac Mon Sep 17 00:00:00 2001
From: Sendou <38327916+Sendouc@users.noreply.github.com>
Date: Tue, 2 Jun 2020 21:28:46 +0300
Subject: [PATCH] weapon pool affects build tab order

---
 .../src/components/user/UserPage.tsx          | 21 ++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

diff --git a/frontend-react/src/components/user/UserPage.tsx b/frontend-react/src/components/user/UserPage.tsx
index c5b157465..8a41f8e8e 100644
--- a/frontend-react/src/components/user/UserPage.tsx
+++ b/frontend-react/src/components/user/UserPage.tsx
@@ -113,9 +113,24 @@ const UserPage: React.FC<RouteComponentProps & UserPageProps> = ({ id }) => {
       content: (
         <TabPanel key={1}>
           <BuildTab
-            builds={builds.sort(
-              (a, b) => weapons.indexOf(a.weapon) - weapons.indexOf(b.weapon)
-            )}
+            builds={builds.sort((a, b) => {
+              const weaponPool = user.weapons
+              if (weaponPool) {
+                if (
+                  weaponPool.includes(a.weapon) &&
+                  weaponPool.includes(b.weapon)
+                ) {
+                  return (
+                    weaponPool.indexOf(a.weapon) - weaponPool.indexOf(b.weapon)
+                  )
+                }
+                const poolComparision =
+                  weaponPool.indexOf(b.weapon) - weaponPool.indexOf(a.weapon)
+
+                if (poolComparision !== 0) return poolComparision
+              }
+              return weapons.indexOf(a.weapon) - weapons.indexOf(b.weapon)
+            })}
             canModifyBuilds={userLean?.discord_id === user.discord_id}
           />
         </TabPanel>