moth/cmd/mothd/handlers.go

293 lines
6.6 KiB
Go
Raw Normal View History

2018-05-02 15:45:45 -06:00
package main
import (
2018-09-17 17:00:08 -06:00
"bufio"
2018-09-24 14:00:18 -06:00
"encoding/json"
2018-05-02 15:45:45 -06:00
"fmt"
2018-09-17 18:02:44 -06:00
"io"
2018-09-17 17:00:08 -06:00
"log"
2018-05-02 15:45:45 -06:00
"net/http"
2018-05-04 17:20:51 -06:00
"os"
"strconv"
2018-09-17 17:00:08 -06:00
"strings"
2018-05-02 15:45:45 -06:00
)
2019-02-22 17:43:04 -07:00
// https://github.com/omniti-labs/jsend
2018-09-24 14:00:18 -06:00
type JSend struct {
2019-02-25 09:07:53 -07:00
Status string `json:"status"`
2019-02-22 17:43:04 -07:00
Data struct {
Short string `json:"short"`
Description string `json:"description"`
2019-02-25 09:07:53 -07:00
} `json:"data"`
2018-09-24 14:00:18 -06:00
}
const (
2019-02-22 17:43:04 -07:00
JSendSuccess = "success"
2019-02-25 09:07:53 -07:00
JSendFail = "fail"
JSendError = "error"
)
2018-09-24 14:00:18 -06:00
2019-02-22 17:43:04 -07:00
func respond(w http.ResponseWriter, req *http.Request, status string, short string, format string, a ...interface{}) {
resp := JSend{}
resp.Status = status
resp.Data.Short = short
resp.Data.Description = fmt.Sprintf(format, a...)
2018-09-24 14:00:18 -06:00
respBytes, err := json.Marshal(resp)
if err != nil {
2018-09-24 14:00:18 -06:00
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
2018-09-24 14:00:18 -06:00
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(http.StatusOK) // RFC2616 makes it pretty clear that 4xx codes are for the user-agent
w.Write(respBytes)
}
2018-09-17 21:32:24 -06:00
// hasLine returns true if line appears in r.
// The entire line must match.
func hasLine(r io.Reader, line string) bool {
2018-09-17 18:02:44 -06:00
scanner := bufio.NewScanner(r)
for scanner.Scan() {
2018-09-17 21:32:24 -06:00
if scanner.Text() == line {
2018-09-17 18:02:44 -06:00
return true
}
}
return false
}
2018-09-19 21:44:34 -06:00
func (ctx *Instance) registerHandler(w http.ResponseWriter, req *http.Request) {
teamName := req.FormValue("name")
teamId := req.FormValue("id")
2018-09-17 17:00:08 -06:00
if !ctx.ValidTeamId(teamId) {
2018-09-17 17:00:08 -06:00
respond(
2019-02-22 17:43:04 -07:00
w, req, JSendFail,
2018-09-17 17:00:08 -06:00
"Invalid Team ID",
"I don't have a record of that team ID. Maybe you used capital letters accidentally?",
)
2018-05-04 17:20:51 -06:00
return
}
2018-09-17 17:00:08 -06:00
f, err := os.OpenFile(ctx.StatePath("teams", teamId), os.O_CREATE|os.O_EXCL|os.O_WRONLY, 0644)
if err != nil {
if os.IsExist(err) {
respond(
w, req, JSendFail,
"Already registered",
"This team ID has already been registered.",
)
} else {
log.Print(err)
respond(
w, req, JSendFail,
"Registration failed",
"Unable to register. Perhaps a teammate has already registered?",
)
}
2018-05-04 17:20:51 -06:00
return
}
defer f.Close()
fmt.Fprintln(f, teamName)
2018-09-17 17:00:08 -06:00
respond(
2019-02-22 17:43:04 -07:00
w, req, JSendSuccess,
2018-09-17 17:00:08 -06:00
"Team registered",
"Your team has been named and you may begin using your team ID!",
2018-09-17 17:00:08 -06:00
)
2018-05-04 17:20:51 -06:00
}
2018-09-19 21:44:34 -06:00
func (ctx *Instance) answerHandler(w http.ResponseWriter, req *http.Request) {
teamId := req.FormValue("id")
category := req.FormValue("cat")
pointstr := req.FormValue("points")
answer := req.FormValue("answer")
2018-05-04 17:20:51 -06:00
if ! ctx.ValidTeamId(teamId) {
respond(
w, req, JSendFail,
"Invalid team ID",
"That team ID is not valid for this event.",
)
return
}
if ctx.TooFast(teamId) {
respond(
w, req, JSendFail,
"Submitting too quickly",
"Your team can only submit one answer every %v", ctx.AttemptInterval,
)
return
}
2018-05-04 17:20:51 -06:00
points, err := strconv.Atoi(pointstr)
if err != nil {
2018-09-17 17:00:08 -06:00
respond(
2019-02-22 17:43:04 -07:00
w, req, JSendFail,
"Cannot parse point value",
"This doesn't look like an integer: %s", pointstr,
2018-09-17 17:00:08 -06:00
)
2018-09-14 18:24:48 -06:00
return
2018-05-04 17:20:51 -06:00
}
haystack, err := ctx.OpenCategoryFile(category, "answers.txt")
2018-09-14 18:24:48 -06:00
if err != nil {
2018-09-17 17:00:08 -06:00
respond(
2019-02-22 17:43:04 -07:00
w, req, JSendFail,
"Cannot list answers",
"Unable to read the list of answers for this category.",
2018-09-17 17:00:08 -06:00
)
2018-09-14 18:24:48 -06:00
return
}
defer haystack.Close()
2018-09-17 17:00:08 -06:00
2018-09-14 18:24:48 -06:00
// Look for the answer
needle := fmt.Sprintf("%d %s", points, answer)
2018-09-17 21:32:24 -06:00
if !hasLine(haystack, needle) {
2018-09-17 17:00:08 -06:00
respond(
2019-02-22 17:43:04 -07:00
w, req, JSendFail,
2018-09-17 17:00:08 -06:00
"Wrong answer",
"That is not the correct answer for %s %d.", category, points,
2018-09-17 17:00:08 -06:00
)
2018-09-14 18:24:48 -06:00
return
2018-05-04 17:20:51 -06:00
}
if err := ctx.AwardPoints(teamId, category, points); err != nil {
2018-09-17 17:00:08 -06:00
respond(
2019-02-22 17:43:04 -07:00
w, req, JSendError,
"Cannot award points",
"The answer is correct, but there was an error awarding points: %v", err.Error(),
2018-09-17 17:00:08 -06:00
)
2018-05-04 17:20:51 -06:00
return
}
2018-09-17 17:00:08 -06:00
respond(
2019-02-22 17:43:04 -07:00
w, req, JSendSuccess,
2018-09-17 17:00:08 -06:00
"Points awarded",
fmt.Sprintf("%d points for %s!", points, teamId),
2018-09-17 17:00:08 -06:00
)
}
2018-09-19 21:44:34 -06:00
func (ctx *Instance) puzzlesHandler(w http.ResponseWriter, req *http.Request) {
teamId := req.FormValue("id")
if _, err := ctx.TeamName(teamId); err != nil {
http.Error(w, "Must provide team ID", http.StatusUnauthorized)
2019-02-22 19:09:38 -07:00
return
}
2018-09-17 17:00:08 -06:00
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(http.StatusOK)
2018-09-19 21:44:34 -06:00
w.Write(ctx.jPuzzleList)
2018-05-08 12:45:50 -06:00
}
2018-09-19 21:44:34 -06:00
func (ctx *Instance) pointsHandler(w http.ResponseWriter, req *http.Request) {
2018-09-17 17:00:08 -06:00
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(http.StatusOK)
2018-09-19 21:44:34 -06:00
w.Write(ctx.jPointsLog)
2018-09-17 17:00:08 -06:00
}
2018-09-19 21:44:34 -06:00
func (ctx *Instance) contentHandler(w http.ResponseWriter, req *http.Request) {
2018-09-19 17:56:26 -06:00
// Prevent directory traversal
if strings.Contains(req.URL.Path, "/.") {
http.Error(w, "Not Found", http.StatusNotFound)
return
}
2018-09-19 21:44:34 -06:00
2018-09-19 17:56:26 -06:00
// Be clever: use only the last three parts of the path. This may prove to be a bad idea.
parts := strings.Split(req.URL.Path, "/")
if len(parts) < 3 {
http.Error(w, "Not Found", http.StatusNotFound)
return
}
2018-09-19 21:44:34 -06:00
2018-09-19 17:56:26 -06:00
fileName := parts[len(parts)-1]
puzzleId := parts[len(parts)-2]
categoryName := parts[len(parts)-3]
2018-09-19 21:44:34 -06:00
mb, ok := ctx.categories[categoryName]
2018-09-19 17:56:26 -06:00
if !ok {
http.Error(w, "Not Found", http.StatusNotFound)
return
}
2018-09-19 21:44:34 -06:00
2018-09-19 17:56:26 -06:00
mbFilename := fmt.Sprintf("content/%s/%s", puzzleId, fileName)
mf, err := mb.Open(mbFilename)
if err != nil {
log.Print(err)
http.Error(w, "Not Found", http.StatusNotFound)
return
}
defer mf.Close()
2018-09-19 21:44:34 -06:00
2018-09-19 17:56:26 -06:00
http.ServeContent(w, req, fileName, mf.ModTime(), mf)
}
2018-09-19 21:44:34 -06:00
func (ctx *Instance) staticHandler(w http.ResponseWriter, req *http.Request) {
path := req.URL.Path
if strings.Contains(path, "..") {
http.Error(w, "Invalid URL path", http.StatusBadRequest)
return
}
if path == "/" {
path = "/index.html"
}
f, err := os.Open(ctx.ThemePath(path))
if err != nil {
http.NotFound(w, req)
return
}
defer f.Close()
d, err := f.Stat()
if err != nil {
http.NotFound(w, req)
return
}
http.ServeContent(w, req, path, d.ModTime(), f)
2018-05-06 21:37:52 -06:00
}
2018-09-14 18:24:48 -06:00
type FurtiveResponseWriter struct {
2019-02-25 09:07:53 -07:00
w http.ResponseWriter
statusCode *int
}
func (w FurtiveResponseWriter) WriteHeader(statusCode int) {
*w.statusCode = statusCode
w.w.WriteHeader(statusCode)
}
func (w FurtiveResponseWriter) Write(buf []byte) (n int, err error) {
n, err = w.w.Write(buf)
return
2018-09-14 18:24:48 -06:00
}
func (w FurtiveResponseWriter) Header() http.Header {
return w.w.Header()
}
// This gives Instances the signature of http.Handler
func (ctx *Instance) ServeHTTP(wOrig http.ResponseWriter, r *http.Request) {
w := FurtiveResponseWriter{
2019-02-25 09:07:53 -07:00
w: wOrig,
statusCode: new(int),
}
2019-02-22 19:09:38 -07:00
ctx.mux.ServeHTTP(w, r)
log.Printf(
"%s %s %s %d\n",
r.RemoteAddr,
r.Method,
r.URL,
*w.statusCode,
)
}
func (ctx *Instance) BindHandlers() {
ctx.mux.HandleFunc(ctx.Base+"/", ctx.staticHandler)
ctx.mux.HandleFunc(ctx.Base+"/register", ctx.registerHandler)
ctx.mux.HandleFunc(ctx.Base+"/answer", ctx.answerHandler)
ctx.mux.HandleFunc(ctx.Base+"/content/", ctx.contentHandler)
ctx.mux.HandleFunc(ctx.Base+"/puzzles.json", ctx.puzzlesHandler)
ctx.mux.HandleFunc(ctx.Base+"/points.json", ctx.pointsHandler)
}