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
70
71
72
73
74
75
76

package csession

import (
	"github.com/pkg/errors"
	"github.com/valyala/fasthttp"

	"{{{ .Package }}}/app/lib/user"
	"{{{ .Package }}}/app/util"
)

const ({{{ if .HasAccount }}}
	WebAuthKey  = "auth"{{{ end }}}
	WebFlashKey = "flash"
	ReferKey    = "refer"
)

func NewCookie(v string) *fasthttp.Cookie {
	ret := &fasthttp.Cookie{}
	ret.SetPath("/")
	ret.SetHTTPOnly(true)
	ret.SetMaxAge(365 * 24 * 60 * 60)
	ret.SetSameSite(fasthttp.CookieSameSiteLaxMode)
	ret.SetKey(util.AppKey)
	ret.SetValue(v)
	return ret
}

func StoreInSession(k string, v string, rc *fasthttp.RequestCtx, websess util.ValueMap, logger util.Logger) error {
	websess[k] = v
	return SaveSession(rc, websess, logger)
}

func RemoveFromSession(k string, rc *fasthttp.RequestCtx, websess util.ValueMap, logger util.Logger) error {
	delete(websess, k)
	return SaveSession(rc, websess, logger)
}

func SaveSession(rc *fasthttp.RequestCtx, websess util.ValueMap, logger util.Logger) error {
	js := util.ToJSONCompact(websess)
	enc, err := util.EncryptMessage(nil, js, logger)
	if err != nil {
		return err
	}
	c := NewCookie(enc)
	rc.Response.Header.SetCookie(c)
	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, rc *fasthttp.RequestCtx, 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", rc, sess, logger), "unable to remove profile from session")
	}
	if n.Name == user.DefaultProfile.Name {
		n.Name = ""
	}
	err := StoreInSession("profile", util.ToJSON(n), rc, sess, logger)
	if err != nil {
		return errors.Wrap(err, "unable to save profile in session")
	}
	return nil
}