Initial commit.

msg/msg.go

@@ -0,0 +1,49 @@
+package msg
+
+import (
+	"github.com/gorilla/sessions"
+	"github.com/labstack/echo-contrib/session"
+	"github.com/labstack/echo/v4"
+)
+
+type Type string
+
+const (
+	Success Type = "success"
+	Info    Type = "info"
+	Warning Type = "warning"
+	Danger  Type = "danger"
+)
+
+// TODO: Error handling
+
+func getSession(c echo.Context) *sessions.Session {
+	sess, _ := session.Get("msg", c)
+	return sess
+}
+
+// Set adds a new message into the cookie storage.
+func Set(c echo.Context, typ Type, value string) {
+	sess := getSession(c)
+	sess.AddFlash(value, string(typ))
+	_ = sess.Save(c.Request(), c.Response())
+}
+
+// Get gets flash messages from the cookie storage.
+func Get(c echo.Context, typ Type) []string {
+	sess := getSession(c)
+	fm := sess.Flashes(string(typ))
+	// If we have some messages.
+	if len(fm) > 0 {
+		_ = sess.Save(c.Request(), c.Response())
+		// Initiate a strings slice to return messages.
+		var flashes []string
+		for _, fl := range fm {
+			// Add message to the slice.
+			flashes = append(flashes, fl.(string))
+		}
+
+		return flashes
+	}
+	return nil
+}