package gophertype

import (
	"log"
	"net/http"

	"git.kirsle.net/apps/gophertype/pkg/models"
	"github.com/gorilla/mux"
	"github.com/jinzhu/gorm"
	"github.com/urfave/negroni"
)

// Site is the master struct for the Gophertype server.
type Site struct {
	n   *negroni.Negroni
	mux *mux.Router
}

// NewSite initializes the Site.
func NewSite() *Site {
	site := &Site{}

	n := negroni.New()
	n.Use(negroni.NewRecovery())
	n.Use(negroni.NewLogger())
	site.n = n

	return site
}

// UseDB specifies the database to use.
func (s *Site) UseDB(driver string, path string) error {
	db, err := gorm.Open(driver, path)
	if err != nil {
		return err
	}

	log.Printf("Using database driver '%s'", driver)
	models.UseDB(db)
	return nil
}

// ListenAndServe starts the HTTP service.
func (s *Site) ListenAndServe(addr string) error {
	return http.ListenAndServe(addr, s.n)
}