89 lines
2.1 KiB
Go
89 lines
2.1 KiB
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/settings"
|
|
"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")),
|
|
}
|
|
|
|
// 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.
|
|
users.DB = blog.DB
|
|
|
|
// Initialize the router.
|
|
r := mux.NewRouter()
|
|
blog.r = r
|
|
|
|
// Blog setup.
|
|
r.HandleFunc("/admin/setup", blog.SetupHandler)
|
|
|
|
// Admin pages that require a logged-in user.
|
|
admin := mux.NewRouter()
|
|
admin.HandleFunc("/admin", blog.AdminHandler)
|
|
r.PathPrefix("/admin").Handler(negroni.New(
|
|
negroni.HandlerFunc(blog.LoginRequired),
|
|
negroni.Wrap(admin),
|
|
))
|
|
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)
|
|
}
|