server.go

package main

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

	"github.com/valyala/fasthttp"
)

var (
	addr     = flag.String("h", "0.0.0.0:8080", "TCP address to listen to")
	root     = flag.String("d", "/usr/share/httpd", "Directory to serve static files from")
	strip    = flag.Int("s", 0, "Number of subdir to remove from the query")
	compress = flag.Bool("c", false, "Enables transparent response compression if set to true")
)

var fsHandler fasthttp.RequestHandler

func reponseNotFound(ctx *fasthttp.RequestCtx) {
	ctx.Response.SetStatusCode(http.StatusNotFound)
	ctx.Response.SetBody([]byte("Not Found\n"))
}

func notFoundHandler(ctx *fasthttp.RequestCtx) {
	if strings.HasSuffix(string(ctx.Request.RequestURI()), ".map") {
		reponseNotFound(ctx)
		return
	}
	ctx.Logger().Printf("File %s not found, defaulting to index.html", ctx.Path())
	ctx.Request.SetRequestURI("/index.html")
	fsHandler(ctx)
}

func createFsHandler() fasthttp.RequestHandler {
	fs := &fasthttp.FS{
		Root:               *root,
		Compress:           *compress,
		IndexNames:         []string{"index.html"},
		PathNotFound:       notFoundHandler,
		PathRewrite:        fasthttp.NewPathSlashesStripper(*strip),
		GenerateIndexPages: false,
		AcceptByteRange:    true,
	}
	return fs.NewRequestHandler()
}

func main() {
	flag.Parse()
	fsHandler = createFsHandler()

	// Start HTTP server.
	if len(*addr) > 0 {
		log.Printf("Starting HTTP server on %q", *addr)
		log.Printf("Serving files from directory %q", *root)
		if err := fasthttp.ListenAndServe(*addr, fsHandler); err != nil {
			log.Fatalf("error in ListenAndServe: %s", err)
		}
	}
}