mirror of https://github.com/Dreamacro/clash.git
47 lines
901 B
Go
47 lines
901 B
Go
|
package auth
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Authenticator interface {
|
||
|
|
Verify(user string, pass string) bool
|
||
|
|
Users() []string
|
||
|
|
}
|
||
|
|
|
||
|
|
type AuthUser struct {
|
||
|
|
User string
|
||
|
|
Pass string
|
||
|
|
}
|
||
|
|
|
||
|
|
type inMemoryAuthenticator struct {
|
||
|
|
storage *sync.Map
|
||
|
|
usernames []string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (au *inMemoryAuthenticator) Verify(user string, pass string) bool {
|
||
|
|
realPass, ok := au.storage.Load(user)
|
||
|
|
return ok && realPass == pass
|
||
|
|
}
|
||
|
|
|
||
|
|
func (au *inMemoryAuthenticator) Users() []string { return au.usernames }
|
||
|
|
|
||
|
|
func NewAuthenticator(users []AuthUser) Authenticator {
|
||
|
|
if len(users) == 0 {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
au := &inMemoryAuthenticator{storage: &sync.Map{}}
|
||
|
|
for _, user := range users {
|
||
|
|
au.storage.Store(user.User, user.Pass)
|
||
|
|
}
|
||
|
|
usernames := make([]string, 0, len(users))
|
||
|
|
au.storage.Range(func(key, value interface{}) bool {
|
||
|
|
usernames = append(usernames, key.(string))
|
||
|
|
return true
|
||
|
|
})
|
||
|
|
au.usernames = usernames
|
||
|
|
|
||
|
|
return au
|
||
|
|
}
|