-
Notifications
You must be signed in to change notification settings - Fork 0
/
session.go
77 lines (67 loc) · 1.86 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
70
71
72
73
74
75
76
77
package data
import "time"
// Session contains all of the information needed to keep track of OAuth
// requests/responses with a 3rd party.
type Session struct {
ResponseType string
Me string
Provider string
ProfileURI string
ClientID string
RedirectURI string
CodeChallenge string
CodeChallengeMethod string
Scope string
State string
CreatedAt time.Time
ExpiresAt time.Time
}
func (s Session) Expired() bool {
return time.Now().After(s.ExpiresAt)
}
func (d *Database) CreateSession(session Session) error {
_, err := d.db.Exec(`
INSERT OR REPLACE INTO session(ResponseType, Me, ClientID, RedirectURI, CodeChallenge, CodeChallengeMethod, Scope, State, Provider, ProfileURI, CreatedAt)
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
`,
session.ResponseType,
session.Me,
session.ClientID,
session.RedirectURI,
session.CodeChallenge,
session.CodeChallengeMethod,
session.Scope,
session.State,
"",
"",
session.CreatedAt)
return err
}
func (d *Database) SetProvider(me, provider, profileURI string) error {
_, err := d.db.Exec(`UPDATE session SET Provider = ?, ProfileURI = ? WHERE Me = ?`,
provider,
profileURI,
me)
return err
}
func (d *Database) Session(me string) (session Session, err error) {
row := d.db.QueryRow(`
SELECT ResponseType, Me, ClientID, RedirectURI, CodeChallenge, CodeChallengeMethod, Scope, State, Provider, ProfileURI, CreatedAt
FROM session
WHERE Me = ?`,
me)
err = row.Scan(
&session.ResponseType,
&session.Me,
&session.ClientID,
&session.RedirectURI,
&session.CodeChallenge,
&session.CodeChallengeMethod,
&session.Scope,
&session.State,
&session.Provider,
&session.ProfileURI,
&session.CreatedAt)
session.ExpiresAt = session.CreatedAt.Add(d.expiry.Session)
return
}