serve/main.go

package main

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

var options struct {
	port     int
	hostname string
}

func bail(status int, template string, args ...interface{}) {
	if !strings.HasSuffix(template, "\n") {
		template += "\n"
	}
	if status == 0 {
		fmt.Fprintf(os.Stdout, template, args...)
	} else {
		fmt.Fprintf(os.Stderr, template, args...)
	}
	os.Exit(status)
}

func main() {
	flag.Parse()
	cwd, err := os.Getwd()
	if err != nil {
		bail(1, "unable to get working directory: %v", err)
	}
	dir := http.Dir(cwd)
	http.Handle("/", &logWrapper{http.FileServer(dir)})

	addr := fmt.Sprintf("%s:%d", options.hostname, options.port)
	if err := http.ListenAndServe(addr, nil); err != nil {
		bail(1, "unable to start server: %v", err)
	}
}

func init() {
	flag.IntVar(&options.port, "port", 8000, "port to serve on")
	flag.StringVar(&options.hostname, "host", "", "hostname")
	log.SetOutput(os.Stdout)
	log.SetFlags(log.Ldate | log.Lmicroseconds)
}