package main

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

	"github.com/gorilla/mux"
)

func main() {

	// define the flags
	ip := flag.String("ip", "0.0.0.0", "ip to listen on")
	port := flag.String("p", "8087", "port to listen on")
	dir := flag.String("dir", "./", "path to host")
	flag.Parse()

	// define the router
	r := mux.NewRouter()
	r.Use(loggingMiddleware)

	// define the subrouter hosting the previously defined directory
	files := r.PathPrefix("/").Subrouter()
	fs := http.FileServer(http.Dir(*dir))
	files.PathPrefix("/").Handler(http.StripPrefix("/", fs))

	// define the http server using the values defined using flags
	httpServer := http.Server{
		Addr:    fmt.Sprintf("%s:%s", *ip, *port),
		Handler: r,
	}

	// start the http server
	log.Printf("http server listening on %s:%s hosting %s", *ip, *port, *dir)
	log.Fatal(httpServer.ListenAndServe())
}

func loggingMiddleware(next http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		log.Println(r.RequestURI)
		next.ServeHTTP(w, r)
	})
}