Noah
ceb42aa4d0
* List comments on a post * Add comments, with preview * Users can edit their own comments (EditToken) * Admin can edit all comments * Delete comments * Comment counts on main blog index pages
66 lines
1.6 KiB
Go
66 lines
1.6 KiB
Go
package gophertype
|
|
|
|
import (
|
|
"html/template"
|
|
"net/http"
|
|
|
|
"git.kirsle.net/apps/gophertype/pkg/console"
|
|
"git.kirsle.net/apps/gophertype/pkg/controllers"
|
|
"git.kirsle.net/apps/gophertype/pkg/models"
|
|
"git.kirsle.net/apps/gophertype/pkg/responses"
|
|
"git.kirsle.net/apps/gophertype/pkg/settings"
|
|
"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(pubroot string) *Site {
|
|
// Initialize the settings.json inside the user root.
|
|
if err := settings.SetFilename(pubroot); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
site := &Site{}
|
|
|
|
n := negroni.New()
|
|
n.Use(negroni.NewRecovery())
|
|
n.Use(negroni.NewLogger())
|
|
site.n = n
|
|
|
|
// Register blog global template functions.
|
|
responses.ExtraFuncs = template.FuncMap{
|
|
"BlogIndex": controllers.PartialBlogIndex,
|
|
"RenderComments": controllers.RenderComments,
|
|
"RenderCommentsRO": controllers.RenderCommentsRO,
|
|
"RenderComment": controllers.RenderComment,
|
|
"RenderCommentForm": controllers.RenderCommentForm,
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
console.Info("Using database driver '%s'", driver)
|
|
models.UseDB(db)
|
|
return nil
|
|
}
|
|
|
|
// ListenAndServe starts the HTTP service.
|
|
func (s *Site) ListenAndServe(addr string) error {
|
|
console.Info("Listening on %s", addr)
|
|
return http.ListenAndServe(addr, s.n)
|
|
}
|