serve.go

   1 // Taken from https://github.com/fogleman/serve and repurposed as a library
   2 package aya
   3
   4 import (
   5         "fmt"
   6         "log"
   7         "net/http"
   8 )
   9
  10 // ResponseWriter wraps http.ResponseWriter to capture the HTTP status code
  11 type ResponseWriter struct {
  12         http.ResponseWriter
  13         StatusCode int
  14 }
  15
  16 func (w *ResponseWriter) WriteHeader(statusCode int) {
  17         w.StatusCode = statusCode
  18         w.ResponseWriter.WriteHeader(statusCode)
  19 }
  20
  21 // Handler wraps http.Handler to log served files
  22 type Handler struct {
  23         http.Handler
  24 }
  25
  26 func (h *Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
  27         rw := &ResponseWriter{w, 0}
  28         h.Handler.ServeHTTP(rw, r)
  29         log.Println(r.RemoteAddr, r.Method, rw.StatusCode, r.URL)
  30 }
  31
  32 // This function is called by the `aya serve` subcommand
  33 func HttpServe(Dir string, Port int) {
  34         handler := &Handler{http.FileServer(http.Dir(Dir))}
  35         http.Handle("/", handler)
  36         addr := fmt.Sprintf("127.0.0.1:%d", Port)
  37         log.Printf("[aya.HttpServe] Listening on %s\n", addr)
  38         log.Fatal(http.ListenAndServe(addr, nil))
  39 }