コードを借りる

Signed-off-by: Izuru Yakumo <yakumo.izuru@chaotic.ninja>

git-svn-id: file:///srv/svn/repo/aya/trunk@73 cec141ff-132a-4243-88a5-ce187bd62f94

serve.go

@@ -0,0 +1,38 @@
+// Taken from https://github.com/fogleman/serve and repurposed as a library
+package aya
+
+import (
+	"fmt"
+	"log"
+	"net/http"
+)
+
+// ResponseWriter wraps http.ResponseWriter to capture the HTTP status code
+type ResponseWriter struct {
+	http.ResponseWriter
+	StatusCode int
+}
+
+func (w *ResponseWriter) WriteHeader(statusCode int) {
+	w.StatusCode = statusCode
+	w.ResponseWriter.WriteHeader(statusCode)
+}
+
+// Handler wraps http.Handler to log served files
+type Handler struct {
+	http.Handler
+}
+
+func (h *Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	rw := &ResponseWriter{w, 0}
+	h.Handler.ServeHTTP(rw, r)
+	log.Println(r.RemoteAddr, r.Method, rw.StatusCode, r.URL)
+}
+
+func HttpServe(Dir string) {
+	handler := &Handler{http.FileServer(http.Dir(Dir))}
+	http.Handle("/", handler)
+	addr := fmt.Sprintf(":%d", 8000)
+	log.Printf("Listening on %s\n", addr)
+	log.Fatal(http.ListenAndServe(addr, nil))
+}