camzalewski/personal-site
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Added easier setter funcs for flash messages.

Browse source
This commit is contained in:
mikestefanello 2021-12-03 16:44:04 -05:00
parent fc1d3cdc33
commit d40244b3d1
4 changed files with 30 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

4
controllers/contact.go
View file

@ -20,7 +20,7 @@ func (a *Contact) Get(c echo.Context) error {
} }
func (a *Contact) Post(c echo.Context) error { func (a *Contact) Post(c echo.Context) error {
msg.Set(c, msg.Success, "Thank you for contacting us!") msg.Success(c, "Thank you for contacting us!")
msg.Set(c, msg.Info, "We will respond to you shortly.") msg.Info(c, "We will respond to you shortly.")
return a.Redirect(c, "home") return a.Redirect(c, "home")
} }

2
controllers/login.go
View file

@ -20,6 +20,6 @@ func (l *Login) Get(c echo.Context) error {
} }
func (l *Login) Post(c echo.Context) error { func (l *Login) Post(c echo.Context) error {
msg.Set(c, msg.Danger, "Invalid credentials. Please try again.") msg.Danger(c, "Invalid credentials. Please try again.")
return l.Get(c) return l.Get(c)
} }

2
controllers/register.go
View file

@ -20,6 +20,6 @@ func (r *Register) Get(c echo.Context) error {
} }
func (r *Register) Post(c echo.Context) error { func (r *Register) Post(c echo.Context) error {
msg.Set(c, msg.Danger, "Registration is currently disabled.") msg.Danger(c, "Registration is currently disabled.")
return r.Get(c) return r.Get(c)
} }

36
msg/msg.go
View file

@ -9,23 +9,34 @@ import (
type Type string type Type string
const ( const (
Success Type = "success" TypeSuccess Type = "success"
Info Type = "info" TypeInfo Type = "info"
Warning Type = "warning" TypeWarning Type = "warning"
Danger Type = "danger" TypeDanger Type = "danger"
) )
// TODO: Error handling // TODO: Error handling and cleanup
func getSession(c echo.Context) *sessions.Session { func Success(c echo.Context, message string) {
sess, _ := session.Get("msg", c) Set(c, TypeSuccess, message)
return sess }
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. // Set adds a new message into the cookie storage.
func Set(c echo.Context, typ Type, value string) { func Set(c echo.Context, typ Type, message string) {
sess := getSession(c) sess := getSession(c)
sess.AddFlash(value, string(typ)) sess.AddFlash(message, string(typ))
_ = sess.Save(c.Request(), c.Response()) _ = sess.Save(c.Request(), c.Response())
} }
@ -47,3 +58,8 @@ func Get(c echo.Context, typ Type) []string {
} }
return nil return nil
} }
func getSession(c echo.Context) *sessions.Session {
sess, _ := session.Get("msg", c)
return sess
}