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/posts"
	"github.com/kirsle/blog/core/models/settings"
	"github.com/kirsle/blog/core/models/users"
	"github.com/shurcooL/github_flavored_markdown/gfmstyle"
	"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

	// Helper singletone
	Posts *PostHelper

	// 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")),
	}
	blog.Posts = InitPostHelper(blog)

	// Load the site config, or start with defaults if not found.
	settings.DB = blog.DB
	config, err := settings.Load()
	if err != nil {
		config = settings.Defaults()
	}

	// Initialize the session cookie store.
	blog.store = sessions.NewCookieStore([]byte(config.Security.SecretKey))
	users.HashCost = config.Security.HashCost

	// Initialize the rest of the models.
	posts.DB = blog.DB
	users.DB = blog.DB

	// Initialize the router.
	r := mux.NewRouter()
	r.HandleFunc("/initial-setup", blog.SetupHandler)
	r.HandleFunc("/login", blog.LoginHandler)
	r.HandleFunc("/logout", blog.LogoutHandler)
	blog.AdminRoutes(r)
	blog.BlogRoutes(r)

	// GitHub Flavored Markdown CSS.
	r.Handle("/css/gfm.css", http.StripPrefix("/css", http.FileServer(gfmstyle.Assets)))

	r.PathPrefix("/").HandlerFunc(blog.PageHandler)
	r.NotFoundHandler = http.HandlerFunc(blog.PageHandler)

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

	// Keep references handy elsewhere in the app.
	blog.n = n
	blog.r = 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)
}