Make relay rebroadcast data

Signed-off-by: Robert Gogete <gogeterobert@yahoo.com>

Author: gogeterobert

src/relay.ts

@@ -17,6 +17,14 @@ const PORT = process.env.PORT ? parseInt(process.env.PORT, 10) : 30878;
 // Graceful shutdown handler
 async function gracefulShutdown(signal: string): Promise<void> {
   console.log(`\n🛑 Received ${signal}, shutting down gracefully...`);
+  
+  // Stop re-broadcast interval
+  if (rebroadcastInterval) {
+    clearInterval(rebroadcastInterval);
+    rebroadcastInterval = null;
+    console.log('✅ Re-broadcast interval stopped');
+  }
+  
   console.log('✅ Gun relay server shutdown complete');
   process.exit(0);
 }
@@ -26,11 +34,59 @@ process.on('SIGTERM', () => gracefulShutdown('SIGTERM'));
 process.on('SIGINT', () => gracefulShutdown('SIGINT'));
 process.on('SIGUSR2', () => gracefulShutdown('SIGUSR2')); // For nodemon
 
+// Store data for re-broadcasting
+const storedData = new Map<string, Record<string, unknown>>();
+let rebroadcastInterval: ReturnType<typeof setInterval> | null = null;
+
+/**
+ * Store data for re-broadcasting
+ */
+function storeData(namespace: string, key: string, data: Record<string, unknown>): void {
+  const dataKey = `${namespace}/${key}`;
+  if (data && typeof data === 'object' && Object.keys(data).filter(k => k !== '_').length > 0) {
+    storedData.set(dataKey, { ...data });
+    console.log(`💾 Stored data for re-broadcasting: ${dataKey}`);
+  } else if (data === null || data === undefined) {
+    storedData.delete(dataKey);
+    console.log(`🗑️ Removed data from re-broadcast store: ${dataKey}`);
+  }
+}
+
+/**
+ * Re-broadcast all stored data
+ */
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+function rebroadcastStoredData(gun: any): void {
+  const timestamp = new Date().toISOString();
+  console.log(`📡 [${timestamp}] Re-broadcasting ${storedData.size} stored entries...`);
+  
+  storedData.forEach((data, dataKey) => {
+    const [namespace, key] = dataKey.split('/', 2);
+    if (namespace && key) {
+      try {
+        // Update lastSeen timestamp for host data
+        const rebroadcastData = { ...data };
+        if (rebroadcastData.storeId && rebroadcastData.lastSeen) {
+          rebroadcastData.lastSeen = Date.now();
+        }
+        
+        // Use gun as any to bypass TypeScript issues with dynamic property access
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        (gun as any).get(namespace).get(key).put(rebroadcastData);
+        console.log(`🔄 Re-broadcasted: ${namespace}/${key}`);
+      } catch (error) {
+        console.warn(`⚠️ Failed to re-broadcast ${dataKey}:`, error);
+      }
+    }
+  });
+}
+
 async function startRelay(): Promise<void> {
   console.log('🚀 Starting Gun relay server...');
   console.log(`📡 Port: ${PORT}`);
   console.log('🔗 WebRTC enabled with mesh networking for peer-to-peer connections');
-  console.log('📊 This relay serves as discovery point only - data flows peer-to-peer');
+  console.log('📊 This relay serves as discovery point with periodic re-broadcasting');
+  console.log('🔄 Re-broadcasting stored data every 5 seconds');
   console.log('📝 Logging enabled for:');
   console.log('   • TCP connections and disconnections');
   console.log('   • Gun.js peer connections (hi/bye events)');
@@ -142,6 +198,9 @@ async function startRelay(): Promise<void> {
         hasData: !!data,
         dataKeys: data ? Object.keys(data).filter(k => k !== '_') : []
       });
+      
+      // Store data for re-broadcasting
+      storeData('dig-nat-tools', key, data);
     }
   });
 
@@ -154,6 +213,9 @@ async function startRelay(): Promise<void> {
         hasData: !!data,
         dataKeys: data ? Object.keys(data).filter(k => k !== '_') : []
       });
+      
+      // Store data for re-broadcasting
+      storeData('dig-nat-tools-test', key, data);
     }
   });
 
@@ -192,6 +254,14 @@ async function startRelay(): Promise<void> {
 
   console.log('✅ Gun.js database initialized with comprehensive logging');
 
+  // Start re-broadcasting interval
+  console.log('🔄 Starting re-broadcast interval (every 5 seconds)...');
+  rebroadcastInterval = setInterval(() => {
+    rebroadcastStoredData(gun);
+  }, 5000); // Re-broadcast every 5 seconds
+
+  console.log('✅ Re-broadcast interval started');
+
   // Handle server errors
   server.on('error', (error) => {
     console.error('❌ Server error:', error);