Update index.js

index.js

@@ -27,54 +27,174 @@ app.use(cors());
 var initialData = {};
 let currentDeviceStates = {}; 
 
-const fetchInitialData = async () => {
-    const devices = [];
-    const rooms = [];
-    const users = [];
+// const fetchInitialData = async () => {
+//     const devices = [];
+//     const rooms = [];
+//     const users = [];
 
-    try {
-        const deviceSnapshot = await db.collection('Devices').get();
-        deviceSnapshot.forEach(doc => devices.push({ id: doc.id, ...doc.data() }));
+//     try {
+//         const deviceSnapshot = await db.collection('Devices').get();
+//         deviceSnapshot.forEach(doc => devices.push({ id: doc.id, ...doc.data() }));
 
-        const roomsSnapshot = await db.collectionGroup('Rooms').get(); // Get all rooms across hostels
-        roomsSnapshot.forEach(doc => rooms.push({ id: doc.id, ...doc.data(), hostelId: doc.ref.parent.parent.id })); // Include hostelId
+//         const roomsSnapshot = await db.collectionGroup('Rooms').get(); // Get all rooms across hostels
+//         roomsSnapshot.forEach(doc => rooms.push({ id: doc.id, ...doc.data(), hostelId: doc.ref.parent.parent.id })); // Include hostelId
 
-        const usersSnapshot = await db.collection('Users').get();
-        usersSnapshot.forEach(doc => users.push({ id: doc.id, ...doc.data() }));
+//         const usersSnapshot = await db.collection('Users').get();
+//         usersSnapshot.forEach(doc => users.push({ id: doc.id, ...doc.data() }));
 
-        initialData = { devices, rooms, users };
+//         initialData = { devices, rooms, users };
 
-        // console.log("Initial data fetched:", initialData);
-        //for each hostel id mantain a list of rooms with devices pertaining to that hostel
+//         // console.log("Initial data fetched:", initialData);
+//         //for each hostel id mantain a list of rooms with devices pertaining to that hostel
         
-        //setall to off
+//         //setall to off
 
-        initialData.devices.forEach(device => {
-            device.state = false;
-        }  ); 
+//         initialData.devices.forEach(device => {
+//             device.state = false;
+//         }  ); 
 
-        initialData.rooms.forEach(room => {
-            if (!currentDeviceStates[room.id]) {
-                currentDeviceStates[room.id] = [];
-            }
+//         initialData.rooms.forEach(room => {
+//             if (!currentDeviceStates[room.id]) {
+//                 currentDeviceStates[room.id] = [];
+//             }
             
-            const roomDevices = initialData.devices.filter(device => device.hostelId === room.hostelId);
-            currentDeviceStates[room.id] = { ...room, devices: roomDevices };
-        });
+//             const roomDevices = initialData.devices.filter(device => device.hostelId === room.hostelId);
+//             currentDeviceStates[room.id] = { ...room, devices: roomDevices };
+//         });
+
+
+
+
+//     } catch (error) {
+//         console.error("Error fetching initial data:", error);
+//         // return { devices: [], rooms: [], users: [] };
+//         initialData = { devices, rooms, users };
+//     }
+// };
+
+
+
+const fetchInitialData = async () => {
+  try {
+      const [deviceSnapshot, roomsSnapshot, usersSnapshot] = await Promise.all([
+          db.collection('Devices').get(),
+          db.collectionGroup('Rooms').get(),
+          db.collection('Users').get()
+      ]);
+
+      const devices = deviceSnapshot.docs.map(doc => ({ id: doc.id, ...doc.data() }));
+      const rooms = roomsSnapshot.docs.map(doc => ({ id: doc.id, ...doc.data(), hostelId: doc.ref.parent.parent.id }));
+      const users = usersSnapshot.docs.map(doc => ({ id: doc.id, ...doc.data() }));
+
+      initialData = { devices, rooms, users };
+
+      initialData.devices.forEach(device => {
+          device.state = false;
+      });
+
+      initialData.rooms.forEach(room => {
+          const roomDevices = initialData.devices.filter(device => device.hostelId === room.hostelId);
+          currentDeviceStates[room.id] = { ...room, devices: roomDevices.map(device => ({...device})) }; // Deep copy to avoid modifying initialData
+      });
+
+      console.log("Initial data fetched and processed.");
+
+
+  } catch (error) {
+      console.error("Error fetching initial data:", error);
+      initialData = { devices: [], rooms: [], users: [] }; // Provide default empty data on error
+      currentDeviceStates = {};
+  }
+};
+
+
+
+fetchInitialData().then(() => {
+  setupFirestoreListeners();
+});
+
+
 
 
 
 
-    } catch (error) {
-        console.error("Error fetching initial data:", error);
-        // return { devices: [], rooms: [], users: [] };
-        initialData = { devices, rooms, users };
-    }
+const updateInitialData = (collectionName, snapshot) => {
+  const changes = snapshot.docChanges();
+  changes.forEach(change => {
+      const docData = { id: change.doc.id, ...change.doc.data() };
+
+      switch (change.type) {
+          case 'added':
+              initialData[collectionName].push(docData);
+              break;
+          case 'modified':
+              const index = initialData[collectionName].findIndex(item => item.id === change.doc.id);
+              if (index !== -1) {
+                  initialData[collectionName][index] = docData;
+
+                  // Handle specific updates for device states in currentDeviceStates
+                  if (collectionName === 'devices') {
+
+                      for (const roomId in currentDeviceStates) {
+                          const roomDevices = currentDeviceStates[roomId].devices;
+                          const deviceIndex = roomDevices.findIndex(dev => dev.id === change.doc.id);
+                          if (deviceIndex !== -1) {
+                              currentDeviceStates[roomId].devices[deviceIndex] = {...docData, state: roomDevices[deviceIndex].state}
+                          }
+                      }
+
+                  }
+
+              }
+
+              break;
+          case 'removed':
+              initialData[collectionName] = initialData[collectionName].filter(item => item.id !== change.doc.id);
+              break;
+      }
+  });
+
+  console.log(`Updated ${collectionName} in initialData.`);
+
+
+  if (collectionName === 'devices' || collectionName === 'rooms') {
+      rebuildCurrentDeviceStates();
+  }
 };
 
 
+const setupFirestoreListeners = () => {
+
+  db.collection('Devices').onSnapshot(snapshot => {
+      updateInitialData('devices', snapshot);
+  });
+  db.collectionGroup('Rooms').onSnapshot(snapshot => {
+      updateInitialData('rooms', snapshot);
+      
+  });
+  db.collection('Users').onSnapshot(snapshot => {
+      updateInitialData('users', snapshot);
+
+  });
+}
+
+
+const rebuildCurrentDeviceStates = () => {
+  currentDeviceStates = {};
+  initialData.rooms.forEach(room => {
+      const roomDevices = initialData.devices.filter(device => device.hostelId === room.hostelId);
+      currentDeviceStates[room.id] = { ...room, devices: roomDevices.map(device => ({ ...device})) };
+  });
+  console.log("currentDeviceStates rebuilt.");
+}
+
+fetchInitialData().then(() => {
+  setupFirestoreListeners();
+});
+
+
+
 
-fetchInitialData();