Refactor user authentication
This commit is contained in:
10
pwd/user.go
10
pwd/user.go
@@ -1,10 +1,14 @@
|
||||
package pwd
|
||||
|
||||
import (
|
||||
"errors"
|
||||
|
||||
"github.com/play-with-docker/play-with-docker/pwd/types"
|
||||
"github.com/play-with-docker/play-with-docker/storage"
|
||||
)
|
||||
|
||||
var userBannedError = errors.New("User is banned")
|
||||
|
||||
func (p *pwd) UserNewLoginRequest(providerName string) (*types.LoginRequest, error) {
|
||||
req := &types.LoginRequest{Id: p.generator.NewId(), Provider: providerName}
|
||||
if err := p.storage.LoginRequestPut(req); err != nil {
|
||||
@@ -40,9 +44,11 @@ func (p *pwd) UserLogin(loginRequest *types.LoginRequest, user *types.User) (*ty
|
||||
return u, nil
|
||||
}
|
||||
func (p *pwd) UserGet(id string) (*types.User, error) {
|
||||
var user *types.User
|
||||
if user, err := p.storage.UserGet(id); err != nil {
|
||||
return nil, err
|
||||
} else {
|
||||
return user, nil
|
||||
} else if user.IsBanned {
|
||||
return user, userBannedError
|
||||
}
|
||||
return user, nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user