73 lines
2.3 KiB
Go
73 lines
2.3 KiB
Go
|
package db
|
||
|
|
|
||
|
|
import "strings"
|
||
|
|
|
||
|
|
func (d *DB) ListAllowBlock(mailboxID int64) ([]MailboxAllowBlockEntry, error) {
|
||
|
|
rows, err := d.Query(`SELECT id, mailbox_id, list_type, pattern, created_at
|
||
|
|
FROM esrv_mailbox_allowblock WHERE mailbox_id = ? ORDER BY list_type, pattern`, mailboxID)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
defer rows.Close()
|
||
|
|
var out []MailboxAllowBlockEntry
|
||
|
|
for rows.Next() {
|
||
|
|
var e MailboxAllowBlockEntry
|
||
|
|
var createdAt string
|
||
|
|
if err := rows.Scan(&e.ID, &e.MailboxID, &e.ListType, &e.Pattern, &createdAt); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
e.CreatedAt, _ = parseTime(createdAt)
|
||
|
|
out = append(out, e)
|
||
|
|
}
|
||
|
|
return out, rows.Err()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *DB) AddAllowBlockEntry(mailboxID int64, listType, pattern string) (int64, error) {
|
||
|
|
res, err := d.Exec(`INSERT OR IGNORE INTO esrv_mailbox_allowblock (mailbox_id, list_type, pattern) VALUES (?, ?, ?)`,
|
||
|
|
mailboxID, listType, strings.ToLower(pattern))
|
||
|
|
if err != nil {
|
||
|
|
return 0, err
|
||
|
|
}
|
||
|
|
return res.LastInsertId()
|
||
|
|
}
|
||
|
|
|
||
|
|
// RemoveAllowBlockEntry deletes an entry, scoped to mailboxID (mirrors RemoveAppPassword/RemoveAlias).
|
||
|
|
func (d *DB) RemoveAllowBlockEntry(id, mailboxID int64) error {
|
||
|
|
_, err := d.Exec(`DELETE FROM esrv_mailbox_allowblock WHERE id = ? AND mailbox_id = ?`, id, mailboxID)
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *DB) IsBlocked(mailboxID int64, senderAddr string) (bool, error) {
|
||
|
|
return matchesAllowBlock(d, mailboxID, "block", senderAddr)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *DB) IsAllowed(mailboxID int64, senderAddr string) (bool, error) {
|
||
|
|
return matchesAllowBlock(d, mailboxID, "allow", senderAddr)
|
||
|
|
}
|
||
|
|
|
||
|
|
// matchesAllowBlock checks senderAddr against every pattern of listType for mailboxID
|
||
|
|
// — an exact address match, or a "@domain.com" wildcard matching senderAddr's domain.
|
||
|
|
func matchesAllowBlock(d *DB, mailboxID int64, listType, senderAddr string) (bool, error) {
|
||
|
|
senderAddr = strings.ToLower(senderAddr)
|
||
|
|
domain := domainPart(senderAddr)
|
||
|
|
rows, err := d.Query(`SELECT pattern FROM esrv_mailbox_allowblock WHERE mailbox_id = ? AND list_type = ?`, mailboxID, listType)
|
||
|
|
if err != nil {
|
||
|
|
return false, err
|
||
|
|
}
|
||
|
|
defer rows.Close()
|
||
|
|
for rows.Next() {
|
||
|
|
var pattern string
|
||
|
|
if err := rows.Scan(&pattern); err != nil {
|
||
|
|
return false, err
|
||
|
|
}
|
||
|
|
if strings.HasPrefix(pattern, "@") {
|
||
|
|
if pattern[1:] == domain {
|
||
|
|
return true, nil
|
||
|
|
}
|
||
|
|
} else if pattern == senderAddr {
|
||
|
|
return true, nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return false, rows.Err()
|
||
|
|
}
|