treat badges as collection

cholladay0816 · Dec 21, 2023 · 753ba0f · 753ba0f
1 parent 4638c01
commit 753ba0f
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/app/Models/User.php b/app/Models/User.php
@@ -8,6 +8,7 @@
 use Illuminate\Database\Eloquent\Relations\HasOne;
 use Illuminate\Foundation\Auth\User as Authenticatable;
 use Illuminate\Notifications\Notifiable;
+use Illuminate\Support\Collection;
 use Illuminate\Support\Facades\Cache;
 use Laravel\Cashier\Billable;
 use Laravel\Fortify\TwoFactorAuthenticatable;
@@ -116,11 +117,12 @@ public function activeAffiliate(): HasOne|Affiliate
             ->where('uses', '>', 0);
     }
 
-    public function getBadges(): array
+    public function getBadges(): array|Collection
     {
-        $badges = [];
-        if($this->activeAffiliate()->count()) {
-            $badges[] .= 'badges.affiliate';
+        $badges = collect();
+        if($this->activeAffiliate()->count())
+        {
+            $badges->push('badges.affiliate');
         }
         return $badges;
     }