make banlist consistent, fixes #54
This commit is contained in:
@@ -119,12 +119,14 @@ func (b *Bot) IsTrusted(addr net.Addr) bool {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
b.log.Debug("address %s is NOT trusted", ip)
|
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
// Ban an address
|
// Ban an address
|
||||||
func (b *Bot) Ban(addr net.Addr) {
|
func (b *Bot) Ban(addr net.Addr) {
|
||||||
|
if !b.cfg.BanlistEnalbed() {
|
||||||
|
return
|
||||||
|
}
|
||||||
if b.IsTrusted(addr) {
|
if b.IsTrusted(addr) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,8 +1,6 @@
|
|||||||
package config
|
package config
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
|
|
||||||
"gitlab.com/etke.cc/go/logger"
|
"gitlab.com/etke.cc/go/logger"
|
||||||
"gitlab.com/etke.cc/linkpearl"
|
"gitlab.com/etke.cc/linkpearl"
|
||||||
"maunium.net/go/mautrix/id"
|
"maunium.net/go/mautrix/id"
|
||||||
@@ -32,15 +30,23 @@ func New(lp *linkpearl.Linkpearl, log *logger.Logger) *Manager {
|
|||||||
return m
|
return m
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// BanlistEnalbed or not
|
||||||
|
func (m *Manager) BanlistEnalbed() bool {
|
||||||
|
return m.ble
|
||||||
|
}
|
||||||
|
|
||||||
// GetBot config
|
// GetBot config
|
||||||
func (m *Manager) GetBot() Bot {
|
func (m *Manager) GetBot() Bot {
|
||||||
config, err := m.lp.GetAccountData(acBotKey)
|
var err error
|
||||||
|
var config Bot
|
||||||
|
config, err = m.lp.GetAccountData(acBotKey)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
m.log.Error("cannot get bot settings: %v", utils.UnwrapError(err))
|
m.log.Error("cannot get bot settings: %v", utils.UnwrapError(err))
|
||||||
}
|
}
|
||||||
if config == nil {
|
if config == nil {
|
||||||
config = make(Bot, 0)
|
config = make(Bot, 0)
|
||||||
}
|
}
|
||||||
|
m.ble = config.BanlistEnabled()
|
||||||
|
|
||||||
return config
|
return config
|
||||||
}
|
}
|
||||||
@@ -88,7 +94,7 @@ func (m *Manager) GetBanlist() List {
|
|||||||
// SetBanlist config
|
// SetBanlist config
|
||||||
func (m *Manager) SetBanlist(cfg List) error {
|
func (m *Manager) SetBanlist(cfg List) error {
|
||||||
if !m.ble {
|
if !m.ble {
|
||||||
return fmt.Errorf("banlist is disabled, kupo")
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
m.mu.Lock("banlist")
|
m.mu.Lock("banlist")
|
||||||
|
|||||||
Reference in New Issue
Block a user