Rename sessions to tokens/tokenData

This commit is contained in:
Anders Pitman committed 2020-10-11 14:27:32 -06:00
1 parent 9da8ca987e
commit 1e25527ab2
3 files changed
+17 -17

No files matched your search

+3 -3
View File
@@ -62,9 +62,9 @@ func (a *Api) handleTunnels(w http.ResponseWriter, r *http.Request) {
w.Write([]byte(body))
case "POST":
a.validateSession(http.HandlerFunc(a.handleCreateTunnel)).ServeHTTP(w, r)
a.validateToken(http.HandlerFunc(a.handleCreateTunnel)).ServeHTTP(w, r)
case "DELETE":
a.validateSession(http.HandlerFunc(a.handleDeleteTunnel)).ServeHTTP(w, r)
a.validateToken(http.HandlerFunc(a.handleDeleteTunnel)).ServeHTTP(w, r)
default:
w.WriteHeader(405)
w.Write([]byte("Invalid method for /tunnels"))
@@ -117,7 +117,7 @@ func (a *Api) handleDeleteTunnel(w http.ResponseWriter, r *http.Request) {
}
}
func (a *Api) validateSession(h http.Handler) http.Handler {
func (a *Api) validateToken(h http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
token, err := extractToken("access_token", r)
+2 -2
View File
@@ -28,7 +28,7 @@ func NewAuth(db *Database) *Auth {
}
func (a *Auth) Authorized(token string) bool {
_, exists := a.db.GetSession(token)
_, exists := a.db.GetTokenData(token)
if exists {
return true
@@ -90,7 +90,7 @@ func (a *Auth) Verify(key string) (string, error) {
return "", errors.New("Error generating key")
}
a.db.SetSession(token, Session{Id: request.Email})
a.db.SetTokenData(token, TokenData{Id: request.Email})
return token, nil
}
+12 -12
View File
@@ -8,12 +8,12 @@ import (
)
type Database struct {
Sessions map[string]Session `json:"sessions"`
Tunnels map[string]Tunnel `json:"tunnels"`
mutex *sync.Mutex
Tokens map[string]TokenData `json:"tokens"`
Tunnels map[string]Tunnel `json:"tunnels"`
mutex *sync.Mutex
}
type Session struct {
type TokenData struct {
Id string `json:"id"`
}
@@ -44,8 +44,8 @@ func NewDatabase() (*Database, error) {
db = &Database{}
}
if db.Sessions == nil {
db.Sessions = make(map[string]Session)
if db.Tokens == nil {
db.Tokens = make(map[string]TokenData)
}
if db.Tunnels == nil {
@@ -61,24 +61,24 @@ func NewDatabase() (*Database, error) {
return db, nil
}
func (d *Database) GetSession(token string) (Session, bool) {
func (d *Database) GetTokenData(token string) (TokenData, bool) {
d.mutex.Lock()
defer d.mutex.Unlock()
sesh, exists := d.Sessions[token]
tokenData, exists := d.Tokens[token]
if !exists {
return Session{}, false
return TokenData{}, false
}
return sesh, true
return tokenData, true
}
func (d *Database) SetSession(token string, sesh Session) {
func (d *Database) SetTokenData(token string, tokenData TokenData) {
d.mutex.Lock()
defer d.mutex.Unlock()
d.Sessions[token] = sesh
d.Tokens[token] = tokenData
d.persist()
}