add ability to share session messages

inertia.go

@@ -10,33 +10,36 @@ import (
 	"strings"
 
 	"github.com/gofiber/fiber/v2"
+	"github.com/gofiber/fiber/v2/middleware/session"
 )
 
 // Inertia type.
 type Inertia struct {
 	url           string
 	rootTemplate  string
 	version       string
+	store         *session.Store
 	sharedProps   map[string]interface{}
 	sharedFuncMap template.FuncMap
 	templateFS    fs.FS
 }
 
 // New function.
-func New(url, rootTemplate, version string) *Inertia {
+func New(url, rootTemplate, version string, store *session.Store) *Inertia {
 	i := new(Inertia)
 	i.url = url
 	i.rootTemplate = rootTemplate
 	i.version = version
+	i.store = store
 	i.sharedProps = make(map[string]interface{})
 	i.sharedFuncMap = template.FuncMap{"marshal": marshal}
 
 	return i
 }
 
 // NewWithFS function.
-func NewWithFS(url, rootTemplate, version string, templateFS fs.FS) *Inertia {
-	i := New(url, rootTemplate, version)
+func NewWithFS(url, rootTemplate, version string, store session.Session, templateFS fs.FS) *Inertia {
+	i := New(url, rootTemplate, version, store)
 	i.templateFS = templateFS
 
 	return i

middleware.go

@@ -15,6 +15,15 @@ func (i *Inertia) Middleware(c *fiber.Ctx) error {
 		return c.SendStatus(fiber.StatusConflict)
 	}
 
+	sess, _ := i.store.Get(c)
+
+	message := sess.Get("message")
+
+	if message != nil {
+		i.Share("message", sess.Get("message"))
+		sess.Delete("message")
+	}
+
 	// Go to next middleware:
 	return c.Next()
 }