blob: 31ad4e390fa9e2c8162be4c2992a487f8415f31b [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,
455 TrapNote: false, AvgCycle: 0, Park: false, OnStage: false, Harmony: false, CollectedBy: "",
456 }
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()
540 }
541 }
542 if cycles != 0 {
543 stat.AvgCycle = overall_time / cycles
544 } else {
545 stat.AvgCycle = 0
546 }
547 return stat, nil
548}
549
550// Handles a Request2024DataScouting request.
551type request2024DataScoutingHandler struct {
552 db Database
553}
554
555func (handler request2024DataScoutingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
556 requestBytes, err := io.ReadAll(req.Body)
557 if err != nil {
558 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
559 return
560 }
561
562 _, success := parseRequest(w, requestBytes, "Request2024DataScouting", request_2024_data_scouting.GetRootAsRequest2024DataScouting)
563 if !success {
564 return
565 }
566
567 stats, err := handler.db.ReturnStats2024()
568 if err != nil {
569 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to query database: ", err))
570 return
571 }
572
573 var response Request2024DataScoutingResponseT
574 for _, stat := range stats {
575 response.StatsList = append(response.StatsList, &request_2024_data_scouting_response.Stats2024T{
576 TeamNumber: stat.TeamNumber,
577 MatchNumber: stat.MatchNumber,
578 SetNumber: stat.SetNumber,
579 CompLevel: stat.CompLevel,
580 StartingQuadrant: stat.StartingQuadrant,
581 SpeakerAuto: stat.SpeakerAuto,
582 AmpAuto: stat.AmpAuto,
583 NotesDroppedAuto: stat.NotesDroppedAuto,
584 MobilityAuto: stat.MobilityAuto,
585 Speaker: stat.Speaker,
586 Amp: stat.Amp,
587 SpeakerAmplified: stat.SpeakerAmplified,
588 AmpAmplified: stat.AmpAmplified,
589 NotesDropped: stat.NotesDropped,
590 Penalties: stat.Penalties,
591 TrapNote: stat.TrapNote,
592 AvgCycle: stat.AvgCycle,
593 Park: stat.Park,
594 OnStage: stat.OnStage,
595 Harmony: stat.Harmony,
596 CollectedBy: stat.CollectedBy,
597 })
598 }
599
600 builder := flatbuffers.NewBuilder(50 * 1024)
601 builder.Finish((&response).Pack(builder))
602 w.Write(builder.FinishedBytes())
603}
604
Emily Markova1abe9782023-03-11 19:45:38 -0800605func ConvertActionsToStat(submitActions *submit_actions.SubmitActions) (db.Stats2023, error) {
606 overall_time := int64(0)
607 cycles := int64(0)
608 picked_up := false
609 lastPlacedTime := int64(0)
Philipp Schrader4b489222023-04-15 16:40:16 -0700610 stat := db.Stats2023{
611 PreScouting: submitActions.PreScouting(),
612 TeamNumber: string(submitActions.TeamNumber()), MatchNumber: submitActions.MatchNumber(), SetNumber: submitActions.SetNumber(), CompLevel: string(submitActions.CompLevel()),
Emily Markova1abe9782023-03-11 19:45:38 -0800613 StartingQuadrant: 0, LowCubesAuto: 0, MiddleCubesAuto: 0, HighCubesAuto: 0, CubesDroppedAuto: 0,
614 LowConesAuto: 0, MiddleConesAuto: 0, HighConesAuto: 0, ConesDroppedAuto: 0, LowCubes: 0, MiddleCubes: 0, HighCubes: 0,
Philipp Schradere11114f2023-04-15 17:04:25 -0700615 CubesDropped: 0, LowCones: 0, MiddleCones: 0, HighCones: 0, ConesDropped: 0, SuperchargedPieces: 0, AvgCycle: 0, CollectedBy: "",
Emily Markova1abe9782023-03-11 19:45:38 -0800616 }
617 // Loop over all actions.
618 for i := 0; i < submitActions.ActionsListLength(); i++ {
619 var action submit_actions.Action
620 if !submitActions.ActionsList(&action, i) {
621 return db.Stats2023{}, errors.New(fmt.Sprintf("Failed to parse submit_actions.Action"))
622 }
623 actionTable := new(flatbuffers.Table)
624 action_type := action.ActionTakenType()
625 if !action.ActionTaken(actionTable) {
626 return db.Stats2023{}, errors.New(fmt.Sprint("Failed to parse sub-action or sub-action was missing"))
627 }
628 if action_type == submit_actions.ActionTypeStartMatchAction {
629 var startMatchAction submit_actions.StartMatchAction
630 startMatchAction.Init(actionTable.Bytes, actionTable.Pos)
631 stat.StartingQuadrant = startMatchAction.Position()
Filip Kujawa0b4b1e52023-04-15 14:05:40 -0700632 } else if action_type == submit_actions.ActionTypeMobilityAction {
633 var mobilityAction submit_actions.MobilityAction
634 mobilityAction.Init(actionTable.Bytes, actionTable.Pos)
635 if mobilityAction.Mobility() {
636 stat.Mobility = true
637 }
638
Emily Markova46a69bf2023-03-22 20:45:52 -0700639 } else if action_type == submit_actions.ActionTypeAutoBalanceAction {
640 var autoBalanceAction submit_actions.AutoBalanceAction
641 autoBalanceAction.Init(actionTable.Bytes, actionTable.Pos)
642 if autoBalanceAction.Docked() {
643 stat.DockedAuto = true
644 }
645 if autoBalanceAction.Engaged() {
646 stat.EngagedAuto = true
647 }
Emily Markova63c63f62023-03-29 20:57:35 -0700648 if autoBalanceAction.BalanceAttempt() {
649 stat.BalanceAttemptAuto = true
650 }
Emily Markova1abe9782023-03-11 19:45:38 -0800651 } else if action_type == submit_actions.ActionTypePickupObjectAction {
652 var pick_up_action submit_actions.PickupObjectAction
653 pick_up_action.Init(actionTable.Bytes, actionTable.Pos)
654 if picked_up == true {
655 object := pick_up_action.ObjectType().String()
656 auto := pick_up_action.Auto()
657 if object == "kCube" && auto == false {
658 stat.CubesDropped += 1
659 } else if object == "kCube" && auto == true {
660 stat.CubesDroppedAuto += 1
661 } else if object == "kCone" && auto == false {
662 stat.ConesDropped += 1
663 } else if object == "kCube" && auto == true {
664 stat.ConesDroppedAuto += 1
665 }
666 } else {
667 picked_up = true
668 }
669 } else if action_type == submit_actions.ActionTypePlaceObjectAction {
670 var place_action submit_actions.PlaceObjectAction
671 place_action.Init(actionTable.Bytes, actionTable.Pos)
672 if !picked_up {
673 return db.Stats2023{}, errors.New(fmt.Sprintf("Got PlaceObjectAction without corresponding PickupObjectAction"))
674 }
675 object := place_action.ObjectType()
676 level := place_action.ScoreLevel()
677 auto := place_action.Auto()
678 if object == 0 && level == 0 && auto == true {
679 stat.LowCubesAuto += 1
680 } else if object == 0 && level == 0 && auto == false {
681 stat.LowCubes += 1
682 } else if object == 0 && level == 1 && auto == true {
683 stat.MiddleCubesAuto += 1
684 } else if object == 0 && level == 1 && auto == false {
685 stat.MiddleCubes += 1
686 } else if object == 0 && level == 2 && auto == true {
687 stat.HighCubesAuto += 1
688 } else if object == 0 && level == 2 && auto == false {
689 stat.HighCubes += 1
690 } else if object == 1 && level == 0 && auto == true {
691 stat.LowConesAuto += 1
692 } else if object == 1 && level == 0 && auto == false {
693 stat.LowCones += 1
694 } else if object == 1 && level == 1 && auto == true {
695 stat.MiddleConesAuto += 1
696 } else if object == 1 && level == 1 && auto == false {
697 stat.MiddleCones += 1
698 } else if object == 1 && level == 2 && auto == true {
699 stat.HighConesAuto += 1
700 } else if object == 1 && level == 2 && auto == false {
701 stat.HighCones += 1
Filip Kujawa7a045e72023-04-13 08:41:09 -0700702 } else if level == 3 {
703 stat.SuperchargedPieces += 1
Emily Markova1abe9782023-03-11 19:45:38 -0800704 } else {
705 return db.Stats2023{}, errors.New(fmt.Sprintf("Got unknown ObjectType/ScoreLevel/Auto combination"))
706 }
707 picked_up = false
708 if lastPlacedTime != int64(0) {
709 // If this is not the first time we place,
710 // start counting cycle time. We define cycle
711 // time as the time between placements.
712 overall_time += int64(action.Timestamp()) - lastPlacedTime
713 cycles += 1
714 }
715 lastPlacedTime = int64(action.Timestamp())
Emily Markova46a69bf2023-03-22 20:45:52 -0700716 } else if action_type == submit_actions.ActionTypeEndMatchAction {
717 var endMatchAction submit_actions.EndMatchAction
718 endMatchAction.Init(actionTable.Bytes, actionTable.Pos)
719 if endMatchAction.Docked() {
720 stat.Docked = true
721 }
722 if endMatchAction.Engaged() {
723 stat.Engaged = true
724 }
Emily Markova63c63f62023-03-29 20:57:35 -0700725 if endMatchAction.BalanceAttempt() {
726 stat.BalanceAttempt = true
727 }
Emily Markova1abe9782023-03-11 19:45:38 -0800728 }
729 }
730 if cycles != 0 {
Philipp Schrader8c878a22023-03-20 22:36:38 -0700731 stat.AvgCycle = overall_time / cycles
Emily Markova1abe9782023-03-11 19:45:38 -0800732 } else {
733 stat.AvgCycle = 0
734 }
735 return stat, nil
736}
737
Emily Markova290147d2023-03-03 22:40:06 -0800738// Handles a Request2023DataScouting request.
739type request2023DataScoutingHandler struct {
740 db Database
741}
742
743func (handler request2023DataScoutingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
744 requestBytes, err := io.ReadAll(req.Body)
745 if err != nil {
746 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
747 return
748 }
749
750 _, success := parseRequest(w, requestBytes, "Request2023DataScouting", request_2023_data_scouting.GetRootAsRequest2023DataScouting)
751 if !success {
752 return
753 }
754
755 stats, err := handler.db.ReturnStats2023()
756 if err != nil {
757 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to query database: ", err))
758 return
759 }
760
761 var response Request2023DataScoutingResponseT
762 for _, stat := range stats {
763 response.StatsList = append(response.StatsList, &request_2023_data_scouting_response.Stats2023T{
Emily Markova63c63f62023-03-29 20:57:35 -0700764 TeamNumber: stat.TeamNumber,
765 MatchNumber: stat.MatchNumber,
766 SetNumber: stat.SetNumber,
767 CompLevel: stat.CompLevel,
768 StartingQuadrant: stat.StartingQuadrant,
769 LowCubesAuto: stat.LowCubesAuto,
770 MiddleCubesAuto: stat.MiddleCubesAuto,
771 HighCubesAuto: stat.HighCubesAuto,
772 CubesDroppedAuto: stat.CubesDroppedAuto,
773 LowConesAuto: stat.LowConesAuto,
774 MiddleConesAuto: stat.MiddleConesAuto,
775 HighConesAuto: stat.HighConesAuto,
776 ConesDroppedAuto: stat.ConesDroppedAuto,
777 LowCubes: stat.LowCubes,
778 MiddleCubes: stat.MiddleCubes,
779 HighCubes: stat.HighCubes,
780 CubesDropped: stat.CubesDropped,
781 LowCones: stat.LowCones,
782 MiddleCones: stat.MiddleCones,
783 HighCones: stat.HighCones,
784 ConesDropped: stat.ConesDropped,
Filip Kujawa7a045e72023-04-13 08:41:09 -0700785 SuperchargedPieces: stat.SuperchargedPieces,
Emily Markova63c63f62023-03-29 20:57:35 -0700786 AvgCycle: stat.AvgCycle,
Filip Kujawa0b4b1e52023-04-15 14:05:40 -0700787 Mobility: stat.Mobility,
Emily Markova63c63f62023-03-29 20:57:35 -0700788 DockedAuto: stat.DockedAuto,
789 EngagedAuto: stat.EngagedAuto,
790 BalanceAttemptAuto: stat.BalanceAttemptAuto,
791 Docked: stat.Docked,
792 Engaged: stat.Engaged,
793 BalanceAttempt: stat.BalanceAttempt,
794 CollectedBy: stat.CollectedBy,
Emily Markova290147d2023-03-03 22:40:06 -0800795 })
796 }
797
798 builder := flatbuffers.NewBuilder(50 * 1024)
799 builder.Finish((&response).Pack(builder))
800 w.Write(builder.FinishedBytes())
801}
802
Emily Markova8e39f452023-12-23 12:17:30 -0800803type requestAllPitImagesHandler struct {
804 db Database
805}
806
807func (handler requestAllPitImagesHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
808 requestBytes, err := io.ReadAll(req.Body)
809 if err != nil {
810 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
811 return
812 }
813
814 _, success := parseRequest(w, requestBytes, "RequestAllPitImages", request_all_pit_images.GetRootAsRequestAllPitImages)
815 if !success {
816 return
817 }
818
819 images, err := handler.db.ReturnPitImages()
820 if err != nil {
821 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to get pit images: %v", err))
822 return
823 }
824
825 var response RequestAllPitImagesResponseT
826 for _, data := range images {
827 response.PitImageList = append(response.PitImageList, &request_all_pit_images_response.PitImageT{
828 TeamNumber: data.TeamNumber,
829 ImagePath: data.ImagePath,
830 CheckSum: data.CheckSum,
831 })
832 }
833
834 builder := flatbuffers.NewBuilder(1024)
835 builder.Finish((&response).Pack(builder))
836 w.Write(builder.FinishedBytes())
837}
838
Emily Markovafaecfe12023-07-01 12:40:03 -0700839type requestPitImagesHandler struct {
840 db Database
841}
842
843func (handler requestPitImagesHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
844 requestBytes, err := io.ReadAll(req.Body)
845 if err != nil {
846 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
847 return
848 }
849
850 request, success := parseRequest(w, requestBytes, "RequestPitImages", request_pit_images.GetRootAsRequestPitImages)
851 if !success {
852 return
853 }
854
855 images, err := handler.db.QueryPitImages(string(request.TeamNumber()))
856 if err != nil {
857 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to query pit images: %v", err))
858 return
859 }
860
861 var response RequestPitImagesResponseT
862 for _, data := range images {
863 response.PitImageList = append(response.PitImageList, &request_pit_images_response.PitImageT{
864 TeamNumber: data.TeamNumber,
865 ImagePath: data.ImagePath,
866 CheckSum: data.CheckSum,
867 })
868 }
869
870 builder := flatbuffers.NewBuilder(1024)
871 builder.Finish((&response).Pack(builder))
872 w.Write(builder.FinishedBytes())
873}
874
Alex Perry81f96ba2022-03-13 18:26:19 -0700875type requestNotesForTeamHandler struct {
876 db Database
877}
878
879func (handler requestNotesForTeamHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
880 requestBytes, err := io.ReadAll(req.Body)
881 if err != nil {
882 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
883 return
884 }
885
Philipp Schraderb7e75932022-03-26 16:18:34 -0700886 request, success := parseRequest(w, requestBytes, "RequestNotesForTeam", request_notes_for_team.GetRootAsRequestNotesForTeam)
Alex Perry81f96ba2022-03-13 18:26:19 -0700887 if !success {
888 return
889 }
890
Emily Markovae68b7632023-12-30 14:17:55 -0800891 notes, err := handler.db.QueryNotes(string(request.Team()))
Alex Perry81f96ba2022-03-13 18:26:19 -0700892 if err != nil {
893 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to query notes: %v", err))
894 return
895 }
896
897 var response RequestNotesForTeamResponseT
Philipp Schradereecb8962022-06-01 21:02:42 -0700898 for _, data := range notes {
Alex Perry81f96ba2022-03-13 18:26:19 -0700899 response.Notes = append(response.Notes, &request_notes_for_team_response.NoteT{data})
900 }
901
902 builder := flatbuffers.NewBuilder(1024)
903 builder.Finish((&response).Pack(builder))
904 w.Write(builder.FinishedBytes())
905}
906
Milo Lin1d59f0c2022-06-22 20:30:58 -0700907type requestShiftScheduleHandler struct {
908 db Database
909}
910
911func (handler requestShiftScheduleHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
912 requestBytes, err := io.ReadAll(req.Body)
913 if err != nil {
914 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
915 return
916 }
917
918 _, success := parseRequest(w, requestBytes, "RequestShiftSchedule", request_shift_schedule.GetRootAsRequestShiftSchedule)
919 if !success {
920 return
921 }
922
923 shiftData, err := handler.db.ReturnAllShifts()
924 if err != nil {
925 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to query shift schedule: %v", err))
926 return
927 }
928
929 var response RequestShiftScheduleResponseT
930 for _, shifts := range shiftData {
931 response.ShiftSchedule = append(response.ShiftSchedule, &request_shift_schedule_response.MatchAssignmentT{
932 MatchNumber: shifts.MatchNumber,
Philipp Schrader2ff455b2023-05-03 22:11:50 -0700933 R1Scouter: shifts.R1scouter,
934 R2Scouter: shifts.R2scouter,
935 R3Scouter: shifts.R3scouter,
936 B1Scouter: shifts.B1scouter,
937 B2Scouter: shifts.B2scouter,
938 B3Scouter: shifts.B3scouter,
Milo Lin1d59f0c2022-06-22 20:30:58 -0700939 })
940 }
941
942 builder := flatbuffers.NewBuilder(1024)
943 builder.Finish((&response).Pack(builder))
944 w.Write(builder.FinishedBytes())
945}
946
947type submitShiftScheduleHandler struct {
948 db Database
949}
950
951func (handler submitShiftScheduleHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
952 // Get the username of the person submitting the data.
953 username := parseUsername(req)
954
955 requestBytes, err := io.ReadAll(req.Body)
956 if err != nil {
957 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
958 return
959 }
960
961 request, success := parseRequest[SubmitShiftSchedule](w, requestBytes, "SubmitShiftSchedule", submit_shift_schedule.GetRootAsSubmitShiftSchedule)
962 if !success {
963 return
964 }
965
966 log.Println("Got shift schedule from", username)
967 shift_schedule_length := request.ShiftScheduleLength()
968 for i := 0; i < shift_schedule_length; i++ {
969 var match_assignment submit_shift_schedule.MatchAssignment
970 request.ShiftSchedule(&match_assignment, i)
971 current_shift := db.Shift{
972 MatchNumber: match_assignment.MatchNumber(),
Philipp Schrader2ff455b2023-05-03 22:11:50 -0700973 R1scouter: string(match_assignment.R1Scouter()),
974 R2scouter: string(match_assignment.R2Scouter()),
975 R3scouter: string(match_assignment.R3Scouter()),
976 B1scouter: string(match_assignment.B1Scouter()),
977 B2scouter: string(match_assignment.B2Scouter()),
978 B3scouter: string(match_assignment.B3Scouter()),
Milo Lin1d59f0c2022-06-22 20:30:58 -0700979 }
980 err = handler.db.AddToShift(current_shift)
981 if err != nil {
982 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to submit shift schedule: ", err))
983 return
984 }
985 }
986
987 builder := flatbuffers.NewBuilder(50 * 1024)
988 builder.Finish((&SubmitShiftScheduleResponseT{}).Pack(builder))
989 w.Write(builder.FinishedBytes())
990}
991
Filip Kujawa210a03b2022-11-24 14:41:11 -0800992type SubmitDriverRankingHandler struct {
993 db Database
994}
995
996func (handler SubmitDriverRankingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
997 requestBytes, err := io.ReadAll(req.Body)
998 if err != nil {
999 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
1000 return
1001 }
1002
1003 request, success := parseRequest(w, requestBytes, "SubmitDriverRanking", submit_driver_ranking.GetRootAsSubmitDriverRanking)
1004 if !success {
1005 return
1006 }
1007
1008 err = handler.db.AddDriverRanking(db.DriverRankingData{
1009 MatchNumber: request.MatchNumber(),
Emily Markovae68b7632023-12-30 14:17:55 -08001010 Rank1: string(request.Rank1()),
1011 Rank2: string(request.Rank2()),
1012 Rank3: string(request.Rank3()),
Filip Kujawa210a03b2022-11-24 14:41:11 -08001013 })
1014
1015 if err != nil {
1016 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to insert driver ranking: %v", err))
1017 return
1018 }
1019
1020 var response SubmitDriverRankingResponseT
1021 builder := flatbuffers.NewBuilder(10)
1022 builder.Finish((&response).Pack(builder))
1023 w.Write(builder.FinishedBytes())
1024}
1025
Filip Kujawaf882e022022-12-14 13:14:08 -08001026type requestAllNotesHandler struct {
1027 db Database
1028}
1029
1030func (handler requestAllNotesHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
1031 requestBytes, err := io.ReadAll(req.Body)
1032 if err != nil {
1033 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
1034 return
1035 }
1036
1037 _, success := parseRequest(w, requestBytes, "RequestAllNotes", request_all_notes.GetRootAsRequestAllNotes)
1038 if !success {
1039 return
1040 }
1041
1042 notes, err := handler.db.ReturnAllNotes()
1043 if err != nil {
1044 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to query database: ", err))
1045 return
1046 }
1047
1048 var response RequestAllNotesResponseT
1049 for _, note := range notes {
1050 response.NoteList = append(response.NoteList, &request_all_notes_response.NoteT{
Filip Kujawa7ddd5652023-03-07 19:56:15 -08001051 Team: note.TeamNumber,
1052 Notes: note.Notes,
1053 GoodDriving: note.GoodDriving,
1054 BadDriving: note.BadDriving,
Filip Kujawa11dc4c92023-04-13 08:55:43 -07001055 SolidPlacing: note.SolidPlacing,
Filip Kujawa7ddd5652023-03-07 19:56:15 -08001056 SketchyPlacing: note.SketchyPlacing,
1057 GoodDefense: note.GoodDefense,
1058 BadDefense: note.BadDefense,
1059 EasilyDefended: note.EasilyDefended,
Filip Kujawaf882e022022-12-14 13:14:08 -08001060 })
1061 }
1062
1063 builder := flatbuffers.NewBuilder(50 * 1024)
1064 builder.Finish((&response).Pack(builder))
1065 w.Write(builder.FinishedBytes())
1066}
1067
1068type requestAllDriverRankingsHandler struct {
1069 db Database
1070}
1071
1072func (handler requestAllDriverRankingsHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
1073 requestBytes, err := io.ReadAll(req.Body)
1074 if err != nil {
1075 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
1076 return
1077 }
1078
1079 _, success := parseRequest(w, requestBytes, "RequestAllDriverRankings", request_all_driver_rankings.GetRootAsRequestAllDriverRankings)
1080 if !success {
1081 return
1082 }
1083
1084 rankings, err := handler.db.ReturnAllDriverRankings()
1085 if err != nil {
1086 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to query database: ", err))
1087 return
1088 }
1089
1090 var response RequestAllDriverRankingsResponseT
1091 for _, ranking := range rankings {
1092 response.DriverRankingList = append(response.DriverRankingList, &request_all_driver_rankings_response.RankingT{
1093 MatchNumber: ranking.MatchNumber,
1094 Rank1: ranking.Rank1,
1095 Rank2: ranking.Rank2,
1096 Rank3: ranking.Rank3,
1097 })
1098 }
1099
1100 builder := flatbuffers.NewBuilder(50 * 1024)
1101 builder.Finish((&response).Pack(builder))
1102 w.Write(builder.FinishedBytes())
1103}
1104
Emily Markova8cb91312024-02-02 12:30:37 -08001105type submit2024ActionsHandler struct {
1106 db Database
1107}
1108
1109func (handler submit2024ActionsHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
1110 // Get the username of the person submitting the data.
1111 username := parseUsername(req)
1112
1113 requestBytes, err := io.ReadAll(req.Body)
1114 if err != nil {
1115 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
1116 return
1117 }
1118
1119 request, success := parseRequest(w, requestBytes, "Submit2024Actions", submit_2024_actions.GetRootAsSubmit2024Actions)
1120 if !success {
1121 return
1122 }
1123
1124 log.Println("Got actions for match", request.MatchNumber(), "team", request.TeamNumber(), "from", username)
1125
1126 for i := 0; i < request.ActionsListLength(); i++ {
1127
1128 var action Action2024
1129 request.ActionsList(&action, i)
1130
1131 dbAction := db.Action{
1132 PreScouting: request.PreScouting(),
1133 TeamNumber: string(request.TeamNumber()),
1134 MatchNumber: request.MatchNumber(),
1135 SetNumber: request.SetNumber(),
1136 CompLevel: string(request.CompLevel()),
1137 //TODO: Serialize CompletedAction
1138 CompletedAction: []byte{},
1139 Timestamp: action.Timestamp(),
1140 CollectedBy: username,
1141 }
1142
1143 // Do some error checking.
1144 if action.Timestamp() < 0 {
1145 respondWithError(w, http.StatusBadRequest, fmt.Sprint(
1146 "Invalid timestamp field value of ", action.Timestamp()))
1147 return
1148 }
1149
1150 err = handler.db.AddAction(dbAction)
1151 if err != nil {
1152 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to add action to database: ", err))
1153 return
1154 }
1155 }
1156
1157 stats, err := ConvertActionsToStat2024(request)
1158 if err != nil {
1159 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to convert actions to stats: ", err))
1160 return
1161 }
1162
1163 stats.CollectedBy = username
1164
1165 err = handler.db.AddToStats2024(stats)
1166 if err != nil {
Emily Markovadcadcb62024-02-03 13:07:17 -08001167 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to submit stats2024: ", stats, ": ", err))
Emily Markova8cb91312024-02-02 12:30:37 -08001168 return
1169 }
1170
1171 builder := flatbuffers.NewBuilder(50 * 1024)
Emily Markovadcadcb62024-02-03 13:07:17 -08001172 builder.Finish((&Submit2024ActionsResponseT{}).Pack(builder))
Emily Markova8cb91312024-02-02 12:30:37 -08001173 w.Write(builder.FinishedBytes())
1174}
1175
Sabina Leaver9b4eb312023-02-20 19:58:17 -08001176type submitActionsHandler struct {
1177 db Database
1178}
1179
1180func (handler submitActionsHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
1181 // Get the username of the person submitting the data.
1182 username := parseUsername(req)
1183
1184 requestBytes, err := io.ReadAll(req.Body)
1185 if err != nil {
1186 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
1187 return
1188 }
1189
1190 request, success := parseRequest(w, requestBytes, "SubmitActions", submit_actions.GetRootAsSubmitActions)
1191 if !success {
1192 return
1193 }
1194
1195 log.Println("Got actions for match", request.MatchNumber(), "team", request.TeamNumber(), "from", username)
1196
1197 for i := 0; i < request.ActionsListLength(); i++ {
1198
1199 var action Action
1200 request.ActionsList(&action, i)
1201
1202 dbAction := db.Action{
Philipp Schrader4b489222023-04-15 16:40:16 -07001203 PreScouting: request.PreScouting(),
Sabina Leaver9b4eb312023-02-20 19:58:17 -08001204 TeamNumber: string(request.TeamNumber()),
1205 MatchNumber: request.MatchNumber(),
1206 SetNumber: request.SetNumber(),
1207 CompLevel: string(request.CompLevel()),
1208 //TODO: Serialize CompletedAction
1209 CompletedAction: []byte{},
Philipp Schrader670a1c82023-05-17 19:42:43 -07001210 Timestamp: action.Timestamp(),
Sabina Leaver9b4eb312023-02-20 19:58:17 -08001211 CollectedBy: username,
1212 }
1213
1214 // Do some error checking.
1215 if action.Timestamp() < 0 {
1216 respondWithError(w, http.StatusBadRequest, fmt.Sprint(
1217 "Invalid timestamp field value of ", action.Timestamp()))
1218 return
1219 }
1220
1221 err = handler.db.AddAction(dbAction)
1222 if err != nil {
1223 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to add action to database: ", err))
1224 return
1225 }
1226 }
1227
Philipp Schradere11114f2023-04-15 17:04:25 -07001228 stats, err := ConvertActionsToStat(request)
1229 if err != nil {
1230 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to convert actions to stats: ", err))
1231 return
1232 }
1233
1234 stats.CollectedBy = username
1235
1236 err = handler.db.AddToStats2023(stats)
1237 if err != nil {
1238 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to submit stats: ", stats, ": ", err))
1239 return
1240 }
1241
Sabina Leaver9b4eb312023-02-20 19:58:17 -08001242 builder := flatbuffers.NewBuilder(50 * 1024)
1243 builder.Finish((&SubmitActionsResponseT{}).Pack(builder))
1244 w.Write(builder.FinishedBytes())
1245}
1246
Emily Markova8cb91312024-02-02 12:30:37 -08001247type Delete2024DataScoutingHandler struct {
1248 db Database
1249}
1250
1251func (handler Delete2024DataScoutingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
1252 requestBytes, err := io.ReadAll(req.Body)
1253 if err != nil {
1254 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
1255 return
1256 }
1257
1258 request, success := parseRequest(w, requestBytes, "Delete2024DataScouting", delete_2024_data_scouting.GetRootAsDelete2024DataScouting)
1259 if !success {
1260 return
1261 }
1262
1263 err = handler.db.DeleteFromStats2024(
1264 string(request.CompLevel()),
1265 request.MatchNumber(),
1266 request.SetNumber(),
1267 string(request.TeamNumber()))
1268
1269 if err != nil {
1270 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to delete from stats2024: %v", err))
1271 return
1272 }
1273
1274 err = handler.db.DeleteFromActions(
1275 string(request.CompLevel()),
1276 request.MatchNumber(),
1277 request.SetNumber(),
1278 string(request.TeamNumber()))
1279
1280 if err != nil {
1281 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to delete from actions: %v", err))
1282 return
1283 }
1284
1285 var response Delete2024DataScoutingResponseT
1286 builder := flatbuffers.NewBuilder(10)
1287 builder.Finish((&response).Pack(builder))
1288 w.Write(builder.FinishedBytes())
1289}
1290
Filip Kujawac1ded372023-05-27 14:33:43 -07001291type Delete2023DataScoutingHandler struct {
1292 db Database
1293}
1294
1295func (handler Delete2023DataScoutingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
1296 requestBytes, err := io.ReadAll(req.Body)
1297 if err != nil {
1298 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
1299 return
1300 }
1301
1302 request, success := parseRequest(w, requestBytes, "Delete2023DataScouting", delete_2023_data_scouting.GetRootAsDelete2023DataScouting)
1303 if !success {
1304 return
1305 }
1306
1307 err = handler.db.DeleteFromStats(
1308 string(request.CompLevel()),
1309 request.MatchNumber(),
1310 request.SetNumber(),
1311 string(request.TeamNumber()))
1312
1313 if err != nil {
1314 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to delete from stats: %v", err))
1315 return
1316 }
1317
1318 err = handler.db.DeleteFromActions(
1319 string(request.CompLevel()),
1320 request.MatchNumber(),
1321 request.SetNumber(),
1322 string(request.TeamNumber()))
1323
1324 if err != nil {
1325 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to delete from actions: %v", err))
1326 return
1327 }
1328
1329 var response Delete2023DataScoutingResponseT
1330 builder := flatbuffers.NewBuilder(10)
1331 builder.Finish((&response).Pack(builder))
1332 w.Write(builder.FinishedBytes())
1333}
1334
Philipp Schrader43c730b2023-02-26 20:27:44 -08001335func HandleRequests(db Database, scoutingServer server.ScoutingServer) {
Philipp Schradercdb5cfc2022-02-20 14:57:07 -08001336 scoutingServer.HandleFunc("/requests", unknown)
Philipp Schradercbf5c6a2022-02-27 23:25:19 -08001337 scoutingServer.Handle("/requests/request/all_matches", requestAllMatchesHandler{db})
Filip Kujawaf882e022022-12-14 13:14:08 -08001338 scoutingServer.Handle("/requests/request/all_notes", requestAllNotesHandler{db})
1339 scoutingServer.Handle("/requests/request/all_driver_rankings", requestAllDriverRankingsHandler{db})
Emily Markova290147d2023-03-03 22:40:06 -08001340 scoutingServer.Handle("/requests/request/2023_data_scouting", request2023DataScoutingHandler{db})
Emily Markova8cb91312024-02-02 12:30:37 -08001341 scoutingServer.Handle("/requests/request/2024_data_scouting", request2024DataScoutingHandler{db})
Alex Perry81f96ba2022-03-13 18:26:19 -07001342 scoutingServer.Handle("/requests/submit/submit_notes", submitNoteScoutingHandler{db})
Emily Markovafaecfe12023-07-01 12:40:03 -07001343 scoutingServer.Handle("/requests/submit/submit_pit_image", submitPitImageScoutingHandler{db})
1344 scoutingServer.Handle("/requests/request/pit_images", requestPitImagesHandler{db})
Emily Markova8e39f452023-12-23 12:17:30 -08001345 scoutingServer.Handle("/requests/request/all_pit_images", requestAllPitImagesHandler{db})
Alex Perry81f96ba2022-03-13 18:26:19 -07001346 scoutingServer.Handle("/requests/request/notes_for_team", requestNotesForTeamHandler{db})
Milo Lin1d59f0c2022-06-22 20:30:58 -07001347 scoutingServer.Handle("/requests/submit/shift_schedule", submitShiftScheduleHandler{db})
1348 scoutingServer.Handle("/requests/request/shift_schedule", requestShiftScheduleHandler{db})
Filip Kujawa210a03b2022-11-24 14:41:11 -08001349 scoutingServer.Handle("/requests/submit/submit_driver_ranking", SubmitDriverRankingHandler{db})
Sabina Leaver9b4eb312023-02-20 19:58:17 -08001350 scoutingServer.Handle("/requests/submit/submit_actions", submitActionsHandler{db})
Emily Markova8cb91312024-02-02 12:30:37 -08001351 scoutingServer.Handle("/requests/submit/submit_2024_actions", submit2024ActionsHandler{db})
Filip Kujawac1ded372023-05-27 14:33:43 -07001352 scoutingServer.Handle("/requests/delete/delete_2023_data_scouting", Delete2023DataScoutingHandler{db})
Emily Markova8cb91312024-02-02 12:30:37 -08001353 scoutingServer.Handle("/requests/delete/delete_2024_data_scouting", Delete2024DataScoutingHandler{db})
Philipp Schradercdb5cfc2022-02-20 14:57:07 -08001354}