add tournaments to big list

quorauk · Sep 28, 2018 · d2b409c · d2b409c
1 parent 10b63ea
commit d2b409c
Show file tree

Hide file tree

Showing 5 changed files with 54 additions and 0 deletions.
diff --git a/api/swagger.yaml b/api/swagger.yaml
@@ -148,6 +148,10 @@ definitions:
         type: string
         format: url
         example:  https://www.acme-corp.com
+      created_at:
+        type: string
+        format: int32
+        example: 2018-09-09T12:28:16.031Z
 # Added by API Auto Mocking Plugin
 host: brackets.superminerbattle.farm
 basePath: /api/1.0.0

diff --git a/gen/models/tournament.go b/gen/models/tournament.go
diff --git a/gen/restapi/embedded_spec.go b/gen/restapi/embedded_spec.go
diff --git a/pkg/challonge/tournaments.go b/pkg/challonge/tournaments.go
@@ -72,3 +72,23 @@ func GetTournamentsForDate(date time.Time) (models []*models.Tournament, err err
 	}
 	return models, nil
 }
+
+func GetTournamentDateMap() (map[string][]*models.Tournament, error) {
+	tournaments, err := GetTournaments()
+	if err != nil {
+		return nil, err
+	}
+	tournamentMap := make(map[string][]*models.Tournament)
+	for _, tournament := range tournaments {
+		start, err := time.Parse("2006-01-02T15:04:05.000-07:00", tournament.CreatedAt)
+		if err == nil {
+			formatted := start.Format("2006-01-02")
+			if val, ok := tournamentMap[formatted]; ok {
+				tournamentMap[formatted] = append(val, tournament)
+			} else {
+				tournamentMap[formatted] = []*models.Tournament{tournament}
+			}
+		}
+	}
+	return tournamentMap, nil
+}
diff --git a/pkg/handlers/events.go b/pkg/handlers/events.go
@@ -32,5 +32,22 @@ func SearchEventsHandler(params operations.SearchEventsParams) middleware.Respon
 		fmt.Println(err)
 		return operations.NewSearchEventsBadRequest()
 	}
+	tournamentMap, err := challonge.GetTournamentDateMap()
+	if err != nil {
+		fmt.Println(err)
+		return operations.NewSearchEventsOK().WithPayload(response)
+	}
+	for _, event := range response {
+		eventStart, err := time.Parse("2006-01-02T15:04:05-0700", event.StartTime)
+		if err != nil {
+			fmt.Println("couldn't parse event date", err)
+		}
+		formatted := eventStart.Format("2006-01-02")
+		if err == nil {
+			if val, ok := tournamentMap[formatted]; ok {
+				event.Tournaments = val
+			}
+		}
+	}
 	return operations.NewSearchEventsOK().WithPayload(response)
 }