main.go

package main

import (
	"flag"
	"log"
	"net/http"
	"time"

	"github.com/didip/tollbooth"
	"github.com/didip/tollbooth/limiter"
	"github.com/gorilla/mux"
	"golang.org/x/crypto/acme/autocert"
)

func main() {
	var (
		imagesPath  = flag.String("static", "./images", "path to images")
		bgImagePath = flag.String("bg", "./images/main.png", "path to images")
		fontPath    = flag.String("font", "./Roboto-Regular.ttf", "path to font")
		repoURL     = flag.String("repo", "https://github.com/veggiedefender/typing#this-readme-is-interactive", "URL of repo to redirect to after typing a letter")
		camoURL     = flag.String("camoURL", "", "URL of github proxied screen image")
		enableHTTPS = flag.Bool("https", false, "enable HTTPS")
		certPath    = flag.String("certs", "./cert-cache", "path to letsencrypt autocert cache directory")
		host        = flag.String("host", "kbd.jse.li", "host to listen TLS on")
	)
	flag.Parse()

	if len(*camoURL) == 0 {
		log.Fatal("Pass in GitHub camo URL with -camoURL <url>")
	}

	scrn, err := NewScreen(*bgImagePath, *fontPath)
	if err != nil {
		log.Fatal(err)
	}

	lmt := tollbooth.NewLimiter(5, &limiter.ExpirableOptions{DefaultExpirationTTL: time.Hour})
	lmt.SetIPLookups([]string{"RemoteAddr"})

	r := mux.NewRouter()
	r.PathPrefix("/k/").Handler(http.StripPrefix("/k/", http.FileServer(http.Dir(*imagesPath))))
	r.Handle("/type/{character:[a-z0-9]|backspace|comma|space|period|enter}", tollbooth.LimitFuncHandler(lmt, TypeHandler(scrn, *repoURL)))
	r.Handle("/screen.gif", RenderHandler(scrn))

	srv := &http.Server{
		Handler:      r,
		ReadTimeout:  10 * time.Second,
		WriteTimeout: 20 * time.Second,
		IdleTimeout:  120 * time.Second,
	}

	go purgeGitHubCache(*camoURL)

	if *enableHTTPS {
		m := &autocert.Manager{
			Cache:      autocert.DirCache(*certPath),
			Prompt:     autocert.AcceptTOS,
			HostPolicy: autocert.HostWhitelist(*host),
		}
		srv.Addr = ":https"
		srv.TLSConfig = m.TLSConfig()
		log.Fatal(srv.ListenAndServeTLS("", ""))
	} else {
		srv.Addr = ":8000"
		log.Fatal(srv.ListenAndServe())
	}
}