Add race control visuals and driver names in laps view

internal/api/openf1.go

@@ -273,24 +273,12 @@ func (c *OpenF1Client) getLatestRaceSessionKey() (int, error) {
 		return 0, errors.New("no Race sessions found")
 	}
 
-	// Walk backwards to find the most recent completed race.
 	now := time.Now()
-	for i := len(sessions) - 1; i >= 0; i-- {
-		s := sessions[i]
-		if s.DateEnd != "" {
-			endTime, err := time.Parse(time.RFC3339, s.DateEnd)
-			if err == nil && endTime.Before(now) {
-				return s.SessionKey, nil
-			}
-		} else if s.DateStart != "" {
-			startTime, err := time.Parse(time.RFC3339, s.DateStart)
-			if err == nil && startTime.Add(3*time.Hour).Before(now) {
-				return s.SessionKey, nil
-			}
-		}
+	latestKey, ok := latestCompletedRaceSessionKey(sessions, now)
+	if !ok {
+		return 0, errors.New("no completed Race sessions found")
 	}
-
-	return 0, errors.New("no completed Race sessions found")
+	return latestKey, nil
 }
 
 // getLatestRaceSessionKeyForYear returns the session_key of the most recent
@@ -311,25 +299,47 @@ func (c *OpenF1Client) getLatestRaceSessionKeyForYear(year int) (int, error) {
 		return 0, fmt.Errorf("no Race sessions found for year %d", year)
 	}
 
-	// Walk backwards to find the most recent completed race.
 	now := time.Now()
-	for i := len(sessions) - 1; i >= 0; i-- {
-		s := sessions[i]
-		if s.DateEnd != "" {
-			endTime, err := time.Parse(time.RFC3339, s.DateEnd)
-			if err == nil && endTime.Before(now) {
-				return s.SessionKey, nil
-			}
-		} else if s.DateStart != "" {
-			// Fallback: if no DateEnd, check DateStart + 3 hours as a rough estimate.
-			startTime, err := time.Parse(time.RFC3339, s.DateStart)
-			if err == nil && startTime.Add(3*time.Hour).Before(now) {
-				return s.SessionKey, nil
-			}
+	latestKey, ok := latestCompletedRaceSessionKey(sessions, now)
+	if !ok {
+		return 0, fmt.Errorf("no completed Race sessions found for year %d", year)
+	}
+	return latestKey, nil
+}
+
+func latestCompletedRaceSessionKey(sessions []models.Session, now time.Time) (int, bool) {
+	var latestKey int
+	var latestTime time.Time
+
+	for _, s := range sessions {
+		completedAt, ok := completedRaceTime(s)
+		if !ok || !completedAt.Before(now) {
+			continue
+		}
+		if latestKey == 0 || completedAt.After(latestTime) {
+			latestKey = s.SessionKey
+			latestTime = completedAt
 		}
 	}
 
-	return 0, fmt.Errorf("no completed Race sessions found for year %d", year)
+	return latestKey, latestKey != 0
+}
+
+func completedRaceTime(s models.Session) (time.Time, bool) {
+	if s.DateEnd != "" {
+		endTime, err := time.Parse(time.RFC3339, s.DateEnd)
+		if err == nil {
+			return endTime, true
+		}
+	}
+	if s.DateStart == "" {
+		return time.Time{}, false
+	}
+	startTime, err := time.Parse(time.RFC3339, s.DateStart)
+	if err != nil {
+		return time.Time{}, false
+	}
+	return startTime.Add(3 * time.Hour), true
 }
 
 // GetLatestDriverChampionship returns championship standings for the most recent

internal/api/session_selection_test.go

@@ -0,0 +1,60 @@
+package api
+
+import (
+	"testing"
+	"time"
+
+	"github.com/AmanTahiliani/box-box/internal/models"
+)
+
+func TestLatestCompletedRaceSessionKeyUsesDatesNotInputOrder(t *testing.T) {
+	now := time.Date(2026, 1, 1, 0, 0, 0, 0, time.UTC)
+	sessions := []models.Session{
+		{
+			SessionKey:  1,
+			SessionName: "Race",
+			DateStart:   "2025-12-01T13:00:00+00:00",
+			DateEnd:     "2025-12-01T15:00:00+00:00",
+		},
+		{
+			SessionKey:  2,
+			SessionName: "Race",
+			DateStart:   "2025-03-01T13:00:00+00:00",
+			DateEnd:     "2025-03-01T15:00:00+00:00",
+		},
+	}
+
+	sessionKey, ok := latestCompletedRaceSessionKey(sessions, now)
+	if !ok {
+		t.Fatal("expected a completed race")
+	}
+	if sessionKey != 1 {
+		t.Fatalf("sessionKey = %d, want 1", sessionKey)
+	}
+}
+
+func TestLatestCompletedRaceSessionKeyIgnoresFutureSessions(t *testing.T) {
+	now := time.Date(2025, 6, 1, 0, 0, 0, 0, time.UTC)
+	sessions := []models.Session{
+		{
+			SessionKey:  1,
+			SessionName: "Race",
+			DateStart:   "2025-12-01T13:00:00+00:00",
+			DateEnd:     "2025-12-01T15:00:00+00:00",
+		},
+		{
+			SessionKey:  2,
+			SessionName: "Race",
+			DateStart:   "2025-05-01T13:00:00+00:00",
+			DateEnd:     "2025-05-01T15:00:00+00:00",
+		},
+	}
+
+	sessionKey, ok := latestCompletedRaceSessionKey(sessions, now)
+	if !ok {
+		t.Fatal("expected a completed race")
+	}
+	if sessionKey != 2 {
+		t.Fatalf("sessionKey = %d, want 2", sessionKey)
+	}
+}