pkg/SugarDB
1
0
Fork 0
mirror of https://github.com/EchoVault/SugarDB.git synced 2025-10-09 09:50:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Implemented connection authentication and ACL WHOAMI command

Browse Source
This commit is contained in:
Kelvin Clement Mwinuka
2023-12-14 23:01:32 +08:00
parent 39bf074e33
commit c3194b0d2d
2 changed files with 57 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/modules/acl/commands.go
View File

@@ -50,7 +50,7 @@ func (p Plugin) HandleCommand(ctx context.Context, cmd []string, server utils.Se
case "deluser":
return p.handleDelUser(ctx, cmd, server)
case "whoami":
return p.handleWhoAmI(ctx, cmd, server)
return p.handleWhoAmI(ctx, cmd, server, conn)
case "list":
return p.handleList(ctx, cmd, server)
case "load":
@@ -92,8 +92,9 @@ func (p Plugin) handleDelUser(ctx context.Context, cmd []string, server utils.Se
return nil, errors.New("ACL DELUSER not implemented")
}
func (p Plugin) handleWhoAmI(ctx context.Context, cmd []string, server utils.Server) ([]byte, error) {
return nil, errors.New("ACL WHOAMI not implemented")
func (p Plugin) handleWhoAmI(ctx context.Context, cmd []string, server utils.Server, conn *net.Conn) ([]byte, error) {
connectionInfo := p.acl.Connections[conn]
return []byte(fmt.Sprintf("+%s\r\n\n", connectionInfo.User.Username)), nil
}
func (p Plugin) handleList(ctx context.Context, cmd []string, server utils.Server) ([]byte, error) {