30 lines
522 B
Go
30 lines
522 B
Go
package app
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/yourorg/ntfywui/internal/security"
|
|
)
|
|
|
|
func (s *Server) setFlash(w http.ResponseWriter, r *http.Request, msg string) {
|
|
sess, ok := s.sessions.Get(r)
|
|
if !ok {
|
|
sess = &security.Session{}
|
|
}
|
|
sess.Flash = msg
|
|
_ = s.sessions.Save(w, sess)
|
|
}
|
|
|
|
func (s *Server) popFlash(w http.ResponseWriter, r *http.Request) string {
|
|
sess, ok := s.sessions.Get(r)
|
|
if !ok {
|
|
return ""
|
|
}
|
|
msg := sess.Flash
|
|
if msg != "" {
|
|
sess.Flash = ""
|
|
_ = s.sessions.Save(w, sess)
|
|
}
|
|
return msg
|
|
}
|