mirror of
https://github.com/Team254/cheesy-arena-lite.git
synced 2026-03-09 13:46:44 -04:00
342 lines
13 KiB
Go
Executable File
342 lines
13 KiB
Go
Executable File
// Copyright 2014 Team 254. All Rights Reserved.
|
|
// Author: pat@patfairbank.com (Patrick Fairbank)
|
|
//
|
|
// Web handlers for generating CSV and PDF reports.
|
|
|
|
package web
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/Team254/cheesy-arena-lite/tournament"
|
|
"github.com/gorilla/mux"
|
|
"github.com/jung-kurt/gofpdf"
|
|
"net/http"
|
|
"strconv"
|
|
)
|
|
|
|
// Generates a CSV-formatted report of the qualification rankings.
|
|
func (web *Web) rankingsCsvReportHandler(w http.ResponseWriter, r *http.Request) {
|
|
rankings, err := web.arena.Database.GetAllRankings()
|
|
if err != nil {
|
|
handleWebErr(w, err)
|
|
return
|
|
}
|
|
|
|
// Don't set the content type as "text/csv", as that will trigger an automatic download in the browser.
|
|
w.Header().Set("Content-Type", "text/plain")
|
|
template, err := web.parseFiles("templates/rankings.csv")
|
|
if err != nil {
|
|
handleWebErr(w, err)
|
|
return
|
|
}
|
|
err = template.ExecuteTemplate(w, "rankings.csv", rankings)
|
|
if err != nil {
|
|
handleWebErr(w, err)
|
|
return
|
|
}
|
|
}
|
|
|
|
// Generates a PDF-formatted report of the qualification rankings.
|
|
func (web *Web) rankingsPdfReportHandler(w http.ResponseWriter, r *http.Request) {
|
|
rankings, err := web.arena.Database.GetAllRankings()
|
|
if err != nil {
|
|
handleWebErr(w, err)
|
|
return
|
|
}
|
|
|
|
// The widths of the table columns in mm, stored here so that they can be referenced for each row.
|
|
colWidths := map[string]float64{"Rank": 13, "Team": 22, "RP": 23, "Auto": 23, "Endgame": 23, "Teleop": 23,
|
|
"W-L-T": 23, "DQ": 23, "Played": 23}
|
|
rowHeight := 6.5
|
|
|
|
pdf := gofpdf.New("P", "mm", "Letter", "font")
|
|
pdf.AddPage()
|
|
|
|
// Render table header row.
|
|
pdf.SetFont("Arial", "B", 10)
|
|
pdf.SetFillColor(220, 220, 220)
|
|
pdf.CellFormat(195, rowHeight, "Team Standings - "+web.arena.EventSettings.Name, "", 1, "C", false, 0, "")
|
|
pdf.CellFormat(colWidths["Rank"], rowHeight, "Rank", "1", 0, "C", true, 0, "")
|
|
pdf.CellFormat(colWidths["Team"], rowHeight, "Team", "1", 0, "C", true, 0, "")
|
|
pdf.CellFormat(colWidths["RP"], rowHeight, "RP", "1", 0, "C", true, 0, "")
|
|
pdf.CellFormat(colWidths["Auto"], rowHeight, "Auto", "1", 0, "C", true, 0, "")
|
|
pdf.CellFormat(colWidths["Endgame"], rowHeight, "Endgame", "1", 0, "C", true, 0, "")
|
|
pdf.CellFormat(colWidths["Teleop"], rowHeight, "Teleop", "1", 0, "C", true, 0, "")
|
|
pdf.CellFormat(colWidths["W-L-T"], rowHeight, "W-L-T", "1", 0, "C", true, 0, "")
|
|
pdf.CellFormat(colWidths["Played"], rowHeight, "Played", "1", 1, "C", true, 0, "")
|
|
for _, ranking := range rankings {
|
|
// Render ranking info row.
|
|
pdf.SetFont("Arial", "B", 10)
|
|
pdf.CellFormat(colWidths["Rank"], rowHeight, strconv.Itoa(ranking.Rank), "1", 0, "C", false, 0, "")
|
|
pdf.SetFont("Arial", "", 10)
|
|
pdf.CellFormat(colWidths["Team"], rowHeight, strconv.Itoa(ranking.TeamId), "1", 0, "C", false, 0, "")
|
|
pdf.CellFormat(colWidths["RP"], rowHeight, strconv.Itoa(ranking.RankingPoints), "1", 0, "C", false, 0, "")
|
|
pdf.CellFormat(colWidths["Auto"], rowHeight, strconv.Itoa(ranking.AutoPoints), "1", 0, "C", false, 0, "")
|
|
pdf.CellFormat(colWidths["Endgame"], rowHeight, strconv.Itoa(ranking.EndgamePoints), "1", 0, "C", false, 0, "")
|
|
pdf.CellFormat(colWidths["Teleop"], rowHeight, strconv.Itoa(ranking.TeleopPoints), "1", 0, "C", false, 0, "")
|
|
record := fmt.Sprintf("%d-%d-%d", ranking.Wins, ranking.Losses, ranking.Ties)
|
|
pdf.CellFormat(colWidths["W-L-T"], rowHeight, record, "1", 0, "C", false, 0, "")
|
|
pdf.CellFormat(colWidths["Played"], rowHeight, strconv.Itoa(ranking.Played), "1", 1, "C", false, 0, "")
|
|
}
|
|
|
|
// Write out the PDF file as the HTTP response.
|
|
w.Header().Set("Content-Type", "application/pdf")
|
|
err = pdf.Output(w)
|
|
if err != nil {
|
|
handleWebErr(w, err)
|
|
return
|
|
}
|
|
}
|
|
|
|
// Generates a CSV-formatted report of the match schedule.
|
|
func (web *Web) scheduleCsvReportHandler(w http.ResponseWriter, r *http.Request) {
|
|
vars := mux.Vars(r)
|
|
matches, err := web.arena.Database.GetMatchesByType(vars["type"])
|
|
if err != nil {
|
|
handleWebErr(w, err)
|
|
return
|
|
}
|
|
|
|
// Don't set the content type as "text/csv", as that will trigger an automatic download in the browser.
|
|
w.Header().Set("Content-Type", "text/plain")
|
|
template, err := web.parseFiles("templates/schedule.csv")
|
|
if err != nil {
|
|
handleWebErr(w, err)
|
|
return
|
|
}
|
|
err = template.ExecuteTemplate(w, "schedule.csv", matches)
|
|
if err != nil {
|
|
handleWebErr(w, err)
|
|
return
|
|
}
|
|
}
|
|
|
|
// Generates a PDF-formatted report of the match schedule.
|
|
func (web *Web) schedulePdfReportHandler(w http.ResponseWriter, r *http.Request) {
|
|
vars := mux.Vars(r)
|
|
matches, err := web.arena.Database.GetMatchesByType(vars["type"])
|
|
if err != nil {
|
|
handleWebErr(w, err)
|
|
return
|
|
}
|
|
teams, err := web.arena.Database.GetAllTeams()
|
|
if err != nil {
|
|
handleWebErr(w, err)
|
|
return
|
|
}
|
|
matchesPerTeam := 0
|
|
if len(teams) > 0 {
|
|
matchesPerTeam = len(matches) * tournament.TeamsPerMatch / len(teams)
|
|
}
|
|
|
|
// The widths of the table columns in mm, stored here so that they can be referenced for each row.
|
|
colWidths := map[string]float64{"Time": 35, "Type": 25, "Match": 15, "Team": 20}
|
|
rowHeight := 6.5
|
|
|
|
pdf := gofpdf.New("P", "mm", "Letter", "font")
|
|
pdf.AddPage()
|
|
|
|
// Render table header row.
|
|
pdf.SetFont("Arial", "B", 10)
|
|
pdf.SetFillColor(220, 220, 220)
|
|
pdf.CellFormat(195, rowHeight, "Match Schedule - "+web.arena.EventSettings.Name, "", 1, "C", false, 0, "")
|
|
pdf.CellFormat(colWidths["Time"], rowHeight, "Time", "1", 0, "C", true, 0, "")
|
|
pdf.CellFormat(colWidths["Type"], rowHeight, "Type", "1", 0, "C", true, 0, "")
|
|
pdf.CellFormat(colWidths["Match"], rowHeight, "Match", "1", 0, "C", true, 0, "")
|
|
pdf.CellFormat(colWidths["Team"], rowHeight, "Red 1", "1", 0, "C", true, 0, "")
|
|
pdf.CellFormat(colWidths["Team"], rowHeight, "Red 2", "1", 0, "C", true, 0, "")
|
|
pdf.CellFormat(colWidths["Team"], rowHeight, "Red 3", "1", 0, "C", true, 0, "")
|
|
pdf.CellFormat(colWidths["Team"], rowHeight, "Blue 1", "1", 0, "C", true, 0, "")
|
|
pdf.CellFormat(colWidths["Team"], rowHeight, "Blue 2", "1", 0, "C", true, 0, "")
|
|
pdf.CellFormat(colWidths["Team"], rowHeight, "Blue 3", "1", 1, "C", true, 0, "")
|
|
pdf.SetFont("Arial", "", 10)
|
|
for _, match := range matches {
|
|
height := rowHeight
|
|
borderStr := "1"
|
|
alignStr := "CM"
|
|
surrogate := false
|
|
if match.Red1IsSurrogate || match.Red2IsSurrogate || match.Red3IsSurrogate ||
|
|
match.Blue1IsSurrogate || match.Blue2IsSurrogate || match.Blue3IsSurrogate {
|
|
// If the match contains surrogates, the row needs to be taller to fit some text beneath team numbers.
|
|
height = 5.0
|
|
borderStr = "LTR"
|
|
alignStr = "CB"
|
|
surrogate = true
|
|
}
|
|
|
|
// Capitalize match types.
|
|
matchType := match.CapitalizedType()
|
|
|
|
formatTeam := func(teamId int) string {
|
|
if teamId == 0 {
|
|
return ""
|
|
} else {
|
|
return strconv.Itoa(teamId)
|
|
}
|
|
}
|
|
|
|
// Render match info row.
|
|
pdf.CellFormat(colWidths["Time"], height, match.Time.Local().Format("Mon 1/02 03:04 PM"), borderStr, 0,
|
|
alignStr, false, 0, "")
|
|
pdf.CellFormat(colWidths["Type"], height, matchType, borderStr, 0, alignStr, false, 0, "")
|
|
pdf.CellFormat(colWidths["Match"], height, match.DisplayName, borderStr, 0, alignStr, false, 0, "")
|
|
pdf.CellFormat(colWidths["Team"], height, formatTeam(match.Red1), borderStr, 0, alignStr, false, 0, "")
|
|
pdf.CellFormat(colWidths["Team"], height, formatTeam(match.Red2), borderStr, 0, alignStr, false, 0, "")
|
|
pdf.CellFormat(colWidths["Team"], height, formatTeam(match.Red3), borderStr, 0, alignStr, false, 0, "")
|
|
pdf.CellFormat(colWidths["Team"], height, formatTeam(match.Blue1), borderStr, 0, alignStr, false, 0, "")
|
|
pdf.CellFormat(colWidths["Team"], height, formatTeam(match.Blue2), borderStr, 0, alignStr, false, 0, "")
|
|
pdf.CellFormat(colWidths["Team"], height, formatTeam(match.Blue3), borderStr, 1, alignStr, false, 0, "")
|
|
if surrogate {
|
|
// Render the text that indicates which teams are surrogates.
|
|
height := 4.0
|
|
pdf.SetFont("Arial", "", 8)
|
|
pdf.CellFormat(colWidths["Time"], height, "", "LBR", 0, "C", false, 0, "")
|
|
pdf.CellFormat(colWidths["Type"], height, "", "LBR", 0, "C", false, 0, "")
|
|
pdf.CellFormat(colWidths["Match"], height, "", "LBR", 0, "C", false, 0, "")
|
|
pdf.CellFormat(colWidths["Team"], height, surrogateText(match.Red1IsSurrogate), "LBR", 0, "CT", false, 0,
|
|
"")
|
|
pdf.CellFormat(colWidths["Team"], height, surrogateText(match.Red2IsSurrogate), "LBR", 0, "CT", false, 0,
|
|
"")
|
|
pdf.CellFormat(colWidths["Team"], height, surrogateText(match.Red3IsSurrogate), "LBR", 0, "CT", false, 0,
|
|
"")
|
|
pdf.CellFormat(colWidths["Team"], height, surrogateText(match.Blue1IsSurrogate), "LBR", 0, "CT", false, 0,
|
|
"")
|
|
pdf.CellFormat(colWidths["Team"], height, surrogateText(match.Blue2IsSurrogate), "LBR", 0, "CT", false, 0,
|
|
"")
|
|
pdf.CellFormat(colWidths["Team"], height, surrogateText(match.Blue3IsSurrogate), "LBR", 1, "CT", false, 0,
|
|
"")
|
|
pdf.SetFont("Arial", "", 10)
|
|
}
|
|
}
|
|
|
|
if vars["type"] != "elimination" {
|
|
// Render some summary info at the bottom.
|
|
pdf.CellFormat(195, 10, fmt.Sprintf("Matches Per Team: %d", matchesPerTeam), "", 1, "L", false, 0, "")
|
|
}
|
|
|
|
// Write out the PDF file as the HTTP response.
|
|
w.Header().Set("Content-Type", "application/pdf")
|
|
err = pdf.Output(w)
|
|
if err != nil {
|
|
handleWebErr(w, err)
|
|
return
|
|
}
|
|
}
|
|
|
|
// Generates a CSV-formatted report of the team list.
|
|
func (web *Web) teamsCsvReportHandler(w http.ResponseWriter, r *http.Request) {
|
|
teams, err := web.arena.Database.GetAllTeams()
|
|
if err != nil {
|
|
handleWebErr(w, err)
|
|
return
|
|
}
|
|
|
|
// Don't set the content type as "text/csv", as that will trigger an automatic download in the browser.
|
|
w.Header().Set("Content-Type", "text/plain")
|
|
template, err := web.parseFiles("templates/teams.csv")
|
|
if err != nil {
|
|
handleWebErr(w, err)
|
|
return
|
|
}
|
|
err = template.ExecuteTemplate(w, "teams.csv", teams)
|
|
if err != nil {
|
|
handleWebErr(w, err)
|
|
return
|
|
}
|
|
}
|
|
|
|
// Generates a PDF-formatted report of the team list.
|
|
func (web *Web) teamsPdfReportHandler(w http.ResponseWriter, r *http.Request) {
|
|
teams, err := web.arena.Database.GetAllTeams()
|
|
if err != nil {
|
|
handleWebErr(w, err)
|
|
return
|
|
}
|
|
|
|
showHasConnected := r.URL.Query().Get("showHasConnected") == "true"
|
|
|
|
// The widths of the table columns in mm, stored here so that they can be referenced for each row.
|
|
var colWidths map[string]float64
|
|
if showHasConnected {
|
|
colWidths = map[string]float64{"Id": 12, "Name": 70, "Location": 65, "RookieYear": 23, "HasConnected": 25}
|
|
} else {
|
|
colWidths = map[string]float64{"Id": 12, "Name": 80, "Location": 80, "RookieYear": 23}
|
|
}
|
|
rowHeight := 6.5
|
|
|
|
pdf := gofpdf.New("P", "mm", "Letter", "font")
|
|
pdf.AddPage()
|
|
pdf.SetFont("Arial", "B", 10)
|
|
pdf.SetFillColor(220, 220, 220)
|
|
|
|
// Render table header row.
|
|
pdf.CellFormat(195, rowHeight, "Team List - "+web.arena.EventSettings.Name, "", 1, "C", false, 0, "")
|
|
pdf.CellFormat(colWidths["Id"], rowHeight, "Team", "1", 0, "C", true, 0, "")
|
|
pdf.CellFormat(colWidths["Name"], rowHeight, "Name", "1", 0, "C", true, 0, "")
|
|
pdf.CellFormat(colWidths["Location"], rowHeight, "Location", "1", 0, "C", true, 0, "")
|
|
if showHasConnected {
|
|
pdf.CellFormat(colWidths["RookieYear"], rowHeight, "Rookie Year", "1", 0, "C", true, 0, "")
|
|
pdf.CellFormat(colWidths["HasConnected"], rowHeight, "Connected?", "1", 1, "C", true, 0, "")
|
|
} else {
|
|
pdf.CellFormat(colWidths["RookieYear"], rowHeight, "Rookie Year", "1", 1, "C", true, 0, "")
|
|
}
|
|
pdf.SetFont("Arial", "", 10)
|
|
for _, team := range teams {
|
|
// Render team info row.
|
|
pdf.CellFormat(colWidths["Id"], rowHeight, strconv.Itoa(team.Id), "1", 0, "L", false, 0, "")
|
|
pdf.CellFormat(colWidths["Name"], rowHeight, team.Nickname, "1", 0, "L", false, 0, "")
|
|
location := fmt.Sprintf("%s, %s, %s", team.City, team.StateProv, team.Country)
|
|
pdf.CellFormat(colWidths["Location"], rowHeight, location, "1", 0, "L", false, 0, "")
|
|
if showHasConnected {
|
|
pdf.CellFormat(colWidths["RookieYear"], rowHeight, strconv.Itoa(team.RookieYear), "1", 0, "L", false, 0, "")
|
|
var hasConnected string
|
|
if team.HasConnected {
|
|
hasConnected = "Yes"
|
|
}
|
|
pdf.CellFormat(colWidths["HasConnected"], rowHeight, hasConnected, "1", 1, "L", false, 0, "")
|
|
} else {
|
|
pdf.CellFormat(colWidths["RookieYear"], rowHeight, strconv.Itoa(team.RookieYear), "1", 1, "L", false, 0, "")
|
|
}
|
|
}
|
|
|
|
// Write out the PDF file as the HTTP response.
|
|
w.Header().Set("Content-Type", "application/pdf")
|
|
err = pdf.Output(w)
|
|
if err != nil {
|
|
handleWebErr(w, err)
|
|
return
|
|
}
|
|
}
|
|
|
|
// Generates a CSV-formatted report of the WPA keys, for import into the radio kiosk.
|
|
func (web *Web) wpaKeysCsvReportHandler(w http.ResponseWriter, r *http.Request) {
|
|
if !web.userIsAdmin(w, r) {
|
|
return
|
|
}
|
|
|
|
teams, err := web.arena.Database.GetAllTeams()
|
|
if err != nil {
|
|
handleWebErr(w, err)
|
|
return
|
|
}
|
|
|
|
w.Header().Set("Content-Type", "text/csv")
|
|
w.Header().Set("Content-Disposition", "attachment; filename=wpa_keys.csv")
|
|
for _, team := range teams {
|
|
_, err := w.Write([]byte(fmt.Sprintf("%d,%s\r\n", team.Id, team.WpaKey)))
|
|
if err != nil {
|
|
handleWebErr(w, err)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
// Returns the text to display if a team is a surrogate.
|
|
func surrogateText(isSurrogate bool) string {
|
|
if isSurrogate {
|
|
return "(surrogate)"
|
|
} else {
|
|
return ""
|
|
}
|
|
}
|