Add reset database script and frontend

inventory-server/src/routes/csv.js

@@ -16,6 +16,7 @@ let importProgress = null;
 // SSE clients for progress updates
 const updateClients = new Set();
 const importClients = new Set();
+const resetClients = new Set();
 
 // Helper to send progress to specific clients
 function sendProgressToClients(clients, progress) {
@@ -85,6 +86,27 @@ router.get('/import/progress', (req, res) => {
     });
 });
 
+router.get('/reset/progress', (req, res) => {
+    res.writeHead(200, {
+        'Content-Type': 'text/event-stream',
+        'Cache-Control': 'no-cache',
+        'Connection': 'keep-alive',
+        'Access-Control-Allow-Origin': req.headers.origin || '*',
+        'Access-Control-Allow-Credentials': 'true'
+    });
+
+    // Send an initial message to test the connection
+    res.write('data: {"status":"running","operation":"Initializing connection..."}\n\n');
+
+    // Add this client to the reset set
+    resetClients.add(res);
+
+    // Remove client when connection closes
+    req.on('close', () => {
+        resetClients.delete(res);
+    });
+});
+
 // Debug endpoint to verify route registration
 router.get('/test', (req, res) => {
     console.log('CSV test endpoint hit');
@@ -296,13 +318,26 @@ router.post('/cancel', (req, res) => {
         activeImport = null;
         importProgress = null;
         
-        // Notify all clients
+        // Get the operation type from the request
+        const { operation } = req.query;
+        
+        // Send cancel message only to the appropriate client set
         const cancelMessage = { 
             status: 'complete',
             operation: 'Operation cancelled'
         };
-        sendProgressToClients(updateClients, cancelMessage);
-        sendProgressToClients(importClients, cancelMessage);
+
+        switch (operation) {
+            case 'update':
+                sendProgressToClients(updateClients, cancelMessage);
+                break;
+            case 'import':
+                sendProgressToClients(importClients, cancelMessage);
+                break;
+            case 'reset':
+                sendProgressToClients(resetClients, cancelMessage);
+                break;
+        }
         
         res.json({ success: true });
     } catch (error) {
@@ -313,4 +348,90 @@ router.post('/cancel', (req, res) => {
     }
 });
 
+// Route to reset database
+router.post('/reset', async (req, res) => {
+    if (activeImport) {
+        return res.status(409).json({ error: 'Import already in progress' });
+    }
+
+    try {
+        const scriptPath = path.join(__dirname, '..', '..', 'scripts', 'reset-db.js');
+        
+        if (!require('fs').existsSync(scriptPath)) {
+            return res.status(500).json({ error: 'Reset script not found' });
+        }
+        
+        activeImport = spawn('node', [scriptPath]);
+        
+        activeImport.stdout.on('data', (data) => {
+            const output = data.toString().trim();
+            
+            try {
+                // Try to parse as JSON
+                const jsonData = JSON.parse(output);
+                sendProgressToClients(resetClients, {
+                    status: 'running',
+                    ...jsonData
+                });
+            } catch (e) {
+                // If not JSON, send as plain progress
+                sendProgressToClients(resetClients, { 
+                    status: 'running',
+                    progress: output
+                });
+            }
+        });
+
+        activeImport.stderr.on('data', (data) => {
+            const error = data.toString().trim();
+            try {
+                // Try to parse as JSON
+                const jsonData = JSON.parse(error);
+                sendProgressToClients(resetClients, { 
+                    status: 'error',
+                    ...jsonData
+                });
+            } catch {
+                sendProgressToClients(resetClients, { 
+                    status: 'error',
+                    error
+                });
+            }
+        });
+
+        await new Promise((resolve, reject) => {
+            activeImport.on('close', (code) => {
+                // Don't treat cancellation (code 143/SIGTERM) as an error
+                if (code === 0 || code === 143) {
+                    sendProgressToClients(resetClients, { 
+                        status: 'complete',
+                        operation: code === 143 ? 'Operation cancelled' : 'Reset complete'
+                    });
+                    resolve();
+                } else {
+                    const errorMsg = `Reset process exited with code ${code}`;
+                    sendProgressToClients(resetClients, { 
+                        status: 'error',
+                        error: errorMsg
+                    });
+                    reject(new Error(errorMsg));
+                }
+                activeImport = null;
+                importProgress = null;
+            });
+        });
+
+        res.json({ success: true });
+    } catch (error) {
+        console.error('Error resetting database:', error);
+        activeImport = null;
+        importProgress = null;
+        sendProgressToClients(resetClients, { 
+            status: 'error',
+            error: error.message
+        });
+        res.status(500).json({ error: 'Failed to reset database', details: error.message });
+    }
+});
+
 module.exports = router;