-
Notifications
You must be signed in to change notification settings - Fork 0
/
session.go
69 lines (59 loc) · 1.88 KB
/
session.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
package csession
import (
"net/http"
"github.com/pkg/errors"
"admini.dev/admini/app/lib/user"
"admini.dev/admini/app/util"
)
const (
WebAuthKey = "auth"
WebFlashKey = "flash"
ReferKey = "refer"
)
func NewCookie(v string) *http.Cookie {
return &http.Cookie{Name: util.AppKey, Value: v, Path: "/", MaxAge: 365 * 24 * 60 * 60, HttpOnly: true, SameSite: http.SameSiteLaxMode}
}
func StoreInSession(k string, v string, w http.ResponseWriter, websess util.ValueMap, logger util.Logger) error {
websess[k] = v
return SaveSession(w, websess, logger)
}
func RemoveFromSession(k string, w http.ResponseWriter, websess util.ValueMap, logger util.Logger) error {
delete(websess, k)
return SaveSession(w, websess, logger)
}
func SaveSession(w http.ResponseWriter, websess util.ValueMap, logger util.Logger) error {
js := util.ToJSONCompact(websess)
enc, err := util.EncryptMessage(nil, js, logger)
if err != nil {
return err
}
http.SetCookie(w, NewCookie(enc))
return nil
}
func GetFromSession(key string, websess util.ValueMap) (string, error) {
value, ok := websess[key]
if !ok {
return "", errors.Errorf("could not find a matching session value with key [%s] for this request", key)
}
s, ok := value.(string)
if !ok {
return "", errors.Errorf("session value with key [%s] is of type [%T], not [string]", key, value)
}
return s, nil
}
func SaveProfile(n *user.Profile, w http.ResponseWriter, sess util.ValueMap, logger util.Logger) error {
if n != nil && n.Name == "" {
n.Name = user.DefaultProfile.Name
}
if n == nil || n.Equals(user.DefaultProfile) {
return errors.Wrap(RemoveFromSession("profile", w, sess, logger), "unable to remove profile from session")
}
if n.Name == user.DefaultProfile.Name {
n.Name = ""
}
err := StoreInSession("profile", util.ToJSON(n), w, sess, logger)
if err != nil {
return errors.Wrap(err, "unable to save profile in session")
}
return nil
}