added database support v1

Author: IAmMyGuy21th

server/index.js

@@ -67,6 +67,32 @@ wss.on('connection', async (ws) => {
             if (message.header === 'LINKS' && db) {
                 const linksCollection = db.collection('links');
 
+                // Handle syncing websocket state with database
+                if (message.type === 'sync_links') {
+                    if (message.links && message.links.length > 0) {
+                        // If websocket has links, replace database content
+                        try {
+                            await linksCollection.deleteMany({});
+                            await linksCollection.insertMany(message.links);
+                            console.log('Database synchronized with WebSocket state');
+                        } catch (err) {
+                            console.error('Error syncing database:', err);
+                        }
+                    } else {
+                        // If no links in websocket, try to get from database
+                        const dbLinks = await linksCollection.find({}).sort({ createdAt: -1 }).toArray();
+                        if (dbLinks.length > 0) {
+                            ws.send(JSON.stringify({
+                                header: 'LINKS',
+                                type: 'links_list',
+                                links: dbLinks
+                            }));
+                            console.log('Sent database links to client');
+                        }
+                    }
+                    return;
+                }
+
                 if (message.type === 'delete_link') {
                     try {
                         // Delete the link from MongoDB

src/pages/LinksGrid.js

@@ -45,8 +45,12 @@ function LinksGrid() {
                 setStatus('Connected');
                 console.log('WebSocket connected');
 
-                // Request existing links
-                newWs.send(JSON.stringify({ header: 'LINKS', type: 'request_links' }));
+                // Send sync request instead of just requesting links
+                newWs.send(JSON.stringify({
+                    header: 'LINKS',
+                    type: 'sync_links',
+                    links: [] // Empty array to trigger database check
+                }));
             };
 
             newWs.onmessage = (event) => {
@@ -121,6 +125,18 @@ function LinksGrid() {
     }, []);
 
     // Effect for WebSocket connection management
+    // Effect for database synchronization
+    useEffect(() => {
+        if (ws && ws.readyState === WebSocket.OPEN) {
+            // Send sync request
+            ws.send(JSON.stringify({
+                header: 'LINKS',
+                type: 'sync_links',
+                links: links
+            }));
+        }
+    }, [ws, links]);
+
     useEffect(() => {
         console.log('Initializing WebSocket connection...');
         const cleanup = connect();