67 lines
1.4 KiB
Go
67 lines
1.4 KiB
Go
package msg
|
|
|
|
import (
|
|
"github.com/gorilla/sessions"
|
|
"github.com/labstack/echo-contrib/session"
|
|
"github.com/labstack/echo/v4"
|
|
)
|
|
|
|
type Type string
|
|
|
|
const (
|
|
TypeSuccess Type = "success"
|
|
TypeInfo Type = "info"
|
|
TypeWarning Type = "warning"
|
|
TypeDanger Type = "danger"
|
|
)
|
|
|
|
// TODO: Error handling and cleanup
|
|
|
|
func Success(c echo.Context, message string) {
|
|
Set(c, TypeSuccess, message)
|
|
}
|
|
|
|
func Info(c echo.Context, message string) {
|
|
Set(c, TypeInfo, message)
|
|
}
|
|
|
|
func Warning(c echo.Context, message string) {
|
|
Set(c, TypeWarning, message)
|
|
}
|
|
|
|
func Danger(c echo.Context, message string) {
|
|
Set(c, TypeDanger, message)
|
|
}
|
|
|
|
// Set adds a new message into the cookie storage.
|
|
func Set(c echo.Context, typ Type, message string) {
|
|
sess := getSession(c)
|
|
sess.AddFlash(message, 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
|
|
}
|
|
|
|
func getSession(c echo.Context) *sessions.Session {
|
|
sess, _ := session.Get("msg", c)
|
|
return sess
|
|
}
|