89 lines
1.7 KiB
Go
89 lines
1.7 KiB
Go
package store
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
type Direction string
|
|
|
|
const (
|
|
DirIn Direction = "in"
|
|
DirOut Direction = "out"
|
|
)
|
|
|
|
func (d Direction) table() (string, error) {
|
|
switch d {
|
|
case DirIn:
|
|
return "whitelist_in", nil
|
|
case DirOut:
|
|
return "whitelist_out", nil
|
|
default:
|
|
return "", fmt.Errorf("invalid direction %q", d)
|
|
}
|
|
}
|
|
|
|
func (s *Store) accountID(name string) (int64, error) {
|
|
a, err := s.GetAccount(name)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return a.ID, nil
|
|
}
|
|
|
|
func (s *Store) AddWhitelist(account string, dir Direction, address string) error {
|
|
tbl, err := dir.table()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
id, err := s.accountID(account)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = s.db.Exec(
|
|
fmt.Sprintf("INSERT OR IGNORE INTO %s(account_id,address) VALUES(?,?)", tbl),
|
|
id, strings.ToLower(address))
|
|
return err
|
|
}
|
|
|
|
func (s *Store) RemoveWhitelist(account string, dir Direction, address string) error {
|
|
tbl, err := dir.table()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
id, err := s.accountID(account)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = s.db.Exec(
|
|
fmt.Sprintf("DELETE FROM %s WHERE account_id=? AND address=?", tbl),
|
|
id, strings.ToLower(address))
|
|
return err
|
|
}
|
|
|
|
func (s *Store) ListWhitelist(account string, dir Direction) ([]string, error) {
|
|
tbl, err := dir.table()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
id, err := s.accountID(account)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
rows, err := s.db.Query(
|
|
fmt.Sprintf("SELECT address FROM %s WHERE account_id=? ORDER BY address", tbl), id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
out := []string{}
|
|
for rows.Next() {
|
|
var a string
|
|
if err := rows.Scan(&a); err != nil {
|
|
return nil, err
|
|
}
|
|
out = append(out, a)
|
|
}
|
|
return out, rows.Err()
|
|
}
|