blob: 8761b7d5f5ff900a9e76e495b6310d0ac8a10365 [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)
Emily Markova9c18e9c2024-04-03 20:06:27 -0700105 ReturnStats2024ForTeam(teamNumber string, matchNumber int32, setNumber int32, compLevel string, compType string) ([]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(
Emily Markova9c18e9c2024-04-03 20:06:27 -0700203 teamNumber, key.MatchNumber, key.SetNumber, key.CompLevel, "Regular")
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()),
Emily Markovacf893f42024-03-13 19:03:10 -0700401 NoShow: bool(request.NoShow()),
402 MatchNumber: request.MatchNumber(),
403 SetNumber: request.SetNumber(),
404 CompLevel: string(request.CompLevel()),
Filip Kujawaf947cb42022-11-21 10:00:30 -0800405 })
Alex Perry81f96ba2022-03-13 18:26:19 -0700406 if err != nil {
407 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to insert notes: %v", err))
408 return
409 }
410
411 var response SubmitNotesResponseT
412 builder := flatbuffers.NewBuilder(10)
413 builder.Finish((&response).Pack(builder))
414 w.Write(builder.FinishedBytes())
415}
416
Emily Markovafaecfe12023-07-01 12:40:03 -0700417type submitPitImageScoutingHandler struct {
418 db Database
419}
420
421func (handler submitPitImageScoutingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
422 requestBytes, err := io.ReadAll(req.Body)
423 if err != nil {
424 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
425 return
426 }
427
428 request, success := parseRequest(w, requestBytes, "SubmitPitImage", submit_pit_image.GetRootAsSubmitPitImage)
429 if !success {
430 return
431 }
432
433 err = handler.db.AddPitImage(db.PitImage{
434 TeamNumber: string(request.TeamNumber()),
435 CheckSum: db.ComputeSha256FromByteArray(request.ImageDataBytes()),
436 ImagePath: string(request.ImagePath()),
437 ImageData: request.ImageDataBytes(),
438 })
439 if err != nil {
440 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to insert notes: %v", err))
441 return
442 }
443
444 var response SubmitPitImageResponseT
445 builder := flatbuffers.NewBuilder(10)
446 builder.Finish((&response).Pack(builder))
447 w.Write(builder.FinishedBytes())
448}
449
Emily Markova8cb91312024-02-02 12:30:37 -0800450func ConvertActionsToStat2024(submit2024Actions *submit_2024_actions.Submit2024Actions) (db.Stats2024, error) {
451 overall_time := int64(0)
452 cycles := int64(0)
453 picked_up := false
454 lastPlacedTime := int64(0)
455 stat := db.Stats2024{
Emily Markova9c18e9c2024-04-03 20:06:27 -0700456 CompType: string(submit2024Actions.CompType()), TeamNumber: string(submit2024Actions.TeamNumber()),
457 MatchNumber: submit2024Actions.MatchNumber(), SetNumber: submit2024Actions.SetNumber(), CompLevel: string(submit2024Actions.CompLevel()),
Emily Markova8cb91312024-02-02 12:30:37 -0800458 StartingQuadrant: 0, SpeakerAuto: 0, AmpAuto: 0, NotesDroppedAuto: 0, MobilityAuto: false,
Emily Markovacd156942024-04-07 19:32:28 -0700459 Speaker: 0, Amp: 0, SpeakerAmplified: 0, NotesDropped: 0, Shuttled: 0, OutOfField: 0, Penalties: 0,
Emily Markova040123c2024-02-27 09:48:37 -0800460 TrapNote: false, Spotlight: false, AvgCycle: 0, Park: false, OnStage: false, Harmony: false, RobotDied: false, CollectedBy: "",
Emily Markova8cb91312024-02-02 12:30:37 -0800461 }
462 // Loop over all actions.
463 for i := 0; i < submit2024Actions.ActionsListLength(); i++ {
464 var action submit_2024_actions.Action
465 if !submit2024Actions.ActionsList(&action, i) {
466 return db.Stats2024{}, errors.New(fmt.Sprintf("Failed to parse submit_2024_actions.Action"))
467 }
468 actionTable := new(flatbuffers.Table)
469 action_type := action.ActionTakenType()
470 if !action.ActionTaken(actionTable) {
471 return db.Stats2024{}, errors.New(fmt.Sprint("Failed to parse sub-action or sub-action was missing"))
472 }
473 if action_type == submit_2024_actions.ActionTypeStartMatchAction {
474 var startMatchAction submit_2024_actions.StartMatchAction
475 startMatchAction.Init(actionTable.Bytes, actionTable.Pos)
476 stat.StartingQuadrant = startMatchAction.Position()
477 } else if action_type == submit_2024_actions.ActionTypeMobilityAction {
478 var mobilityAction submit_2024_actions.MobilityAction
479 mobilityAction.Init(actionTable.Bytes, actionTable.Pos)
480 if mobilityAction.Mobility() {
481 stat.MobilityAuto = true
482 }
Emily Markova8cb91312024-02-02 12:30:37 -0800483 } else if action_type == submit_2024_actions.ActionTypePenaltyAction {
484 var penaltyAction submit_2024_actions.PenaltyAction
485 penaltyAction.Init(actionTable.Bytes, actionTable.Pos)
Emily Markovadcadcb62024-02-03 13:07:17 -0800486 stat.Penalties += penaltyAction.Penalties()
Emily Markova8cb91312024-02-02 12:30:37 -0800487
Emily Markova040123c2024-02-27 09:48:37 -0800488 } else if action_type == submit_2024_actions.ActionTypeRobotDeathAction {
489 var robotDeathAction submit_2024_actions.RobotDeathAction
490 robotDeathAction.Init(actionTable.Bytes, actionTable.Pos)
491 stat.RobotDied = true
492
Emily Markova8cb91312024-02-02 12:30:37 -0800493 } else if action_type == submit_2024_actions.ActionTypePickupNoteAction {
494 var pick_up_action submit_2024_actions.PickupNoteAction
495 pick_up_action.Init(actionTable.Bytes, actionTable.Pos)
Emily Markova040123c2024-02-27 09:48:37 -0800496 picked_up = true
Emily Markova8cb91312024-02-02 12:30:37 -0800497 } 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()
Emily Markovacd156942024-04-07 19:32:28 -0700505 count_in_cycle := true
Emily Markova8cb91312024-02-02 12:30:37 -0800506 if score_type == submit_2024_actions.ScoreTypekAMP && auto {
507 stat.AmpAuto += 1
508 } else if score_type == submit_2024_actions.ScoreTypekAMP && !auto {
509 stat.Amp += 1
Emily Markova8cb91312024-02-02 12:30:37 -0800510 } else if score_type == submit_2024_actions.ScoreTypekSPEAKER && !auto {
511 stat.Speaker += 1
512 } else if score_type == submit_2024_actions.ScoreTypekSPEAKER && auto {
513 stat.SpeakerAuto += 1
514 } else if score_type == submit_2024_actions.ScoreTypekSPEAKER_AMPLIFIED && !auto {
515 stat.SpeakerAmplified += 1
Emily Markova040123c2024-02-27 09:48:37 -0800516 } else if score_type == submit_2024_actions.ScoreTypekDROPPED && auto {
517 stat.NotesDroppedAuto += 1
Emily Markovacd156942024-04-07 19:32:28 -0700518 count_in_cycle = false
Emily Markova040123c2024-02-27 09:48:37 -0800519 } else if score_type == submit_2024_actions.ScoreTypekDROPPED && !auto {
520 stat.NotesDropped += 1
Emily Markovacd156942024-04-07 19:32:28 -0700521 count_in_cycle = false
522 } else if score_type == submit_2024_actions.ScoreTypekSHUTTLED {
523 stat.Shuttled += 1
524 count_in_cycle = false
525 } else if score_type == submit_2024_actions.ScoreTypekOUT_OF_FIELD {
526 stat.OutOfField += 1
527 count_in_cycle = false
Emily Markova8cb91312024-02-02 12:30:37 -0800528 } else {
529 return db.Stats2024{}, errors.New(fmt.Sprintf("Got unknown ObjectType/ScoreLevel/Auto combination"))
530 }
531 picked_up = false
Emily Markovacd156942024-04-07 19:32:28 -0700532 if count_in_cycle {
533 // Assuming dropped, shuttled, and out of field
534 // notes are not counted in total cycle time.
535 if lastPlacedTime != int64(0) {
536 // If this is not the first time we place,
537 // start counting cycle time. We define cycle
538 // time as the time between placements.
539 overall_time += int64(action.Timestamp()) - lastPlacedTime
540 }
Emily Markova8cb91312024-02-02 12:30:37 -0800541 cycles += 1
Emily Markovacd156942024-04-07 19:32:28 -0700542 lastPlacedTime = int64(action.Timestamp())
Emily Markova8cb91312024-02-02 12:30:37 -0800543 }
Emily Markova8cb91312024-02-02 12:30:37 -0800544 } else if action_type == submit_2024_actions.ActionTypeEndMatchAction {
545 var endMatchAction submit_2024_actions.EndMatchAction
546 endMatchAction.Init(actionTable.Bytes, actionTable.Pos)
547 if endMatchAction.StageType() == submit_2024_actions.StageTypekON_STAGE {
548 stat.OnStage = true
549 } else if endMatchAction.StageType() == submit_2024_actions.StageTypekPARK {
550 stat.Park = true
551 } else if endMatchAction.StageType() == submit_2024_actions.StageTypekHARMONY {
552 stat.Harmony = true
553 }
554 stat.TrapNote = endMatchAction.TrapNote()
Emily Markova6079e2f2024-02-17 13:17:24 -0800555 stat.Spotlight = endMatchAction.Spotlight()
Emily Markova8cb91312024-02-02 12:30:37 -0800556 }
557 }
558 if cycles != 0 {
559 stat.AvgCycle = overall_time / cycles
560 } else {
561 stat.AvgCycle = 0
562 }
563 return stat, nil
564}
565
566// Handles a Request2024DataScouting request.
567type request2024DataScoutingHandler struct {
568 db Database
569}
570
571func (handler request2024DataScoutingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
572 requestBytes, err := io.ReadAll(req.Body)
573 if err != nil {
574 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
575 return
576 }
577
578 _, success := parseRequest(w, requestBytes, "Request2024DataScouting", request_2024_data_scouting.GetRootAsRequest2024DataScouting)
579 if !success {
580 return
581 }
582
583 stats, err := handler.db.ReturnStats2024()
584 if err != nil {
585 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to query database: ", err))
586 return
587 }
588
589 var response Request2024DataScoutingResponseT
590 for _, stat := range stats {
591 response.StatsList = append(response.StatsList, &request_2024_data_scouting_response.Stats2024T{
592 TeamNumber: stat.TeamNumber,
593 MatchNumber: stat.MatchNumber,
594 SetNumber: stat.SetNumber,
595 CompLevel: stat.CompLevel,
596 StartingQuadrant: stat.StartingQuadrant,
597 SpeakerAuto: stat.SpeakerAuto,
598 AmpAuto: stat.AmpAuto,
599 NotesDroppedAuto: stat.NotesDroppedAuto,
600 MobilityAuto: stat.MobilityAuto,
601 Speaker: stat.Speaker,
602 Amp: stat.Amp,
603 SpeakerAmplified: stat.SpeakerAmplified,
Emily Markova8cb91312024-02-02 12:30:37 -0800604 NotesDropped: stat.NotesDropped,
Emily Markovacd156942024-04-07 19:32:28 -0700605 Shuttled: stat.Shuttled,
606 OutOfField: stat.OutOfField,
Emily Markova8cb91312024-02-02 12:30:37 -0800607 Penalties: stat.Penalties,
608 TrapNote: stat.TrapNote,
Emily Markova6079e2f2024-02-17 13:17:24 -0800609 Spotlight: stat.Spotlight,
Emily Markova8cb91312024-02-02 12:30:37 -0800610 AvgCycle: stat.AvgCycle,
611 Park: stat.Park,
612 OnStage: stat.OnStage,
613 Harmony: stat.Harmony,
Emily Markova040123c2024-02-27 09:48:37 -0800614 RobotDied: stat.RobotDied,
Emily Markova8cb91312024-02-02 12:30:37 -0800615 CollectedBy: stat.CollectedBy,
Emily Markova9c18e9c2024-04-03 20:06:27 -0700616 CompType: stat.CompType,
Emily Markova8cb91312024-02-02 12:30:37 -0800617 })
618 }
619
620 builder := flatbuffers.NewBuilder(50 * 1024)
621 builder.Finish((&response).Pack(builder))
622 w.Write(builder.FinishedBytes())
623}
624
Emily Markova1abe9782023-03-11 19:45:38 -0800625func ConvertActionsToStat(submitActions *submit_actions.SubmitActions) (db.Stats2023, error) {
626 overall_time := int64(0)
627 cycles := int64(0)
628 picked_up := false
629 lastPlacedTime := int64(0)
Philipp Schrader4b489222023-04-15 16:40:16 -0700630 stat := db.Stats2023{
631 PreScouting: submitActions.PreScouting(),
632 TeamNumber: string(submitActions.TeamNumber()), MatchNumber: submitActions.MatchNumber(), SetNumber: submitActions.SetNumber(), CompLevel: string(submitActions.CompLevel()),
Emily Markova1abe9782023-03-11 19:45:38 -0800633 StartingQuadrant: 0, LowCubesAuto: 0, MiddleCubesAuto: 0, HighCubesAuto: 0, CubesDroppedAuto: 0,
634 LowConesAuto: 0, MiddleConesAuto: 0, HighConesAuto: 0, ConesDroppedAuto: 0, LowCubes: 0, MiddleCubes: 0, HighCubes: 0,
Philipp Schradere11114f2023-04-15 17:04:25 -0700635 CubesDropped: 0, LowCones: 0, MiddleCones: 0, HighCones: 0, ConesDropped: 0, SuperchargedPieces: 0, AvgCycle: 0, CollectedBy: "",
Emily Markova1abe9782023-03-11 19:45:38 -0800636 }
637 // Loop over all actions.
638 for i := 0; i < submitActions.ActionsListLength(); i++ {
639 var action submit_actions.Action
640 if !submitActions.ActionsList(&action, i) {
641 return db.Stats2023{}, errors.New(fmt.Sprintf("Failed to parse submit_actions.Action"))
642 }
643 actionTable := new(flatbuffers.Table)
644 action_type := action.ActionTakenType()
645 if !action.ActionTaken(actionTable) {
646 return db.Stats2023{}, errors.New(fmt.Sprint("Failed to parse sub-action or sub-action was missing"))
647 }
648 if action_type == submit_actions.ActionTypeStartMatchAction {
649 var startMatchAction submit_actions.StartMatchAction
650 startMatchAction.Init(actionTable.Bytes, actionTable.Pos)
651 stat.StartingQuadrant = startMatchAction.Position()
Filip Kujawa0b4b1e52023-04-15 14:05:40 -0700652 } else if action_type == submit_actions.ActionTypeMobilityAction {
653 var mobilityAction submit_actions.MobilityAction
654 mobilityAction.Init(actionTable.Bytes, actionTable.Pos)
655 if mobilityAction.Mobility() {
656 stat.Mobility = true
657 }
658
Emily Markova46a69bf2023-03-22 20:45:52 -0700659 } else if action_type == submit_actions.ActionTypeAutoBalanceAction {
660 var autoBalanceAction submit_actions.AutoBalanceAction
661 autoBalanceAction.Init(actionTable.Bytes, actionTable.Pos)
662 if autoBalanceAction.Docked() {
663 stat.DockedAuto = true
664 }
665 if autoBalanceAction.Engaged() {
666 stat.EngagedAuto = true
667 }
Emily Markova63c63f62023-03-29 20:57:35 -0700668 if autoBalanceAction.BalanceAttempt() {
669 stat.BalanceAttemptAuto = true
670 }
Emily Markova1abe9782023-03-11 19:45:38 -0800671 } else if action_type == submit_actions.ActionTypePickupObjectAction {
672 var pick_up_action submit_actions.PickupObjectAction
673 pick_up_action.Init(actionTable.Bytes, actionTable.Pos)
674 if picked_up == true {
675 object := pick_up_action.ObjectType().String()
676 auto := pick_up_action.Auto()
677 if object == "kCube" && auto == false {
678 stat.CubesDropped += 1
679 } else if object == "kCube" && auto == true {
680 stat.CubesDroppedAuto += 1
681 } else if object == "kCone" && auto == false {
682 stat.ConesDropped += 1
683 } else if object == "kCube" && auto == true {
684 stat.ConesDroppedAuto += 1
685 }
686 } else {
687 picked_up = true
688 }
689 } else if action_type == submit_actions.ActionTypePlaceObjectAction {
690 var place_action submit_actions.PlaceObjectAction
691 place_action.Init(actionTable.Bytes, actionTable.Pos)
692 if !picked_up {
693 return db.Stats2023{}, errors.New(fmt.Sprintf("Got PlaceObjectAction without corresponding PickupObjectAction"))
694 }
695 object := place_action.ObjectType()
696 level := place_action.ScoreLevel()
697 auto := place_action.Auto()
698 if object == 0 && level == 0 && auto == true {
699 stat.LowCubesAuto += 1
700 } else if object == 0 && level == 0 && auto == false {
701 stat.LowCubes += 1
702 } else if object == 0 && level == 1 && auto == true {
703 stat.MiddleCubesAuto += 1
704 } else if object == 0 && level == 1 && auto == false {
705 stat.MiddleCubes += 1
706 } else if object == 0 && level == 2 && auto == true {
707 stat.HighCubesAuto += 1
708 } else if object == 0 && level == 2 && auto == false {
709 stat.HighCubes += 1
710 } else if object == 1 && level == 0 && auto == true {
711 stat.LowConesAuto += 1
712 } else if object == 1 && level == 0 && auto == false {
713 stat.LowCones += 1
714 } else if object == 1 && level == 1 && auto == true {
715 stat.MiddleConesAuto += 1
716 } else if object == 1 && level == 1 && auto == false {
717 stat.MiddleCones += 1
718 } else if object == 1 && level == 2 && auto == true {
719 stat.HighConesAuto += 1
720 } else if object == 1 && level == 2 && auto == false {
721 stat.HighCones += 1
Filip Kujawa7a045e72023-04-13 08:41:09 -0700722 } else if level == 3 {
723 stat.SuperchargedPieces += 1
Emily Markova1abe9782023-03-11 19:45:38 -0800724 } else {
725 return db.Stats2023{}, errors.New(fmt.Sprintf("Got unknown ObjectType/ScoreLevel/Auto combination"))
726 }
727 picked_up = false
728 if lastPlacedTime != int64(0) {
729 // If this is not the first time we place,
730 // start counting cycle time. We define cycle
731 // time as the time between placements.
732 overall_time += int64(action.Timestamp()) - lastPlacedTime
733 cycles += 1
734 }
735 lastPlacedTime = int64(action.Timestamp())
Emily Markova46a69bf2023-03-22 20:45:52 -0700736 } else if action_type == submit_actions.ActionTypeEndMatchAction {
737 var endMatchAction submit_actions.EndMatchAction
738 endMatchAction.Init(actionTable.Bytes, actionTable.Pos)
739 if endMatchAction.Docked() {
740 stat.Docked = true
741 }
742 if endMatchAction.Engaged() {
743 stat.Engaged = true
744 }
Emily Markova63c63f62023-03-29 20:57:35 -0700745 if endMatchAction.BalanceAttempt() {
746 stat.BalanceAttempt = true
747 }
Emily Markova1abe9782023-03-11 19:45:38 -0800748 }
749 }
750 if cycles != 0 {
Philipp Schrader8c878a22023-03-20 22:36:38 -0700751 stat.AvgCycle = overall_time / cycles
Emily Markova1abe9782023-03-11 19:45:38 -0800752 } else {
753 stat.AvgCycle = 0
754 }
755 return stat, nil
756}
757
Emily Markova290147d2023-03-03 22:40:06 -0800758// Handles a Request2023DataScouting request.
759type request2023DataScoutingHandler struct {
760 db Database
761}
762
763func (handler request2023DataScoutingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
764 requestBytes, err := io.ReadAll(req.Body)
765 if err != nil {
766 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
767 return
768 }
769
770 _, success := parseRequest(w, requestBytes, "Request2023DataScouting", request_2023_data_scouting.GetRootAsRequest2023DataScouting)
771 if !success {
772 return
773 }
774
775 stats, err := handler.db.ReturnStats2023()
776 if err != nil {
777 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to query database: ", err))
778 return
779 }
780
781 var response Request2023DataScoutingResponseT
782 for _, stat := range stats {
783 response.StatsList = append(response.StatsList, &request_2023_data_scouting_response.Stats2023T{
Emily Markova63c63f62023-03-29 20:57:35 -0700784 TeamNumber: stat.TeamNumber,
785 MatchNumber: stat.MatchNumber,
786 SetNumber: stat.SetNumber,
787 CompLevel: stat.CompLevel,
788 StartingQuadrant: stat.StartingQuadrant,
789 LowCubesAuto: stat.LowCubesAuto,
790 MiddleCubesAuto: stat.MiddleCubesAuto,
791 HighCubesAuto: stat.HighCubesAuto,
792 CubesDroppedAuto: stat.CubesDroppedAuto,
793 LowConesAuto: stat.LowConesAuto,
794 MiddleConesAuto: stat.MiddleConesAuto,
795 HighConesAuto: stat.HighConesAuto,
796 ConesDroppedAuto: stat.ConesDroppedAuto,
797 LowCubes: stat.LowCubes,
798 MiddleCubes: stat.MiddleCubes,
799 HighCubes: stat.HighCubes,
800 CubesDropped: stat.CubesDropped,
801 LowCones: stat.LowCones,
802 MiddleCones: stat.MiddleCones,
803 HighCones: stat.HighCones,
804 ConesDropped: stat.ConesDropped,
Filip Kujawa7a045e72023-04-13 08:41:09 -0700805 SuperchargedPieces: stat.SuperchargedPieces,
Emily Markova63c63f62023-03-29 20:57:35 -0700806 AvgCycle: stat.AvgCycle,
Filip Kujawa0b4b1e52023-04-15 14:05:40 -0700807 Mobility: stat.Mobility,
Emily Markova63c63f62023-03-29 20:57:35 -0700808 DockedAuto: stat.DockedAuto,
809 EngagedAuto: stat.EngagedAuto,
810 BalanceAttemptAuto: stat.BalanceAttemptAuto,
811 Docked: stat.Docked,
812 Engaged: stat.Engaged,
813 BalanceAttempt: stat.BalanceAttempt,
814 CollectedBy: stat.CollectedBy,
Emily Markova290147d2023-03-03 22:40:06 -0800815 })
816 }
817
818 builder := flatbuffers.NewBuilder(50 * 1024)
819 builder.Finish((&response).Pack(builder))
820 w.Write(builder.FinishedBytes())
821}
822
Emily Markova8e39f452023-12-23 12:17:30 -0800823type requestAllPitImagesHandler struct {
824 db Database
825}
826
827func (handler requestAllPitImagesHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
828 requestBytes, err := io.ReadAll(req.Body)
829 if err != nil {
830 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
831 return
832 }
833
834 _, success := parseRequest(w, requestBytes, "RequestAllPitImages", request_all_pit_images.GetRootAsRequestAllPitImages)
835 if !success {
836 return
837 }
838
839 images, err := handler.db.ReturnPitImages()
840 if err != nil {
841 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to get pit images: %v", err))
842 return
843 }
844
845 var response RequestAllPitImagesResponseT
846 for _, data := range images {
847 response.PitImageList = append(response.PitImageList, &request_all_pit_images_response.PitImageT{
848 TeamNumber: data.TeamNumber,
849 ImagePath: data.ImagePath,
850 CheckSum: data.CheckSum,
851 })
852 }
853
854 builder := flatbuffers.NewBuilder(1024)
855 builder.Finish((&response).Pack(builder))
856 w.Write(builder.FinishedBytes())
857}
858
Emily Markovafaecfe12023-07-01 12:40:03 -0700859type requestPitImagesHandler struct {
860 db Database
861}
862
863func (handler requestPitImagesHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
864 requestBytes, err := io.ReadAll(req.Body)
865 if err != nil {
866 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
867 return
868 }
869
870 request, success := parseRequest(w, requestBytes, "RequestPitImages", request_pit_images.GetRootAsRequestPitImages)
871 if !success {
872 return
873 }
874
875 images, err := handler.db.QueryPitImages(string(request.TeamNumber()))
876 if err != nil {
877 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to query pit images: %v", err))
878 return
879 }
880
881 var response RequestPitImagesResponseT
882 for _, data := range images {
883 response.PitImageList = append(response.PitImageList, &request_pit_images_response.PitImageT{
884 TeamNumber: data.TeamNumber,
885 ImagePath: data.ImagePath,
886 CheckSum: data.CheckSum,
887 })
888 }
889
890 builder := flatbuffers.NewBuilder(1024)
891 builder.Finish((&response).Pack(builder))
892 w.Write(builder.FinishedBytes())
893}
894
Alex Perry81f96ba2022-03-13 18:26:19 -0700895type requestNotesForTeamHandler struct {
896 db Database
897}
898
899func (handler requestNotesForTeamHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
900 requestBytes, err := io.ReadAll(req.Body)
901 if err != nil {
902 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
903 return
904 }
905
Philipp Schraderb7e75932022-03-26 16:18:34 -0700906 request, success := parseRequest(w, requestBytes, "RequestNotesForTeam", request_notes_for_team.GetRootAsRequestNotesForTeam)
Alex Perry81f96ba2022-03-13 18:26:19 -0700907 if !success {
908 return
909 }
910
Emily Markovae68b7632023-12-30 14:17:55 -0800911 notes, err := handler.db.QueryNotes(string(request.Team()))
Alex Perry81f96ba2022-03-13 18:26:19 -0700912 if err != nil {
913 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to query notes: %v", err))
914 return
915 }
916
917 var response RequestNotesForTeamResponseT
Philipp Schradereecb8962022-06-01 21:02:42 -0700918 for _, data := range notes {
Alex Perry81f96ba2022-03-13 18:26:19 -0700919 response.Notes = append(response.Notes, &request_notes_for_team_response.NoteT{data})
920 }
921
922 builder := flatbuffers.NewBuilder(1024)
923 builder.Finish((&response).Pack(builder))
924 w.Write(builder.FinishedBytes())
925}
926
Milo Lin1d59f0c2022-06-22 20:30:58 -0700927type requestShiftScheduleHandler struct {
928 db Database
929}
930
931func (handler requestShiftScheduleHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
932 requestBytes, err := io.ReadAll(req.Body)
933 if err != nil {
934 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
935 return
936 }
937
938 _, success := parseRequest(w, requestBytes, "RequestShiftSchedule", request_shift_schedule.GetRootAsRequestShiftSchedule)
939 if !success {
940 return
941 }
942
943 shiftData, err := handler.db.ReturnAllShifts()
944 if err != nil {
945 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to query shift schedule: %v", err))
946 return
947 }
948
949 var response RequestShiftScheduleResponseT
950 for _, shifts := range shiftData {
951 response.ShiftSchedule = append(response.ShiftSchedule, &request_shift_schedule_response.MatchAssignmentT{
952 MatchNumber: shifts.MatchNumber,
Philipp Schrader2ff455b2023-05-03 22:11:50 -0700953 R1Scouter: shifts.R1scouter,
954 R2Scouter: shifts.R2scouter,
955 R3Scouter: shifts.R3scouter,
956 B1Scouter: shifts.B1scouter,
957 B2Scouter: shifts.B2scouter,
958 B3Scouter: shifts.B3scouter,
Milo Lin1d59f0c2022-06-22 20:30:58 -0700959 })
960 }
961
962 builder := flatbuffers.NewBuilder(1024)
963 builder.Finish((&response).Pack(builder))
964 w.Write(builder.FinishedBytes())
965}
966
967type submitShiftScheduleHandler struct {
968 db Database
969}
970
971func (handler submitShiftScheduleHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
972 // Get the username of the person submitting the data.
973 username := parseUsername(req)
974
975 requestBytes, err := io.ReadAll(req.Body)
976 if err != nil {
977 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
978 return
979 }
980
981 request, success := parseRequest[SubmitShiftSchedule](w, requestBytes, "SubmitShiftSchedule", submit_shift_schedule.GetRootAsSubmitShiftSchedule)
982 if !success {
983 return
984 }
985
986 log.Println("Got shift schedule from", username)
987 shift_schedule_length := request.ShiftScheduleLength()
988 for i := 0; i < shift_schedule_length; i++ {
989 var match_assignment submit_shift_schedule.MatchAssignment
990 request.ShiftSchedule(&match_assignment, i)
991 current_shift := db.Shift{
992 MatchNumber: match_assignment.MatchNumber(),
Philipp Schrader2ff455b2023-05-03 22:11:50 -0700993 R1scouter: string(match_assignment.R1Scouter()),
994 R2scouter: string(match_assignment.R2Scouter()),
995 R3scouter: string(match_assignment.R3Scouter()),
996 B1scouter: string(match_assignment.B1Scouter()),
997 B2scouter: string(match_assignment.B2Scouter()),
998 B3scouter: string(match_assignment.B3Scouter()),
Milo Lin1d59f0c2022-06-22 20:30:58 -0700999 }
1000 err = handler.db.AddToShift(current_shift)
1001 if err != nil {
1002 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to submit shift schedule: ", err))
1003 return
1004 }
1005 }
1006
1007 builder := flatbuffers.NewBuilder(50 * 1024)
1008 builder.Finish((&SubmitShiftScheduleResponseT{}).Pack(builder))
1009 w.Write(builder.FinishedBytes())
1010}
1011
Filip Kujawa210a03b2022-11-24 14:41:11 -08001012type SubmitDriverRankingHandler struct {
1013 db Database
1014}
1015
1016func (handler SubmitDriverRankingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
1017 requestBytes, err := io.ReadAll(req.Body)
1018 if err != nil {
1019 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
1020 return
1021 }
1022
1023 request, success := parseRequest(w, requestBytes, "SubmitDriverRanking", submit_driver_ranking.GetRootAsSubmitDriverRanking)
1024 if !success {
1025 return
1026 }
1027
1028 err = handler.db.AddDriverRanking(db.DriverRankingData{
1029 MatchNumber: request.MatchNumber(),
Emily Markovae68b7632023-12-30 14:17:55 -08001030 Rank1: string(request.Rank1()),
1031 Rank2: string(request.Rank2()),
1032 Rank3: string(request.Rank3()),
Filip Kujawa210a03b2022-11-24 14:41:11 -08001033 })
1034
1035 if err != nil {
1036 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to insert driver ranking: %v", err))
1037 return
1038 }
1039
1040 var response SubmitDriverRankingResponseT
1041 builder := flatbuffers.NewBuilder(10)
1042 builder.Finish((&response).Pack(builder))
1043 w.Write(builder.FinishedBytes())
1044}
1045
Filip Kujawaf882e022022-12-14 13:14:08 -08001046type requestAllNotesHandler struct {
1047 db Database
1048}
1049
1050func (handler requestAllNotesHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
1051 requestBytes, err := io.ReadAll(req.Body)
1052 if err != nil {
1053 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
1054 return
1055 }
1056
1057 _, success := parseRequest(w, requestBytes, "RequestAllNotes", request_all_notes.GetRootAsRequestAllNotes)
1058 if !success {
1059 return
1060 }
1061
1062 notes, err := handler.db.ReturnAllNotes()
1063 if err != nil {
1064 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to query database: ", err))
1065 return
1066 }
1067
1068 var response RequestAllNotesResponseT
1069 for _, note := range notes {
1070 response.NoteList = append(response.NoteList, &request_all_notes_response.NoteT{
Filip Kujawa7ddd5652023-03-07 19:56:15 -08001071 Team: note.TeamNumber,
1072 Notes: note.Notes,
1073 GoodDriving: note.GoodDriving,
1074 BadDriving: note.BadDriving,
Filip Kujawa11dc4c92023-04-13 08:55:43 -07001075 SolidPlacing: note.SolidPlacing,
Filip Kujawa7ddd5652023-03-07 19:56:15 -08001076 SketchyPlacing: note.SketchyPlacing,
1077 GoodDefense: note.GoodDefense,
1078 BadDefense: note.BadDefense,
1079 EasilyDefended: note.EasilyDefended,
Emily Markovacf893f42024-03-13 19:03:10 -07001080 NoShow: note.NoShow,
1081 MatchNumber: note.MatchNumber,
1082 CompLevel: note.CompLevel,
1083 SetNumber: note.SetNumber,
Filip Kujawaf882e022022-12-14 13:14:08 -08001084 })
1085 }
1086
1087 builder := flatbuffers.NewBuilder(50 * 1024)
1088 builder.Finish((&response).Pack(builder))
1089 w.Write(builder.FinishedBytes())
1090}
1091
1092type requestAllDriverRankingsHandler struct {
1093 db Database
1094}
1095
1096func (handler requestAllDriverRankingsHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
1097 requestBytes, err := io.ReadAll(req.Body)
1098 if err != nil {
1099 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
1100 return
1101 }
1102
1103 _, success := parseRequest(w, requestBytes, "RequestAllDriverRankings", request_all_driver_rankings.GetRootAsRequestAllDriverRankings)
1104 if !success {
1105 return
1106 }
1107
1108 rankings, err := handler.db.ReturnAllDriverRankings()
1109 if err != nil {
1110 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to query database: ", err))
1111 return
1112 }
1113
1114 var response RequestAllDriverRankingsResponseT
1115 for _, ranking := range rankings {
1116 response.DriverRankingList = append(response.DriverRankingList, &request_all_driver_rankings_response.RankingT{
1117 MatchNumber: ranking.MatchNumber,
1118 Rank1: ranking.Rank1,
1119 Rank2: ranking.Rank2,
1120 Rank3: ranking.Rank3,
1121 })
1122 }
1123
1124 builder := flatbuffers.NewBuilder(50 * 1024)
1125 builder.Finish((&response).Pack(builder))
1126 w.Write(builder.FinishedBytes())
1127}
1128
Emily Markova8cb91312024-02-02 12:30:37 -08001129type submit2024ActionsHandler struct {
1130 db Database
1131}
1132
1133func (handler submit2024ActionsHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
1134 // Get the username of the person submitting the data.
1135 username := parseUsername(req)
1136
1137 requestBytes, err := io.ReadAll(req.Body)
1138 if err != nil {
1139 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
1140 return
1141 }
1142
1143 request, success := parseRequest(w, requestBytes, "Submit2024Actions", submit_2024_actions.GetRootAsSubmit2024Actions)
1144 if !success {
1145 return
1146 }
1147
Philipp Schradera6f629b2024-02-25 20:45:01 -08001148 log.Println("Got actions for match", request.MatchNumber(), "team", string(request.TeamNumber()), "from", username)
Emily Markova8cb91312024-02-02 12:30:37 -08001149
1150 for i := 0; i < request.ActionsListLength(); i++ {
1151
1152 var action Action2024
1153 request.ActionsList(&action, i)
1154
1155 dbAction := db.Action{
Emily Markova9c18e9c2024-04-03 20:06:27 -07001156 CompType: string(request.CompType()),
Emily Markova8cb91312024-02-02 12:30:37 -08001157 TeamNumber: string(request.TeamNumber()),
1158 MatchNumber: request.MatchNumber(),
1159 SetNumber: request.SetNumber(),
1160 CompLevel: string(request.CompLevel()),
1161 //TODO: Serialize CompletedAction
1162 CompletedAction: []byte{},
1163 Timestamp: action.Timestamp(),
1164 CollectedBy: username,
1165 }
1166
1167 // Do some error checking.
1168 if action.Timestamp() < 0 {
1169 respondWithError(w, http.StatusBadRequest, fmt.Sprint(
1170 "Invalid timestamp field value of ", action.Timestamp()))
1171 return
1172 }
1173
1174 err = handler.db.AddAction(dbAction)
1175 if err != nil {
1176 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to add action to database: ", err))
1177 return
1178 }
1179 }
1180
1181 stats, err := ConvertActionsToStat2024(request)
1182 if err != nil {
1183 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to convert actions to stats: ", err))
1184 return
1185 }
1186
1187 stats.CollectedBy = username
1188
1189 err = handler.db.AddToStats2024(stats)
1190 if err != nil {
Emily Markovadcadcb62024-02-03 13:07:17 -08001191 respondWithError(w, http.StatusInternalServerError, fmt.Sprint("Failed to submit stats2024: ", stats, ": ", err))
Emily Markova8cb91312024-02-02 12:30:37 -08001192 return
1193 }
1194
1195 builder := flatbuffers.NewBuilder(50 * 1024)
Emily Markovadcadcb62024-02-03 13:07:17 -08001196 builder.Finish((&Submit2024ActionsResponseT{}).Pack(builder))
Emily Markova8cb91312024-02-02 12:30:37 -08001197 w.Write(builder.FinishedBytes())
1198}
1199
Emily Markova8cb91312024-02-02 12:30:37 -08001200type Delete2024DataScoutingHandler struct {
1201 db Database
1202}
1203
1204func (handler Delete2024DataScoutingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
1205 requestBytes, err := io.ReadAll(req.Body)
1206 if err != nil {
1207 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
1208 return
1209 }
1210
1211 request, success := parseRequest(w, requestBytes, "Delete2024DataScouting", delete_2024_data_scouting.GetRootAsDelete2024DataScouting)
1212 if !success {
1213 return
1214 }
1215
1216 err = handler.db.DeleteFromStats2024(
1217 string(request.CompLevel()),
1218 request.MatchNumber(),
1219 request.SetNumber(),
1220 string(request.TeamNumber()))
1221
1222 if err != nil {
1223 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to delete from stats2024: %v", err))
1224 return
1225 }
1226
1227 err = handler.db.DeleteFromActions(
1228 string(request.CompLevel()),
1229 request.MatchNumber(),
1230 request.SetNumber(),
1231 string(request.TeamNumber()))
1232
1233 if err != nil {
1234 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to delete from actions: %v", err))
1235 return
1236 }
1237
1238 var response Delete2024DataScoutingResponseT
1239 builder := flatbuffers.NewBuilder(10)
1240 builder.Finish((&response).Pack(builder))
1241 w.Write(builder.FinishedBytes())
1242}
1243
Filip Kujawac1ded372023-05-27 14:33:43 -07001244type Delete2023DataScoutingHandler struct {
1245 db Database
1246}
1247
1248func (handler Delete2023DataScoutingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
1249 requestBytes, err := io.ReadAll(req.Body)
1250 if err != nil {
1251 respondWithError(w, http.StatusBadRequest, fmt.Sprint("Failed to read request bytes:", err))
1252 return
1253 }
1254
1255 request, success := parseRequest(w, requestBytes, "Delete2023DataScouting", delete_2023_data_scouting.GetRootAsDelete2023DataScouting)
1256 if !success {
1257 return
1258 }
1259
1260 err = handler.db.DeleteFromStats(
1261 string(request.CompLevel()),
1262 request.MatchNumber(),
1263 request.SetNumber(),
1264 string(request.TeamNumber()))
1265
1266 if err != nil {
1267 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to delete from stats: %v", err))
1268 return
1269 }
1270
1271 err = handler.db.DeleteFromActions(
1272 string(request.CompLevel()),
1273 request.MatchNumber(),
1274 request.SetNumber(),
1275 string(request.TeamNumber()))
1276
1277 if err != nil {
1278 respondWithError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to delete from actions: %v", err))
1279 return
1280 }
1281
1282 var response Delete2023DataScoutingResponseT
1283 builder := flatbuffers.NewBuilder(10)
1284 builder.Finish((&response).Pack(builder))
1285 w.Write(builder.FinishedBytes())
1286}
1287
Philipp Schrader43c730b2023-02-26 20:27:44 -08001288func HandleRequests(db Database, scoutingServer server.ScoutingServer) {
Philipp Schradercdb5cfc2022-02-20 14:57:07 -08001289 scoutingServer.HandleFunc("/requests", unknown)
Philipp Schradercbf5c6a2022-02-27 23:25:19 -08001290 scoutingServer.Handle("/requests/request/all_matches", requestAllMatchesHandler{db})
Filip Kujawaf882e022022-12-14 13:14:08 -08001291 scoutingServer.Handle("/requests/request/all_notes", requestAllNotesHandler{db})
1292 scoutingServer.Handle("/requests/request/all_driver_rankings", requestAllDriverRankingsHandler{db})
Emily Markova290147d2023-03-03 22:40:06 -08001293 scoutingServer.Handle("/requests/request/2023_data_scouting", request2023DataScoutingHandler{db})
Emily Markova8cb91312024-02-02 12:30:37 -08001294 scoutingServer.Handle("/requests/request/2024_data_scouting", request2024DataScoutingHandler{db})
Alex Perry81f96ba2022-03-13 18:26:19 -07001295 scoutingServer.Handle("/requests/submit/submit_notes", submitNoteScoutingHandler{db})
Emily Markovafaecfe12023-07-01 12:40:03 -07001296 scoutingServer.Handle("/requests/submit/submit_pit_image", submitPitImageScoutingHandler{db})
1297 scoutingServer.Handle("/requests/request/pit_images", requestPitImagesHandler{db})
Emily Markova8e39f452023-12-23 12:17:30 -08001298 scoutingServer.Handle("/requests/request/all_pit_images", requestAllPitImagesHandler{db})
Alex Perry81f96ba2022-03-13 18:26:19 -07001299 scoutingServer.Handle("/requests/request/notes_for_team", requestNotesForTeamHandler{db})
Milo Lin1d59f0c2022-06-22 20:30:58 -07001300 scoutingServer.Handle("/requests/submit/shift_schedule", submitShiftScheduleHandler{db})
1301 scoutingServer.Handle("/requests/request/shift_schedule", requestShiftScheduleHandler{db})
Filip Kujawa210a03b2022-11-24 14:41:11 -08001302 scoutingServer.Handle("/requests/submit/submit_driver_ranking", SubmitDriverRankingHandler{db})
Emily Markova8cb91312024-02-02 12:30:37 -08001303 scoutingServer.Handle("/requests/submit/submit_2024_actions", submit2024ActionsHandler{db})
Filip Kujawac1ded372023-05-27 14:33:43 -07001304 scoutingServer.Handle("/requests/delete/delete_2023_data_scouting", Delete2023DataScoutingHandler{db})
Emily Markova8cb91312024-02-02 12:30:37 -08001305 scoutingServer.Handle("/requests/delete/delete_2024_data_scouting", Delete2024DataScoutingHandler{db})
Philipp Schradercdb5cfc2022-02-20 14:57:07 -08001306}