mirror of
https://github.com/m1k1o/neko.git
synced 2024-07-24 14:40:50 +12:00
move server to server directory.
This commit is contained in:
124
server/internal/member/object/provider.go
Normal file
124
server/internal/member/object/provider.go
Normal file
@ -0,0 +1,124 @@
|
||||
package object
|
||||
|
||||
import (
|
||||
"github.com/demodesk/neko/pkg/types"
|
||||
)
|
||||
|
||||
func New(config Config) types.MemberProvider {
|
||||
return &MemberProviderCtx{
|
||||
config: config,
|
||||
entries: make(map[string]*memberEntry),
|
||||
}
|
||||
}
|
||||
|
||||
type MemberProviderCtx struct {
|
||||
config Config
|
||||
entries map[string]*memberEntry
|
||||
}
|
||||
|
||||
func (provider *MemberProviderCtx) Connect() error {
|
||||
var err error
|
||||
|
||||
for _, entry := range provider.config.Users {
|
||||
_, err = provider.Insert(entry.Username, entry.Password, entry.Profile)
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func (provider *MemberProviderCtx) Disconnect() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (provider *MemberProviderCtx) Authenticate(username string, password string) (string, types.MemberProfile, error) {
|
||||
// id will be also username
|
||||
id := username
|
||||
|
||||
entry, ok := provider.entries[id]
|
||||
if !ok {
|
||||
return "", types.MemberProfile{}, types.ErrMemberDoesNotExist
|
||||
}
|
||||
|
||||
// TODO: Use hash function.
|
||||
if !entry.CheckPassword(password) {
|
||||
return "", types.MemberProfile{}, types.ErrMemberInvalidPassword
|
||||
}
|
||||
|
||||
return id, entry.profile, nil
|
||||
}
|
||||
|
||||
func (provider *MemberProviderCtx) Insert(username string, password string, profile types.MemberProfile) (string, error) {
|
||||
// id will be also username
|
||||
id := username
|
||||
|
||||
_, ok := provider.entries[id]
|
||||
if ok {
|
||||
return "", types.ErrMemberAlreadyExists
|
||||
}
|
||||
|
||||
provider.entries[id] = &memberEntry{
|
||||
// TODO: Use hash function.
|
||||
password: password,
|
||||
profile: profile,
|
||||
}
|
||||
|
||||
return id, nil
|
||||
}
|
||||
|
||||
func (provider *MemberProviderCtx) UpdateProfile(id string, profile types.MemberProfile) error {
|
||||
entry, ok := provider.entries[id]
|
||||
if !ok {
|
||||
return types.ErrMemberDoesNotExist
|
||||
}
|
||||
|
||||
entry.profile = profile
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (provider *MemberProviderCtx) UpdatePassword(id string, password string) error {
|
||||
entry, ok := provider.entries[id]
|
||||
if !ok {
|
||||
return types.ErrMemberDoesNotExist
|
||||
}
|
||||
|
||||
// TODO: Use hash function.
|
||||
entry.password = password
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (provider *MemberProviderCtx) Select(id string) (types.MemberProfile, error) {
|
||||
entry, ok := provider.entries[id]
|
||||
if !ok {
|
||||
return types.MemberProfile{}, types.ErrMemberDoesNotExist
|
||||
}
|
||||
|
||||
return entry.profile, nil
|
||||
}
|
||||
|
||||
func (provider *MemberProviderCtx) SelectAll(limit int, offset int) (map[string]types.MemberProfile, error) {
|
||||
profiles := make(map[string]types.MemberProfile)
|
||||
|
||||
i := 0
|
||||
for id, entry := range provider.entries {
|
||||
if i >= offset && (limit == 0 || i < offset+limit) {
|
||||
profiles[id] = entry.profile
|
||||
}
|
||||
|
||||
i = i + 1
|
||||
}
|
||||
|
||||
return profiles, nil
|
||||
}
|
||||
|
||||
func (provider *MemberProviderCtx) Delete(id string) error {
|
||||
_, ok := provider.entries[id]
|
||||
if !ok {
|
||||
return types.ErrMemberDoesNotExist
|
||||
}
|
||||
|
||||
delete(provider.entries, id)
|
||||
|
||||
return nil
|
||||
}
|
24
server/internal/member/object/types.go
Normal file
24
server/internal/member/object/types.go
Normal file
@ -0,0 +1,24 @@
|
||||
package object
|
||||
|
||||
import (
|
||||
"github.com/demodesk/neko/pkg/types"
|
||||
)
|
||||
|
||||
type memberEntry struct {
|
||||
password string
|
||||
profile types.MemberProfile
|
||||
}
|
||||
|
||||
func (m *memberEntry) CheckPassword(password string) bool {
|
||||
return m.password == password
|
||||
}
|
||||
|
||||
type User struct {
|
||||
Username string
|
||||
Password string
|
||||
Profile types.MemberProfile
|
||||
}
|
||||
|
||||
type Config struct {
|
||||
Users []User
|
||||
}
|
Reference in New Issue
Block a user