fix https://github.com/kataras/iris/issues/1542

Former-commit-id: 23f795a6b6b64745311d0630d8e630edc3f65244

_examples/bootstrapper/bootstrap/bootstrapper.go

@@ -0,0 +1,123 @@
+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(StaticAssets[1:len(StaticAssets)-1], 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...)
+}