server-manager/backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed invite bug
Some checks failed
Build and Push Docker Image / Build image (push) Has been cancelled
Details

Browse Source
This commit is contained in:
ACoolName 2025-04-09 03:34:05 +03:00
parent b8e028b80d
commit 440896ec69
1 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
dbhandler/mongo/user_pass_authentication.go
View File

@ -25,8 +25,8 @@ type UserPassAuthenticationDbHandler struct {
collection *mongo.Collection collection *mongo.Collection
} }
func (self *UserPassAuthenticationDbHandler) ListUsers(ctx context.Context) ([]models.User, error) { func (i *UserPassAuthenticationDbHandler) ListUsers(ctx context.Context) ([]models.User, error) {
cursor, err := self.collection.Find(ctx, bson.M{}) cursor, err := i.collection.Find(ctx, bson.M{})
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -49,9 +49,9 @@ func (self *UserPassAuthenticationDbHandler) ListUsers(ctx context.Context) ([]m
return modelUsers, nil return modelUsers, nil
} }
func (self *UserPassAuthenticationDbHandler) AuthenticateUser(ctx context.Context, username string, password string) (*models.User, error) { func (i *UserPassAuthenticationDbHandler) AuthenticateUser(ctx context.Context, username string, password string) (*models.User, error) {
var user AuthUser var user AuthUser
err := self.collection.FindOne(ctx, bson.M{"username": username}).Decode(&user) err := i.collection.FindOne(ctx, bson.M{"username": username}).Decode(&user)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -70,7 +70,7 @@ func (self *UserPassAuthenticationDbHandler) AuthenticateUser(ctx context.Contex
}, nil }, nil
} }
func (self *UserPassAuthenticationDbHandler) CreateUser( func (i *UserPassAuthenticationDbHandler) CreateUser(
ctx context.Context, ctx context.Context,
username string, username string,
password string, password string,
@ -83,7 +83,7 @@ func (self *UserPassAuthenticationDbHandler) CreateUser(
return err return err
} }
_, err = self.collection.InsertOne(ctx, &AuthUser{ _, err = i.collection.InsertOne(ctx, &AuthUser{
Username: username, Username: username,
HashedPassword: hashedPassword, HashedPassword: hashedPassword,
Permissions: permissions, Permissions: permissions,
@ -94,16 +94,16 @@ func (self *UserPassAuthenticationDbHandler) CreateUser(
return err return err
} }
func (self *UserPassAuthenticationDbHandler) CountUsers(ctx context.Context) (uint, error) { func (i *UserPassAuthenticationDbHandler) CountUsers(ctx context.Context) (uint, error) {
count, err := self.collection.CountDocuments(ctx, bson.M{}) count, err := i.collection.CountDocuments(ctx, bson.M{})
if err != nil { if err != nil {
return 0, err return 0, err
} }
return uint(count), nil return uint(count), nil
} }
func (self *UserPassAuthenticationDbHandler) RemoveUser(ctx context.Context, username string) error { func (i *UserPassAuthenticationDbHandler) RemoveUser(ctx context.Context, username string) error {
_, err := self.collection.DeleteOne( _, err := i.collection.DeleteOne(
ctx, ctx,
bson.M{ bson.M{
"username": username, "username": username,
@ -113,12 +113,12 @@ func (self *UserPassAuthenticationDbHandler) RemoveUser(ctx context.Context, use
return err return err
} }
func (self *UserPassAuthenticationDbHandler) SetPermissions( func (i *UserPassAuthenticationDbHandler) SetPermissions(
ctx context.Context, ctx context.Context,
username string, username string,
permissions models.Permission, permissions models.Permission,
) error { ) error {
_, err := self.collection.UpdateOne( _, err := i.collection.UpdateOne(
ctx, ctx,
bson.M{ bson.M{
"username": username, "username": username,
@ -155,10 +155,10 @@ type InviteTokenDbHandler struct {
collection *mongo.Collection collection *mongo.Collection
} }
func (self *ServersDbHandler) SaveInviteToken(ctx context.Context, email string, permissions models.Permission) (string, error) { func (i *InviteTokenDbHandler) SaveInviteToken(ctx context.Context, email string, permissions models.Permission) (string, error) {
token := uuid.NewString() token := uuid.NewString()
_, err := self.collection.InsertOne(ctx, &InviteToken{ _, err := i.collection.InsertOne(ctx, &InviteToken{
Permissions: permissions, Permissions: permissions,
Email: email, Email: email,
Token: token, Token: token,
@ -170,9 +170,9 @@ func (self *ServersDbHandler) SaveInviteToken(ctx context.Context, email string,
return token, nil return token, nil
} }
func (self *ServersDbHandler) GetInviteToken(ctx context.Context, token string) (*dbhandler.InviteToken, error) { func (i *InviteTokenDbHandler) GetInviteToken(ctx context.Context, token string) (*dbhandler.InviteToken, error) {
var inviteToken InviteToken var inviteToken InviteToken
err := self.collection.FindOne(ctx, bson.M{"token": token}).Decode(&inviteToken) err := i.collection.FindOne(ctx, bson.M{"token": token}).Decode(&inviteToken)
if err != nil { if err != nil {
return nil, err return nil, err
} }