cake-order-tracker/router/handler.go

package router

import (
	"fmt"
	"html/template"
	"net/http"
	"strconv"
	"strings"
	"time"

	"github.com/Bronku/iroon/store"
)

type Server struct {
	tmpl *template.Template
	s    *store.Store
	http.Handler
}

func (h *Server) Close() {
	if h.s != nil {
		h.s.Close()
	}
}

func (h *Server) openStore() error {
	var err error
	h.s, err = store.OpenStore("./foo.db")
	if err != nil {
		h.s.Close()
	}
	return err
}

// #todo embed templates, and load them in init function
func (h *Server) loadTemplates() error {
	var err error
	h.tmpl, err = template.ParseFiles("index.html", "order.html")
	return err
}

func (h *Server) loadHandler() {
	mux := http.NewServeMux()

	mux.HandleFunc("GET /order/", h.Form)
	mux.HandleFunc("GET /", h.Index)
	mux.HandleFunc("POST /", h.AddOrder)

	h.Handler = mux
}

func New() (*Server, error) {
	var server Server
	var err error

	err = server.loadTemplates()
	if err != nil {
		return nil, err
	}

	err = server.openStore()
	if err != nil {
		return nil, err
	}

	server.loadHandler()

	return &server, nil
}

func (h *Server) Form(w http.ResponseWriter, r *http.Request) {
	url := strings.Split(r.URL.String(), "/")
	o := store.Order{
		ID:   -1,
		Date: time.Now(),
	}
	id, err := strconv.Atoi(url[2])
	if err == nil {
		newOrder, err := h.s.GetOrder(id)
		if err == nil {
			o = newOrder
		}
	}
	type formData struct {
		Order     store.Order
		Catalogue []store.Cake
	}

	cakes, err := h.s.GetCakes()
	if err != nil {
		fmt.Fprint(w, "server side error getting available cakes: ", err)
		return
	}
	data := formData{o, cakes}
	//fmt.Println(data)

	w.Header().Set("content-type", "text/html")
	err = h.tmpl.ExecuteTemplate(w, "order.html", data)
	if err != nil {
		fmt.Println("error executing the template: ", err)
	}
}

func (h *Server) AddOrder(w http.ResponseWriter, r *http.Request) {
	err := r.ParseForm()
	if err != nil {
		fmt.Println("can't parse the form")
		w.WriteHeader(http.StatusBadRequest)
		return
	}
	fmt.Println("received form: ", r.Form)

	var n store.Order
	n.ID, err = strconv.Atoi(r.FormValue("id"))
	if err != nil {
		w.WriteHeader(http.StatusBadRequest)
		fmt.Println("can't parse order id: ", err)
		return
	}

	n.Name = strings.TrimSpace(r.FormValue("name"))

	n.Surname = strings.TrimSpace(r.FormValue("surname"))

	n.Phone = strings.TrimSpace(r.FormValue("phone"))

	n.Location = strings.TrimSpace(r.FormValue("location"))

	n.Date, err = time.Parse("2006-01-02", r.FormValue("date"))
	if err != nil {
		w.WriteHeader(http.StatusBadRequest)
		fmt.Println("can't parse order date: ", err)
		return
	}

	n.Status = strings.TrimSpace(r.FormValue("status"))

	n.Paid, err = strconv.Atoi(r.FormValue("paid"))
	if err != nil {
		w.WriteHeader(http.StatusBadRequest)
		fmt.Println("can't parse order paid: ", err)
		return
	}

	n.Accepted = time.Now()

	n.Cakes = make([]store.Cake, 0)
	cakes, err := h.s.GetCakes()
	if err != nil {
		fmt.Fprint(w, "server side error getting available cakes: ", err)
		return
	}
	for _, e := range cakes {
		value := r.FormValue(fmt.Sprintf("cake[%d]", e.ID))
		if value == "" {
			continue
		}
		amount, err := strconv.Atoi(value)
		e.Amount = amount
		if err != nil {
			continue
		}
		n.Cakes = append(n.Cakes, e)
	}

	fmt.Println("parsed order: ", n)

	h.s.SaveOrder(n)

	w.Header().Set("content-type", "text/html")
	w.WriteHeader(http.StatusAccepted)
	w.Write([]byte("accepted <a href='/'>back</a>"))
}

func (h *Server) Index(w http.ResponseWriter, r *http.Request) {
	w.Header().Set("content-type", "text/html")
	orders, err := h.s.GetOrders()
	if err != nil {
		fmt.Fprint(w, "server side error getting orders: ", err)
		return
	}
	err = h.tmpl.ExecuteTemplate(w, "index.html", orders)
	if err != nil {
		fmt.Println("error executing the template: ", err)
	}
}