Philipp Schrader | cdb5cfc | 2022-02-20 14:57:07 -0800 | [diff] [blame] | 1 | package requests |
| 2 | |
| 3 | import ( |
Philipp Schrader | fae8a7e | 2022-03-13 22:51:54 -0700 | [diff] [blame] | 4 | "encoding/base64" |
Philipp Schrader | d3fac19 | 2022-03-02 20:35:46 -0800 | [diff] [blame] | 5 | "errors" |
Philipp Schrader | cdb5cfc | 2022-02-20 14:57:07 -0800 | [diff] [blame] | 6 | "fmt" |
| 7 | "io" |
Philipp Schrader | fae8a7e | 2022-03-13 22:51:54 -0700 | [diff] [blame] | 8 | "log" |
Philipp Schrader | cdb5cfc | 2022-02-20 14:57:07 -0800 | [diff] [blame] | 9 | "net/http" |
Philipp Schrader | d3fac19 | 2022-03-02 20:35:46 -0800 | [diff] [blame] | 10 | "strconv" |
| 11 | "strings" |
Philipp Schrader | cdb5cfc | 2022-02-20 14:57:07 -0800 | [diff] [blame] | 12 | |
Philipp Schrader | 8747f1b | 2022-02-23 23:56:22 -0800 | [diff] [blame] | 13 | "github.com/frc971/971-Robot-Code/scouting/db" |
Philipp Schrader | d3fac19 | 2022-03-02 20:35:46 -0800 | [diff] [blame] | 14 | "github.com/frc971/971-Robot-Code/scouting/scraping" |
Philipp Schrader | cdb5cfc | 2022-02-20 14:57:07 -0800 | [diff] [blame] | 15 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/error_response" |
Philipp Schrader | d3fac19 | 2022-03-02 20:35:46 -0800 | [diff] [blame] | 16 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/refresh_match_list" |
| 17 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/refresh_match_list_response" |
Filip Kujawa | f882e02 | 2022-12-14 13:14:08 -0800 | [diff] [blame] | 18 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_all_driver_rankings" |
| 19 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_all_driver_rankings_response" |
Philipp Schrader | cbf5c6a | 2022-02-27 23:25:19 -0800 | [diff] [blame] | 20 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_all_matches" |
| 21 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_all_matches_response" |
Filip Kujawa | f882e02 | 2022-12-14 13:14:08 -0800 | [diff] [blame] | 22 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_all_notes" |
| 23 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_all_notes_response" |
Philipp Schrader | acf9623 | 2022-03-01 22:03:30 -0800 | [diff] [blame] | 24 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_data_scouting" |
| 25 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_data_scouting_response" |
Alex Perry | 81f96ba | 2022-03-13 18:26:19 -0700 | [diff] [blame] | 26 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_notes_for_team" |
| 27 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_notes_for_team_response" |
Milo Lin | 1d59f0c | 2022-06-22 20:30:58 -0700 | [diff] [blame] | 28 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_shift_schedule" |
| 29 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_shift_schedule_response" |
Philipp Schrader | cdb5cfc | 2022-02-20 14:57:07 -0800 | [diff] [blame] | 30 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/submit_data_scouting" |
Philipp Schrader | 30005e4 | 2022-03-06 13:53:58 -0800 | [diff] [blame] | 31 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/submit_data_scouting_response" |
Filip Kujawa | 210a03b | 2022-11-24 14:41:11 -0800 | [diff] [blame] | 32 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/submit_driver_ranking" |
| 33 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/submit_driver_ranking_response" |
Alex Perry | 81f96ba | 2022-03-13 18:26:19 -0700 | [diff] [blame] | 34 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/submit_notes" |
| 35 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/submit_notes_response" |
Milo Lin | 1d59f0c | 2022-06-22 20:30:58 -0700 | [diff] [blame] | 36 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/submit_shift_schedule" |
| 37 | "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/submit_shift_schedule_response" |
Philipp Schrader | cdb5cfc | 2022-02-20 14:57:07 -0800 | [diff] [blame] | 38 | "github.com/frc971/971-Robot-Code/scouting/webserver/server" |
| 39 | flatbuffers "github.com/google/flatbuffers/go" |
| 40 | ) |
| 41 | |
Philipp Schrader | cbf5c6a | 2022-02-27 23:25:19 -0800 | [diff] [blame] | 42 | type SubmitDataScouting = submit_data_scouting.SubmitDataScouting |
Philipp Schrader | 30005e4 | 2022-03-06 13:53:58 -0800 | [diff] [blame] | 43 | type SubmitDataScoutingResponseT = submit_data_scouting_response.SubmitDataScoutingResponseT |
Philipp Schrader | cbf5c6a | 2022-02-27 23:25:19 -0800 | [diff] [blame] | 44 | type RequestAllMatches = request_all_matches.RequestAllMatches |
| 45 | type RequestAllMatchesResponseT = request_all_matches_response.RequestAllMatchesResponseT |
Filip Kujawa | f882e02 | 2022-12-14 13:14:08 -0800 | [diff] [blame] | 46 | type RequestAllDriverRankings = request_all_driver_rankings.RequestAllDriverRankings |
| 47 | type RequestAllDriverRankingsResponseT = request_all_driver_rankings_response.RequestAllDriverRankingsResponseT |
| 48 | type RequestAllNotes = request_all_notes.RequestAllNotes |
| 49 | type RequestAllNotesResponseT = request_all_notes_response.RequestAllNotesResponseT |
Philipp Schrader | acf9623 | 2022-03-01 22:03:30 -0800 | [diff] [blame] | 50 | type RequestDataScouting = request_data_scouting.RequestDataScouting |
| 51 | type RequestDataScoutingResponseT = request_data_scouting_response.RequestDataScoutingResponseT |
Philipp Schrader | d3fac19 | 2022-03-02 20:35:46 -0800 | [diff] [blame] | 52 | type RefreshMatchList = refresh_match_list.RefreshMatchList |
| 53 | type RefreshMatchListResponseT = refresh_match_list_response.RefreshMatchListResponseT |
Alex Perry | 81f96ba | 2022-03-13 18:26:19 -0700 | [diff] [blame] | 54 | type SubmitNotes = submit_notes.SubmitNotes |
| 55 | type SubmitNotesResponseT = submit_notes_response.SubmitNotesResponseT |
| 56 | type RequestNotesForTeam = request_notes_for_team.RequestNotesForTeam |
| 57 | type RequestNotesForTeamResponseT = request_notes_for_team_response.RequestNotesForTeamResponseT |
Milo Lin | 1d59f0c | 2022-06-22 20:30:58 -0700 | [diff] [blame] | 58 | type RequestShiftSchedule = request_shift_schedule.RequestShiftSchedule |
| 59 | type RequestShiftScheduleResponseT = request_shift_schedule_response.RequestShiftScheduleResponseT |
| 60 | type SubmitShiftSchedule = submit_shift_schedule.SubmitShiftSchedule |
| 61 | type SubmitShiftScheduleResponseT = submit_shift_schedule_response.SubmitShiftScheduleResponseT |
Filip Kujawa | 210a03b | 2022-11-24 14:41:11 -0800 | [diff] [blame] | 62 | type SubmitDriverRanking = submit_driver_ranking.SubmitDriverRanking |
| 63 | type SubmitDriverRankingResponseT = submit_driver_ranking_response.SubmitDriverRankingResponseT |
Philipp Schrader | cbf5c6a | 2022-02-27 23:25:19 -0800 | [diff] [blame] | 64 | |
Philipp Schrader | 8747f1b | 2022-02-23 23:56:22 -0800 | [diff] [blame] | 65 | // The interface we expect the database abstraction to conform to. |
| 66 | // We use an interface here because it makes unit testing easier. |
| 67 | type Database interface { |
| 68 | AddToMatch(db.Match) error |
Milo Lin | 1d59f0c | 2022-06-22 20:30:58 -0700 | [diff] [blame] | 69 | AddToShift(db.Shift) error |
Philipp Schrader | 8747f1b | 2022-02-23 23:56:22 -0800 | [diff] [blame] | 70 | AddToStats(db.Stats) error |
| 71 | ReturnMatches() ([]db.Match, error) |
Filip Kujawa | f882e02 | 2022-12-14 13:14:08 -0800 | [diff] [blame] | 72 | ReturnAllNotes() ([]db.NotesData, error) |
| 73 | ReturnAllDriverRankings() ([]db.DriverRankingData, error) |
Milo Lin | 1d59f0c | 2022-06-22 20:30:58 -0700 | [diff] [blame] | 74 | ReturnAllShifts() ([]db.Shift, error) |
Philipp Schrader | 8747f1b | 2022-02-23 23:56:22 -0800 | [diff] [blame] | 75 | ReturnStats() ([]db.Stats, error) |
Philipp Schrader | d1c4bef | 2022-02-28 22:51:30 -0800 | [diff] [blame] | 76 | QueryMatches(int32) ([]db.Match, error) |
Milo Lin | 1d59f0c | 2022-06-22 20:30:58 -0700 | [diff] [blame] | 77 | QueryAllShifts(int) ([]db.Shift, error) |
Philipp Schrader | 8747f1b | 2022-02-23 23:56:22 -0800 | [diff] [blame] | 78 | QueryStats(int) ([]db.Stats, error) |
Philipp Schrader | eecb896 | 2022-06-01 21:02:42 -0700 | [diff] [blame] | 79 | QueryNotes(int32) ([]string, error) |
Filip Kujawa | f947cb4 | 2022-11-21 10:00:30 -0800 | [diff] [blame] | 80 | AddNotes(db.NotesData) error |
Filip Kujawa | 210a03b | 2022-11-24 14:41:11 -0800 | [diff] [blame] | 81 | AddDriverRanking(db.DriverRankingData) error |
Philipp Schrader | 8747f1b | 2022-02-23 23:56:22 -0800 | [diff] [blame] | 82 | } |
| 83 | |
Philipp Schrader | d3fac19 | 2022-03-02 20:35:46 -0800 | [diff] [blame] | 84 | type ScrapeMatchList func(int32, string) ([]scraping.Match, error) |
| 85 | |
Philipp Schrader | cdb5cfc | 2022-02-20 14:57:07 -0800 | [diff] [blame] | 86 | // Handles unknown requests. Just returns a 404. |
| 87 | func unknown(w http.ResponseWriter, req *http.Request) { |
| 88 | w.WriteHeader(http.StatusNotFound) |
| 89 | } |
| 90 | |
| 91 | func respondWithError(w http.ResponseWriter, statusCode int, errorMessage string) { |
| 92 | builder := flatbuffers.NewBuilder(1024) |
| 93 | builder.Finish((&error_response.ErrorResponseT{ |
| 94 | ErrorMessage: errorMessage, |
| 95 | }).Pack(builder)) |
| 96 | w.WriteHeader(statusCode) |
| 97 | w.Write(builder.FinishedBytes()) |
| 98 | } |
| 99 | |
| 100 | func respondNotImplemented(w http.ResponseWriter) { |
| 101 | respondWithError(w, http.StatusNotImplemented, "") |
| 102 | } |
| 103 | |
Philipp Schrader | b7e7593 | 2022-03-26 16:18:34 -0700 | [diff] [blame] | 104 | func parseRequest[T interface{}](w http.ResponseWriter, buf []byte, requestName string, parser func([]byte, flatbuffers.UOffsetT) *T) (*T, bool) { |
Philipp Schrader | cdb5cfc | 2022-02-20 14:57:07 -0800 | [diff] [blame] | 105 | success := true |
| 106 | defer func() { |
| 107 | if r := recover(); r != nil { |
Philipp Schrader | b7e7593 | 2022-03-26 16:18:34 -0700 | [diff] [blame] | 108 | respondWithError(w, http.StatusBadRequest, fmt.Sprintf("Failed to parse %s: %v", requestName, r)) |
Philipp Schrader | cdb5cfc | 2022-02-20 14:57:07 -0800 | [diff] [blame] | 109 | success = false |
| 110 | } |
| 111 | }() |
Philipp Schrader | b7e7593 | 2022-03-26 16:18:34 -0700 | [diff] [blame] | 112 | result := parser(buf, 0) |
Philipp Schrader | cdb5cfc | 2022-02-20 14:57:07 -0800 | [diff] [blame] | 113 | return result, success |
| 114 | } |
| 115 | |
Philipp Schrader | fae8a7e | 2022-03-13 22:51:54 -0700 | [diff] [blame] | 116 | // Parses the authorization information that the browser inserts into the |
| 117 | // headers. The authorization follows this format: |
| 118 | // |
| 119 | // req.Headers["Authorization"] = []string{"Basic <base64 encoded username:password>"} |
| 120 | func parseUsername(req *http.Request) string { |
| 121 | auth, ok := req.Header["Authorization"] |
| 122 | if !ok { |
| 123 | return "unknown" |
| 124 | } |
| 125 | |
| 126 | parts := strings.Split(auth[0], " ") |
| 127 | if !(len(parts) == 2 && parts[0] == "Basic") { |
| 128 | return "unknown" |
| 129 | } |
| 130 | |
| 131 | info, err := base64.StdEncoding.DecodeString(parts[1]) |
| 132 | if err != nil { |
| 133 | log.Println("ERROR: Failed to parse Basic authentication.") |
| 134 | return "unknown" |
| 135 | } |
| 136 | |
| 137 | loginParts := strings.Split(string(info), ":") |
| 138 | if len(loginParts) != 2 { |
| 139 | return "unknown" |
| 140 | } |
| 141 | return loginParts[0] |
| 142 | } |
| 143 | |
Philipp Schrader | cdb5cfc | 2022-02-20 14:57:07 -0800 | [diff] [blame] | 144 | // Handles a SubmitDataScouting request. |
Philipp Schrader | 8747f1b | 2022-02-23 23:56:22 -0800 | [diff] [blame] | 145 | type submitDataScoutingHandler struct { |
| 146 | db Database |
| 147 | } |
| 148 | |
| 149 | func (handler submitDataScoutingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) { |
Philipp Schrader | fae8a7e | 2022-03-13 22:51:54 -0700 | [diff] [blame] | 150 | // Get the username of the person submitting the data. |
| 151 | username := parseUsername(req) |
Philipp Schrader | fae8a7e | 2022-03-13 22:51:54 -0700 | [diff] [blame] | 152 | |
Philipp Schrader | cdb5cfc | 2022-02-20 14:57:07 -0800 | [diff] [blame] | 153 | requestBytes, err := io.ReadAll(req.Body) |
| 154 | if err != nil { |
| 155 | respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err)) |
| 156 | return |
| 157 | } |
| 158 | |
Philipp Schrader | b7e7593 | 2022-03-26 16:18:34 -0700 | [diff] [blame] | 159 | request, success := parseRequest[SubmitDataScouting](w, requestBytes, "SubmitDataScouting", submit_data_scouting.GetRootAsSubmitDataScouting) |
Philipp Schrader | cdb5cfc | 2022-02-20 14:57:07 -0800 | [diff] [blame] | 160 | if !success { |
| 161 | return |
| 162 | } |
| 163 | |
Philipp Schrader | d7b6eba | 2022-03-18 22:21:25 -0700 | [diff] [blame] | 164 | log.Println("Got data scouting data for match", request.Match(), "team", request.Team(), "from", username) |
| 165 | |
Philipp Schrader | 30005e4 | 2022-03-06 13:53:58 -0800 | [diff] [blame] | 166 | stats := db.Stats{ |
Philipp Schrader | fee07e1 | 2022-03-17 22:19:47 -0700 | [diff] [blame] | 167 | TeamNumber: request.Team(), |
| 168 | MatchNumber: request.Match(), |
Philipp Schrader | 30b4a68 | 2022-04-16 14:36:17 -0700 | [diff] [blame] | 169 | SetNumber: request.SetNumber(), |
Philipp Schrader | 4535b7e | 2022-04-08 20:27:00 -0700 | [diff] [blame] | 170 | CompLevel: string(request.CompLevel()), |
Philipp Schrader | fee07e1 | 2022-03-17 22:19:47 -0700 | [diff] [blame] | 171 | StartingQuadrant: request.StartingQuadrant(), |
| 172 | AutoBallPickedUp: [5]bool{ |
| 173 | request.AutoBall1(), request.AutoBall2(), request.AutoBall3(), |
| 174 | request.AutoBall4(), request.AutoBall5(), |
| 175 | }, |
Philipp Schrader | fa45d74 | 2022-03-18 19:29:05 -0700 | [diff] [blame] | 176 | ShotsMissedAuto: request.MissedShotsAuto(), |
| 177 | UpperGoalAuto: request.UpperGoalAuto(), |
| 178 | LowerGoalAuto: request.LowerGoalAuto(), |
| 179 | ShotsMissed: request.MissedShotsTele(), |
| 180 | UpperGoalShots: request.UpperGoalTele(), |
| 181 | LowerGoalShots: request.LowerGoalTele(), |
| 182 | PlayedDefense: request.DefenseRating(), |
| 183 | DefenseReceivedScore: request.DefenseReceivedRating(), |
| 184 | Climbing: int32(request.ClimbLevel()), |
| 185 | CollectedBy: username, |
| 186 | Comment: string(request.Comment()), |
Philipp Schrader | 30005e4 | 2022-03-06 13:53:58 -0800 | [diff] [blame] | 187 | } |
Philipp Schrader | cdb5cfc | 2022-02-20 14:57:07 -0800 | [diff] [blame] | 188 | |
Philipp Schrader | fee07e1 | 2022-03-17 22:19:47 -0700 | [diff] [blame] | 189 | // Do some error checking. |
| 190 | if stats.StartingQuadrant < 1 || stats.StartingQuadrant > 4 { |
| 191 | respondWithError(w, http.StatusBadRequest, fmt.Sprint( |
| 192 | "Invalid starting_quadrant field value of ", stats.StartingQuadrant)) |
| 193 | return |
| 194 | } |
| 195 | |
Philipp Schrader | 30005e4 | 2022-03-06 13:53:58 -0800 | [diff] [blame] | 196 | err = handler.db.AddToStats(stats) |
| 197 | if err != nil { |
| 198 | respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to submit datascouting data: ", err)) |
Philipp Schrader | fee07e1 | 2022-03-17 22:19:47 -0700 | [diff] [blame] | 199 | return |
Philipp Schrader | 30005e4 | 2022-03-06 13:53:58 -0800 | [diff] [blame] | 200 | } |
| 201 | |
| 202 | builder := flatbuffers.NewBuilder(50 * 1024) |
| 203 | builder.Finish((&SubmitDataScoutingResponseT{}).Pack(builder)) |
| 204 | w.Write(builder.FinishedBytes()) |
Philipp Schrader | cdb5cfc | 2022-02-20 14:57:07 -0800 | [diff] [blame] | 205 | } |
| 206 | |
Philipp Schrader | cbf5c6a | 2022-02-27 23:25:19 -0800 | [diff] [blame] | 207 | // Handles a RequestAllMaches request. |
| 208 | type requestAllMatchesHandler struct { |
| 209 | db Database |
| 210 | } |
| 211 | |
| 212 | func (handler requestAllMatchesHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) { |
| 213 | requestBytes, err := io.ReadAll(req.Body) |
| 214 | if err != nil { |
| 215 | respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err)) |
| 216 | return |
| 217 | } |
| 218 | |
Philipp Schrader | b7e7593 | 2022-03-26 16:18:34 -0700 | [diff] [blame] | 219 | _, success := parseRequest(w, requestBytes, "RequestAllMatches", request_all_matches.GetRootAsRequestAllMatches) |
Philipp Schrader | cbf5c6a | 2022-02-27 23:25:19 -0800 | [diff] [blame] | 220 | if !success { |
| 221 | return |
| 222 | } |
| 223 | |
| 224 | matches, err := handler.db.ReturnMatches() |
| 225 | if err != nil { |
Philipp Schrader | fae8a7e | 2022-03-13 22:51:54 -0700 | [diff] [blame] | 226 | respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to query database: ", err)) |
Philipp Schrader | 2e7eb000 | 2022-03-02 22:52:39 -0800 | [diff] [blame] | 227 | return |
Philipp Schrader | cbf5c6a | 2022-02-27 23:25:19 -0800 | [diff] [blame] | 228 | } |
| 229 | |
| 230 | var response RequestAllMatchesResponseT |
| 231 | for _, match := range matches { |
| 232 | response.MatchList = append(response.MatchList, &request_all_matches_response.MatchT{ |
| 233 | MatchNumber: match.MatchNumber, |
Philipp Schrader | 30b4a68 | 2022-04-16 14:36:17 -0700 | [diff] [blame] | 234 | SetNumber: match.SetNumber, |
Philipp Schrader | cbf5c6a | 2022-02-27 23:25:19 -0800 | [diff] [blame] | 235 | CompLevel: match.CompLevel, |
| 236 | R1: match.R1, |
| 237 | R2: match.R2, |
| 238 | R3: match.R3, |
| 239 | B1: match.B1, |
| 240 | B2: match.B2, |
| 241 | B3: match.B3, |
| 242 | }) |
| 243 | } |
| 244 | |
| 245 | builder := flatbuffers.NewBuilder(50 * 1024) |
| 246 | builder.Finish((&response).Pack(builder)) |
| 247 | w.Write(builder.FinishedBytes()) |
| 248 | } |
| 249 | |
Philipp Schrader | acf9623 | 2022-03-01 22:03:30 -0800 | [diff] [blame] | 250 | // Handles a RequestDataScouting request. |
| 251 | type requestDataScoutingHandler struct { |
| 252 | db Database |
| 253 | } |
| 254 | |
| 255 | func (handler requestDataScoutingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) { |
| 256 | requestBytes, err := io.ReadAll(req.Body) |
| 257 | if err != nil { |
| 258 | respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err)) |
| 259 | return |
| 260 | } |
| 261 | |
Philipp Schrader | b7e7593 | 2022-03-26 16:18:34 -0700 | [diff] [blame] | 262 | _, success := parseRequest(w, requestBytes, "RequestDataScouting", request_data_scouting.GetRootAsRequestDataScouting) |
Philipp Schrader | acf9623 | 2022-03-01 22:03:30 -0800 | [diff] [blame] | 263 | if !success { |
| 264 | return |
| 265 | } |
| 266 | |
| 267 | stats, err := handler.db.ReturnStats() |
| 268 | if err != nil { |
| 269 | respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Faled to query database: ", err)) |
Philipp Schrader | 2e7eb000 | 2022-03-02 22:52:39 -0800 | [diff] [blame] | 270 | return |
Philipp Schrader | acf9623 | 2022-03-01 22:03:30 -0800 | [diff] [blame] | 271 | } |
| 272 | |
| 273 | var response RequestDataScoutingResponseT |
| 274 | for _, stat := range stats { |
| 275 | response.StatsList = append(response.StatsList, &request_data_scouting_response.StatsT{ |
Philipp Schrader | fa45d74 | 2022-03-18 19:29:05 -0700 | [diff] [blame] | 276 | Team: stat.TeamNumber, |
| 277 | Match: stat.MatchNumber, |
Philipp Schrader | 30b4a68 | 2022-04-16 14:36:17 -0700 | [diff] [blame] | 278 | SetNumber: stat.SetNumber, |
Philipp Schrader | 4535b7e | 2022-04-08 20:27:00 -0700 | [diff] [blame] | 279 | CompLevel: stat.CompLevel, |
Philipp Schrader | fa45d74 | 2022-03-18 19:29:05 -0700 | [diff] [blame] | 280 | StartingQuadrant: stat.StartingQuadrant, |
| 281 | AutoBall1: stat.AutoBallPickedUp[0], |
| 282 | AutoBall2: stat.AutoBallPickedUp[1], |
| 283 | AutoBall3: stat.AutoBallPickedUp[2], |
| 284 | AutoBall4: stat.AutoBallPickedUp[3], |
| 285 | AutoBall5: stat.AutoBallPickedUp[4], |
| 286 | MissedShotsAuto: stat.ShotsMissedAuto, |
| 287 | UpperGoalAuto: stat.UpperGoalAuto, |
| 288 | LowerGoalAuto: stat.LowerGoalAuto, |
| 289 | MissedShotsTele: stat.ShotsMissed, |
| 290 | UpperGoalTele: stat.UpperGoalShots, |
| 291 | LowerGoalTele: stat.LowerGoalShots, |
| 292 | DefenseRating: stat.PlayedDefense, |
| 293 | DefenseReceivedRating: stat.DefenseReceivedScore, |
| 294 | ClimbLevel: request_data_scouting_response.ClimbLevel(stat.Climbing), |
| 295 | CollectedBy: stat.CollectedBy, |
| 296 | Comment: stat.Comment, |
Philipp Schrader | acf9623 | 2022-03-01 22:03:30 -0800 | [diff] [blame] | 297 | }) |
| 298 | } |
| 299 | |
| 300 | builder := flatbuffers.NewBuilder(50 * 1024) |
| 301 | builder.Finish((&response).Pack(builder)) |
| 302 | w.Write(builder.FinishedBytes()) |
| 303 | } |
| 304 | |
Philipp Schrader | d3fac19 | 2022-03-02 20:35:46 -0800 | [diff] [blame] | 305 | func parseTeamKey(teamKey string) (int, error) { |
| 306 | // TBA prefixes teams with "frc". Not sure why. Get rid of that. |
| 307 | teamKey = strings.TrimPrefix(teamKey, "frc") |
Philipp Schrader | 84fe778 | 2022-11-12 08:10:36 -0800 | [diff] [blame] | 308 | magnitude := 0 |
| 309 | if strings.HasSuffix(teamKey, "A") { |
Filip Kujawa | 08057fa | 2022-11-12 09:54:49 -0800 | [diff] [blame] | 310 | magnitude = 0 |
Philipp Schrader | 84fe778 | 2022-11-12 08:10:36 -0800 | [diff] [blame] | 311 | teamKey = strings.TrimSuffix(teamKey, "A") |
| 312 | } else if strings.HasSuffix(teamKey, "B") { |
Filip Kujawa | 08057fa | 2022-11-12 09:54:49 -0800 | [diff] [blame] | 313 | magnitude = 9 |
Philipp Schrader | 84fe778 | 2022-11-12 08:10:36 -0800 | [diff] [blame] | 314 | teamKey = strings.TrimSuffix(teamKey, "B") |
| 315 | } else if strings.HasSuffix(teamKey, "C") { |
Filip Kujawa | 08057fa | 2022-11-12 09:54:49 -0800 | [diff] [blame] | 316 | magnitude = 8 |
Philipp Schrader | 84fe778 | 2022-11-12 08:10:36 -0800 | [diff] [blame] | 317 | teamKey = strings.TrimSuffix(teamKey, "C") |
| 318 | } else if strings.HasSuffix(teamKey, "D") { |
Filip Kujawa | 08057fa | 2022-11-12 09:54:49 -0800 | [diff] [blame] | 319 | magnitude = 7 |
Philipp Schrader | 84fe778 | 2022-11-12 08:10:36 -0800 | [diff] [blame] | 320 | teamKey = strings.TrimSuffix(teamKey, "D") |
| 321 | } else if strings.HasSuffix(teamKey, "E") { |
Filip Kujawa | 08057fa | 2022-11-12 09:54:49 -0800 | [diff] [blame] | 322 | magnitude = 6 |
Philipp Schrader | 84fe778 | 2022-11-12 08:10:36 -0800 | [diff] [blame] | 323 | teamKey = strings.TrimSuffix(teamKey, "E") |
| 324 | } else if strings.HasSuffix(teamKey, "F") { |
Filip Kujawa | 08057fa | 2022-11-12 09:54:49 -0800 | [diff] [blame] | 325 | magnitude = 5 |
Philipp Schrader | 84fe778 | 2022-11-12 08:10:36 -0800 | [diff] [blame] | 326 | teamKey = strings.TrimSuffix(teamKey, "F") |
| 327 | } |
Filip Kujawa | 4de6feb | 2022-11-12 08:41:49 -0800 | [diff] [blame] | 328 | |
| 329 | if magnitude != 0 { |
| 330 | teamKey = strconv.Itoa(magnitude) + teamKey |
Philipp Schrader | 84fe778 | 2022-11-12 08:10:36 -0800 | [diff] [blame] | 331 | } |
Filip Kujawa | 4de6feb | 2022-11-12 08:41:49 -0800 | [diff] [blame] | 332 | |
| 333 | result, err := strconv.Atoi(teamKey) |
Philipp Schrader | 84fe778 | 2022-11-12 08:10:36 -0800 | [diff] [blame] | 334 | return result, err |
Philipp Schrader | d3fac19 | 2022-03-02 20:35:46 -0800 | [diff] [blame] | 335 | } |
| 336 | |
| 337 | // Parses the alliance data from the specified match and returns the three red |
| 338 | // teams and the three blue teams. |
| 339 | func parseTeamKeys(match *scraping.Match) ([3]int32, [3]int32, error) { |
| 340 | redKeys := match.Alliances.Red.TeamKeys |
| 341 | blueKeys := match.Alliances.Blue.TeamKeys |
| 342 | |
| 343 | if len(redKeys) != 3 || len(blueKeys) != 3 { |
| 344 | return [3]int32{}, [3]int32{}, errors.New(fmt.Sprintf( |
| 345 | "Found %d red teams and %d blue teams.", len(redKeys), len(blueKeys))) |
| 346 | } |
| 347 | |
| 348 | var red [3]int32 |
| 349 | for i, key := range redKeys { |
| 350 | team, err := parseTeamKey(key) |
| 351 | if err != nil { |
| 352 | return [3]int32{}, [3]int32{}, errors.New(fmt.Sprintf( |
| 353 | "Failed to parse red %d team '%s' as integer: %v", i+1, key, err)) |
| 354 | } |
| 355 | red[i] = int32(team) |
| 356 | } |
| 357 | var blue [3]int32 |
| 358 | for i, key := range blueKeys { |
| 359 | team, err := parseTeamKey(key) |
| 360 | if err != nil { |
| 361 | return [3]int32{}, [3]int32{}, errors.New(fmt.Sprintf( |
| 362 | "Failed to parse blue %d team '%s' as integer: %v", i+1, key, err)) |
| 363 | } |
| 364 | blue[i] = int32(team) |
| 365 | } |
| 366 | return red, blue, nil |
| 367 | } |
| 368 | |
| 369 | type refreshMatchListHandler struct { |
| 370 | db Database |
| 371 | scrape ScrapeMatchList |
| 372 | } |
| 373 | |
| 374 | func (handler refreshMatchListHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) { |
| 375 | requestBytes, err := io.ReadAll(req.Body) |
| 376 | if err != nil { |
| 377 | respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err)) |
| 378 | return |
| 379 | } |
| 380 | |
Philipp Schrader | b7e7593 | 2022-03-26 16:18:34 -0700 | [diff] [blame] | 381 | request, success := parseRequest(w, requestBytes, "RefreshMatchList", refresh_match_list.GetRootAsRefreshMatchList) |
Philipp Schrader | d3fac19 | 2022-03-02 20:35:46 -0800 | [diff] [blame] | 382 | if !success { |
| 383 | return |
| 384 | } |
| 385 | |
| 386 | matches, err := handler.scrape(request.Year(), string(request.EventCode())) |
| 387 | if err != nil { |
| 388 | respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Faled to scrape match list: ", err)) |
| 389 | return |
| 390 | } |
| 391 | |
| 392 | for _, match := range matches { |
| 393 | // Make sure the data is valid. |
| 394 | red, blue, err := parseTeamKeys(&match) |
| 395 | if err != nil { |
| 396 | respondWithError(w, http.StatusInternalServerError, fmt.Sprintf( |
| 397 | "TheBlueAlliance data for match %d is malformed: %v", match.MatchNumber, err)) |
| 398 | return |
| 399 | } |
| 400 | // Add the match to the database. |
Philipp Schrader | 7365d32 | 2022-03-06 16:40:08 -0800 | [diff] [blame] | 401 | err = handler.db.AddToMatch(db.Match{ |
Philipp Schrader | d3fac19 | 2022-03-02 20:35:46 -0800 | [diff] [blame] | 402 | MatchNumber: int32(match.MatchNumber), |
Philipp Schrader | 30b4a68 | 2022-04-16 14:36:17 -0700 | [diff] [blame] | 403 | SetNumber: int32(match.SetNumber), |
Philipp Schrader | 45befdd | 2022-04-08 19:12:44 -0700 | [diff] [blame] | 404 | CompLevel: match.CompLevel, |
| 405 | R1: red[0], |
| 406 | R2: red[1], |
| 407 | R3: red[2], |
| 408 | B1: blue[0], |
| 409 | B2: blue[1], |
| 410 | B3: blue[2], |
Philipp Schrader | d3fac19 | 2022-03-02 20:35:46 -0800 | [diff] [blame] | 411 | }) |
Philipp Schrader | 7365d32 | 2022-03-06 16:40:08 -0800 | [diff] [blame] | 412 | if err != nil { |
| 413 | respondWithError(w, http.StatusInternalServerError, fmt.Sprintf( |
| 414 | "Failed to add match %d to the database: %v", match.MatchNumber, err)) |
| 415 | return |
| 416 | } |
Philipp Schrader | d3fac19 | 2022-03-02 20:35:46 -0800 | [diff] [blame] | 417 | } |
| 418 | |
| 419 | var response RefreshMatchListResponseT |
| 420 | builder := flatbuffers.NewBuilder(1024) |
| 421 | builder.Finish((&response).Pack(builder)) |
| 422 | w.Write(builder.FinishedBytes()) |
| 423 | } |
| 424 | |
Alex Perry | 81f96ba | 2022-03-13 18:26:19 -0700 | [diff] [blame] | 425 | type submitNoteScoutingHandler struct { |
| 426 | db Database |
| 427 | } |
| 428 | |
| 429 | func (handler submitNoteScoutingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) { |
| 430 | requestBytes, err := io.ReadAll(req.Body) |
| 431 | if err != nil { |
| 432 | respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err)) |
| 433 | return |
| 434 | } |
| 435 | |
Philipp Schrader | b7e7593 | 2022-03-26 16:18:34 -0700 | [diff] [blame] | 436 | request, success := parseRequest(w, requestBytes, "SubmitNotes", submit_notes.GetRootAsSubmitNotes) |
Alex Perry | 81f96ba | 2022-03-13 18:26:19 -0700 | [diff] [blame] | 437 | if !success { |
| 438 | return |
| 439 | } |
| 440 | |
Filip Kujawa | f947cb4 | 2022-11-21 10:00:30 -0800 | [diff] [blame] | 441 | err = handler.db.AddNotes(db.NotesData{ |
| 442 | TeamNumber: request.Team(), |
| 443 | Notes: string(request.Notes()), |
| 444 | GoodDriving: bool(request.GoodDriving()), |
| 445 | BadDriving: bool(request.BadDriving()), |
| 446 | SketchyClimb: bool(request.SketchyClimb()), |
| 447 | SolidClimb: bool(request.SolidClimb()), |
| 448 | GoodDefense: bool(request.GoodDefense()), |
| 449 | BadDefense: bool(request.BadDefense()), |
| 450 | }) |
Alex Perry | 81f96ba | 2022-03-13 18:26:19 -0700 | [diff] [blame] | 451 | if err != nil { |
| 452 | respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to insert notes: %v", err)) |
| 453 | return |
| 454 | } |
| 455 | |
| 456 | var response SubmitNotesResponseT |
| 457 | builder := flatbuffers.NewBuilder(10) |
| 458 | builder.Finish((&response).Pack(builder)) |
| 459 | w.Write(builder.FinishedBytes()) |
| 460 | } |
| 461 | |
Alex Perry | 81f96ba | 2022-03-13 18:26:19 -0700 | [diff] [blame] | 462 | type requestNotesForTeamHandler struct { |
| 463 | db Database |
| 464 | } |
| 465 | |
| 466 | func (handler requestNotesForTeamHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) { |
| 467 | requestBytes, err := io.ReadAll(req.Body) |
| 468 | if err != nil { |
| 469 | respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err)) |
| 470 | return |
| 471 | } |
| 472 | |
Philipp Schrader | b7e7593 | 2022-03-26 16:18:34 -0700 | [diff] [blame] | 473 | request, success := parseRequest(w, requestBytes, "RequestNotesForTeam", request_notes_for_team.GetRootAsRequestNotesForTeam) |
Alex Perry | 81f96ba | 2022-03-13 18:26:19 -0700 | [diff] [blame] | 474 | if !success { |
| 475 | return |
| 476 | } |
| 477 | |
Philipp Schrader | eecb896 | 2022-06-01 21:02:42 -0700 | [diff] [blame] | 478 | notes, err := handler.db.QueryNotes(request.Team()) |
Alex Perry | 81f96ba | 2022-03-13 18:26:19 -0700 | [diff] [blame] | 479 | if err != nil { |
| 480 | respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to query notes: %v", err)) |
| 481 | return |
| 482 | } |
| 483 | |
| 484 | var response RequestNotesForTeamResponseT |
Philipp Schrader | eecb896 | 2022-06-01 21:02:42 -0700 | [diff] [blame] | 485 | for _, data := range notes { |
Alex Perry | 81f96ba | 2022-03-13 18:26:19 -0700 | [diff] [blame] | 486 | response.Notes = append(response.Notes, &request_notes_for_team_response.NoteT{data}) |
| 487 | } |
| 488 | |
| 489 | builder := flatbuffers.NewBuilder(1024) |
| 490 | builder.Finish((&response).Pack(builder)) |
| 491 | w.Write(builder.FinishedBytes()) |
| 492 | } |
| 493 | |
Milo Lin | 1d59f0c | 2022-06-22 20:30:58 -0700 | [diff] [blame] | 494 | type requestShiftScheduleHandler struct { |
| 495 | db Database |
| 496 | } |
| 497 | |
| 498 | func (handler requestShiftScheduleHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) { |
| 499 | requestBytes, err := io.ReadAll(req.Body) |
| 500 | if err != nil { |
| 501 | respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err)) |
| 502 | return |
| 503 | } |
| 504 | |
| 505 | _, success := parseRequest(w, requestBytes, "RequestShiftSchedule", request_shift_schedule.GetRootAsRequestShiftSchedule) |
| 506 | if !success { |
| 507 | return |
| 508 | } |
| 509 | |
| 510 | shiftData, err := handler.db.ReturnAllShifts() |
| 511 | if err != nil { |
| 512 | respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to query shift schedule: %v", err)) |
| 513 | return |
| 514 | } |
| 515 | |
| 516 | var response RequestShiftScheduleResponseT |
| 517 | for _, shifts := range shiftData { |
| 518 | response.ShiftSchedule = append(response.ShiftSchedule, &request_shift_schedule_response.MatchAssignmentT{ |
| 519 | MatchNumber: shifts.MatchNumber, |
| 520 | R1scouter: shifts.R1scouter, |
| 521 | R2scouter: shifts.R2scouter, |
| 522 | R3scouter: shifts.R3scouter, |
| 523 | B1scouter: shifts.B1scouter, |
| 524 | B2scouter: shifts.B2scouter, |
| 525 | B3scouter: shifts.B3scouter, |
| 526 | }) |
| 527 | } |
| 528 | |
| 529 | builder := flatbuffers.NewBuilder(1024) |
| 530 | builder.Finish((&response).Pack(builder)) |
| 531 | w.Write(builder.FinishedBytes()) |
| 532 | } |
| 533 | |
| 534 | type submitShiftScheduleHandler struct { |
| 535 | db Database |
| 536 | } |
| 537 | |
| 538 | func (handler submitShiftScheduleHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) { |
| 539 | // Get the username of the person submitting the data. |
| 540 | username := parseUsername(req) |
| 541 | |
| 542 | requestBytes, err := io.ReadAll(req.Body) |
| 543 | if err != nil { |
| 544 | respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err)) |
| 545 | return |
| 546 | } |
| 547 | |
| 548 | request, success := parseRequest[SubmitShiftSchedule](w, requestBytes, "SubmitShiftSchedule", submit_shift_schedule.GetRootAsSubmitShiftSchedule) |
| 549 | if !success { |
| 550 | return |
| 551 | } |
| 552 | |
| 553 | log.Println("Got shift schedule from", username) |
| 554 | shift_schedule_length := request.ShiftScheduleLength() |
| 555 | for i := 0; i < shift_schedule_length; i++ { |
| 556 | var match_assignment submit_shift_schedule.MatchAssignment |
| 557 | request.ShiftSchedule(&match_assignment, i) |
| 558 | current_shift := db.Shift{ |
| 559 | MatchNumber: match_assignment.MatchNumber(), |
| 560 | R1scouter: string(match_assignment.R1scouter()), |
| 561 | R2scouter: string(match_assignment.R2scouter()), |
| 562 | R3scouter: string(match_assignment.R3scouter()), |
| 563 | B1scouter: string(match_assignment.B1scouter()), |
| 564 | B2scouter: string(match_assignment.B2scouter()), |
| 565 | B3scouter: string(match_assignment.B3scouter()), |
| 566 | } |
| 567 | err = handler.db.AddToShift(current_shift) |
| 568 | if err != nil { |
| 569 | respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to submit shift schedule: ", err)) |
| 570 | return |
| 571 | } |
| 572 | } |
| 573 | |
| 574 | builder := flatbuffers.NewBuilder(50 * 1024) |
| 575 | builder.Finish((&SubmitShiftScheduleResponseT{}).Pack(builder)) |
| 576 | w.Write(builder.FinishedBytes()) |
| 577 | } |
| 578 | |
Filip Kujawa | 210a03b | 2022-11-24 14:41:11 -0800 | [diff] [blame] | 579 | type SubmitDriverRankingHandler struct { |
| 580 | db Database |
| 581 | } |
| 582 | |
| 583 | func (handler SubmitDriverRankingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) { |
| 584 | requestBytes, err := io.ReadAll(req.Body) |
| 585 | if err != nil { |
| 586 | respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err)) |
| 587 | return |
| 588 | } |
| 589 | |
| 590 | request, success := parseRequest(w, requestBytes, "SubmitDriverRanking", submit_driver_ranking.GetRootAsSubmitDriverRanking) |
| 591 | if !success { |
| 592 | return |
| 593 | } |
| 594 | |
| 595 | err = handler.db.AddDriverRanking(db.DriverRankingData{ |
| 596 | MatchNumber: request.MatchNumber(), |
| 597 | Rank1: request.Rank1(), |
| 598 | Rank2: request.Rank2(), |
| 599 | Rank3: request.Rank3(), |
| 600 | }) |
| 601 | |
| 602 | if err != nil { |
| 603 | respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to insert driver ranking: %v", err)) |
| 604 | return |
| 605 | } |
| 606 | |
| 607 | var response SubmitDriverRankingResponseT |
| 608 | builder := flatbuffers.NewBuilder(10) |
| 609 | builder.Finish((&response).Pack(builder)) |
| 610 | w.Write(builder.FinishedBytes()) |
| 611 | } |
| 612 | |
Filip Kujawa | f882e02 | 2022-12-14 13:14:08 -0800 | [diff] [blame] | 613 | type requestAllNotesHandler struct { |
| 614 | db Database |
| 615 | } |
| 616 | |
| 617 | func (handler requestAllNotesHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) { |
| 618 | requestBytes, err := io.ReadAll(req.Body) |
| 619 | if err != nil { |
| 620 | respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err)) |
| 621 | return |
| 622 | } |
| 623 | |
| 624 | _, success := parseRequest(w, requestBytes, "RequestAllNotes", request_all_notes.GetRootAsRequestAllNotes) |
| 625 | if !success { |
| 626 | return |
| 627 | } |
| 628 | |
| 629 | notes, err := handler.db.ReturnAllNotes() |
| 630 | if err != nil { |
| 631 | respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to query database: ", err)) |
| 632 | return |
| 633 | } |
| 634 | |
| 635 | var response RequestAllNotesResponseT |
| 636 | for _, note := range notes { |
| 637 | response.NoteList = append(response.NoteList, &request_all_notes_response.NoteT{ |
| 638 | Team: note.TeamNumber, |
| 639 | Notes: note.Notes, |
| 640 | GoodDriving: note.GoodDriving, |
| 641 | BadDriving: note.BadDriving, |
| 642 | SketchyClimb: note.SketchyClimb, |
| 643 | SolidClimb: note.SolidClimb, |
| 644 | GoodDefense: note.GoodDefense, |
| 645 | BadDefense: note.BadDefense, |
| 646 | }) |
| 647 | } |
| 648 | |
| 649 | builder := flatbuffers.NewBuilder(50 * 1024) |
| 650 | builder.Finish((&response).Pack(builder)) |
| 651 | w.Write(builder.FinishedBytes()) |
| 652 | } |
| 653 | |
| 654 | type requestAllDriverRankingsHandler struct { |
| 655 | db Database |
| 656 | } |
| 657 | |
| 658 | func (handler requestAllDriverRankingsHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) { |
| 659 | requestBytes, err := io.ReadAll(req.Body) |
| 660 | if err != nil { |
| 661 | respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err)) |
| 662 | return |
| 663 | } |
| 664 | |
| 665 | _, success := parseRequest(w, requestBytes, "RequestAllDriverRankings", request_all_driver_rankings.GetRootAsRequestAllDriverRankings) |
| 666 | if !success { |
| 667 | return |
| 668 | } |
| 669 | |
| 670 | rankings, err := handler.db.ReturnAllDriverRankings() |
| 671 | if err != nil { |
| 672 | respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to query database: ", err)) |
| 673 | return |
| 674 | } |
| 675 | |
| 676 | var response RequestAllDriverRankingsResponseT |
| 677 | for _, ranking := range rankings { |
| 678 | response.DriverRankingList = append(response.DriverRankingList, &request_all_driver_rankings_response.RankingT{ |
| 679 | MatchNumber: ranking.MatchNumber, |
| 680 | Rank1: ranking.Rank1, |
| 681 | Rank2: ranking.Rank2, |
| 682 | Rank3: ranking.Rank3, |
| 683 | }) |
| 684 | } |
| 685 | |
| 686 | builder := flatbuffers.NewBuilder(50 * 1024) |
| 687 | builder.Finish((&response).Pack(builder)) |
| 688 | w.Write(builder.FinishedBytes()) |
| 689 | } |
| 690 | |
Philipp Schrader | d3fac19 | 2022-03-02 20:35:46 -0800 | [diff] [blame] | 691 | func HandleRequests(db Database, scrape ScrapeMatchList, scoutingServer server.ScoutingServer) { |
Philipp Schrader | cdb5cfc | 2022-02-20 14:57:07 -0800 | [diff] [blame] | 692 | scoutingServer.HandleFunc("/requests", unknown) |
Philipp Schrader | 8747f1b | 2022-02-23 23:56:22 -0800 | [diff] [blame] | 693 | scoutingServer.Handle("/requests/submit/data_scouting", submitDataScoutingHandler{db}) |
Philipp Schrader | cbf5c6a | 2022-02-27 23:25:19 -0800 | [diff] [blame] | 694 | scoutingServer.Handle("/requests/request/all_matches", requestAllMatchesHandler{db}) |
Filip Kujawa | f882e02 | 2022-12-14 13:14:08 -0800 | [diff] [blame] | 695 | scoutingServer.Handle("/requests/request/all_notes", requestAllNotesHandler{db}) |
| 696 | scoutingServer.Handle("/requests/request/all_driver_rankings", requestAllDriverRankingsHandler{db}) |
Philipp Schrader | acf9623 | 2022-03-01 22:03:30 -0800 | [diff] [blame] | 697 | scoutingServer.Handle("/requests/request/data_scouting", requestDataScoutingHandler{db}) |
Philipp Schrader | d3fac19 | 2022-03-02 20:35:46 -0800 | [diff] [blame] | 698 | scoutingServer.Handle("/requests/refresh_match_list", refreshMatchListHandler{db, scrape}) |
Alex Perry | 81f96ba | 2022-03-13 18:26:19 -0700 | [diff] [blame] | 699 | scoutingServer.Handle("/requests/submit/submit_notes", submitNoteScoutingHandler{db}) |
| 700 | scoutingServer.Handle("/requests/request/notes_for_team", requestNotesForTeamHandler{db}) |
Milo Lin | 1d59f0c | 2022-06-22 20:30:58 -0700 | [diff] [blame] | 701 | scoutingServer.Handle("/requests/submit/shift_schedule", submitShiftScheduleHandler{db}) |
| 702 | scoutingServer.Handle("/requests/request/shift_schedule", requestShiftScheduleHandler{db}) |
Filip Kujawa | 210a03b | 2022-11-24 14:41:11 -0800 | [diff] [blame] | 703 | scoutingServer.Handle("/requests/submit/submit_driver_ranking", SubmitDriverRankingHandler{db}) |
Philipp Schrader | cdb5cfc | 2022-02-20 14:57:07 -0800 | [diff] [blame] | 704 | } |