package main

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

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

/*
func TemplateHandler(h http.Handler) http.Handler {
	h.handler
}
*/

func uploadHandler(dir string) http.Handler {
	fn := func(w http.ResponseWriter, r *http.Request) {
		file, header, err := r.FormFile("file")

		if err != nil {
			w.WriteHeader(500)
			w.Write([]byte(err.Error()))
			log.Println("ERROR", err.Error())
			return
		}

		defer file.Close()

		if !strings.HasSuffix(dir, "/") {
			dir = dir + "/"
		}

		out, err := os.Create(dir + header.Filename)
		if err != nil {
			w.WriteHeader(500)
			w.Write([]byte(err.Error()))
			log.Println("ERROR", err.Error())
			return
		}

		defer out.Close()
		_, err = io.Copy(out, file)
		if err != nil {
			w.WriteHeader(500)
			w.Write([]byte(err.Error()))
			log.Println("ERROR", err.Error())
			return
		}

		log.Println(r.Method, r.URL.Path, header.Filename, r.RemoteAddr)
		w.Write([]byte("Uploaded " + header.Filename))
	}
	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.")

	flag.Parse()

	port := os.Getenv("PORT")
	if port == "" {
		port = *port_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))
	}
	mux.Handle("/", accessLog(http.FileServer(http.Dir(*dir_f))))
	log.Fatal(http.ListenAndServe(*ip_f+":"+port, mux))
}