From 95448832c3d317bcc8640dc0e948205d72098eaf Mon Sep 17 00:00:00 2001
From: Sem van der Hoeven <semmer99@gmail.com>
Date: Wed, 21 Oct 2020 20:42:00 +0200
Subject: [PATCH] [ADD] added get lobby for users method

---
 Client/ViewModels/ViewModel.cs       |  3 ---
 Server/Models/ServerCommunication.cs | 12 ++++++++++++
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/Client/ViewModels/ViewModel.cs b/Client/ViewModels/ViewModel.cs
index 1ebcc7e..bbb9843 100644
--- a/Client/ViewModels/ViewModel.cs
+++ b/Client/ViewModels/ViewModel.cs
@@ -55,9 +55,6 @@ namespace Client
             wantToBeHost = true;
             wantToBeHostId = id;
             client.OnLobbiesReceivedAndWaitingForHost = hostLobbiesReceived;
-            // lobby id krijgen waarvan je host bent
-            // wachten totdat list van lobbies is gekomen
-            // uit die list de id zoeken en die joinen
 
         }
 
diff --git a/Server/Models/ServerCommunication.cs b/Server/Models/ServerCommunication.cs
index f0eb010..a069725 100644
--- a/Server/Models/ServerCommunication.cs
+++ b/Server/Models/ServerCommunication.cs
@@ -121,6 +121,18 @@ namespace Server.Models
             }
         }
 
+        public Lobby GetLobbyForUser(User user)
+        {
+            foreach (Lobby l in lobbies)
+            {
+                if (l.Users.Contains(user))
+                {
+                    return l;
+                }
+            }
+            return null;
+        }
+
         public void AddToLobby(Lobby lobby, User user)
         {
             foreach (Lobby l in lobbies)