Improve rate limit handling and query efficiency

Refactored Supabase rate limit logic for better exponential backoff and global state management. Updated cachedAuth to synchronize session cache and handle rate limit cooldowns more robustly. Optimized database queries in RebalanceHistoryTable and UnifiedAnalysisHistory to select only necessary fields, improving performance. Enhanced UI loading state in RebalanceHistoryTable and fixed day formatting in ScheduleListModal. Improved next run calculation in timeUtils to always return the earliest valid candidate.

Author: BWJ2310

src/components/Header.tsx

@@ -124,15 +124,15 @@ export default function Header() {
         <div className="container mx-auto px-6 py-4">
           <div className="flex items-center justify-between">
             <div className="flex items-center gap-2 sm:gap-4">
-              <div className="flex items-center gap-2 sm:gap-3">
+              <a className="flex items-center gap-2 sm:gap-3" href="/">
                 <div className="p-2 rounded-lg">
                   <img src="/goose.png" alt="TradingGoose Logo" className="h-8 w-8 sm:h-10 sm:w-10" />
                 </div>
                 <div>
                   <h1 className="text-xl sm:text-2xl font-bold" style={{ color: '#FFCC00' }}>TradingGoose</h1>
                   <p className="text-xs sm:text-sm text-muted-foreground">AI-Powered Portfolio Management</p>
                 </div>
-              </div>
+              </a>
 
               {isAuthenticated && (
                 <div className="hidden md:flex items-center gap-1">

src/components/RebalanceHistoryTable.tsx

@@ -63,7 +63,7 @@ interface RebalanceAnalysis {
 
 interface RebalanceRequest {
   id: string;
-  user_id: string;
+  user_id?: string;
   status: RebalanceStatus | string; // Support both new RebalanceStatus and legacy strings
   created_at: string;
   total_stocks: number;
@@ -185,7 +185,7 @@ export default function RebalanceHistoryTable() {
     try {
       const { data, error } = await supabase
         .from('analysis_history')
-        .select('*')
+        .select('id,ticker,analysis_status,full_analysis,created_at')
         .eq('rebalance_request_id', rebalanceId);
 
       if (error) throw error;
@@ -214,7 +214,7 @@ export default function RebalanceHistoryTable() {
       // Query only rebalances from the selected date
       const { data, error } = await supabase
         .from('rebalance_requests')
-        .select('*')
+        .select('id,status,created_at,total_stocks,stocks_analyzed,plan_generated_at,error_message')
         .eq('user_id', user.id)
         .gte('created_at', startOfDay.toISOString())
         .lte('created_at', endOfDay.toISOString())
@@ -274,12 +274,19 @@ export default function RebalanceHistoryTable() {
       setCancelledRebalances(cancelled);
 
       // Fetch analysis data for running rebalances to calculate progress
-      const analysisDataMap: { [key: string]: any[] } = {};
-      for (const rebalance of running) {
-        const analyses = await fetchAnalysisDataForRebalance(rebalance.id);
-        analysisDataMap[rebalance.id] = analyses;
+      if (running.length > 0) {
+        const analysisResults = await Promise.all(
+          running.map((rebalance) => fetchAnalysisDataForRebalance(rebalance.id))
+        );
+
+        const analysisDataMap: { [key: string]: any[] } = {};
+        running.forEach((rebalance, index) => {
+          analysisDataMap[rebalance.id] = analysisResults[index];
+        });
+        setAnalysisData(analysisDataMap);
+      } else {
+        setAnalysisData({});
       }
-      setAnalysisData(analysisDataMap);
 
     } catch (error) {
       console.error('Error fetching rebalance requests:', error);
@@ -688,13 +695,96 @@ export default function RebalanceHistoryTable() {
   if (loading) {
     return (
       <Card>
-        <CardHeader>
-          <CardTitle>Rebalance History</CardTitle>
-        </CardHeader>
-        <CardContent>
-          <div className="flex items-center justify-center py-8">
-            <Loader2 className="h-6 w-6 animate-spin" />
+        <CardContent className="pt-6">
+          <div className="flex items-center justify-between mb-4">
+            <h3 className="text-lg font-semibold">Rebalance History</h3>
+            <div className="flex items-center gap-2">
+              {/* Manual refresh button */}
+              <Button
+                variant="ghost"
+                size="sm"
+                className="h-8 w-8 p-0 hover:bg-[#fc0]/10 hover:text-[#fc0]"
+                title="Refresh"
+                disabled
+              >
+                <RefreshCw className="h-4 w-4" />
+              </Button>
+
+              <div className="w-px h-6 bg-border" />
+
+              <div className="flex items-center gap-1">
+                <Button
+                  variant="ghost"
+                  size="sm"
+                  disabled
+                  className="h-8 w-8 p-0 hover:bg-[#fc0]/10 hover:text-[#fc0]"
+                >
+                  <ChevronLeft className="h-4 w-4" />
+                </Button>
+                <Popover>
+                  <PopoverTrigger asChild>
+                    <Button
+                      variant="outline"
+                      size="sm"
+                      disabled
+                      className="justify-start text-left"
+                    >
+                      <CalendarIcon className="h-4 w-4 mr-2" />
+                      {getDateDisplay()}
+                    </Button>
+                  </PopoverTrigger>
+                  <PopoverContent className="w-auto p-0" align="start">
+                    <Calendar
+                      mode="single"
+                      selected={new Date(selectedDate)}
+                      initialFocus
+                    />
+                  </PopoverContent>
+                </Popover>
+                <Button
+                  variant="ghost"
+                  size="sm"
+                  disabled
+                  className="h-8 w-8 p-0 hover:bg-[#fc0]/10 hover:text-[#fc0]"
+                >
+                  <ChevronRight className="h-4 w-4" />
+                </Button>
+              </div>
+            </div>
           </div>
+
+          <Tabs value={activeTab} className="space-y-4">
+            <TabsList className="grid w-full grid-cols-4">
+              <TabsTrigger value="all">All</TabsTrigger>
+              <TabsTrigger value="running">Active</TabsTrigger>
+              <TabsTrigger value="completed">Completed</TabsTrigger>
+              <TabsTrigger value="canceled">Canceled</TabsTrigger>
+            </TabsList>
+
+            <TabsContent value="all" className="space-y-4">
+              <div className="flex items-center justify-center py-8">
+                <Loader2 className="h-6 w-6 animate-spin" />
+              </div>
+            </TabsContent>
+
+            <TabsContent value="running">
+              <div className="flex items-center justify-center py-8">
+                <Loader2 className="h-6 w-6 animate-spin" />
+              </div>
+            </TabsContent>
+
+            <TabsContent value="completed">
+              <div className="flex items-center justify-center py-8">
+                <Loader2 className="h-6 w-6 animate-spin" />
+              </div>
+            </TabsContent>
+
+            <TabsContent value="canceled">
+              <div className="flex items-center justify-center py-8">
+                <Loader2 className="h-6 w-6 animate-spin" />
+              </div>
+            </TabsContent>
+          </Tabs>
         </CardContent>
       </Card>
     );

src/components/ScheduleListModal.tsx

@@ -271,7 +271,10 @@ export default function ScheduleListModal({ isOpen, onClose }: ScheduleListModal
   const formatDaysList = (days?: number[]) => {
     if (!days || days.length === 0) return '';
     const dayNames = ['Sun', 'Mon', 'Tue', 'Wed', 'Thu', 'Fri', 'Sat'];
-    return days.map(d => dayNames[d]).join(', ');
+    return [...new Set(days)]
+      .sort((a, b) => a - b)
+      .map(d => dayNames[d])
+      .join(', ');
   };
 
   // State for storing calculated next run times and true UTC time
@@ -630,4 +633,4 @@ export default function ScheduleListModal({ isOpen, onClose }: ScheduleListModal
       </AlertDialog>
     </>
   );
-}
+}

src/components/UnifiedAnalysisHistory.tsx

@@ -61,6 +61,8 @@ interface AnalysisHistoryItem {
   full_analysis?: any;
   created_at: string;
   analysis_status?: AnalysisStatus | number; // Support both new and legacy formats
+  rebalance_request_id?: string | null;
+  is_canceled?: boolean;
 }
 
 interface RunningAnalysisItem {
@@ -235,10 +237,10 @@ export default function UnifiedAnalysisHistory() {
       const startOfDay = new Date(year, month - 1, day, 0, 0, 0, 0);
       const endOfDay = new Date(year, month - 1, day, 23, 59, 59, 999);
 
-      // Query only analyses from the selected date
+      // Query only analyses from the selected date (summary fields only)
       const { data, error } = await supabase
         .from('analysis_history')
-        .select('*')
+        .select('id,ticker,analysis_date,decision,confidence,agent_insights,created_at,analysis_status,rebalance_request_id,is_canceled')
         .eq('user_id', user.id)
         .gte('created_at', startOfDay.toISOString())
         .lte('created_at', endOfDay.toISOString())
@@ -266,61 +268,92 @@ export default function UnifiedAnalysisHistory() {
         throw error;
       }
 
-      // Separate running, completed, and canceled analyses based on database data
-      const runningAnalyses: RunningAnalysisItem[] = [];
+      const baseRows = data ?? [];
+      const runningIds: string[] = [];
+      const runningBase: Array<Omit<RunningAnalysisItem, 'full_analysis'>> = [];
       const completedAnalyses: AnalysisHistoryItem[] = [];
       const canceledAnalyses: AnalysisHistoryItem[] = [];
 
-      for (const item of data || []) {
-        // Use analysis_status field if available, otherwise fall back to old logic
-        if ('analysis_status' in item) {
-          // Convert legacy numeric status to new string format
-          const status: AnalysisStatus = typeof item.analysis_status === 'number'
-            ? convertLegacyAnalysisStatus(item.analysis_status)
-            : item.analysis_status as AnalysisStatus;
-
-          if (status === ANALYSIS_STATUS.RUNNING || status === ANALYSIS_STATUS.PENDING) {
-            runningAnalyses.push({
-              id: item.id,
-              ticker: item.ticker,
-              created_at: item.created_at,
-              full_analysis: item.full_analysis,
-              agent_insights: item.agent_insights,  // Add this!
-              rebalance_request_id: item.rebalance_request_id,
-              status: status
-            });
-          } else if (status === ANALYSIS_STATUS.COMPLETED) {
-            completedAnalyses.push(item);
-          } else if (status === ANALYSIS_STATUS.ERROR || status === ANALYSIS_STATUS.CANCELLED) {
-            // Show canceled/error analyses in the canceled section
-            canceledAnalyses.push({
-              ...item,
-              decision: status === ANALYSIS_STATUS.ERROR ? 'ERROR' : 'CANCELED',
-              confidence: item.confidence || 0
-            });
-          }
-        } else {
-          // Fall back to old logic for backward compatibility
-          const hasAgentInsights = item.agent_insights && Object.keys(item.agent_insights).length > 0;
-          const isRunning = item.analysis_status === ANALYSIS_STATUS.RUNNING ||
-            (item.confidence === 0 && !hasAgentInsights);
-
-          if (isRunning) {
-            runningAnalyses.push({
-              id: item.id,
-              ticker: item.ticker,
-              created_at: item.created_at,
-              full_analysis: item.full_analysis,
-              rebalance_request_id: item.rebalance_request_id,
-              status: ANALYSIS_STATUS.RUNNING  // Default to running for fallback logic
-            });
-          } else if ((item.confidence > 0 || hasAgentInsights) && item.decision && ['BUY', 'SELL', 'HOLD'].includes(item.decision)) {
-            completedAnalyses.push(item);
+      for (const item of baseRows) {
+        let status: AnalysisStatus | null = null;
+
+        if (typeof item.analysis_status === 'number') {
+          status = convertLegacyAnalysisStatus(item.analysis_status);
+        } else if (typeof item.analysis_status === 'string') {
+          status = item.analysis_status.toLowerCase() as AnalysisStatus;
+        }
+
+        if (!status) {
+          const hasAgentInsights = item.agent_insights && Object.keys(item.agent_insights || {}).length > 0;
+
+          if (item.is_canceled) {
+            status = ANALYSIS_STATUS.CANCELLED;
+          } else if (item.confidence === 0 && !hasAgentInsights) {
+            status = ANALYSIS_STATUS.RUNNING;
+          } else if (item.decision && ['BUY', 'SELL', 'HOLD'].includes(item.decision)) {
+            status = ANALYSIS_STATUS.COMPLETED;
+          } else {
+            status = ANALYSIS_STATUS.RUNNING;
           }
         }
+
+        if (status === ANALYSIS_STATUS.RUNNING || status === ANALYSIS_STATUS.PENDING) {
+          runningIds.push(item.id);
+          runningBase.push({
+            id: item.id,
+            ticker: item.ticker,
+            created_at: item.created_at,
+            agent_insights: item.agent_insights,
+            rebalance_request_id: item.rebalance_request_id || undefined,
+            status
+          });
+        } else if (status === ANALYSIS_STATUS.COMPLETED) {
+          completedAnalyses.push({
+            id: item.id,
+            ticker: item.ticker,
+            analysis_date: item.analysis_date,
+            decision: item.decision,
+            confidence: item.confidence,
+            agent_insights: item.agent_insights,
+            created_at: item.created_at,
+            analysis_status: status,
+            rebalance_request_id: item.rebalance_request_id || undefined
+          });
+        } else if (status === ANALYSIS_STATUS.ERROR || status === ANALYSIS_STATUS.CANCELLED) {
+          canceledAnalyses.push({
+            id: item.id,
+            ticker: item.ticker,
+            analysis_date: item.analysis_date,
+            decision: status === ANALYSIS_STATUS.ERROR ? 'ERROR' : 'CANCELED',
+            confidence: item.confidence || 0,
+            agent_insights: item.agent_insights,
+            created_at: item.created_at,
+            analysis_status: status,
+            rebalance_request_id: item.rebalance_request_id || undefined
+          });
+        }
       }
 
-      setRunningAnalyses(runningAnalyses);
+      let runningDetailsMap = new Map<string, any>();
+      if (runningIds.length > 0) {
+        const { data: runningDetails, error: runningDetailsError } = await supabase
+          .from('analysis_history')
+          .select('id, full_analysis')
+          .in('id', runningIds);
+
+        if (!runningDetailsError && runningDetails) {
+          runningDetailsMap = new Map(
+            runningDetails.map(detail => [detail.id, detail.full_analysis])
+          );
+        }
+      }
+
+      const hydratedRunningAnalyses: RunningAnalysisItem[] = runningBase.map(item => ({
+        ...item,
+        full_analysis: runningDetailsMap.get(item.id) || null
+      }));
+
+      setRunningAnalyses(hydratedRunningAnalyses);
       setHistory(completedAnalyses);
       setCanceledAnalyses(canceledAnalyses);
 
@@ -390,7 +423,7 @@ export default function UnifiedAnalysisHistory() {
         // Only fetch the specific running analyses to check their status
         const { data, error } = await supabase
           .from('analysis_history')
-          .select('*')
+          .select('id,analysis_status,full_analysis,agent_insights,created_at')
           .eq('user_id', user.id)
           .in('id', runningIds)
           .gte('created_at', startOfDay.toISOString())