From fbe82f53fb5f474ca9dfff190e736cff4d12f28c Mon Sep 17 00:00:00 2001
From: Alessandro Cifani <alessandro.cifani@adevinta.com>
Date: Fri, 10 Jan 2025 17:04:36 +0100
Subject: [PATCH] refactor: use current year starting from march

---
 pkg/api/ergast.go | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

diff --git a/pkg/api/ergast.go b/pkg/api/ergast.go
index 10fffb2..11d0cd9 100644
--- a/pkg/api/ergast.go
+++ b/pkg/api/ergast.go
@@ -9,11 +9,21 @@ import (
 
 const baseURL = "https://api.jolpi.ca/ergast/f1"
 
-var currentYear = time.Now().Year()
+// getCurrentSeason returns the active F1 season year
+func getCurrentSeason() int {
+	now := time.Now()
+	year := now.Year()
+
+	// F1 season typically starts in March
+	if now.Month() < time.March {
+		return year - 1
+	}
+	return year
+}
 
 func GetLatestRaceResult() (*RaceTable, error) {
 	result := RaceResultResponse{}
-	err := apiCall(fmt.Sprintf("/%d/last/results", currentYear), &result)
+	err := apiCall(fmt.Sprintf("/%d/last/results", getCurrentSeason()), &result)
 	if err != nil {
 		return nil, err
 	}
@@ -22,7 +32,7 @@ func GetLatestRaceResult() (*RaceTable, error) {
 
 func GetCurrentDriverStandings() (*DriverStandingsTable, error) {
 	result := DriverStandingsResponse{}
-	err := apiCall(fmt.Sprintf("/%d/driverstandings", currentYear), &result)
+	err := apiCall(fmt.Sprintf("/%d/driverstandings", getCurrentSeason()), &result)
 	if err != nil {
 		return nil, err
 	}
@@ -31,7 +41,7 @@ func GetCurrentDriverStandings() (*DriverStandingsTable, error) {
 
 func GetCurrentConstructorStandings() (*ConstructorStandingsTable, error) {
 	result := ConstructorStandingsResponse{}
-	err := apiCall(fmt.Sprintf("/%d/constructorstandings", currentYear), &result)
+	err := apiCall(fmt.Sprintf("/%d/constructorstandings", getCurrentSeason()), &result)
 	if err != nil {
 		return nil, err
 	}
@@ -40,7 +50,7 @@ func GetCurrentConstructorStandings() (*ConstructorStandingsTable, error) {
 
 func GetCurrentSeasonSchedule() (*ScheduleTable, error) {
 	result := ScheduleResponse{}
-	err := apiCall("/current", &result)
+	err := apiCall(fmt.Sprintf("/%d", getCurrentSeason()), &result)
 	if err != nil {
 		return nil, err
 	}