mirror of
https://github.com/boringproxy/boringproxy.git
synced 2026-11-04 13:17:30 -06:00
Rename sessions to tokens/tokenData
This commit is contained in:
1 parent
9da8ca987e
commit
1e25527ab2
3 files changed
+17
-17
No files matched your search
@@ -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)
|
||||
|
||||
@@ -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
@@ -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()
|
||||
}
|
||||
|
||||
|
||||
Reference in new issue
Block a user