From 47ff412c2b7cb8d0027a0216847785ab4f68bfd7 Mon Sep 17 00:00:00 2001
From: Justin Edmund <justin@jedmund.com>
Date: Thu, 3 Mar 2022 03:14:00 -0800
Subject: [PATCH] Change the weapon's image for null element weapons

---
 components/WeaponUnit/index.tsx | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/components/WeaponUnit/index.tsx b/components/WeaponUnit/index.tsx
index cf3d0826..21df410e 100644
--- a/components/WeaponUnit/index.tsx
+++ b/components/WeaponUnit/index.tsx
@@ -43,10 +43,17 @@ const WeaponUnit = (props: Props) => {
         if (props.gridWeapon) {
             const weapon = props.gridWeapon.object!
     
-            if (props.unitType == 0)
-                imgSrc = `${process.env.NEXT_PUBLIC_SIERO_IMG_URL}/weapon-main/${weapon.granblue_id}.jpg`
-            else
-                imgSrc = `${process.env.NEXT_PUBLIC_SIERO_IMG_URL}/weapon-grid/${weapon.granblue_id}.jpg`
+            if (props.unitType == 0) {
+                if (props.gridWeapon.object.element == 0 && props.gridWeapon.element)
+                    imgSrc = `${process.env.NEXT_PUBLIC_SIERO_IMG_URL}/weapon-main/${weapon.granblue_id}_${props.gridWeapon.element}.jpg`
+                else
+                    imgSrc = `${process.env.NEXT_PUBLIC_SIERO_IMG_URL}/weapon-main/${weapon.granblue_id}.jpg`
+            } else {
+                if (props.gridWeapon.object.element == 0 && props.gridWeapon.element)
+                    imgSrc = `${process.env.NEXT_PUBLIC_SIERO_IMG_URL}/weapon-grid/${weapon.granblue_id}_${props.gridWeapon.element}.jpg`
+                else
+                    imgSrc = `${process.env.NEXT_PUBLIC_SIERO_IMG_URL}/weapon-grid/${weapon.granblue_id}.jpg`
+            }
         }
         
         setImageUrl(imgSrc)