package main
import (
"errors"
"html/template"
"log"
"net/http"
"github.com/masterminds/sprig"
)
const (
addr = ":8080"
)
type data map[string]any
/*
@app.route("/")
def index():
*/
func index(w http.ResponseWriter, r *http.Request) {
http.Redirect(w, r, "/contacts", http.StatusSeeOther)
}
/*
@app.route("/contacts")
*/
func contacts(cs *ContactsStore) http.HandlerFunc {
tpl := makeTemplate(
"layout",
"index",
)
return func(w http.ResponseWriter, r *http.Request) {
search := r.URL.Query().Get("q")
var contacts_set []Contact
if search == "" {
contacts_set = cs.GetAll()
} else {
contacts_set = cs.Get(search)
}
if err := tpl.ExecuteTemplate(w, "layout.html", data{
"contacts": contacts_set,
"search": search,
}); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
}
}
}
/*
@app.route("/contacts/new", methods=['GET']) (1)
def contacts_new_get():
return render_template("new.html", contact=Contact()
*/
func contactsNewGet(cs *ContactsStore) http.HandlerFunc {
tpl := makeTemplate(
"layout",
"new",
)
return func(w http.ResponseWriter, r *http.Request) {
c := cs.New()
if err := tpl.ExecuteTemplate(w, "layout.html", c); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
}
}
}
func makeTemplate(first string, others ...string) *template.Template {
paths := append([]string{first}, others...)
for i, path := range paths {
paths[i] = "./templates/" + path + ".gohtml"
}
tpl := template.Must(template.ParseFiles(paths...)).
Funcs(sprig.FuncMap())
return tpl
}
func setupRoutes(mux *http.ServeMux, cs *ContactsStore) {
mux.HandleFunc("/", index)
mux.HandleFunc("/favicon.ico", func(w http.ResponseWriter, r *http.Request) {
http.ServeFile(w, r, "./static/img/favicon.ico")
})
mux.Handle("GET /static/", http.StripPrefix("/static/", http.FileServer(http.Dir("./static"))))
mux.Handle("GET /contacts", contacts(cs))
mux.Handle("GET /contacts/new", contactsNewGet(cs))
}
func main() {
cs, err := NewContactsStore()
if err != nil {
log.Fatalf("initializing contacts: %v\n", err)
}
mux := http.NewServeMux()
setupRoutes(mux, cs)
log.Printf("Listening on http://localhost%s", addr)
if err := http.ListenAndServe(addr, mux); err != nil {
if !errors.Is(err, http.ErrServerClosed) {
log.Printf("Server error: %v\n", err)
return
}
}
log.Println("Server shutdown")
}