simpleserver/server.go

package simpleserver

import (
	"fmt"
	"io/ioutil"
	"net/http"
	"os"
	"os/signal"
	"path/filepath"
)

func viewHandler(root string, w http.ResponseWriter, r *http.Request) {
	file := filepath.Join(root, filepath.FromSlash(r.URL.Path[1:]))
	info, err := os.Stat(file)
	if err != nil {
		fmt.Fprint(w, err)
		return
	}
	if info.IsDir() {
		file = filepath.Join(file, "index.html")
	}
	fmt.Println("GET " + file)
	content, _ := ioutil.ReadFile(file)
	fmt.Fprint(w, string(content))
}

func Serve(root string) {
	address := "127.0.0.1:8000"
	fmt.Println("Serving " + root + " at http://" + address + " ...")
	fmt.Println("Use Ctrl+C to exit")
	fmt.Println("(This is not a production web server, don't get any funny ideas)")
	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { viewHandler(root, w, r) })
	srv := &http.Server{Addr: address}
	c := make(chan os.Signal, 1)
	signal.Notify(c, os.Interrupt)
	go func() {
		<-c
		fmt.Println("Received interrupt, closing...")
		srv.Close()
	}()
	err := srv.ListenAndServe()
	if err != nil && err != http.ErrServerClosed {
		fmt.Fprintln(os.Stderr, err)
	}
}