97 lines
2.3 KiB
Go
97 lines
2.3 KiB
Go
package msg
|
|
|
|
import (
|
|
"github.com/gorilla/sessions"
|
|
"github.com/labstack/echo/v4"
|
|
"github.com/camzawacki/personal-site/pkg/log"
|
|
"github.com/camzawacki/personal-site/pkg/session"
|
|
)
|
|
|
|
// Type is a message type.
|
|
type Type string
|
|
|
|
const (
|
|
// TypeSuccess represents a success message type.
|
|
TypeSuccess Type = "success"
|
|
|
|
// TypeInfo represents a info message type.
|
|
TypeInfo Type = "info"
|
|
|
|
// TypeWarning represents a warning message type.
|
|
TypeWarning Type = "warning"
|
|
|
|
// TypeError represents an error message type.
|
|
TypeError Type = "error"
|
|
)
|
|
|
|
const (
|
|
// sessionName stores the name of the session which contains flash messages.
|
|
sessionName = "msg"
|
|
)
|
|
|
|
// Success sets a success flash message.
|
|
func Success(ctx echo.Context, message string) {
|
|
Set(ctx, TypeSuccess, message)
|
|
}
|
|
|
|
// Info sets an info flash message.
|
|
func Info(ctx echo.Context, message string) {
|
|
Set(ctx, TypeInfo, message)
|
|
}
|
|
|
|
// Warning sets a warning flash message.
|
|
func Warning(ctx echo.Context, message string) {
|
|
Set(ctx, TypeWarning, message)
|
|
}
|
|
|
|
// Error sets an error flash message.
|
|
func Error(ctx echo.Context, message string) {
|
|
Set(ctx, TypeError, message)
|
|
}
|
|
|
|
// Set adds a new flash message of a given type into the session storage.
|
|
// Errors will be logged and not returned.
|
|
func Set(ctx echo.Context, typ Type, message string) {
|
|
if sess, err := getSession(ctx); err == nil {
|
|
sess.AddFlash(message, string(typ))
|
|
save(ctx, sess)
|
|
}
|
|
}
|
|
|
|
// Get gets flash messages of a given type from the session storage.
|
|
// Errors will be logged and not returned.
|
|
func Get(ctx echo.Context, typ Type) []string {
|
|
if sess, err := getSession(ctx); err == nil {
|
|
if flash := sess.Flashes(string(typ)); len(flash) > 0 {
|
|
save(ctx, sess)
|
|
|
|
msgs := make([]string, 0, len(flash))
|
|
for _, m := range flash {
|
|
msgs = append(msgs, m.(string))
|
|
}
|
|
return msgs
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// getSession gets the flash message session.
|
|
func getSession(ctx echo.Context) (*sessions.Session, error) {
|
|
sess, err := session.Get(ctx, sessionName)
|
|
if err != nil {
|
|
log.Ctx(ctx).Error("cannot load flash message session",
|
|
"error", err,
|
|
)
|
|
}
|
|
return sess, err
|
|
}
|
|
|
|
// save saves the flash message session.
|
|
func save(ctx echo.Context, sess *sessions.Session) {
|
|
if err := sess.Save(ctx.Request(), ctx.Response()); err != nil {
|
|
log.Ctx(ctx).Error("failed to set flash message",
|
|
"error", err,
|
|
)
|
|
}
|
|
}
|