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)
}