mirrors/maddy
1
0
Fork 0
mirror of https://github.com/foxcpp/maddy.git synced 2025-04-04 13:37:41 +03:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #566 from arisudesu-forks/loop-extract

Browse source 
Extract mapping lookup from loop
This commit is contained in:
Max Mazurov 2023-03-12 14:09:31 +03:00 • committed by GitHub
commit ca44d0e036
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

34
internal/authz/lookup.go
View file

@ -9,28 +9,30 @@ import (
) )
func AuthorizeEmailUse(ctx context.Context, username string, addrs []string, mapping module.Table) (bool, error) { func AuthorizeEmailUse(ctx context.Context, username string, addrs []string, mapping module.Table) (bool, error) {
var validEmails []string
if multi, ok := mapping.(module.MultiTable); ok {
var err error
validEmails, err = multi.LookupMulti(ctx, username)
if err != nil {
return false, fmt.Errorf("authz: %w", err)
}
} else {
validEmail, ok, err := mapping.Lookup(ctx, username)
if err != nil {
return false, fmt.Errorf("authz: %w", err)
}
if ok {
validEmails = []string{validEmail}
}
}
for _, addr := range addrs { for _, addr := range addrs {
_, domain, err := address.Split(addr) _, domain, err := address.Split(addr)
if err != nil { if err != nil {
return false, fmt.Errorf("authz: %w", err) return false, fmt.Errorf("authz: %w", err)
} }
var validEmails []string
if multi, ok := mapping.(module.MultiTable); ok {
validEmails, err = multi.LookupMulti(ctx, username)
if err != nil {
return false, fmt.Errorf("authz: %w", err)
}
} else {
validEmail, ok, err := mapping.Lookup(ctx, username)
if err != nil {
return false, fmt.Errorf("authz: %w", err)
}
if ok {
validEmails = []string{validEmail}
}
}
for _, ent := range validEmails { for _, ent := range validEmails {
if ent == domain || ent == "*" || ent == addr { if ent == domain || ent == "*" || ent == addr {
return true, nil return true, nil