package main

import (
	"context"
	"flag"
	"fmt"
	"log"
	"net/http"
	"os"
	"time"
    "nonsense-time/api"
)


var logger *log.Logger

func timeoutMiddleware(next http.Handler, duration time.Duration) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		ctx, cancel := context.WithTimeout(r.Context(), duration)
		defer cancel()

		next.ServeHTTP(w, r.WithContext(ctx))
	})
}

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")
	debug := flag.Bool("v", false, "verbose output")

	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()

	addr := fmt.Sprintf("%s:%d", *bindAddr, *port)
	loggerFlags := 0
	loggerPrefix := ""
	if *debug {
        loggerPrefix = "Nonsense Time: "
		loggerFlags = log.Ldate | log.Ltime | log.Lshortfile
	}
	logger = log.New(os.Stdout, loggerPrefix, loggerFlags)

	vtt := timeoutMiddleware(http.HandlerFunc(api.VttOnline), *waitTime)
	site := timeoutMiddleware(http.HandlerFunc(api.SiteOnline), *waitTime)

	mux.Handle("GET /vtt/status", vtt)
	mux.HandleFunc("GET /vtt", api.VttRedirect)
	mux.Handle("GET /site/status", site)

	logger.Println("Listening on ", addr)
	logger.Fatal(http.ListenAndServe(addr, mux))
}