-
Notifications
You must be signed in to change notification settings - Fork 4
/
contest.go
45 lines (40 loc) · 1.35 KB
/
contest.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
package middleware
import (
"fmt"
"github.com/google/logger"
"github.com/jsannemo/omogenjudge/frontend/request"
"github.com/jsannemo/omogenjudge/storage/contests"
)
// readContest is a processor that stores the current contest data in the request context.
func readContest(r *request.Request) (request.Response, error) {
hostname := r.Request.Host
cs, err := contests.ListContests(r.Request.Context(), contests.ListArgs{WithProblems: true}, contests.ListFilter{HostName: hostname})
if err != nil {
return nil, fmt.Errorf("could not retrieve current contest: %v", err)
} else if len(cs) > 0 {
r.Context.Contest = cs.Latest()
}
return nil, nil
}
// readTeam is a processor that stores the logged-in team in the request context.
func readTeam(r *request.Request) (request.Response, error) {
if r.Context.Contest == nil || r.Context.User == nil {
return nil, nil
}
teams, err := contests.ListTeams(r.Request.Context(), contests.TeamFilter{
ContestID: r.Context.Contest.ContestID,
AccountID: r.Context.User.AccountID,
})
if err != nil {
return nil, fmt.Errorf("could not retrieve current team: %v", err)
} else if len(teams) > 0 {
r.Context.Team = teams[0]
}
if len(teams) > 1 {
logger.Fatalf("Found too many teams for account %d in contest %d (%d)",
r.Context.User.AccountID,
r.Context.Contest.ContestID,
len(teams))
}
return nil, nil
}