From 6d36f7b21e07fc47c65301058512ee2aa7074528 Mon Sep 17 00:00:00 2001
From: Laurent <laurent@staxriver.com>
Date: Fri, 21 Jun 2024 10:32:24 +0200
Subject: [PATCH] Fix club not being added to the user after setting the
 creator

---
 PadelClub/Data/DataStore.swift | 1 +
 PadelClub/Data/User.swift      | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/PadelClub/Data/DataStore.swift b/PadelClub/Data/DataStore.swift
index 590d913..9a4738b 100644
--- a/PadelClub/Data/DataStore.swift
+++ b/PadelClub/Data/DataStore.swift
@@ -141,6 +141,7 @@ class DataStore: ObservableObject {
         for club in clubsCollection {
             if let userId = Store.main.userId, club.creator == nil {
                 club.creator = userId
+                self.userStorage.item()?.addClub(club)
                 clubsCollection.writeChangeAndInsertOnServer(instance: club)
             }
         }
diff --git a/PadelClub/Data/User.swift b/PadelClub/Data/User.swift
index 29de3e6..ae24386 100644
--- a/PadelClub/Data/User.swift
+++ b/PadelClub/Data/User.swift
@@ -102,6 +102,12 @@ class User: ModelObject, UserBase, Storable {
         }
     }
     
+    func addClub(_ club: Club) {
+        if !self.clubs.contains(where: { $0.id == club.id }) {
+            self.clubs.append(club.id)
+        }
+    }
+    
     enum CodingKeys: String, CodingKey {
         case _id = "id"
         case _username = "username"