02.02-01.gist

main.go

package main

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

	"snippetbox.org/pkg/handlers/web"
)

var (
	addr   = flag.String("addr", ":4000", "HTTP service address")
	public = flag.String("public", "./public", "Public asset directory")
)

func main() {
	flag.Parse()

	f, err := os.OpenFile("/tmp/error.log", os.O_RDWR|os.O_CREATE|os.O_RDWR, 0666)
	if err != nil {
		log.Fatal(err)
	}
	defer f.Close()

	errorLog := log.New(f, "ERROR\t", log.Ldate|log.Ltime|log.Llongfile)

	fileServer := http.FileServer(http.Dir(*public))

	mux := http.NewServeMux()
	mux.HandleFunc("/", web.Home)
	mux.HandleFunc("/snippets", web.Create)
	mux.HandleFunc("/snippets/", web.Show)
	mux.Handle("/public/", http.StripPrefix("/public", fileServer))

	srv := http.Server{
		Addr:     *addr,
		ErrorLog: errorLog,
		Handler:  mux,
	}
	err = srv.ListenAndServe()
	errorLog.Fatal(err)
}