blob: a9919af64f897c78d3152b0937f870170002d2e5 [file] [log] [blame]
Philipp Schradercdb5cfc2022-02-20 14:57:07 -08001package requests
2
3import (
Philipp Schraderfae8a7e2022-03-13 22:51:54 -07004 "encoding/base64"
Philipp Schraderd3fac192022-03-02 20:35:46 -08005 "errors"
Philipp Schradercdb5cfc2022-02-20 14:57:07 -08006 "fmt"
7 "io"
Philipp Schraderfae8a7e2022-03-13 22:51:54 -07008 "log"
Philipp Schradercdb5cfc2022-02-20 14:57:07 -08009 "net/http"
Emily Markovabf24c9e2023-02-08 20:31:11 -080010 "sort"
Philipp Schraderd3fac192022-03-02 20:35:46 -080011 "strconv"
12 "strings"
Philipp Schradercdb5cfc2022-02-20 14:57:07 -080013
Philipp Schrader8747f1b2022-02-23 23:56:22 -080014 "github.com/frc971/971-Robot-Code/scouting/db"
Filip Kujawac1ded372023-05-27 14:33:43 -070015 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/delete_2023_data_scouting"
16 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/delete_2023_data_scouting_response"
Emily Markova8cb91312024-02-02 12:30:37 -080017 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/delete_2024_data_scouting"
18 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/delete_2024_data_scouting_response"
Philipp Schradercdb5cfc2022-02-20 14:57:07 -080019 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/error_response"
Emily Markova290147d2023-03-03 22:40:06 -080020 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_2023_data_scouting"
21 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_2023_data_scouting_response"
Emily Markova8cb91312024-02-02 12:30:37 -080022 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_2024_data_scouting"
23 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_2024_data_scouting_response"
Filip Kujawaf882e022022-12-14 13:14:08 -080024 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_all_driver_rankings"
25 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_all_driver_rankings_response"
Philipp Schradercbf5c6a2022-02-27 23:25:19 -080026 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_all_matches"
27 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_all_matches_response"
Filip Kujawaf882e022022-12-14 13:14:08 -080028 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_all_notes"
29 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_all_notes_response"
Emily Markova8e39f452023-12-23 12:17:30 -080030 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_all_pit_images"
31 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_all_pit_images_response"
Alex Perry81f96ba2022-03-13 18:26:19 -070032 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_notes_for_team"
33 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_notes_for_team_response"
Emily Markovafaecfe12023-07-01 12:40:03 -070034 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_pit_images"
35 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_pit_images_response"
Milo Lin1d59f0c2022-06-22 20:30:58 -070036 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_shift_schedule"
37 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/request_shift_schedule_response"
Emily Markova8cb91312024-02-02 12:30:37 -080038 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/submit_2024_actions"
39 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/submit_2024_actions_response"
Sabina Leaver759090b2023-01-14 20:42:56 -080040 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/submit_actions"
41 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/submit_actions_response"
Filip Kujawa210a03b2022-11-24 14:41:11 -080042 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/submit_driver_ranking"
43 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/submit_driver_ranking_response"
Alex Perry81f96ba2022-03-13 18:26:19 -070044 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/submit_notes"
45 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/submit_notes_response"
Emily Markovafaecfe12023-07-01 12:40:03 -070046 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/submit_pit_image"
47 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/submit_pit_image_response"
Milo Lin1d59f0c2022-06-22 20:30:58 -070048 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/submit_shift_schedule"
49 "github.com/frc971/971-Robot-Code/scouting/webserver/requests/messages/submit_shift_schedule_response"
Philipp Schradercdb5cfc2022-02-20 14:57:07 -080050 "github.com/frc971/971-Robot-Code/scouting/webserver/server"
51 flatbuffers "github.com/google/flatbuffers/go"
52)
53
Philipp Schradercbf5c6a2022-02-27 23:25:19 -080054type RequestAllMatches = request_all_matches.RequestAllMatches
55type RequestAllMatchesResponseT = request_all_matches_response.RequestAllMatchesResponseT
Filip Kujawaf882e022022-12-14 13:14:08 -080056type RequestAllDriverRankings = request_all_driver_rankings.RequestAllDriverRankings
57type RequestAllDriverRankingsResponseT = request_all_driver_rankings_response.RequestAllDriverRankingsResponseT
58type RequestAllNotes = request_all_notes.RequestAllNotes
59type RequestAllNotesResponseT = request_all_notes_response.RequestAllNotesResponseT
Emily Markova290147d2023-03-03 22:40:06 -080060type Request2023DataScouting = request_2023_data_scouting.Request2023DataScouting
61type Request2023DataScoutingResponseT = request_2023_data_scouting_response.Request2023DataScoutingResponseT
Emily Markova8cb91312024-02-02 12:30:37 -080062type Request2024DataScouting = request_2024_data_scouting.Request2024DataScouting
63type Request2024DataScoutingResponseT = request_2024_data_scouting_response.Request2024DataScoutingResponseT
Alex Perry81f96ba2022-03-13 18:26:19 -070064type SubmitNotes = submit_notes.SubmitNotes
65type SubmitNotesResponseT = submit_notes_response.SubmitNotesResponseT
Emily Markovafaecfe12023-07-01 12:40:03 -070066type SubmitPitImage = submit_pit_image.SubmitPitImage
67type SubmitPitImageResponseT = submit_pit_image_response.SubmitPitImageResponseT
68type RequestPitImages = request_pit_images.RequestPitImages
69type RequestPitImagesResponseT = request_pit_images_response.RequestPitImagesResponseT
Emily Markova8e39f452023-12-23 12:17:30 -080070type RequestAllPitImages = request_all_pit_images.RequestAllPitImages
71type RequestAllPitImagesResponseT = request_all_pit_images_response.RequestAllPitImagesResponseT
Alex Perry81f96ba2022-03-13 18:26:19 -070072type RequestNotesForTeam = request_notes_for_team.RequestNotesForTeam
73type RequestNotesForTeamResponseT = request_notes_for_team_response.RequestNotesForTeamResponseT
Milo Lin1d59f0c2022-06-22 20:30:58 -070074type RequestShiftSchedule = request_shift_schedule.RequestShiftSchedule
75type RequestShiftScheduleResponseT = request_shift_schedule_response.RequestShiftScheduleResponseT
76type SubmitShiftSchedule = submit_shift_schedule.SubmitShiftSchedule
77type SubmitShiftScheduleResponseT = submit_shift_schedule_response.SubmitShiftScheduleResponseT
Filip Kujawa210a03b2022-11-24 14:41:11 -080078type SubmitDriverRanking = submit_driver_ranking.SubmitDriverRanking
79type SubmitDriverRankingResponseT = submit_driver_ranking_response.SubmitDriverRankingResponseT
Sabina Leaver759090b2023-01-14 20:42:56 -080080type SubmitActions = submit_actions.SubmitActions
Sabina Leaver9b4eb312023-02-20 19:58:17 -080081type Action = submit_actions.Action
Emily Markova8cb91312024-02-02 12:30:37 -080082type Action2024 = submit_2024_actions.Action
Sabina Leaver759090b2023-01-14 20:42:56 -080083type SubmitActionsResponseT = submit_actions_response.SubmitActionsResponseT
Emily Markova8cb91312024-02-02 12:30:37 -080084type Submit2024Actions = submit_2024_actions.Submit2024Actions
85type Submit2024ActionsResponseT = submit_2024_actions_response.Submit2024ActionsResponseT
Filip Kujawac1ded372023-05-27 14:33:43 -070086type Delete2023DataScouting = delete_2023_data_scouting.Delete2023DataScouting
87type Delete2023DataScoutingResponseT = delete_2023_data_scouting_response.Delete2023DataScoutingResponseT
Emily Markova8cb91312024-02-02 12:30:37 -080088type Delete2024DataScouting = delete_2024_data_scouting.Delete2024DataScouting
89type Delete2024DataScoutingResponseT = delete_2024_data_scouting_response.Delete2024DataScoutingResponseT
Philipp Schradercbf5c6a2022-02-27 23:25:19 -080090
Philipp Schrader8747f1b2022-02-23 23:56:22 -080091// The interface we expect the database abstraction to conform to.
92// We use an interface here because it makes unit testing easier.
93type Database interface {
Emily Markovabf24c9e2023-02-08 20:31:11 -080094 AddToMatch(db.TeamMatch) error
Milo Lin1d59f0c2022-06-22 20:30:58 -070095 AddToShift(db.Shift) error
Emily Markova290147d2023-03-03 22:40:06 -080096 AddToStats2023(db.Stats2023) error
Emily Markova8cb91312024-02-02 12:30:37 -080097 AddToStats2024(db.Stats2024) error
Emily Markovabf24c9e2023-02-08 20:31:11 -080098 ReturnMatches() ([]db.TeamMatch, error)
Filip Kujawaf882e022022-12-14 13:14:08 -080099 ReturnAllNotes() ([]db.NotesData, error)
100 ReturnAllDriverRankings() ([]db.DriverRankingData, error)
Milo Lin1d59f0c2022-06-22 20:30:58 -0700101 ReturnAllShifts() ([]db.Shift, error)
Emily Markova290147d2023-03-03 22:40:06 -0800102 ReturnStats2023() ([]db.Stats2023, error)
Filip Kujawaf3f9def2023-04-20 13:46:46 -0700103 ReturnStats2023ForTeam(teamNumber string, matchNumber int32, setNumber int32, compLevel string, preScouting bool) ([]db.Stats2023, error)
Emily Markova8cb91312024-02-02 12:30:37 -0800104 ReturnStats2024() ([]db.Stats2024, error)
105 ReturnStats2024ForTeam(teamNumber string, matchNumber int32, setNumber int32, compLevel string, preScouting bool) ([]db.Stats2024, error)
Milo Lin1d59f0c2022-06-22 20:30:58 -0700106 QueryAllShifts(int) ([]db.Shift, error)
Emily Markovae68b7632023-12-30 14:17:55 -0800107 QueryNotes(string) ([]string, error)
Emily Markovafaecfe12023-07-01 12:40:03 -0700108 QueryPitImages(string) ([]db.RequestedPitImage, error)
Emily Markova8e39f452023-12-23 12:17:30 -0800109 ReturnPitImages() ([]db.PitImage, error)
Filip Kujawaf947cb42022-11-21 10:00:30 -0800110 AddNotes(db.NotesData) error
Emily Markovafaecfe12023-07-01 12:40:03 -0700111 AddPitImage(db.PitImage) error
Filip Kujawa210a03b2022-11-24 14:41:11 -0800112 AddDriverRanking(db.DriverRankingData) error
Sabina Leaver9b4eb312023-02-20 19:58:17 -0800113 AddAction(db.Action) error
Filip Kujawac1ded372023-05-27 14:33:43 -0700114 DeleteFromStats(string, int32, int32, string) error
Emily Markova8cb91312024-02-02 12:30:37 -0800115 DeleteFromStats2024(string, int32, int32, string) error
Filip Kujawac1ded372023-05-27 14:33:43 -0700116 DeleteFromActions(string, int32, int32, string) error
Philipp Schrader8747f1b2022-02-23 23:56:22 -0800117}
118
Philipp Schradercdb5cfc2022-02-20 14:57:07 -0800119// Handles unknown requests. Just returns a 404.
120func unknown(w http.ResponseWriter, req *http.Request) {
121 w.WriteHeader(http.StatusNotFound)
122}
123
124func respondWithError(w http.ResponseWriter, statusCode int, errorMessage string) {
125 builder := flatbuffers.NewBuilder(1024)
126 builder.Finish((&error_response.ErrorResponseT{
127 ErrorMessage: errorMessage,
128 }).Pack(builder))
129 w.WriteHeader(statusCode)
130 w.Write(builder.FinishedBytes())
131}
132
133func respondNotImplemented(w http.ResponseWriter) {
134 respondWithError(w, http.StatusNotImplemented, "")
135}
136
Philipp Schraderb7e75932022-03-26 16:18:34 -0700137func parseRequest[T interface{}](w http.ResponseWriter, buf []byte, requestName string, parser func([]byte, flatbuffers.UOffsetT) *T) (*T, bool) {
Philipp Schradercdb5cfc2022-02-20 14:57:07 -0800138 success := true
139 defer func() {
140 if r := recover(); r != nil {
Philipp Schraderb7e75932022-03-26 16:18:34 -0700141 respondWithError(w, http.StatusBadRequest, fmt.Sprintf("Failed to parse %s: %v", requestName, r))
Philipp Schradercdb5cfc2022-02-20 14:57:07 -0800142 success = false
143 }
144 }()
Philipp Schraderb7e75932022-03-26 16:18:34 -0700145 result := parser(buf, 0)
Philipp Schradercdb5cfc2022-02-20 14:57:07 -0800146 return result, success
147}
148
Philipp Schraderfae8a7e2022-03-13 22:51:54 -0700149// Parses the authorization information that the browser inserts into the
150// headers. The authorization follows this format:
151//
Philipp Schrader35bb1532023-03-05 13:49:12 -0800152// req.Headers["Authorization"] = []string{"Basic <base64 encoded username:password>"}
Philipp Schraderfae8a7e2022-03-13 22:51:54 -0700153func parseUsername(req *http.Request) string {
154 auth, ok := req.Header["Authorization"]
155 if !ok {
156 return "unknown"
157 }
158
159 parts := strings.Split(auth[0], " ")
160 if !(len(parts) == 2 && parts[0] == "Basic") {
161 return "unknown"
162 }
163
164 info, err := base64.StdEncoding.DecodeString(parts[1])
165 if err != nil {
166 log.Println("ERROR: Failed to parse Basic authentication.")
167 return "unknown"
168 }
169
170 loginParts := strings.Split(string(info), ":")
171 if len(loginParts) != 2 {
172 return "unknown"
173 }
174 return loginParts[0]
175}
176
Philipp Schradercbf5c6a2022-02-27 23:25:19 -0800177// Handles a RequestAllMaches request.
178type requestAllMatchesHandler struct {
179 db Database
180}
181
Philipp Schrader0f7b6362023-03-11 14:02:48 -0800182// Change structure of match objects in the database(1 per team) to
183// the old match structure(1 per match) that the webserver uses.
184// We use the information in this struct to identify which match object
185// corresponds to which old match structure object.
186type MatchAssemblyKey struct {
187 MatchNumber int32
188 SetNumber int32
189 CompLevel string
190}
191
Emily Markovabf24c9e2023-02-08 20:31:11 -0800192func findIndexInList(list []string, comp_level string) (int, error) {
193 for index, value := range list {
194 if value == comp_level {
195 return index, nil
196 }
197 }
198 return -1, errors.New(fmt.Sprint("Failed to find comp level ", comp_level, " in list ", list))
199}
200
Emily Markovab8551572023-03-22 19:49:39 -0700201func (handler requestAllMatchesHandler) teamHasBeenDataScouted(key MatchAssemblyKey, teamNumber string) (bool, error) {
Emily Markovadcadcb62024-02-03 13:07:17 -0800202 stats, err := handler.db.ReturnStats2024ForTeam(
Filip Kujawaf3f9def2023-04-20 13:46:46 -0700203 teamNumber, key.MatchNumber, key.SetNumber, key.CompLevel, false)
Philipp Schrader0f7b6362023-03-11 14:02:48 -0800204 if err != nil {
205 return false, err
206 }
207 return (len(stats) > 0), nil
208}
209
Philipp Schradercbf5c6a2022-02-27 23:25:19 -0800210func (handler requestAllMatchesHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
211 requestBytes, err := io.ReadAll(req.Body)
212 if err != nil {
213 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
214 return
215 }
216
Philipp Schraderb7e75932022-03-26 16:18:34 -0700217 _, success := parseRequest(w, requestBytes, "RequestAllMatches", request_all_matches.GetRootAsRequestAllMatches)
Philipp Schradercbf5c6a2022-02-27 23:25:19 -0800218 if !success {
219 return
220 }
221
222 matches, err := handler.db.ReturnMatches()
223 if err != nil {
Philipp Schraderfae8a7e2022-03-13 22:51:54 -0700224 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to query database: ", err))
Philipp Schrader2e7eb0002022-03-02 22:52:39 -0800225 return
Philipp Schradercbf5c6a2022-02-27 23:25:19 -0800226 }
227
Philipp Schrader0f7b6362023-03-11 14:02:48 -0800228 assembledMatches := map[MatchAssemblyKey]request_all_matches_response.MatchT{}
Emily Markovabf24c9e2023-02-08 20:31:11 -0800229
Philipp Schradercbf5c6a2022-02-27 23:25:19 -0800230 for _, match := range matches {
Philipp Schrader0f7b6362023-03-11 14:02:48 -0800231 key := MatchAssemblyKey{match.MatchNumber, match.SetNumber, match.CompLevel}
232
233 // Retrieve the converted match structure we have assembled so
234 // far. If we haven't started assembling one yet, then start a
235 // new one.
Emily Markovabf24c9e2023-02-08 20:31:11 -0800236 entry, ok := assembledMatches[key]
237 if !ok {
238 entry = request_all_matches_response.MatchT{
239 MatchNumber: match.MatchNumber,
240 SetNumber: match.SetNumber,
241 CompLevel: match.CompLevel,
Philipp Schrader0f7b6362023-03-11 14:02:48 -0800242 DataScouted: &request_all_matches_response.ScoutedLevelT{},
Emily Markovabf24c9e2023-02-08 20:31:11 -0800243 }
244 }
Philipp Schrader0f7b6362023-03-11 14:02:48 -0800245
Emily Markovab8551572023-03-22 19:49:39 -0700246 var team *string
Philipp Schrader0f7b6362023-03-11 14:02:48 -0800247 var dataScoutedTeam *bool
248
249 // Fill in the field for the match that we have in in the
250 // database. In the database, each match row only has 1 team
251 // number.
Emily Markovabf24c9e2023-02-08 20:31:11 -0800252 switch match.Alliance {
253 case "R":
254 switch match.AlliancePosition {
255 case 1:
Philipp Schrader0f7b6362023-03-11 14:02:48 -0800256 team = &entry.R1
257 dataScoutedTeam = &entry.DataScouted.R1
Emily Markovabf24c9e2023-02-08 20:31:11 -0800258 case 2:
Philipp Schrader0f7b6362023-03-11 14:02:48 -0800259 team = &entry.R2
260 dataScoutedTeam = &entry.DataScouted.R2
Emily Markovabf24c9e2023-02-08 20:31:11 -0800261 case 3:
Philipp Schrader0f7b6362023-03-11 14:02:48 -0800262 team = &entry.R3
263 dataScoutedTeam = &entry.DataScouted.R3
Emily Markovabf24c9e2023-02-08 20:31:11 -0800264 default:
265 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Unknown red position ", strconv.Itoa(int(match.AlliancePosition)), " in match ", strconv.Itoa(int(match.MatchNumber))))
266 return
267 }
268 case "B":
269 switch match.AlliancePosition {
270 case 1:
Philipp Schrader0f7b6362023-03-11 14:02:48 -0800271 team = &entry.B1
272 dataScoutedTeam = &entry.DataScouted.B1
Emily Markovabf24c9e2023-02-08 20:31:11 -0800273 case 2:
Philipp Schrader0f7b6362023-03-11 14:02:48 -0800274 team = &entry.B2
275 dataScoutedTeam = &entry.DataScouted.B2
Emily Markovabf24c9e2023-02-08 20:31:11 -0800276 case 3:
Philipp Schrader0f7b6362023-03-11 14:02:48 -0800277 team = &entry.B3
278 dataScoutedTeam = &entry.DataScouted.B3
Emily Markovabf24c9e2023-02-08 20:31:11 -0800279 default:
280 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Unknown blue position ", strconv.Itoa(int(match.AlliancePosition)), " in match ", strconv.Itoa(int(match.MatchNumber))))
281 return
282 }
283 default:
284 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Unknown alliance ", match.Alliance, " in match ", strconv.Itoa(int(match.AlliancePosition))))
285 return
286 }
Philipp Schrader0f7b6362023-03-11 14:02:48 -0800287
288 *team = match.TeamNumber
289
290 // Figure out if this team has been data scouted already.
291 *dataScoutedTeam, err = handler.teamHasBeenDataScouted(key, match.TeamNumber)
292 if err != nil {
293 respondWithError(w, http.StatusInternalServerError, fmt.Sprint(
294 "Failed to determine data scouting status for team ",
295 strconv.Itoa(int(match.AlliancePosition)),
296 " in match ",
297 strconv.Itoa(int(match.MatchNumber)),
298 err))
299 return
300 }
301
Emily Markovabf24c9e2023-02-08 20:31:11 -0800302 assembledMatches[key] = entry
303 }
304
305 var response RequestAllMatchesResponseT
306 for _, match := range assembledMatches {
307 copied_match := match
308 response.MatchList = append(response.MatchList, &copied_match)
309 }
310
311 var MATCH_TYPE_ORDERING = []string{"qm", "ef", "qf", "sf", "f"}
312
313 err = nil
314 sort.Slice(response.MatchList, func(i, j int) bool {
315 if err != nil {
316 return false
317 }
318 a := response.MatchList[i]
319 b := response.MatchList[j]
320
Emily Markovaabcac6e2023-02-18 17:50:03 -0800321 aMatchTypeIndex, err2 := findIndexInList(MATCH_TYPE_ORDERING, a.CompLevel)
322 if err2 != nil {
323 err = errors.New(fmt.Sprint("Comp level ", a.CompLevel, " not found in sorting list ", MATCH_TYPE_ORDERING, " : ", err2))
Emily Markovabf24c9e2023-02-08 20:31:11 -0800324 return false
325 }
Emily Markovaabcac6e2023-02-18 17:50:03 -0800326 bMatchTypeIndex, err2 := findIndexInList(MATCH_TYPE_ORDERING, b.CompLevel)
327 if err2 != nil {
328 err = errors.New(fmt.Sprint("Comp level ", b.CompLevel, " not found in sorting list ", MATCH_TYPE_ORDERING, " : ", err2))
Emily Markovabf24c9e2023-02-08 20:31:11 -0800329 return false
330 }
331
332 if aMatchTypeIndex < bMatchTypeIndex {
333 return true
334 }
335 if aMatchTypeIndex > bMatchTypeIndex {
336 return false
337 }
338
339 // Then sort by match number. E.g. in semi finals, all match 1 rounds
340 // are done first. Then come match 2 rounds. And then, if necessary,
341 // the match 3 rounds.
342 aMatchNumber := a.MatchNumber
343 bMatchNumber := b.MatchNumber
344 if aMatchNumber < bMatchNumber {
345 return true
346 }
347 if aMatchNumber > bMatchNumber {
348 return false
349 }
350 // Lastly, sort by set number. I.e. Semi Final 1 Match 1 happens first.
351 // Then comes Semi Final 2 Match 1. Then comes Semi Final 1 Match 2. Then
352 // Semi Final 2 Match 2.
353 aSetNumber := a.SetNumber
354 bSetNumber := b.SetNumber
355 if aSetNumber < bSetNumber {
356 return true
357 }
358 if aSetNumber > bSetNumber {
359 return false
360 }
361 return true
362 })
363
364 if err != nil {
365 // check if error happened during sorting and notify webpage if that
366 respondWithError(w, http.StatusInternalServerError, fmt.Sprint(err))
367 return
Philipp Schradercbf5c6a2022-02-27 23:25:19 -0800368 }
369
370 builder := flatbuffers.NewBuilder(50 * 1024)
371 builder.Finish((&response).Pack(builder))
372 w.Write(builder.FinishedBytes())
373}
374
Alex Perry81f96ba2022-03-13 18:26:19 -0700375type submitNoteScoutingHandler struct {
376 db Database
377}
378
379func (handler submitNoteScoutingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
380 requestBytes, err := io.ReadAll(req.Body)
381 if err != nil {
382 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
383 return
384 }
385
Philipp Schraderb7e75932022-03-26 16:18:34 -0700386 request, success := parseRequest(w, requestBytes, "SubmitNotes", submit_notes.GetRootAsSubmitNotes)
Alex Perry81f96ba2022-03-13 18:26:19 -0700387 if !success {
388 return
389 }
390
Filip Kujawaf947cb42022-11-21 10:00:30 -0800391 err = handler.db.AddNotes(db.NotesData{
Emily Markovae68b7632023-12-30 14:17:55 -0800392 TeamNumber: string(request.Team()),
Filip Kujawa7ddd5652023-03-07 19:56:15 -0800393 Notes: string(request.Notes()),
394 GoodDriving: bool(request.GoodDriving()),
395 BadDriving: bool(request.BadDriving()),
Filip Kujawa11dc4c92023-04-13 08:55:43 -0700396 SolidPlacing: bool(request.SolidPlacing()),
Filip Kujawa7ddd5652023-03-07 19:56:15 -0800397 SketchyPlacing: bool(request.SketchyPlacing()),
398 GoodDefense: bool(request.GoodDefense()),
399 BadDefense: bool(request.BadDefense()),
400 EasilyDefended: bool(request.EasilyDefended()),
Filip Kujawaf947cb42022-11-21 10:00:30 -0800401 })
Alex Perry81f96ba2022-03-13 18:26:19 -0700402 if err != nil {
403 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to insert notes: %v", err))
404 return
405 }
406
407 var response SubmitNotesResponseT
408 builder := flatbuffers.NewBuilder(10)
409 builder.Finish((&response).Pack(builder))
410 w.Write(builder.FinishedBytes())
411}
412
Emily Markovafaecfe12023-07-01 12:40:03 -0700413type submitPitImageScoutingHandler struct {
414 db Database
415}
416
417func (handler submitPitImageScoutingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
418 requestBytes, err := io.ReadAll(req.Body)
419 if err != nil {
420 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
421 return
422 }
423
424 request, success := parseRequest(w, requestBytes, "SubmitPitImage", submit_pit_image.GetRootAsSubmitPitImage)
425 if !success {
426 return
427 }
428
429 err = handler.db.AddPitImage(db.PitImage{
430 TeamNumber: string(request.TeamNumber()),
431 CheckSum: db.ComputeSha256FromByteArray(request.ImageDataBytes()),
432 ImagePath: string(request.ImagePath()),
433 ImageData: request.ImageDataBytes(),
434 })
435 if err != nil {
436 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to insert notes: %v", err))
437 return
438 }
439
440 var response SubmitPitImageResponseT
441 builder := flatbuffers.NewBuilder(10)
442 builder.Finish((&response).Pack(builder))
443 w.Write(builder.FinishedBytes())
444}
445
Emily Markova8cb91312024-02-02 12:30:37 -0800446func ConvertActionsToStat2024(submit2024Actions *submit_2024_actions.Submit2024Actions) (db.Stats2024, error) {
447 overall_time := int64(0)
448 cycles := int64(0)
449 picked_up := false
450 lastPlacedTime := int64(0)
451 stat := db.Stats2024{
452 PreScouting: submit2024Actions.PreScouting(), TeamNumber: string(submit2024Actions.TeamNumber()), MatchNumber: submit2024Actions.MatchNumber(), SetNumber: submit2024Actions.SetNumber(), CompLevel: string(submit2024Actions.CompLevel()),
453 StartingQuadrant: 0, SpeakerAuto: 0, AmpAuto: 0, NotesDroppedAuto: 0, MobilityAuto: false,
454 Speaker: 0, Amp: 0, SpeakerAmplified: 0, AmpAmplified: 0, NotesDropped: 0, Penalties: 0,
Emily Markova6079e2f2024-02-17 13:17:24 -0800455 TrapNote: false, Spotlight: false, AvgCycle: 0, Park: false, OnStage: false, Harmony: false, CollectedBy: "",
Emily Markova8cb91312024-02-02 12:30:37 -0800456 }
457 // Loop over all actions.
458 for i := 0; i < submit2024Actions.ActionsListLength(); i++ {
459 var action submit_2024_actions.Action
460 if !submit2024Actions.ActionsList(&action, i) {
461 return db.Stats2024{}, errors.New(fmt.Sprintf("Failed to parse submit_2024_actions.Action"))
462 }
463 actionTable := new(flatbuffers.Table)
464 action_type := action.ActionTakenType()
465 if !action.ActionTaken(actionTable) {
466 return db.Stats2024{}, errors.New(fmt.Sprint("Failed to parse sub-action or sub-action was missing"))
467 }
468 if action_type == submit_2024_actions.ActionTypeStartMatchAction {
469 var startMatchAction submit_2024_actions.StartMatchAction
470 startMatchAction.Init(actionTable.Bytes, actionTable.Pos)
471 stat.StartingQuadrant = startMatchAction.Position()
472 } else if action_type == submit_2024_actions.ActionTypeMobilityAction {
473 var mobilityAction submit_2024_actions.MobilityAction
474 mobilityAction.Init(actionTable.Bytes, actionTable.Pos)
475 if mobilityAction.Mobility() {
476 stat.MobilityAuto = true
477 }
478
479 } else if action_type == submit_2024_actions.ActionTypePenaltyAction {
480 var penaltyAction submit_2024_actions.PenaltyAction
481 penaltyAction.Init(actionTable.Bytes, actionTable.Pos)
Emily Markovadcadcb62024-02-03 13:07:17 -0800482 stat.Penalties += penaltyAction.Penalties()
Emily Markova8cb91312024-02-02 12:30:37 -0800483
484 } else if action_type == submit_2024_actions.ActionTypePickupNoteAction {
485 var pick_up_action submit_2024_actions.PickupNoteAction
486 pick_up_action.Init(actionTable.Bytes, actionTable.Pos)
487 if picked_up == true {
488 auto := pick_up_action.Auto()
489 if auto == false {
490 stat.NotesDropped += 1
491 } else {
492 stat.NotesDroppedAuto += 1
493 }
494 } else {
495 picked_up = true
496 }
497 } else if action_type == submit_2024_actions.ActionTypePlaceNoteAction {
498 var place_action submit_2024_actions.PlaceNoteAction
499 place_action.Init(actionTable.Bytes, actionTable.Pos)
500 if !picked_up {
501 return db.Stats2024{}, errors.New(fmt.Sprintf("Got PlaceNoteAction without corresponding PickupObjectAction"))
502 }
503 score_type := place_action.ScoreType()
504 auto := place_action.Auto()
505 if score_type == submit_2024_actions.ScoreTypekAMP && auto {
506 stat.AmpAuto += 1
507 } else if score_type == submit_2024_actions.ScoreTypekAMP && !auto {
508 stat.Amp += 1
509 } else if score_type == submit_2024_actions.ScoreTypekAMP_AMPLIFIED && !auto {
510 stat.AmpAmplified += 1
511 } else if score_type == submit_2024_actions.ScoreTypekSPEAKER && !auto {
512 stat.Speaker += 1
513 } else if score_type == submit_2024_actions.ScoreTypekSPEAKER && auto {
514 stat.SpeakerAuto += 1
515 } else if score_type == submit_2024_actions.ScoreTypekSPEAKER_AMPLIFIED && !auto {
516 stat.SpeakerAmplified += 1
517 } else {
518 return db.Stats2024{}, errors.New(fmt.Sprintf("Got unknown ObjectType/ScoreLevel/Auto combination"))
519 }
520 picked_up = false
521 if lastPlacedTime != int64(0) {
522 // If this is not the first time we place,
523 // start counting cycle time. We define cycle
524 // time as the time between placements.
525 overall_time += int64(action.Timestamp()) - lastPlacedTime
526 cycles += 1
527 }
528 lastPlacedTime = int64(action.Timestamp())
529 } else if action_type == submit_2024_actions.ActionTypeEndMatchAction {
530 var endMatchAction submit_2024_actions.EndMatchAction
531 endMatchAction.Init(actionTable.Bytes, actionTable.Pos)
532 if endMatchAction.StageType() == submit_2024_actions.StageTypekON_STAGE {
533 stat.OnStage = true
534 } else if endMatchAction.StageType() == submit_2024_actions.StageTypekPARK {
535 stat.Park = true
536 } else if endMatchAction.StageType() == submit_2024_actions.StageTypekHARMONY {
537 stat.Harmony = true
538 }
539 stat.TrapNote = endMatchAction.TrapNote()
Emily Markova6079e2f2024-02-17 13:17:24 -0800540 stat.Spotlight = endMatchAction.Spotlight()
Emily Markova8cb91312024-02-02 12:30:37 -0800541 }
542 }
543 if cycles != 0 {
544 stat.AvgCycle = overall_time / cycles
545 } else {
546 stat.AvgCycle = 0
547 }
548 return stat, nil
549}
550
551// Handles a Request2024DataScouting request.
552type request2024DataScoutingHandler struct {
553 db Database
554}
555
556func (handler request2024DataScoutingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
557 requestBytes, err := io.ReadAll(req.Body)
558 if err != nil {
559 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
560 return
561 }
562
563 _, success := parseRequest(w, requestBytes, "Request2024DataScouting", request_2024_data_scouting.GetRootAsRequest2024DataScouting)
564 if !success {
565 return
566 }
567
568 stats, err := handler.db.ReturnStats2024()
569 if err != nil {
570 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to query database: ", err))
571 return
572 }
573
574 var response Request2024DataScoutingResponseT
575 for _, stat := range stats {
576 response.StatsList = append(response.StatsList, &request_2024_data_scouting_response.Stats2024T{
577 TeamNumber: stat.TeamNumber,
578 MatchNumber: stat.MatchNumber,
579 SetNumber: stat.SetNumber,
580 CompLevel: stat.CompLevel,
581 StartingQuadrant: stat.StartingQuadrant,
582 SpeakerAuto: stat.SpeakerAuto,
583 AmpAuto: stat.AmpAuto,
584 NotesDroppedAuto: stat.NotesDroppedAuto,
585 MobilityAuto: stat.MobilityAuto,
586 Speaker: stat.Speaker,
587 Amp: stat.Amp,
588 SpeakerAmplified: stat.SpeakerAmplified,
589 AmpAmplified: stat.AmpAmplified,
590 NotesDropped: stat.NotesDropped,
591 Penalties: stat.Penalties,
592 TrapNote: stat.TrapNote,
Emily Markova6079e2f2024-02-17 13:17:24 -0800593 Spotlight: stat.Spotlight,
Emily Markova8cb91312024-02-02 12:30:37 -0800594 AvgCycle: stat.AvgCycle,
595 Park: stat.Park,
596 OnStage: stat.OnStage,
597 Harmony: stat.Harmony,
598 CollectedBy: stat.CollectedBy,
599 })
600 }
601
602 builder := flatbuffers.NewBuilder(50 * 1024)
603 builder.Finish((&response).Pack(builder))
604 w.Write(builder.FinishedBytes())
605}
606
Emily Markova1abe9782023-03-11 19:45:38 -0800607func ConvertActionsToStat(submitActions *submit_actions.SubmitActions) (db.Stats2023, error) {
608 overall_time := int64(0)
609 cycles := int64(0)
610 picked_up := false
611 lastPlacedTime := int64(0)
Philipp Schrader4b489222023-04-15 16:40:16 -0700612 stat := db.Stats2023{
613 PreScouting: submitActions.PreScouting(),
614 TeamNumber: string(submitActions.TeamNumber()), MatchNumber: submitActions.MatchNumber(), SetNumber: submitActions.SetNumber(), CompLevel: string(submitActions.CompLevel()),
Emily Markova1abe9782023-03-11 19:45:38 -0800615 StartingQuadrant: 0, LowCubesAuto: 0, MiddleCubesAuto: 0, HighCubesAuto: 0, CubesDroppedAuto: 0,
616 LowConesAuto: 0, MiddleConesAuto: 0, HighConesAuto: 0, ConesDroppedAuto: 0, LowCubes: 0, MiddleCubes: 0, HighCubes: 0,
Philipp Schradere11114f2023-04-15 17:04:25 -0700617 CubesDropped: 0, LowCones: 0, MiddleCones: 0, HighCones: 0, ConesDropped: 0, SuperchargedPieces: 0, AvgCycle: 0, CollectedBy: "",
Emily Markova1abe9782023-03-11 19:45:38 -0800618 }
619 // Loop over all actions.
620 for i := 0; i < submitActions.ActionsListLength(); i++ {
621 var action submit_actions.Action
622 if !submitActions.ActionsList(&action, i) {
623 return db.Stats2023{}, errors.New(fmt.Sprintf("Failed to parse submit_actions.Action"))
624 }
625 actionTable := new(flatbuffers.Table)
626 action_type := action.ActionTakenType()
627 if !action.ActionTaken(actionTable) {
628 return db.Stats2023{}, errors.New(fmt.Sprint("Failed to parse sub-action or sub-action was missing"))
629 }
630 if action_type == submit_actions.ActionTypeStartMatchAction {
631 var startMatchAction submit_actions.StartMatchAction
632 startMatchAction.Init(actionTable.Bytes, actionTable.Pos)
633 stat.StartingQuadrant = startMatchAction.Position()
Filip Kujawa0b4b1e52023-04-15 14:05:40 -0700634 } else if action_type == submit_actions.ActionTypeMobilityAction {
635 var mobilityAction submit_actions.MobilityAction
636 mobilityAction.Init(actionTable.Bytes, actionTable.Pos)
637 if mobilityAction.Mobility() {
638 stat.Mobility = true
639 }
640
Emily Markova46a69bf2023-03-22 20:45:52 -0700641 } else if action_type == submit_actions.ActionTypeAutoBalanceAction {
642 var autoBalanceAction submit_actions.AutoBalanceAction
643 autoBalanceAction.Init(actionTable.Bytes, actionTable.Pos)
644 if autoBalanceAction.Docked() {
645 stat.DockedAuto = true
646 }
647 if autoBalanceAction.Engaged() {
648 stat.EngagedAuto = true
649 }
Emily Markova63c63f62023-03-29 20:57:35 -0700650 if autoBalanceAction.BalanceAttempt() {
651 stat.BalanceAttemptAuto = true
652 }
Emily Markova1abe9782023-03-11 19:45:38 -0800653 } else if action_type == submit_actions.ActionTypePickupObjectAction {
654 var pick_up_action submit_actions.PickupObjectAction
655 pick_up_action.Init(actionTable.Bytes, actionTable.Pos)
656 if picked_up == true {
657 object := pick_up_action.ObjectType().String()
658 auto := pick_up_action.Auto()
659 if object == "kCube" && auto == false {
660 stat.CubesDropped += 1
661 } else if object == "kCube" && auto == true {
662 stat.CubesDroppedAuto += 1
663 } else if object == "kCone" && auto == false {
664 stat.ConesDropped += 1
665 } else if object == "kCube" && auto == true {
666 stat.ConesDroppedAuto += 1
667 }
668 } else {
669 picked_up = true
670 }
671 } else if action_type == submit_actions.ActionTypePlaceObjectAction {
672 var place_action submit_actions.PlaceObjectAction
673 place_action.Init(actionTable.Bytes, actionTable.Pos)
674 if !picked_up {
675 return db.Stats2023{}, errors.New(fmt.Sprintf("Got PlaceObjectAction without corresponding PickupObjectAction"))
676 }
677 object := place_action.ObjectType()
678 level := place_action.ScoreLevel()
679 auto := place_action.Auto()
680 if object == 0 && level == 0 && auto == true {
681 stat.LowCubesAuto += 1
682 } else if object == 0 && level == 0 && auto == false {
683 stat.LowCubes += 1
684 } else if object == 0 && level == 1 && auto == true {
685 stat.MiddleCubesAuto += 1
686 } else if object == 0 && level == 1 && auto == false {
687 stat.MiddleCubes += 1
688 } else if object == 0 && level == 2 && auto == true {
689 stat.HighCubesAuto += 1
690 } else if object == 0 && level == 2 && auto == false {
691 stat.HighCubes += 1
692 } else if object == 1 && level == 0 && auto == true {
693 stat.LowConesAuto += 1
694 } else if object == 1 && level == 0 && auto == false {
695 stat.LowCones += 1
696 } else if object == 1 && level == 1 && auto == true {
697 stat.MiddleConesAuto += 1
698 } else if object == 1 && level == 1 && auto == false {
699 stat.MiddleCones += 1
700 } else if object == 1 && level == 2 && auto == true {
701 stat.HighConesAuto += 1
702 } else if object == 1 && level == 2 && auto == false {
703 stat.HighCones += 1
Filip Kujawa7a045e72023-04-13 08:41:09 -0700704 } else if level == 3 {
705 stat.SuperchargedPieces += 1
Emily Markova1abe9782023-03-11 19:45:38 -0800706 } else {
707 return db.Stats2023{}, errors.New(fmt.Sprintf("Got unknown ObjectType/ScoreLevel/Auto combination"))
708 }
709 picked_up = false
710 if lastPlacedTime != int64(0) {
711 // If this is not the first time we place,
712 // start counting cycle time. We define cycle
713 // time as the time between placements.
714 overall_time += int64(action.Timestamp()) - lastPlacedTime
715 cycles += 1
716 }
717 lastPlacedTime = int64(action.Timestamp())
Emily Markova46a69bf2023-03-22 20:45:52 -0700718 } else if action_type == submit_actions.ActionTypeEndMatchAction {
719 var endMatchAction submit_actions.EndMatchAction
720 endMatchAction.Init(actionTable.Bytes, actionTable.Pos)
721 if endMatchAction.Docked() {
722 stat.Docked = true
723 }
724 if endMatchAction.Engaged() {
725 stat.Engaged = true
726 }
Emily Markova63c63f62023-03-29 20:57:35 -0700727 if endMatchAction.BalanceAttempt() {
728 stat.BalanceAttempt = true
729 }
Emily Markova1abe9782023-03-11 19:45:38 -0800730 }
731 }
732 if cycles != 0 {
Philipp Schrader8c878a22023-03-20 22:36:38 -0700733 stat.AvgCycle = overall_time / cycles
Emily Markova1abe9782023-03-11 19:45:38 -0800734 } else {
735 stat.AvgCycle = 0
736 }
737 return stat, nil
738}
739
Emily Markova290147d2023-03-03 22:40:06 -0800740// Handles a Request2023DataScouting request.
741type request2023DataScoutingHandler struct {
742 db Database
743}
744
745func (handler request2023DataScoutingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
746 requestBytes, err := io.ReadAll(req.Body)
747 if err != nil {
748 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
749 return
750 }
751
752 _, success := parseRequest(w, requestBytes, "Request2023DataScouting", request_2023_data_scouting.GetRootAsRequest2023DataScouting)
753 if !success {
754 return
755 }
756
757 stats, err := handler.db.ReturnStats2023()
758 if err != nil {
759 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to query database: ", err))
760 return
761 }
762
763 var response Request2023DataScoutingResponseT
764 for _, stat := range stats {
765 response.StatsList = append(response.StatsList, &request_2023_data_scouting_response.Stats2023T{
Emily Markova63c63f62023-03-29 20:57:35 -0700766 TeamNumber: stat.TeamNumber,
767 MatchNumber: stat.MatchNumber,
768 SetNumber: stat.SetNumber,
769 CompLevel: stat.CompLevel,
770 StartingQuadrant: stat.StartingQuadrant,
771 LowCubesAuto: stat.LowCubesAuto,
772 MiddleCubesAuto: stat.MiddleCubesAuto,
773 HighCubesAuto: stat.HighCubesAuto,
774 CubesDroppedAuto: stat.CubesDroppedAuto,
775 LowConesAuto: stat.LowConesAuto,
776 MiddleConesAuto: stat.MiddleConesAuto,
777 HighConesAuto: stat.HighConesAuto,
778 ConesDroppedAuto: stat.ConesDroppedAuto,
779 LowCubes: stat.LowCubes,
780 MiddleCubes: stat.MiddleCubes,
781 HighCubes: stat.HighCubes,
782 CubesDropped: stat.CubesDropped,
783 LowCones: stat.LowCones,
784 MiddleCones: stat.MiddleCones,
785 HighCones: stat.HighCones,
786 ConesDropped: stat.ConesDropped,
Filip Kujawa7a045e72023-04-13 08:41:09 -0700787 SuperchargedPieces: stat.SuperchargedPieces,
Emily Markova63c63f62023-03-29 20:57:35 -0700788 AvgCycle: stat.AvgCycle,
Filip Kujawa0b4b1e52023-04-15 14:05:40 -0700789 Mobility: stat.Mobility,
Emily Markova63c63f62023-03-29 20:57:35 -0700790 DockedAuto: stat.DockedAuto,
791 EngagedAuto: stat.EngagedAuto,
792 BalanceAttemptAuto: stat.BalanceAttemptAuto,
793 Docked: stat.Docked,
794 Engaged: stat.Engaged,
795 BalanceAttempt: stat.BalanceAttempt,
796 CollectedBy: stat.CollectedBy,
Emily Markova290147d2023-03-03 22:40:06 -0800797 })
798 }
799
800 builder := flatbuffers.NewBuilder(50 * 1024)
801 builder.Finish((&response).Pack(builder))
802 w.Write(builder.FinishedBytes())
803}
804
Emily Markova8e39f452023-12-23 12:17:30 -0800805type requestAllPitImagesHandler struct {
806 db Database
807}
808
809func (handler requestAllPitImagesHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
810 requestBytes, err := io.ReadAll(req.Body)
811 if err != nil {
812 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
813 return
814 }
815
816 _, success := parseRequest(w, requestBytes, "RequestAllPitImages", request_all_pit_images.GetRootAsRequestAllPitImages)
817 if !success {
818 return
819 }
820
821 images, err := handler.db.ReturnPitImages()
822 if err != nil {
823 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to get pit images: %v", err))
824 return
825 }
826
827 var response RequestAllPitImagesResponseT
828 for _, data := range images {
829 response.PitImageList = append(response.PitImageList, &request_all_pit_images_response.PitImageT{
830 TeamNumber: data.TeamNumber,
831 ImagePath: data.ImagePath,
832 CheckSum: data.CheckSum,
833 })
834 }
835
836 builder := flatbuffers.NewBuilder(1024)
837 builder.Finish((&response).Pack(builder))
838 w.Write(builder.FinishedBytes())
839}
840
Emily Markovafaecfe12023-07-01 12:40:03 -0700841type requestPitImagesHandler struct {
842 db Database
843}
844
845func (handler requestPitImagesHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
846 requestBytes, err := io.ReadAll(req.Body)
847 if err != nil {
848 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
849 return
850 }
851
852 request, success := parseRequest(w, requestBytes, "RequestPitImages", request_pit_images.GetRootAsRequestPitImages)
853 if !success {
854 return
855 }
856
857 images, err := handler.db.QueryPitImages(string(request.TeamNumber()))
858 if err != nil {
859 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to query pit images: %v", err))
860 return
861 }
862
863 var response RequestPitImagesResponseT
864 for _, data := range images {
865 response.PitImageList = append(response.PitImageList, &request_pit_images_response.PitImageT{
866 TeamNumber: data.TeamNumber,
867 ImagePath: data.ImagePath,
868 CheckSum: data.CheckSum,
869 })
870 }
871
872 builder := flatbuffers.NewBuilder(1024)
873 builder.Finish((&response).Pack(builder))
874 w.Write(builder.FinishedBytes())
875}
876
Alex Perry81f96ba2022-03-13 18:26:19 -0700877type requestNotesForTeamHandler struct {
878 db Database
879}
880
881func (handler requestNotesForTeamHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
882 requestBytes, err := io.ReadAll(req.Body)
883 if err != nil {
884 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
885 return
886 }
887
Philipp Schraderb7e75932022-03-26 16:18:34 -0700888 request, success := parseRequest(w, requestBytes, "RequestNotesForTeam", request_notes_for_team.GetRootAsRequestNotesForTeam)
Alex Perry81f96ba2022-03-13 18:26:19 -0700889 if !success {
890 return
891 }
892
Emily Markovae68b7632023-12-30 14:17:55 -0800893 notes, err := handler.db.QueryNotes(string(request.Team()))
Alex Perry81f96ba2022-03-13 18:26:19 -0700894 if err != nil {
895 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to query notes: %v", err))
896 return
897 }
898
899 var response RequestNotesForTeamResponseT
Philipp Schradereecb8962022-06-01 21:02:42 -0700900 for _, data := range notes {
Alex Perry81f96ba2022-03-13 18:26:19 -0700901 response.Notes = append(response.Notes, &request_notes_for_team_response.NoteT{data})
902 }
903
904 builder := flatbuffers.NewBuilder(1024)
905 builder.Finish((&response).Pack(builder))
906 w.Write(builder.FinishedBytes())
907}
908
Milo Lin1d59f0c2022-06-22 20:30:58 -0700909type requestShiftScheduleHandler struct {
910 db Database
911}
912
913func (handler requestShiftScheduleHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
914 requestBytes, err := io.ReadAll(req.Body)
915 if err != nil {
916 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
917 return
918 }
919
920 _, success := parseRequest(w, requestBytes, "RequestShiftSchedule", request_shift_schedule.GetRootAsRequestShiftSchedule)
921 if !success {
922 return
923 }
924
925 shiftData, err := handler.db.ReturnAllShifts()
926 if err != nil {
927 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to query shift schedule: %v", err))
928 return
929 }
930
931 var response RequestShiftScheduleResponseT
932 for _, shifts := range shiftData {
933 response.ShiftSchedule = append(response.ShiftSchedule, &request_shift_schedule_response.MatchAssignmentT{
934 MatchNumber: shifts.MatchNumber,
Philipp Schrader2ff455b2023-05-03 22:11:50 -0700935 R1Scouter: shifts.R1scouter,
936 R2Scouter: shifts.R2scouter,
937 R3Scouter: shifts.R3scouter,
938 B1Scouter: shifts.B1scouter,
939 B2Scouter: shifts.B2scouter,
940 B3Scouter: shifts.B3scouter,
Milo Lin1d59f0c2022-06-22 20:30:58 -0700941 })
942 }
943
944 builder := flatbuffers.NewBuilder(1024)
945 builder.Finish((&response).Pack(builder))
946 w.Write(builder.FinishedBytes())
947}
948
949type submitShiftScheduleHandler struct {
950 db Database
951}
952
953func (handler submitShiftScheduleHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
954 // Get the username of the person submitting the data.
955 username := parseUsername(req)
956
957 requestBytes, err := io.ReadAll(req.Body)
958 if err != nil {
959 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
960 return
961 }
962
963 request, success := parseRequest[SubmitShiftSchedule](w, requestBytes, "SubmitShiftSchedule", submit_shift_schedule.GetRootAsSubmitShiftSchedule)
964 if !success {
965 return
966 }
967
968 log.Println("Got shift schedule from", username)
969 shift_schedule_length := request.ShiftScheduleLength()
970 for i := 0; i < shift_schedule_length; i++ {
971 var match_assignment submit_shift_schedule.MatchAssignment
972 request.ShiftSchedule(&match_assignment, i)
973 current_shift := db.Shift{
974 MatchNumber: match_assignment.MatchNumber(),
Philipp Schrader2ff455b2023-05-03 22:11:50 -0700975 R1scouter: string(match_assignment.R1Scouter()),
976 R2scouter: string(match_assignment.R2Scouter()),
977 R3scouter: string(match_assignment.R3Scouter()),
978 B1scouter: string(match_assignment.B1Scouter()),
979 B2scouter: string(match_assignment.B2Scouter()),
980 B3scouter: string(match_assignment.B3Scouter()),
Milo Lin1d59f0c2022-06-22 20:30:58 -0700981 }
982 err = handler.db.AddToShift(current_shift)
983 if err != nil {
984 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to submit shift schedule: ", err))
985 return
986 }
987 }
988
989 builder := flatbuffers.NewBuilder(50 * 1024)
990 builder.Finish((&SubmitShiftScheduleResponseT{}).Pack(builder))
991 w.Write(builder.FinishedBytes())
992}
993
Filip Kujawa210a03b2022-11-24 14:41:11 -0800994type SubmitDriverRankingHandler struct {
995 db Database
996}
997
998func (handler SubmitDriverRankingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
999 requestBytes, err := io.ReadAll(req.Body)
1000 if err != nil {
1001 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
1002 return
1003 }
1004
1005 request, success := parseRequest(w, requestBytes, "SubmitDriverRanking", submit_driver_ranking.GetRootAsSubmitDriverRanking)
1006 if !success {
1007 return
1008 }
1009
1010 err = handler.db.AddDriverRanking(db.DriverRankingData{
1011 MatchNumber: request.MatchNumber(),
Emily Markovae68b7632023-12-30 14:17:55 -08001012 Rank1: string(request.Rank1()),
1013 Rank2: string(request.Rank2()),
1014 Rank3: string(request.Rank3()),
Filip Kujawa210a03b2022-11-24 14:41:11 -08001015 })
1016
1017 if err != nil {
1018 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to insert driver ranking: %v", err))
1019 return
1020 }
1021
1022 var response SubmitDriverRankingResponseT
1023 builder := flatbuffers.NewBuilder(10)
1024 builder.Finish((&response).Pack(builder))
1025 w.Write(builder.FinishedBytes())
1026}
1027
Filip Kujawaf882e022022-12-14 13:14:08 -08001028type requestAllNotesHandler struct {
1029 db Database
1030}
1031
1032func (handler requestAllNotesHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
1033 requestBytes, err := io.ReadAll(req.Body)
1034 if err != nil {
1035 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
1036 return
1037 }
1038
1039 _, success := parseRequest(w, requestBytes, "RequestAllNotes", request_all_notes.GetRootAsRequestAllNotes)
1040 if !success {
1041 return
1042 }
1043
1044 notes, err := handler.db.ReturnAllNotes()
1045 if err != nil {
1046 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to query database: ", err))
1047 return
1048 }
1049
1050 var response RequestAllNotesResponseT
1051 for _, note := range notes {
1052 response.NoteList = append(response.NoteList, &request_all_notes_response.NoteT{
Filip Kujawa7ddd5652023-03-07 19:56:15 -08001053 Team: note.TeamNumber,
1054 Notes: note.Notes,
1055 GoodDriving: note.GoodDriving,
1056 BadDriving: note.BadDriving,
Filip Kujawa11dc4c92023-04-13 08:55:43 -07001057 SolidPlacing: note.SolidPlacing,
Filip Kujawa7ddd5652023-03-07 19:56:15 -08001058 SketchyPlacing: note.SketchyPlacing,
1059 GoodDefense: note.GoodDefense,
1060 BadDefense: note.BadDefense,
1061 EasilyDefended: note.EasilyDefended,
Filip Kujawaf882e022022-12-14 13:14:08 -08001062 })
1063 }
1064
1065 builder := flatbuffers.NewBuilder(50 * 1024)
1066 builder.Finish((&response).Pack(builder))
1067 w.Write(builder.FinishedBytes())
1068}
1069
1070type requestAllDriverRankingsHandler struct {
1071 db Database
1072}
1073
1074func (handler requestAllDriverRankingsHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
1075 requestBytes, err := io.ReadAll(req.Body)
1076 if err != nil {
1077 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
1078 return
1079 }
1080
1081 _, success := parseRequest(w, requestBytes, "RequestAllDriverRankings", request_all_driver_rankings.GetRootAsRequestAllDriverRankings)
1082 if !success {
1083 return
1084 }
1085
1086 rankings, err := handler.db.ReturnAllDriverRankings()
1087 if err != nil {
1088 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to query database: ", err))
1089 return
1090 }
1091
1092 var response RequestAllDriverRankingsResponseT
1093 for _, ranking := range rankings {
1094 response.DriverRankingList = append(response.DriverRankingList, &request_all_driver_rankings_response.RankingT{
1095 MatchNumber: ranking.MatchNumber,
1096 Rank1: ranking.Rank1,
1097 Rank2: ranking.Rank2,
1098 Rank3: ranking.Rank3,
1099 })
1100 }
1101
1102 builder := flatbuffers.NewBuilder(50 * 1024)
1103 builder.Finish((&response).Pack(builder))
1104 w.Write(builder.FinishedBytes())
1105}
1106
Emily Markova8cb91312024-02-02 12:30:37 -08001107type submit2024ActionsHandler struct {
1108 db Database
1109}
1110
1111func (handler submit2024ActionsHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
1112 // Get the username of the person submitting the data.
1113 username := parseUsername(req)
1114
1115 requestBytes, err := io.ReadAll(req.Body)
1116 if err != nil {
1117 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
1118 return
1119 }
1120
1121 request, success := parseRequest(w, requestBytes, "Submit2024Actions", submit_2024_actions.GetRootAsSubmit2024Actions)
1122 if !success {
1123 return
1124 }
1125
1126 log.Println("Got actions for match", request.MatchNumber(), "team", request.TeamNumber(), "from", username)
1127
1128 for i := 0; i < request.ActionsListLength(); i++ {
1129
1130 var action Action2024
1131 request.ActionsList(&action, i)
1132
1133 dbAction := db.Action{
1134 PreScouting: request.PreScouting(),
1135 TeamNumber: string(request.TeamNumber()),
1136 MatchNumber: request.MatchNumber(),
1137 SetNumber: request.SetNumber(),
1138 CompLevel: string(request.CompLevel()),
1139 //TODO: Serialize CompletedAction
1140 CompletedAction: []byte{},
1141 Timestamp: action.Timestamp(),
1142 CollectedBy: username,
1143 }
1144
1145 // Do some error checking.
1146 if action.Timestamp() < 0 {
1147 respondWithError(w, http.StatusBadRequest, fmt.Sprint(
1148 "Invalid timestamp field value of ", action.Timestamp()))
1149 return
1150 }
1151
1152 err = handler.db.AddAction(dbAction)
1153 if err != nil {
1154 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to add action to database: ", err))
1155 return
1156 }
1157 }
1158
1159 stats, err := ConvertActionsToStat2024(request)
1160 if err != nil {
1161 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to convert actions to stats: ", err))
1162 return
1163 }
1164
1165 stats.CollectedBy = username
1166
1167 err = handler.db.AddToStats2024(stats)
1168 if err != nil {
Emily Markovadcadcb62024-02-03 13:07:17 -08001169 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to submit stats2024: ", stats, ": ", err))
Emily Markova8cb91312024-02-02 12:30:37 -08001170 return
1171 }
1172
1173 builder := flatbuffers.NewBuilder(50 * 1024)
Emily Markovadcadcb62024-02-03 13:07:17 -08001174 builder.Finish((&Submit2024ActionsResponseT{}).Pack(builder))
Emily Markova8cb91312024-02-02 12:30:37 -08001175 w.Write(builder.FinishedBytes())
1176}
1177
Sabina Leaver9b4eb312023-02-20 19:58:17 -08001178type submitActionsHandler struct {
1179 db Database
1180}
1181
1182func (handler submitActionsHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
1183 // Get the username of the person submitting the data.
1184 username := parseUsername(req)
1185
1186 requestBytes, err := io.ReadAll(req.Body)
1187 if err != nil {
1188 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
1189 return
1190 }
1191
1192 request, success := parseRequest(w, requestBytes, "SubmitActions", submit_actions.GetRootAsSubmitActions)
1193 if !success {
1194 return
1195 }
1196
1197 log.Println("Got actions for match", request.MatchNumber(), "team", request.TeamNumber(), "from", username)
1198
1199 for i := 0; i < request.ActionsListLength(); i++ {
1200
1201 var action Action
1202 request.ActionsList(&action, i)
1203
1204 dbAction := db.Action{
Philipp Schrader4b489222023-04-15 16:40:16 -07001205 PreScouting: request.PreScouting(),
Sabina Leaver9b4eb312023-02-20 19:58:17 -08001206 TeamNumber: string(request.TeamNumber()),
1207 MatchNumber: request.MatchNumber(),
1208 SetNumber: request.SetNumber(),
1209 CompLevel: string(request.CompLevel()),
1210 //TODO: Serialize CompletedAction
1211 CompletedAction: []byte{},
Philipp Schrader670a1c82023-05-17 19:42:43 -07001212 Timestamp: action.Timestamp(),
Sabina Leaver9b4eb312023-02-20 19:58:17 -08001213 CollectedBy: username,
1214 }
1215
1216 // Do some error checking.
1217 if action.Timestamp() < 0 {
1218 respondWithError(w, http.StatusBadRequest, fmt.Sprint(
1219 "Invalid timestamp field value of ", action.Timestamp()))
1220 return
1221 }
1222
1223 err = handler.db.AddAction(dbAction)
1224 if err != nil {
1225 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to add action to database: ", err))
1226 return
1227 }
1228 }
1229
Philipp Schradere11114f2023-04-15 17:04:25 -07001230 stats, err := ConvertActionsToStat(request)
1231 if err != nil {
1232 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to convert actions to stats: ", err))
1233 return
1234 }
1235
1236 stats.CollectedBy = username
1237
1238 err = handler.db.AddToStats2023(stats)
1239 if err != nil {
1240 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to submit stats: ", stats, ": ", err))
1241 return
1242 }
1243
Sabina Leaver9b4eb312023-02-20 19:58:17 -08001244 builder := flatbuffers.NewBuilder(50 * 1024)
1245 builder.Finish((&SubmitActionsResponseT{}).Pack(builder))
1246 w.Write(builder.FinishedBytes())
1247}
1248
Emily Markova8cb91312024-02-02 12:30:37 -08001249type Delete2024DataScoutingHandler struct {
1250 db Database
1251}
1252
1253func (handler Delete2024DataScoutingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
1254 requestBytes, err := io.ReadAll(req.Body)
1255 if err != nil {
1256 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
1257 return
1258 }
1259
1260 request, success := parseRequest(w, requestBytes, "Delete2024DataScouting", delete_2024_data_scouting.GetRootAsDelete2024DataScouting)
1261 if !success {
1262 return
1263 }
1264
1265 err = handler.db.DeleteFromStats2024(
1266 string(request.CompLevel()),
1267 request.MatchNumber(),
1268 request.SetNumber(),
1269 string(request.TeamNumber()))
1270
1271 if err != nil {
1272 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to delete from stats2024: %v", err))
1273 return
1274 }
1275
1276 err = handler.db.DeleteFromActions(
1277 string(request.CompLevel()),
1278 request.MatchNumber(),
1279 request.SetNumber(),
1280 string(request.TeamNumber()))
1281
1282 if err != nil {
1283 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to delete from actions: %v", err))
1284 return
1285 }
1286
1287 var response Delete2024DataScoutingResponseT
1288 builder := flatbuffers.NewBuilder(10)
1289 builder.Finish((&response).Pack(builder))
1290 w.Write(builder.FinishedBytes())
1291}
1292
Filip Kujawac1ded372023-05-27 14:33:43 -07001293type Delete2023DataScoutingHandler struct {
1294 db Database
1295}
1296
1297func (handler Delete2023DataScoutingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
1298 requestBytes, err := io.ReadAll(req.Body)
1299 if err != nil {
1300 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
1301 return
1302 }
1303
1304 request, success := parseRequest(w, requestBytes, "Delete2023DataScouting", delete_2023_data_scouting.GetRootAsDelete2023DataScouting)
1305 if !success {
1306 return
1307 }
1308
1309 err = handler.db.DeleteFromStats(
1310 string(request.CompLevel()),
1311 request.MatchNumber(),
1312 request.SetNumber(),
1313 string(request.TeamNumber()))
1314
1315 if err != nil {
1316 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to delete from stats: %v", err))
1317 return
1318 }
1319
1320 err = handler.db.DeleteFromActions(
1321 string(request.CompLevel()),
1322 request.MatchNumber(),
1323 request.SetNumber(),
1324 string(request.TeamNumber()))
1325
1326 if err != nil {
1327 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to delete from actions: %v", err))
1328 return
1329 }
1330
1331 var response Delete2023DataScoutingResponseT
1332 builder := flatbuffers.NewBuilder(10)
1333 builder.Finish((&response).Pack(builder))
1334 w.Write(builder.FinishedBytes())
1335}
1336
Philipp Schrader43c730b2023-02-26 20:27:44 -08001337func HandleRequests(db Database, scoutingServer server.ScoutingServer) {
Philipp Schradercdb5cfc2022-02-20 14:57:07 -08001338 scoutingServer.HandleFunc("/requests", unknown)
Philipp Schradercbf5c6a2022-02-27 23:25:19 -08001339 scoutingServer.Handle("/requests/request/all_matches", requestAllMatchesHandler{db})
Filip Kujawaf882e022022-12-14 13:14:08 -08001340 scoutingServer.Handle("/requests/request/all_notes", requestAllNotesHandler{db})
1341 scoutingServer.Handle("/requests/request/all_driver_rankings", requestAllDriverRankingsHandler{db})
Emily Markova290147d2023-03-03 22:40:06 -08001342 scoutingServer.Handle("/requests/request/2023_data_scouting", request2023DataScoutingHandler{db})
Emily Markova8cb91312024-02-02 12:30:37 -08001343 scoutingServer.Handle("/requests/request/2024_data_scouting", request2024DataScoutingHandler{db})
Alex Perry81f96ba2022-03-13 18:26:19 -07001344 scoutingServer.Handle("/requests/submit/submit_notes", submitNoteScoutingHandler{db})
Emily Markovafaecfe12023-07-01 12:40:03 -07001345 scoutingServer.Handle("/requests/submit/submit_pit_image", submitPitImageScoutingHandler{db})
1346 scoutingServer.Handle("/requests/request/pit_images", requestPitImagesHandler{db})
Emily Markova8e39f452023-12-23 12:17:30 -08001347 scoutingServer.Handle("/requests/request/all_pit_images", requestAllPitImagesHandler{db})
Alex Perry81f96ba2022-03-13 18:26:19 -07001348 scoutingServer.Handle("/requests/request/notes_for_team", requestNotesForTeamHandler{db})
Milo Lin1d59f0c2022-06-22 20:30:58 -07001349 scoutingServer.Handle("/requests/submit/shift_schedule", submitShiftScheduleHandler{db})
1350 scoutingServer.Handle("/requests/request/shift_schedule", requestShiftScheduleHandler{db})
Filip Kujawa210a03b2022-11-24 14:41:11 -08001351 scoutingServer.Handle("/requests/submit/submit_driver_ranking", SubmitDriverRankingHandler{db})
Sabina Leaver9b4eb312023-02-20 19:58:17 -08001352 scoutingServer.Handle("/requests/submit/submit_actions", submitActionsHandler{db})
Emily Markova8cb91312024-02-02 12:30:37 -08001353 scoutingServer.Handle("/requests/submit/submit_2024_actions", submit2024ActionsHandler{db})
Filip Kujawac1ded372023-05-27 14:33:43 -07001354 scoutingServer.Handle("/requests/delete/delete_2023_data_scouting", Delete2023DataScoutingHandler{db})
Emily Markova8cb91312024-02-02 12:30:37 -08001355 scoutingServer.Handle("/requests/delete/delete_2024_data_scouting", Delete2024DataScoutingHandler{db})
Philipp Schradercdb5cfc2022-02-20 14:57:07 -08001356}