blog/core/app.go

package core

import (
	"net/http"
	"path/filepath"

	"github.com/gorilla/mux"
	"github.com/gorilla/sessions"
	"github.com/kirsle/blog/core/jsondb"
	"github.com/kirsle/blog/core/models/users"
	"github.com/urfave/negroni"
)

// Blog is the root application object that maintains the app configuration
// and helper objects.
type Blog struct {
	Debug bool

	// DocumentRoot is the core static files root; UserRoot masks over it.
	DocumentRoot string
	UserRoot     string

	DB *jsondb.DB

	// Web app objects.
	n     *negroni.Negroni // Negroni middleware manager
	r     *mux.Router      // Router
	store sessions.Store
}

// New initializes the Blog application.
func New(documentRoot, userRoot string) *Blog {
	blog := &Blog{
		DocumentRoot: documentRoot,
		UserRoot:     userRoot,
		DB:           jsondb.New(filepath.Join(userRoot, ".private")),

		store: sessions.NewCookieStore([]byte("secret-key")), // TODO configurable!
	}

	// Initialize all the models.
	users.DB = blog.DB

	r := mux.NewRouter()
	blog.r = r
	r.HandleFunc("/admin/setup", blog.SetupHandler)
	r.HandleFunc("/login", blog.LoginHandler)
	r.HandleFunc("/logout", blog.LogoutHandler)
	r.HandleFunc("/", blog.PageHandler)
	r.NotFoundHandler = http.HandlerFunc(blog.PageHandler)

	n := negroni.New(
		negroni.NewRecovery(),
		negroni.NewLogger(),
	)
	blog.n = n
	n.Use(negroni.HandlerFunc(blog.AuthMiddleware))
	n.UseHandler(r)

	return blog
}

// ListenAndServe begins listening on the given bind address.
func (b *Blog) ListenAndServe(address string) {
	log.Info("Listening on %s", address)
	http.ListenAndServe(address, b.n)
}