diff --git a/bridge/config/config.go b/bridge/config/config.go index aa881a79..89222894 100644 --- a/bridge/config/config.go +++ b/bridge/config/config.go @@ -34,6 +34,8 @@ type Protocol struct { Password string // IRC,mattermost,XMPP PrefixMessagesWithNick bool // mattemost, slack Protocol string //all protocols + MessageQueue int // IRC, size of message queue for flood control + MessageDelay int // IRC, time in millisecond to wait between messages RemoteNickFormat string // all protocols Server string // IRC,mattermost,XMPP,discord ShowJoinPart bool // all protocols diff --git a/bridge/irc/irc.go b/bridge/irc/irc.go index e5b6858c..30f27724 100644 --- a/bridge/irc/irc.go +++ b/bridge/irc/irc.go @@ -23,6 +23,7 @@ type Birc struct { protocol string Remote chan config.Message connected chan struct{} + Local chan config.Message // local queue for flood control } var flog *log.Entry @@ -32,15 +33,22 @@ func init() { flog = log.WithFields(log.Fields{"module": protocol}) } -func New(config config.Protocol, origin string, c chan config.Message) *Birc { +func New(cfg config.Protocol, origin string, c chan config.Message) *Birc { b := &Birc{} - b.Config = &config + b.Config = &cfg b.Nick = b.Config.Nick b.Remote = c b.names = make(map[string][]string) b.origin = origin b.protocol = protocol b.connected = make(chan struct{}) + if b.Config.MessageDelay == 0 { + b.Config.MessageDelay = 1300 + } + if b.Config.MessageQueue == 0 { + b.Config.MessageQueue = 30 + } + b.Local = make(chan config.Message, b.Config.MessageQueue+10) return b } @@ -81,6 +89,7 @@ func (b *Birc) Connect() error { return fmt.Errorf("connection timed out") } i.Debug = false + go b.doSend() return nil } @@ -115,11 +124,27 @@ func (b *Birc) Send(msg config.Message) error { return nil } for _, text := range strings.Split(msg.Text, "\n") { - b.i.Privmsg(msg.Channel, msg.Username+text) + if len(b.Local) < b.Config.MessageQueue { + if len(b.Local) == b.Config.MessageQueue-1 { + text = text + " " + } + b.Local <- config.Message{Text: text, Username: msg.Username, Channel: msg.Channel} + } else { + flog.Debugf("flooding, dropping message (queue at %d)", len(b.Local)) + } } return nil } +func (b *Birc) doSend() { + rate := time.Millisecond * time.Duration(b.Config.MessageDelay) + throttle := time.Tick(rate) + for msg := range b.Local { + <-throttle + b.i.Privmsg(msg.Channel, msg.Username+msg.Text) + } +} + func (b *Birc) endNames(event *irc.Event) { channel := event.Arguments[1] sort.Strings(b.names[channel])