Skip to content
Snippets Groups Projects
main.go 2.26 KiB
Newer Older
gbe's avatar
gbe committed
package main

import (
	"embed"
gbe's avatar
gbe committed
	"fmt"
gbe's avatar
gbe committed
	"html/template"
	"log"
	"net/http"
gbe's avatar
gbe committed
	"strconv"
gbe's avatar
gbe committed

	"github.com/jmoiron/sqlx"
	_ "modernc.org/sqlite" // Imported for side effects: registers DB driver
gbe's avatar
gbe committed
)

//go:embed templates/*.tpl
var templateFS embed.FS
var templates = template.Must(template.ParseFS(templateFS, "templates/*.tpl"))

//go:embed static/*
var staticFS embed.FS

func main() {
gbe's avatar
gbe committed
	db, err := sqlx.Open("sqlite", "vino.sqlite")
	if err != nil {
		log.Fatalln("can't open DB:", err)
	}
	defer db.Close()

	err = initDB(db)
	if err != nil {
		log.Fatalln("can't initialize DB:", err)
	}

gbe's avatar
gbe committed
	http.Handle("/static/", http.FileServer(http.FS(staticFS)))

	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		log.Println("handling", r.Method, r.URL, "from", r.RemoteAddr)

		if r.Method == "GET" {
gbe's avatar
gbe committed
			wines, err := ListWines(r.Context(), db)
			if err != nil {
				http.Error(w, "can't list wines: "+err.Error(), http.StatusInternalServerError)
				return
			}

gbe's avatar
gbe committed
			w.Header().Add("content-type", "text/html")

gbe's avatar
gbe committed
			data := struct {
				Wines []Vino
			}{wines}

			err = templates.ExecuteTemplate(w, "index.tpl", data)
gbe's avatar
gbe committed
			if err != nil {
				log.Println("can't execute index template:", err)
			}

			return
		}
gbe's avatar
gbe committed

		if r.Method != "POST" {
			http.Error(w, "invalid method:"+r.Method, http.StatusMethodNotAllowed)
			return
		}

		// TODO: load/store image

		name := r.FormValue("name")
		if name == "" {
			http.Error(w, "name empty or missing", http.StatusBadRequest)
			return
		}

		if len(name) > 80 {
			http.Error(w, "name too long, max length is 80", http.StatusBadRequest)
			return
		}

gbe's avatar
gbe committed
		ratingVal := r.FormValue("rating")
		rating, err := strconv.Atoi(ratingVal)
		if err != nil {
			http.Error(w, "can't convert rating:"+err.Error(), http.StatusBadRequest)
			return
		}

		vino := Vino{
			Name:   name,
			Rating: rating,
		}

		err = vino.Store(r.Context(), db)
		if err != nil {
			http.Error(w, fmt.Sprintf("can't store wine %q: %s", vino, err), http.StatusInternalServerError)
			return
		}

		http.Redirect(w, r, "/", http.StatusSeeOther) // TODO: Is this the correct status?
gbe's avatar
gbe committed
	})

	const listenAddr = "127.0.0.1:7878"

	log.Printf("here we go, listening on http://%s", listenAddr)

gbe's avatar
gbe committed
	err = http.ListenAndServe(listenAddr, nil)
gbe's avatar
gbe committed
	if err != nil {
gbe's avatar
gbe committed
		log.Fatalln("http handler failed:", err)
gbe's avatar
gbe committed
	}
}