upd: historical race data

Author: pxndey

drs

@@ -0,0 +1,12 @@
+package helpers
+
+import (
+	"fmt"
+)
+
+func Help() {
+	fmt.Printf("  -year <year>: Specify the race year (default is current year).")
+	fmt.Printf("  -race <race number>: Specify the race data flag.")
+	fmt.Printf("  --drivers: Display driver's championship standings for the given year.")
+	fmt.Printf("  --constructors: Display constructor's championship standings for the given year.")
+}

drs.exe

@@ -0,0 +1,71 @@
+package helpers
+
+import (
+	"encoding/xml"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"os"
+	"text/tabwriter"
+)
+
+type RaceResult struct {
+	RaceName string `xml:"RaceTable>Race>RaceName"`
+	Result   []struct {
+		Driver struct {
+			GivenName  string `xml:"GivenName"`
+			FamilyName string `xml:"FamilyName"`
+		} `xml:"Driver"`
+		Constructor struct {
+			Name string `xml:"Name"`
+		} `xml:"Constructor"`
+	} `xml:"RaceTable>Race>ResultsList>Result"`
+}
+
+func getRace(year int, race int) [][]string {
+	raceURL := fmt.Sprintf("https://ergast.com/api/f1/%d/%d/results", year, race)
+
+	resp, err := http.Get(raceURL)
+	if err != nil {
+		fmt.Println("error fetching: ", err)
+		return nil
+	}
+
+	defer resp.Body.Close()
+
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		fmt.Println("error reading resp: ", err)
+	}
+
+	var data RaceResult
+
+	err = xml.Unmarshal(body, &data)
+	if err != nil {
+		fmt.Printf("error parsing: ", err)
+	}
+
+	fmt.Printf("%d %s\n", year, data.RaceName)
+	var standings [][]string
+	for _, standing := range data.Result {
+		row := []string{
+			fmt.Sprintf(standing.Driver.GivenName + " " + standing.Driver.FamilyName),
+			standing.Constructor.Name,
+		}
+		standings = append(standings, row)
+	}
+
+	return standings
+}
+
+// Greet prints a greeting message.
+func Race(year int, race int) {
+	data := getRace(year, race)
+	w := tabwriter.NewWriter(os.Stdout, 0, 0, 2, ' ', tabwriter.TabIndent)
+
+	for i, row := range data {
+		fmt.Fprintln(w, fmt.Sprintf("%d.", i+1)+"\t"+row[0]+"\t"+row[1])
+	}
+	w.Flush()
+
+}

internal/helpers/help.go

