mirror of https://github.com/dirtbags/moth.git
88 lines
1.6 KiB
Go
88 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"log"
|
|
"math/rand"
|
|
"mime"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
func setup() error {
|
|
rand.Seed(time.Now().UnixNano())
|
|
return nil
|
|
}
|
|
|
|
func main() {
|
|
ctx := &Instance{}
|
|
|
|
flag.StringVar(
|
|
&ctx.Base,
|
|
"base",
|
|
"/",
|
|
"Base URL of this instance",
|
|
)
|
|
flag.StringVar(
|
|
&ctx.MothballDir,
|
|
"mothballs",
|
|
"/mothballs",
|
|
"Path to read mothballs",
|
|
)
|
|
flag.StringVar(
|
|
&ctx.StateDir,
|
|
"state",
|
|
"/state",
|
|
"Path to write state",
|
|
)
|
|
flag.StringVar(
|
|
&ctx.ThemeDir,
|
|
"theme",
|
|
"/theme",
|
|
"Path to static theme resources (HTML, images, css, ...)",
|
|
)
|
|
flag.DurationVar(
|
|
&ctx.AttemptInterval,
|
|
"attempt",
|
|
500*time.Millisecond,
|
|
"Per-team time required between answer attempts",
|
|
)
|
|
maintenanceInterval := flag.Duration(
|
|
"maint",
|
|
20*time.Second,
|
|
"Time between maintenance tasks",
|
|
)
|
|
flag.BoolVar(
|
|
&ctx.UseXForwarded,
|
|
"x-forwarded-for",
|
|
false,
|
|
"Emit IPs from the X-Forwarded-For header in logs, when available, instead of the source IP. Use this when running behind a load-balancer or proxy",
|
|
)
|
|
listen := flag.String(
|
|
"listen",
|
|
":8080",
|
|
"[host]:port to bind and listen",
|
|
)
|
|
flag.Parse()
|
|
|
|
if err := setup(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
err := ctx.Initialize()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer ctx.eventLogWriter.Close()
|
|
|
|
// Add some MIME extensions
|
|
// Doing this avoids decompressing a mothball entry twice per request
|
|
mime.AddExtensionType(".json", "application/json")
|
|
mime.AddExtensionType(".zip", "application/zip")
|
|
|
|
go ctx.Maintenance(*maintenanceInterval)
|
|
|
|
log.Printf("Listening on %s", *listen)
|
|
log.Fatal(http.ListenAndServe(*listen, ctx))
|
|
}
|