add registration and post count

handlers.go

@@ -2,6 +2,7 @@ package main

import (
	"errors"
	"labix.org/v2/mgo/bson"
	"net/http"
)

@@ -44,6 +45,12 @@ func sign(w http.ResponseWriter, req *http.Request, ctx *Context) (err error) {
		return
	}

	//ignore errors: it's ok if the post count is wrong. we can always look at
	//the entries table to fix.
	ctx.C("users").Update(bson.M{"_id": ctx.User.ID}, bson.M{
		"$inc": bson.M{"posts": 1},
	})

	http.Redirect(w, req, reverse("index"), http.StatusSeeOther)
	return
}
@@ -74,3 +81,29 @@ func logout(w http.ResponseWriter, req *http.Request, ctx *Context) error {
	http.Redirect(w, req, reverse("index"), http.StatusSeeOther)
	return nil
}

func registerForm(w http.ResponseWriter, req *http.Request, ctx *Context) (err error) {
	return T("register.html").Execute(w, map[string]interface{}{
		"ctx": ctx,
	})
}

func register(w http.ResponseWriter, req *http.Request, ctx *Context) error {
	username, password := req.FormValue("username"), req.FormValue("password")

	u := &User{
		Username: username,
		ID:       bson.NewObjectId(),
	}
	u.SetPassword(password)

	if err := ctx.C("users").Insert(u); err != nil {
		ctx.Session.AddFlash("Problem registering user.")
		return registerForm(w, req, ctx)
	}

	//store the user id in the values and redirect to index
	ctx.Session.Values["user"] = u.ID
	http.Redirect(w, req, reverse("index"), http.StatusSeeOther)
	return nil
}

main.go

@@ -42,13 +42,27 @@ func main() {
	}
	database = session.DB("").Name

	//create an index for the username field on the users collection
	if err := session.DB("").C("users").EnsureIndex(mgo.Index{
		Key:    []string{"username"},
		Unique: true,
	}); err != nil {
		panic(err)
	}

	store = sessions.NewCookieStore([]byte(os.Getenv("KEY")))

	router = pat.New()
	router.Add("GET", "/login", handler(loginForm)).Name("login")
	router.Add("GET", "/logout", handler(logout)).Name("logout")
	router.Add("POST", "/login", handler(login))

	router.Add("GET", "/register", handler(registerForm)).Name("register")
	router.Add("POST", "/register", handler(register))

	router.Add("GET", "/logout", handler(logout)).Name("logout")

	router.Add("GET", "/", handler(hello)).Name("index")

	router.Add("POST", "/sign", handler(sign)).Name("sign")

	if err = http.ListenAndServe(":"+os.Getenv("PORT"), router); err != nil {

templates/index.html

@@ -22,6 +22,9 @@ <h1>Guestbook</h1>
        <p><button>Sign</button></p>
    </form>
    {{ else }}
        <h1>Please <a href="{{ reverse "login" }}">login</a> to sign.</h1>
        <h1>Please
            <a href="{{ reverse "login" }}">login</a> or
            <a href="{{ reverse "register" }}">register</a>
        to sign.</h1>
    {{ end }}
{{ end }}

templates/register.html

@@ -0,0 +1,15 @@
{{ define "title" }}Guestbook - Register{{ end }}

{{ define "content" }}
    <h1>Register</h1>

    {{ range .ctx.Session.Flashes }}
        <h2>{{ . }}</h2>
    {{ end }}

    <form action="{{ reverse "register" }}" method="POST">
        <p>Username: <input type="text" name="username"></p>
        <p>Password: <input type="password" name="password"></p>
        <p><button>Login</button></p>
    </form>
{{ end }}