From f8112994ce9578b2f82a9b91837796720174001e Mon Sep 17 00:00:00 2001
From: William Melo <william.quintas@gmail.com>
Date: Thu, 2 Jun 2022 01:15:16 -0300
Subject: [PATCH] #13: Updates map when session db is modified

---
 .../DynamicModeContainer.tsx                  | 35 ++++++++++++++++++-
 features/sessions/slice.ts                    | 12 +++++--
 tsconfig.json                                 |  3 +-
 3 files changed, 46 insertions(+), 4 deletions(-)

diff --git a/components/DynamicModeContainer/DynamicModeContainer.tsx b/components/DynamicModeContainer/DynamicModeContainer.tsx
index bf9816b..e614153 100644
--- a/components/DynamicModeContainer/DynamicModeContainer.tsx
+++ b/components/DynamicModeContainer/DynamicModeContainer.tsx
@@ -8,7 +8,12 @@ import {
   Typography,
 } from "@mui/material";
 import { grey } from "@mui/material/colors";
-import { Fragment, FunctionComponent, useState } from "react";
+import { collection, doc, onSnapshot } from "firebase/firestore";
+import { Fragment, FunctionComponent, useEffect, useState } from "react";
+import { db } from "../../config/firebaseClient";
+import { useAppDispatch, useAppSelector } from "../../config/hooks";
+import { selectSession, updateEntities } from "../../features/sessions/slice";
+import { ICoordinatesData } from "../../models/ICoordinatesData";
 import MapContainer from "../MapContainer/MapContainer";
 
 interface DynamicModeContainerProps {
@@ -18,6 +23,34 @@ interface DynamicModeContainerProps {
 const DynamicModeContainer: FunctionComponent<DynamicModeContainerProps> = ({
   sessionId,
 }) => {
+  const dispatch = useAppDispatch();
+  const session = useAppSelector(selectSession);
+
+  const buildEntities = (coordinatesData: ICoordinatesData[]): IEntity[] => {
+    const entitiesIds = [...new Set(coordinatesData?.map(({ id }) => id))];
+    return entitiesIds.map((entityId) => ({
+      id: entityId,
+      color: "#ff0000",
+      coordinates: coordinatesData.filter(({ id }) => id === entityId),
+    }));
+  };
+
+  useEffect(() => {
+    if (sessionId) {
+      const sessionRef = doc(db, "sessions", sessionId);
+      const coordinatesRef = collection(sessionRef, "coordinates");
+
+      const unsubscribe = onSnapshot(coordinatesRef, ({ docs }) => {
+        const coordinatesData = docs.map((doc) =>
+          doc.data()
+        ) as ICoordinatesData[];
+
+        dispatch(updateEntities(buildEntities(coordinatesData)));
+      });
+      return () => unsubscribe();
+    }
+  }, [sessionId]);
+
   return (
     <Container component="main" maxWidth="lg">
       <Paper sx={{ my: 2, p: 1 }}>
diff --git a/features/sessions/slice.ts b/features/sessions/slice.ts
index a6c6aab..6f40d3d 100644
--- a/features/sessions/slice.ts
+++ b/features/sessions/slice.ts
@@ -40,6 +40,9 @@ export const sessionSlice = createSlice({
         state.entities = currentEntities;
       }
     },
+    updateEntities: (state, action: PayloadAction<IEntity[]>) => {
+      state.entities = action.payload;
+    },
     deleteEntity: (state, action: PayloadAction<string>) => {
       const currentEntities = [...state.entities];
 
@@ -50,8 +53,13 @@ export const sessionSlice = createSlice({
   },
 });
 
-export const { createSession, createEntity, editEntity, deleteEntity } =
-  sessionSlice.actions;
+export const {
+  createSession,
+  createEntity,
+  editEntity,
+  updateEntities,
+  deleteEntity,
+} = sessionSlice.actions;
 
 export const selectSession = (state: AppState) => state.session;
 export const selectEntities = (state: AppState) => state.session.entities;
diff --git a/tsconfig.json b/tsconfig.json
index 438839e..905f1bf 100644
--- a/tsconfig.json
+++ b/tsconfig.json
@@ -16,7 +16,8 @@
     "noEmit": true,
     "noFallthroughCasesInSwitch": true,
     "jsx": "preserve",
-    "incremental": true
+    "incremental": true,
+    "downlevelIteration": true
   },
   "include": ["next-env.d.ts", "**/*.ts", "**/*.tsx"],
   "exclude": ["node_modules"]