authenticator.go

// Only contains code for running the service. Brings together logic from various
// other services.
package main

import (
	"database/sql"
	"log"
	"net/http"
	"text/template"

	"github.com/codegangsta/negroni"
	"github.com/coopernurse/gorp"
	_ "github.com/go-sql-driver/mysql"
	"github.com/gorilla/context"
	"github.com/gorilla/sessions"
	"github.com/julienschmidt/httprouter"
	"github.com/nyc-camp/authenticator/libtmpl"
	"github.com/nyc-camp/authenticator/libuser"
)

var userStorageMySQL UserStorageMySQL
var sessionStorage sessions.Store

func init() {
	db, err := sql.Open("mysql", "authenticator:authenticator@tcp(localhost:3306)/authenticator?parseTime=true")
	if err != nil {
		log.Printf("%v\n", err)
		log.Panic("An error occured while connecting to the database.")
	}

	dbmap := &gorp.DbMap{Db: db, Dialect: gorp.MySQLDialect{"InnoDB", "UTF8"}}
	dbmap.AddTableWithName(libuser.User{}, "user").SetKeys(false, "uid")
	userStorageMySQL = UserStorageMySQL{Dbmap: dbmap}
	sessionStorage = sessions.NewCookieStore([]byte("7Iow7KmwXj5x9e3q41396e4pd1A31Rme"), []byte("3vC4APPo2HoBY9AhguVz8EU24D0n0I5G"))
}

func main() {
	tmplCfg := libtmpl.HTMLTemplateConfig{TemplateDir: "templates/", DefaultErrorFunc: libuser.HandleError}
	router := httprouter.New()
	router.NotFound = func(w http.ResponseWriter, r *http.Request) {
		t, err := template.ParseFiles("templates/404.html")
		if err != nil {
			return
		}

		t.ExecuteTemplate(w, "html", nil)
	}
	recoveryMiddleware := negroni.NewRecovery()
	recoveryMiddleware.PrintStack = false
	n := negroni.New(recoveryMiddleware, negroni.NewLogger(), negroni.NewStatic(http.Dir("public")))

	userRegistration := libuser.UserRegistration{
		Storage:        userStorageMySQL,
		TemplateConfig: tmplCfg,
		SessionStore:   sessionStorage,
	}
	userLogin := libuser.UserLogin{
		Storage:        userStorageMySQL,
		TemplateConfig: tmplCfg,
		SessionStore:   sessionStorage,
	}
	userAccount := libuser.UserAccount{
		Storage:        userStorageMySQL,
		TemplateConfig: tmplCfg,
		SessionStore:   sessionStorage,
	}
	userLogout := libuser.UserLogout{
		Storage:        userStorageMySQL,
		TemplateConfig: tmplCfg,
		SessionStore:   sessionStorage,
	}

	router.GET("/register", userRegistration.GetRegistrationForm)
	router.POST("/register", userRegistration.HandleRegistrationSubmission)
	router.GET("/login", userLogin.LoginForm)
	router.POST("/login", userLogin.LoginSubmission)
	router.GET("/account", userAccount.AccountPage)
	router.GET("/logout", userLogout.Logout)

	n.UseHandler(context.ClearHandler(router))

	n.Run(":4567")
}