Quick and dirty file server

statichttp.go

package main

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

var (
    flagLog  = flag.Bool("log", false, "Log requests")
    flagBind = flag.String("bind", ":9999", "Bind to address")
    flagDir  = flag.String("dir", ".", "Directory to serve")
)

type loggingHandler struct {
    http.Handler
}

type loggingWriter struct {
    http.ResponseWriter
    code int
    wh   bool
}

func (l *loggingWriter) WriteHeader(code int) {
    l.code = code
    l.wh = true
    l.ResponseWriter.WriteHeader(code)
}

func (l *loggingWriter) Write(p []byte) (int, error) {
    if !l.wh {
        l.WriteHeader(200)
    }
    return l.ResponseWriter.Write(p)
}

func (l loggingHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    lw := loggingWriter{ResponseWriter: w}
    l.Handler.ServeHTTP(&lw, r)
    log.Println(r.Host, r.RequestURI, lw.code)
}

func main() {
    flag.Parse()

    h := http.FileServer(http.Dir(*flagDir))
    if *flagLog {
        h = loggingHandler{h}
    }
    log.Fatal(http.ListenAndServe(*flagBind, h))
}