From 399dc97808ffbebc2c11c9cf9f6f9fa11416f362 Mon Sep 17 00:00:00 2001
From: Saman Pandey <singularity.1@hotmail.com>
Date: Sun, 3 May 2026 20:00:17 +0530
Subject: [PATCH] fix: the decision modal is now working fine

---
 .../app/components/alerts/AlertToast.jsx      |  15 ++-
 .../app/components/decision/DecisionModal.jsx |   9 +-
 dashboard/app/demo/page.js                    | 101 ++++++++----------
 dashboard/app/hooks/useDemoInject.js          |  95 ++++++++++++++++
 4 files changed, 162 insertions(+), 58 deletions(-)
 create mode 100644 dashboard/app/hooks/useDemoInject.js

diff --git a/dashboard/app/components/alerts/AlertToast.jsx b/dashboard/app/components/alerts/AlertToast.jsx
index 207c6ef..b2c7555 100644
--- a/dashboard/app/components/alerts/AlertToast.jsx
+++ b/dashboard/app/components/alerts/AlertToast.jsx
@@ -3,6 +3,7 @@
 import { useEffect, useRef } from 'react';
 import { toast } from 'sonner';
 import { useAlertStore } from '../../store/alertStore.js';
+import { useDemoInject } from '../../hooks/useDemoInject.js';
 
 const TYPE_ICONS = {
   WEATHER: '🌊',
@@ -20,6 +21,8 @@ export default function AlertToastController() {
   const disruptions = useAlertStore((s) => s.disruptions);
   const prevLengthRef = useRef(null);
   const seenIdsRef = useRef(new Set());
+  const activeInjectionsRef = useRef(new Set());
+  const { injectDisruption } = useDemoInject(null);
 
   useEffect(() => {
     // Record initial snapshot as baseline so existing disruptions do not trigger toasts.
@@ -68,7 +71,15 @@ export default function AlertToastController() {
           <button
             className="mt-2 text-[10px] font-bold uppercase tracking-widest bg-[var(--bg-elevated)] hover:bg-[var(--bg-overlay)] text-[var(--text-primary)] px-3 py-1.5 rounded-xl border border-[var(--border-subtle)] transition-all text-center active:scale-95"
             onClick={() => {
-              useAlertStore.getState().setActiveDisruptionId(id);
+              // Prevent duplicate injections
+              if (activeInjectionsRef.current.has(id)) return;
+              activeInjectionsRef.current.add(id);
+
+              // Trigger full injection flow
+              injectDisruption(id, null);
+
+              // Dismiss toast after action
+              setTimeout(() => toast.dismiss(toastId), 300);
             }}
           >
             Protocol Analysis →
@@ -89,7 +100,7 @@ export default function AlertToastController() {
       );
     }
     prevLengthRef.current = disruptions.length;
-  }, [disruptions]);
+  }, [disruptions, injectDisruption]);
 
   return null;
 }
