Defined own handler type to reduce repetive jobs.

1 files changed, 65 insertions, 0 deletions

diff --git a/handler.go b/handler.go
new file mode 100644
index 0000000..d710b2a
--- /dev/null
+++ b/handler.go
@@ -0,0 +1,65 @@
+package main
+
+import (
+	"github.com/gorilla/sessions"
+	"html/template"
+	"kch42.de/gostuff/mailremind/model"
+	"log"
+	"net/http"
+)
+
+type Handler func(user model.User, sess *sessions.Session, req *http.Request) interface{}
+
+func getSess(req *http.Request) (*sessions.Session, error) {
+	sess, err := SessionStorage.Get(req, "mailremind-sess")
+	if err != nil {
+		sess, err = SessionStorage.New(req, "mailremind-sess")
+	}
+
+	return sess, err
+}
+
+func userFromSess(sess *sessions.Session) model.User {
+	_id, ok := sess.Values["uid"]
+	if !ok {
+		return nil
+	}
+
+	id, ok := _id.(string)
+	if !ok {
+		return nil
+	}
+
+	uid, err := db.ParseDBID(id)
+	if err != nil {
+		return nil
+	}
+
+	user, err := dbcon.UserByID(uid)
+	if err != nil {
+		return nil
+	}
+
+	return user
+}
+
+func mkHttpHandler(h Handler, tpl *template.Template) http.HandlerFunc {
+	return func(rw http.ResponseWriter, req *http.Request) {
+		sess, err := getSess(req)
+		if err != nil {
+			log.Printf("Error while getting session: %s", err)
+			rw.Write([]byte("Unable to create session")) // TODO: Better error message...
+		}
+
+		user := userFromSess(sess)
+		outdata := h(user, sess, req)
+
+		if err := sess.Save(req, rw); err != nil {
+			log.Printf("Error while saving session: %s", err)
+		}
+
+		if err := tpl.Execute(rw, outdata); err != nil {
+			log.Printf("Error executing template %s: %s", tpl.Name, err)
+		}
+	}
+}