41 lines
1005 B
Go
41 lines
1005 B
Go
|
package html
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"forge.cadoles.com/wpetit/goweb/service"
|
||
|
|
"forge.cadoles.com/wpetit/goweb/service/session"
|
||
|
|
"forge.cadoles.com/wpetit/goweb/service/template"
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
)
|
||
|
|
|
||
|
|
// WithFlashes extends the template's data with session's flash messages
|
||
|
|
func WithFlashes(w http.ResponseWriter, r *http.Request, container *service.Container) template.DataExtFunc {
|
||
|
|
return func(data template.Data) (template.Data, error) {
|
||
|
|
|
||
|
|
sessionService, err := session.From(container)
|
||
|
|
if err != nil {
|
||
|
|
return nil, errors.Wrap(err, "error while retrieving session service")
|
||
|
|
}
|
||
|
|
|
||
|
|
sess, err := sessionService.Get(w, r)
|
||
|
|
if err != nil {
|
||
|
|
return nil, errors.Wrap(err, "error while retrieving session")
|
||
|
|
}
|
||
|
|
|
||
|
|
flashes := sess.Flashes(
|
||
|
|
session.FlashError, session.FlashWarn,
|
||
|
|
session.FlashSuccess, session.FlashInfo,
|
||
|
|
)
|
||
|
|
|
||
|
|
data["Flashes"] = flashes
|
||
|
|
|
||
|
|
if err := sess.Save(w, r); err != nil {
|
||
|
|
return nil, errors.Wrap(err, "error while saving session")
|
||
|
|
}
|
||
|
|
|
||
|
|
return data, nil
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|