package main

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

func accessLog(h http.Handler, quiet bool) http.Handler {
	t := &TemplateHandler{handler: h}
	fn := func(w http.ResponseWriter, r *http.Request) {
		if !quiet {
			log.Println(r.Method, r.URL.Path, r.RemoteAddr)
		}
		t.ServeHTTP(w, r)
	}
	return http.HandlerFunc(fn)
}

func main() {
	ip_f := flag.String("ip", "0.0.0.0", "IP adress to listen on.")
	port_f := flag.String("port", "3000", "Port to listen on.")
	dir_f := flag.String("dir", ".", "Directory to serve.")
	disallow_upl_f := flag.Bool("disallow-upload", false, "Disallow uploads to /upload.")
	upl_dir_f := flag.String("upload-dir", ".", "Directory to store uploaded files.")
	quiet_f := flag.Bool("quiet", false, "Be quiet, suppress output.")

	flag.Parse()

	port := os.Getenv("PORT")
	if port == "" {
		port = *port_f
	}

	if !*quiet_f {
		fmt.Println("Starting uhttpd serving \"" + *dir_f + "\" on " + *ip_f + ":" + port + ".")
	}

	mux := http.NewServeMux()
	if !*disallow_upl_f {
		os.MkdirAll(*upl_dir_f, 0755)
		mux.Handle("/upload", uploadHandler(*upl_dir_f, *quiet_f))
	}
	mux.Handle("/", accessLog(http.FileServer(http.Dir(*dir_f)), *quiet_f))
	log.Fatal(http.ListenAndServe(*ip_f+":"+port, mux))
}