From cbcc9ada7288a708b880062863d380506f2b5adf Mon Sep 17 00:00:00 2001
From: Mitchell McCaffrey <mitchemmc@gmail.com>
Date: Fri, 1 Jan 2021 13:56:12 +1100
Subject: [PATCH] Fix file maps not appearing on refresh for map owner

---
 src/network/NetworkedMapAndTokens.js | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/src/network/NetworkedMapAndTokens.js b/src/network/NetworkedMapAndTokens.js
index e3eb650..9e45998 100644
--- a/src/network/NetworkedMapAndTokens.js
+++ b/src/network/NetworkedMapAndTokens.js
@@ -386,13 +386,20 @@ function NetworkedMapAndTokens({ session }) {
       assetProgressUpdate({ id, total, count });
     }
 
-    function handleSocketMap(map) {
+    async function handleSocketMap(map) {
       if (map) {
-        setCurrentMap(map);
+        // If we're the owner get the full map from the database
+        if (map.type === "file" && map.owner === userId) {
+          const fullMap = await getMapFromDB(map.id);
+          setCurrentMap(fullMap);
+        } else {
+          setCurrentMap(map);
+        }
       } else {
         setCurrentMap(null);
       }
     }
+
     session.on("peerData", handlePeerData);
     session.on("peerDataProgress", handlePeerDataProgress);
     session.socket?.on("map", handleSocketMap);