diff --git a/dashboard/app/components/decision/DecisionModal.jsx b/dashboard/app/components/decision/DecisionModal.jsx
index 6ae6d62..6540501 100644
--- a/dashboard/app/components/decision/DecisionModal.jsx
+++ b/dashboard/app/components/decision/DecisionModal.jsx
@@ -171,7 +171,14 @@ export default function DecisionModal() {
     if (isApproving || !traceId || approvedRank || isExecuted) return;
     setIsApproving(true);
     try {
-      const res = await fetch('/api/execute', { method: 'POST', headers: { 'Content-Type': 'application/json' }, body: JSON.stringify({ traceId, rank }) });
+      // Get disruptionId from activeResolution (set by demo page bridge)
+      const disruptionId = activeResolution?.disruptionId;
+
+      const res = await fetch('/api/execute', { 
+        method: 'POST', 
+        headers: { 'Content-Type': 'application/json' }, 
+        body: JSON.stringify({ traceId, rank, disruptionId }) 
+      });
       const result = await res.json();
       if (!res.ok) throw new Error(result.error || `HTTP ${res.status}`);
       setApprovedRank(rank);
diff --git a/dashboard/app/demo/page.js b/dashboard/app/demo/page.js
index 156f711..0fad2e5 100644
--- a/dashboard/app/demo/page.js
+++ b/dashboard/app/demo/page.js
@@ -1,6 +1,7 @@
 'use client';
 
 import { useCallback, useEffect, useRef, useState } from 'react';
+import dynamic from 'next/dynamic';
 import {
   collection,
   getDocs,
@@ -11,7 +12,8 @@ import {
   where,
 } from 'firebase/firestore';
 import { db, isFirebaseConfigured } from '../lib/firebase.js';
-import { watchDisruptionSupabase, watchImpactSupabase, watchResolutionSupabase } from '../lib/supabaseWatcher.js';
+import { useAlertStore } from '../store/alertStore.js';
+import { useDemoInject } from '../hooks/useDemoInject.js';
 import {
   Wind,
   Anchor,
@@ -32,6 +34,11 @@ import {
 } from 'lucide-react';
 import NavBar from '../components/NavBar.jsx';
 
+const DecisionModal = dynamic(
+  () => import('../components/decision/DecisionModal.jsx'),
+  { ssr: false, loading: () => null }
+);
+
 const SCENARIOS = [
   {
     id: 'pacific_storm',
@@ -417,6 +424,14 @@ export default function DemoPage() {
   const unsubsRef = useRef([]);
   const logEndRef = useRef(null);
 
+  // Extract store actions
+  const setActiveDisruptionId = useAlertStore((s) => s.setActiveDisruptionId);
+  const setResolutionWithOptions = useAlertStore((s) => s.setResolutionWithOptions);
+  const clearActiveDisruption = useAlertStore((s) => s.clearActiveDisruption);
+
+  // Injection hook - handles watcher chain
+  const { injectDisruption } = useDemoInject((msg, type) => log(msg, type));
+
   const log = useCallback((msg, type = 'info') => {
     const ts = new Date().toLocaleTimeString('en-US', { hour12: false });
     setLogs((prev) => [...prev.slice(-60), { ts, msg, type }]);
@@ -430,59 +445,15 @@ export default function DemoPage() {
     return () => unsubsRef.current.forEach((u) => u?.());
   }, []);
 
-  const watchResolutionRef = useRef(null);
-  const watchImpactRef = useRef(null);
-
-  const watchResolution = useCallback(
-    (dId) => {
-      log('Impact complete — AI resolution agent active…', 'info');
-      const unsub = watchResolutionSupabase(dId, ({ resolution: res, options: opts }) => {
-        setResolution(res);
-        setOptions(opts);
-        setStage('resolution');
-        log(`✓ AI generated ${opts.length} resolution strategies`, 'success');
-        setTimeout(() => setStage('decision'), 1200);
-      });
-      unsubsRef.current.push(unsub);
-    },
-    [log]
-  );
-
-  const watchImpact = useCallback(
-    (dId) => {
-      log('Monitor agent processing… watching impact_reports…', 'info');
-      const unsub = watchImpactSupabase(dId, (data) => {
-        setImpactReport(data);
-        setStage('impact');
-        log(
-          `✓ Impact scored: $${Number(data.totalCargoAtRiskUSD || 0).toLocaleString()} at risk across ${(data.affectedShipments || []).length} shipments`,
-          'success'
-        );
-        if (watchResolutionRef.current) watchResolutionRef.current(dId);
-      });
-      unsubsRef.current.push(unsub);
-    },
-    [log]
-  );
+  // Update local demo page state when stages progress
+  const handleStageChange = useCallback((newStage) => {
+    setStage(newStage);
+  }, []);
 
-  useEffect(() => {
-    watchResolutionRef.current = watchResolution;
-    watchImpactRef.current = watchImpact;
-  }, [watchResolution, watchImpact]);
-
-  const watchDisruption = useCallback(
-    (dId) => {
-      log(`Listening for disruption ${dId} in Supabase…`, 'info');
-      const unsub = watchDisruptionSupabase(dId, (data) => {
-        setDisruption(data);
-        setStage('monitoring');
-        log(`✓ Disruption confirmed: ${data.title || data.type || dId}`, 'success');
-        if (watchImpactRef.current) watchImpactRef.current(dId);
-      });
-      unsubsRef.current.push(unsub);
-    },
-    [log]
-  );
+  const handleResolutionReady = useCallback((res, opts) => {
+    setResolution(res);
+    setOptions(opts);
+  }, []);
 
   const handleLaunch = useCallback(async () => {
     if (!selectedScenario) return;
@@ -520,11 +491,27 @@ export default function DemoPage() {
       setDisruptionId(dId);
       setTraceId(tId);
 
+      // Bridge into alertStore — activates DecisionModal pipeline
+      setActiveDisruptionId(dId);
+
       log(`✓ Disruption published — ID: ${dId}`, 'success');
       log(`  TraceId: ${tId}`, 'info');
       log('Event bus fan-out complete. Agent pipeline starting…', 'info');
 
-      watchDisruption(dId);
+      // Inject into alert system - handles full watcher chain
+      const unsubscribers = injectDisruption(dId, (stage, data) => {
+        handleStageChange(stage);
+        // Update local state with data from watchers
+        if (stage === 'monitoring' && data) {
+          setDisruption(data);
+        } else if (stage === 'impact' && data) {
+          setImpactReport(data);
+        } else if (stage === 'resolution' && data?.resolution && data?.options) {
+          setResolution(data.resolution);
+          setOptions(data.options);
+        }
+      });
+      unsubsRef.current.push(...unsubscribers);
     } catch (err) {
       setError(err.message);
       setStage('idle');
@@ -532,7 +519,7 @@ export default function DemoPage() {
     } finally {
       setLaunching(false);
     }
-  }, [selectedScenario, log, watchDisruption]);
+  }, [selectedScenario, log, injectDisruption, handleStageChange]);
 
   // Auto-launch once scenario is pre-selected from URL
   useEffect(() => {
@@ -626,6 +613,7 @@ export default function DemoPage() {
   const handleReset = () => {
     unsubsRef.current.forEach((u) => u?.());
     unsubsRef.current = [];
+    clearActiveDisruption();
     setStage('idle');
     setSelectedScenario(null);
     setDisruptionId(null);
@@ -1300,6 +1288,9 @@ export default function DemoPage() {
             </div>
           </div>
         </main>
+
+        <DecisionModal />
+
       </div>
     </>
   );
diff --git a/dashboard/app/hooks/useDemoInject.js b/dashboard/app/hooks/useDemoInject.js
new file mode 100644
index 0000000..e4a9cfa
--- /dev/null
+++ b/dashboard/app/hooks/useDemoInject.js
@@ -0,0 +1,95 @@
+'use client';
+
+import { useCallback } from 'react';
+import { watchDisruptionSupabase, watchImpactSupabase, watchResolutionSupabase } from '../lib/supabaseWatcher.js';
+import { useAlertStore } from '../store/alertStore.js';
+
+/**
+ * Hook: useDemoInject
+ * 
+ * Provides reusable injection logic for triggering demo/alert flows.
+ * Handles:
+ * - Starting Supabase watchers
+ * - Bridging data into alertStore
+ * - Opening DecisionModal on resolution
+ */
+export function useDemoInject(onLog) {
+  const setActiveDisruptionId = useAlertStore((s) => s.setActiveDisruptionId);
+  const setResolutionWithOptions = useAlertStore((s) => s.setResolutionWithOptions);
+
+  /**
+   * Injects a disruption into the decision flow
+   * @param {string} disruptionId - The disruption ID
+   * @param {Function} onStageChange - Callback when stage changes
+   * @returns {Array} Array of unsubscribe functions for cleanup
+   */
+  const injectDisruption = useCallback(
+    (disruptionId, onStageChange) => {
+      if (!disruptionId) return [];
+
+      const unsubscribers = [];
+
+      // Notify the store that this disruption is active
+      setActiveDisruptionId(disruptionId);
+      onLog?.('Injecting disruption into alert system…', 'info');
+
+      // Watch disruption
+      const unsubDisruption = watchDisruptionSupabase(disruptionId, (data) => {
+        onLog?.(
+          `✓ Disruption confirmed: ${data.title || data.type || disruptionId}`,
+          'success'
+        );
+        onStageChange?.('monitoring', data);
+
+        // Watch impact after disruption confirmed
+        const unsubImpact = watchImpactSupabase(disruptionId, (impactData) => {
+          onLog?.(
+            `✓ Impact scored: $${Number(impactData.totalCargoAtRiskUSD || 0).toLocaleString()} at risk across ${(impactData.affectedShipments || []).length} shipments`,
+            'success'
+          );
+          onStageChange?.('impact', impactData);
+
+          // Watch resolution after impact available
+          const unsubResolution = watchResolutionSupabase(
+            disruptionId,
+            ({ resolution: res, options: opts }) => {
+              onLog?.(
+                `✓ AI generated ${opts.length} resolution strategies`,
+                'success'
+              );
+              onStageChange?.('resolution', { resolution: res, options: opts });
+
+              // Bridge into alertStore — opens DecisionModal
+              setResolutionWithOptions({
+                ...res,
+                disruptionId,
+                options: opts,
+                urgency: res.urgency ?? 8,
+                analysisText: res.analysisText,
+                impactReport: {
+                  totalCargoAtRiskUSD: res.totalCargoAtRiskUSD,
+                  cascadeRisk: res.cascadeRisk,
+                  urgency: res.urgency,
+                  analysisText: res.analysisText,
+                  affectedShipments: Array.from({
+                    length: res.shipmentCount || 0,
+                  }),
+                },
+              });
+
+              setTimeout(() => onStageChange?.('decision'), 1200);
+            }
+          );
+          unsubscribers.push(unsubResolution);
+        });
+        unsubscribers.push(unsubImpact);
+      });
+      unsubscribers.push(unsubDisruption);
+
+      return unsubscribers;
+    },
+    [setActiveDisruptionId, setResolutionWithOptions, onLog]
+  );
+
+  return { injectDisruption };
+}