@@ -0,0 +1,136 @@
+package helpers
+
+import (
+	"encoding/xml"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"os"
+	"text/tabwriter"
+)
+
+type constructorData struct {
+	StandingsTable struct {
+		StandingsList struct {
+			ConstructorStandings []struct {
+				Points      string `xml:"points,attr"`
+				Constructor struct {
+					Name string `xml:"Name"`
+				} `xml:"Constructor"`
+			} `xml:"ConstructorStanding"`
+		} `xml:"StandingsList"`
+	} `xml:"StandingsTable"`
+}
+
+type driverStruct struct {
+	DriverStandings []struct {
+		Points string `xml:"points,attr"`
+		Driver struct {
+			GivenName  string `xml:"GivenName"`
+			FamilyName string `xml:"FamilyName"`
+		} `xml:"Driver"`
+		Team struct {
+			Name string `xml:"Name"`
+		} `xml:"Constructor"`
+	} `xml:"StandingsTable>StandingsList>DriverStanding"`
+}
+
+// Greet prints a greeting message.
+
+func getDriverStandings(year int) [][]string {
+	driverURL := fmt.Sprintf("http://ergast.com/api/f1/%d/driverstandings", year)
+
+	resp, err := http.Get(driverURL)
+	if err != nil {
+		fmt.Println("error fetching: ", err)
+		return nil
+	}
+
+	defer resp.Body.Close()
+
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		fmt.Printf("error reading resp: ", err)
+	}
+
+	var data driverStruct
+
+	err = xml.Unmarshal(body, &data)
+	if err != nil {
+		fmt.Printf("error parsing resp: ", err)
+	}
+
+	var standings [][]string
+
+	for _, standing := range data.DriverStandings {
+		row := []string{
+			fmt.Sprintf(standing.Driver.GivenName + " " + standing.Driver.FamilyName),
+			standing.Team.Name,
+			standing.Points,
+		}
+		standings = append(standings, row)
+	}
+
+	return standings
+}
+
+func getConstructorStandings(year int) [][]string {
+	constructorURL := fmt.Sprintf("http://ergast.com/api/f1/%d/constructorStandings", year)
+
+	resp, err := http.Get(constructorURL)
+	if err != nil {
+		fmt.Println("error fetching: ", err)
+		return nil
+	}
+
+	defer resp.Body.Close()
+
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		fmt.Printf("error reading resp: ", err)
+	}
+
+	var data constructorData
+
+	err = xml.Unmarshal(body, &data)
+	if err != nil {
+		fmt.Printf("error parsing resp: ", err)
+	}
+
+	var standings [][]string
+
+	for _, standing := range data.StandingsTable.StandingsList.ConstructorStandings {
+		row := []string{
+			fmt.Sprintf(standing.Constructor.Name),
+			standing.Points,
+		}
+		standings = append(standings, row)
+	}
+
+	return standings
+}
+
+// TODO: format returned into table: input:year
+func DriverStandings(year int) {
+	data := getDriverStandings(year)
+	w := tabwriter.NewWriter(os.Stdout, 0, 0, 2, ' ', tabwriter.TabIndent)
+
+	for i, row := range data {
+		fmt.Fprintln(w, fmt.Sprintf("%d.", i+1)+"\t"+row[0]+"\t"+row[1]+"\t"+row[2])
+	}
+	w.Flush()
+}
+
+// TODO: constructor standings: input:year
+func ConstructorStandings(year int) {
+	// return fmt.Sprintf("constructr standings")
+	data := getConstructorStandings(year)
+	w := tabwriter.NewWriter(os.Stdout, 0, 0, 2, ' ', tabwriter.TabIndent)
+
+	for i, row := range data {
+		fmt.Fprintln(w, fmt.Sprintf("%d.", i+1)+"\t"+row[0]+"\t"+row[1])
+	}
+	fmt.Println("")
+
+	w.Flush()
+}

internal/helpers/history.go

@@ -1,7 +1,43 @@
 package main
 
-import "fmt"
+import (
+	"flag"
+	"fmt"
+	"time"
+
+	"example.com/drs/internal/helpers"
+)
 
 func main() {
-	fmt.Println("Hello, Go!")
+	yearFlag := flag.Int("year", time.Now().Year(), "race year")
+	raceFlag := flag.Int("race", 0, "race data flag")
+	driversChampionshipFlag := flag.Bool("drivers", false, "drvier's championship for given year")
+	constructorsChampionshipFlag := flag.Bool("constructors", false, "constructor's championship for given year")
+	helpFlag := flag.Bool("help", false, "show help")
+	flag.Parse()
+
+	if flag.NFlag() == 0 {
+		fmt.Printf("Welcome to DRS!\nUsage:")
+		helpers.Help() // Call Help function to show the usage
+		return
+	}
+
+	if *helpFlag {
+		helpers.Help()
+		return
+	}
+
+	if *raceFlag != 0 {
+		helpers.Race(*yearFlag, *raceFlag)
+	}
+	if *driversChampionshipFlag {
+		helpers.DriverStandings(*yearFlag)
+		fmt.Printf("\n")
+	}
+
+	fmt.Printf("\n")
+	if *constructorsChampionshipFlag {
+		helpers.ConstructorStandings(*yearFlag)
+		fmt.Printf("\n")
+	}
 }