From bae6b49eac155fb5f338e0272e398701ccfc5eeb Mon Sep 17 00:00:00 2001
From: hujiabin <519660157@qq.com>
Date: Wed, 3 Jul 2024 15:41:51 +0800
Subject: [PATCH] 1

---
 webapp/models/Hero.php | 28 ++++++++++++++++++++++++++--
 1 file changed, 26 insertions(+), 2 deletions(-)

diff --git a/webapp/models/Hero.php b/webapp/models/Hero.php
index b5c15d6f..fc3e45eb 100644
--- a/webapp/models/Hero.php
+++ b/webapp/models/Hero.php
@@ -17,6 +17,7 @@ require_once('models/User.php');
 require_once('models/Avatar.php');
 require_once('models/ChipPlugin.php');
 require_once('models/ChipPage.php');
+require_once('models/Nft.php');
 require_once('services/NftService.php');
 require_once('services/FormulaService.php');
 require_once('services/ContractConfig.php');
@@ -186,7 +187,7 @@ class Hero extends BaseModel {
         $heroAtteMeta = \mt\EconomyAttribute::findByGrade($heroMeta['relationship'],$row['quality']);
         $unsealTime = $row['unseal_time'] ? $row['unseal_time'] : $row['createtime'];
         $validTime = $unsealTime + 86400 * $heroAtteMeta['validTime'] ;
-        return  array(
+        $dto = array(
             'idx' => $row['idx'],
             'token_id' => $row['token_id'],
             'hero_uniid' => $row['idx'],
@@ -211,7 +212,18 @@ class Hero extends BaseModel {
             "current_times" => myself()->_getDailyV(TN_DAILY_GOLD_MODE_BATTLE_TIMES,$row['idx']),
             "total_times" => $heroAtteMeta['roundPerDay'],
         );
-
+        $dto['is_avatar'] = 0;
+        if ($dto['token_id']){
+            $nftList = Nft::getNftListByType(myself()->_getAddress(),Nft::HERO_TYPE);
+            if (count($nftList) > 0){
+                foreach ($nftList as $value){
+                    if ($value['token_id'] == $dto['token_id']){
+                        $dto['is_avatar'] = 1;
+                    }
+                }
+            }
+        }
+        return $dto;
     }
 
     public static function toDto($row)
@@ -302,6 +314,18 @@ class Hero extends BaseModel {
 //            $nft_address = $contract ? $contract['hero'] : "";
 //        }
 //        $dto['nft_address'] = $nft_address;
+        $dto['is_avatar'] = 0;
+        if ($dto['token_id']){
+            $nftList = Nft::getNftListByType(myself()->_getAddress(),Nft::HERO_TYPE);
+            if (count($nftList) > 0){
+                foreach ($nftList as $value){
+                    if ($value['token_id'] == $dto['token_id']){
+                        $dto['is_avatar'] = 1;
+                    }
+                }
+            }
+        }
+
         return $dto;
     }