-
Notifications
You must be signed in to change notification settings - Fork 10
/
sessions.go
93 lines (77 loc) · 1.82 KB
/
sessions.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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
package sessions
import (
"net/http"
"github.com/gorilla/sessions"
)
// Session is an interface that stores values and configurations for a session.
type Session interface {
Get(key interface{}) interface{}
Set(key interface{}, val interface{})
Delete(key interface{})
Clear()
AddFlash(value interface{}, vars ...string)
Flashes(vars ...string) []interface{}
Save() error
Options(Options)
}
// Store is an interface for custom session stores.
type Store interface {
sessions.Store
}
// Options stores configurations of a session
type Options struct {
Path string
Domain string
MaxAge int
Secure bool
HttpOnly bool
}
type session struct {
name string
request *http.Request
writer http.ResponseWriter
store Store
session *sessions.Session
}
func (s *session) Get(key interface{}) interface{} {
return s.Session().Values[key]
}
func (s *session) Set(key interface{}, value interface{}) {
s.Session().Values[key] = value
}
func (s *session) Delete(key interface{}) {
delete(s.Session().Values, key)
}
func (s *session) Clear() {
for key := range s.Session().Values {
s.Delete(key)
}
}
func (s *session) AddFlash(value interface{}, vars ...string) {
s.Session().AddFlash(value, vars...)
}
func (s *session) Flashes(vars ...string) []interface{} {
return s.Session().Flashes(vars...)
}
func (s *session) Save() error {
return s.Session().Save(s.request, s.writer)
}
func (s *session) Session() *sessions.Session {
if s.session == nil {
if session, err := s.store.Get(s.request, s.name); err != nil {
panic(err)
} else {
s.session = session
}
}
return s.session
}
func (s *session) Options(options Options) {
s.Session().Options = &sessions.Options{
Path: options.Path,
Domain: options.Domain,
MaxAge: options.MaxAge,
Secure: options.Secure,
HttpOnly: options.HttpOnly,
}
}