package main import ( "flag" "fmt" "log/slog" "net/http" "nonsense-time/api" "nonsense-time/middleware" "os" "time" ) var logger *slog.Logger func main() { port := flag.Int("p", 8080, "the port to listen on") bindAddr := flag.String("b", "", "the adress to bind to (leave empty for all interfaces)") waitTime := flag.Duration("w", 2*time.Second, "the maximum time for a request, unit defaults to ns") logLevel := flag.String("l", "info", "log level (debug, info, warn, error)") flag.Usage = func() { fmt.Fprintf(os.Stderr, "Usage: %s [options]\n", os.Args[0]) fmt.Fprintln(os.Stderr, "Options:") flag.PrintDefaults() } flag.Parse() mux := http.NewServeMux() loggerOpts := new(slog.HandlerOptions) switch *logLevel { case "debug": loggerOpts.Level = slog.LevelDebug loggerOpts.AddSource = true case "info": loggerOpts.Level = slog.LevelInfo case "warn": loggerOpts.Level = slog.LevelDebug case "error": loggerOpts.Level = slog.LevelError default: panic(fmt.Sprintf("Unkown log level %s", *logLevel)) } addr := fmt.Sprintf("%s:%d", *bindAddr, *port) logger = slog.New(slog.NewTextHandler(os.Stdout, loggerOpts)) api.Logger = logger vtt := middleware.Timeout(http.HandlerFunc(api.VttOnline), *waitTime) vttLogs := middleware.Gzip(http.HandlerFunc(api.VttLogs)) mux.Handle("GET /vtt/status", vtt) mux.HandleFunc("GET /vtt", api.VttRedirect) mux.Handle("GET /vtt/logs", vttLogs) logger.Info(fmt.Sprint("Listening on ", addr)) logger.Info(http.ListenAndServe(addr, mux).Error()) }