mirror of
https://github.com/kataras/iris.git
synced 2025-01-23 18:51:03 +01:00
c3205dafa1
Read HISTORY.md it contains a breaking change, second parameter of HandleDir should be iris.Dir(...) instead of just a string relative to: https://github.com/kataras/iris/issues/1556#issuecomment-661057446 Former-commit-id: 14b48a06fb3b99287dff543932be2937a64233b9
124 lines
3.0 KiB
Go
124 lines
3.0 KiB
Go
package bootstrap
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/gorilla/securecookie"
|
|
|
|
"github.com/kataras/iris/v12"
|
|
"github.com/kataras/iris/v12/middleware/logger"
|
|
"github.com/kataras/iris/v12/middleware/recover"
|
|
"github.com/kataras/iris/v12/sessions"
|
|
"github.com/kataras/iris/v12/websocket"
|
|
)
|
|
|
|
type Configurator func(*Bootstrapper)
|
|
|
|
type Bootstrapper struct {
|
|
*iris.Application
|
|
AppName string
|
|
AppOwner string
|
|
AppSpawnDate time.Time
|
|
|
|
Sessions *sessions.Sessions
|
|
}
|
|
|
|
// New returns a new Bootstrapper.
|
|
func New(appName, appOwner string, cfgs ...Configurator) *Bootstrapper {
|
|
b := &Bootstrapper{
|
|
AppName: appName,
|
|
AppOwner: appOwner,
|
|
AppSpawnDate: time.Now(),
|
|
Application: iris.New(),
|
|
}
|
|
|
|
for _, cfg := range cfgs {
|
|
cfg(b)
|
|
}
|
|
|
|
return b
|
|
}
|
|
|
|
// SetupViews loads the templates.
|
|
func (b *Bootstrapper) SetupViews(viewsDir string) {
|
|
b.RegisterView(iris.HTML(viewsDir, ".html").Layout("shared/layout.html"))
|
|
}
|
|
|
|
// SetupSessions initializes the sessions, optionally.
|
|
func (b *Bootstrapper) SetupSessions(expires time.Duration, cookieHashKey, cookieBlockKey []byte) {
|
|
b.Sessions = sessions.New(sessions.Config{
|
|
Cookie: "SECRET_SESS_COOKIE_" + b.AppName,
|
|
Expires: expires,
|
|
Encoding: securecookie.New(cookieHashKey, cookieBlockKey),
|
|
})
|
|
}
|
|
|
|
// SetupWebsockets prepares the websocket server.
|
|
func (b *Bootstrapper) SetupWebsockets(endpoint string, handler websocket.ConnHandler) {
|
|
ws := websocket.New(websocket.DefaultGorillaUpgrader, handler)
|
|
|
|
b.Get(endpoint, websocket.Handler(ws))
|
|
}
|
|
|
|
// SetupErrorHandlers prepares the http error handlers
|
|
// `(context.StatusCodeNotSuccessful`, which defaults to >=400 (but you can change it).
|
|
func (b *Bootstrapper) SetupErrorHandlers() {
|
|
b.OnAnyErrorCode(func(ctx iris.Context) {
|
|
err := iris.Map{
|
|
"app": b.AppName,
|
|
"status": ctx.GetStatusCode(),
|
|
"message": ctx.Values().GetString("message"),
|
|
}
|
|
|
|
if jsonOutput := ctx.URLParamExists("json"); jsonOutput {
|
|
ctx.JSON(err)
|
|
return
|
|
}
|
|
|
|
ctx.ViewData("Err", err)
|
|
ctx.ViewData("Title", "Error")
|
|
ctx.View("shared/error.html")
|
|
})
|
|
}
|
|
|
|
const (
|
|
// StaticAssets is the root directory for public assets like images, css, js.
|
|
StaticAssets = "./public/"
|
|
// Favicon is the relative 9to the "StaticAssets") favicon path for our app.
|
|
Favicon = "favicon.ico"
|
|
)
|
|
|
|
// Configure accepts configurations and runs them inside the Bootstraper's context.
|
|
func (b *Bootstrapper) Configure(cs ...Configurator) {
|
|
for _, c := range cs {
|
|
c(b)
|
|
}
|
|
}
|
|
|
|
// Bootstrap prepares our application.
|
|
//
|
|
// Returns itself.
|
|
func (b *Bootstrapper) Bootstrap() *Bootstrapper {
|
|
b.SetupViews("./views")
|
|
b.SetupSessions(24*time.Hour,
|
|
[]byte("the-big-and-secret-fash-key-here"),
|
|
[]byte("lot-secret-of-characters-big-too"),
|
|
)
|
|
b.SetupErrorHandlers()
|
|
|
|
// static files
|
|
b.Favicon(StaticAssets + Favicon)
|
|
b.HandleDir("/public", iris.Dir(StaticAssets))
|
|
|
|
// middleware, after static files
|
|
b.Use(recover.New())
|
|
b.Use(logger.New())
|
|
|
|
return b
|
|
}
|
|
|
|
// Listen starts the http server with the specified "addr".
|
|
func (b *Bootstrapper) Listen(addr string, cfgs ...iris.Configurator) {
|
|
b.Run(iris.Addr(addr), cfgs...)
|
|
}
|