Make the deadline configurable
parent
c31e8d9f51
commit
90dc6edfad
|
@ -22,6 +22,8 @@ type CommandDeadline struct {
|
|||
Deadline time.Time
|
||||
}
|
||||
|
||||
var Cooldown = 2 * time.Minute // default cooldown is 2 minutes
|
||||
|
||||
func NewCommand(next hugot.CommandHandler) hugot.CommandHandler {
|
||||
return &CommandDeadline{CommandHandler: next, Deadline: time.Now().UTC()}
|
||||
}
|
||||
|
@ -33,7 +35,7 @@ func NewHears(next hugot.HearsHandler) hugot.HearsHandler {
|
|||
func (i *HearsDeadline) Heard(ctx context.Context, w hugot.ResponseWriter, m *hugot.Message, matches [][]string) {
|
||||
now := time.Now().UTC()
|
||||
if now.After(i.Deadline) {
|
||||
i.Deadline = now.Add(2 * time.Minute)
|
||||
i.Deadline = now.Add(Cooldown)
|
||||
i.HearsHandler.Heard(ctx, w, m, matches)
|
||||
return
|
||||
}
|
||||
|
@ -45,7 +47,7 @@ func (i *HearsDeadline) Heard(ctx context.Context, w hugot.ResponseWriter, m *hu
|
|||
func (i *CommandDeadline) Command(ctx context.Context, w hugot.ResponseWriter, m *hugot.Message) error {
|
||||
now := time.Now().UTC()
|
||||
if now.After(i.Deadline) {
|
||||
i.Deadline = now.Add(2 * time.Minute)
|
||||
i.Deadline = now.Add(Cooldown)
|
||||
return i.CommandHandler.Command(ctx, w, m)
|
||||
}
|
||||
|
||||
|
|
|
@ -21,6 +21,7 @@ import (
|
|||
|
||||
var (
|
||||
channels = flag.String("channels", "##testing,##test", "Channels to join (separated by comma)")
|
||||
cooldown = flag.Int("cooldown", 120, "Command rate-limit, in seconds")
|
||||
host = flag.String("host", "chat.freenode.net", "Server host[:port]")
|
||||
ident = flag.String("ident", "lysenko", "Lysenko Bot")
|
||||
nick = flag.String("nick", "lysenko", "Lysenko Bot")
|
||||
|
@ -71,6 +72,8 @@ func main() {
|
|||
}
|
||||
defer db.DB.Close()
|
||||
|
||||
deadline.Cooldown = time.Second * time.Duration(*cooldown)
|
||||
|
||||
mux := hugot.DefaultMux
|
||||
mux.HandleHears("edb.AddQuoteHandler{QuoteDB: db})
|
||||
mux.HandleHears(deadline.NewHears("edb.LastQuoteHandler{QuoteDB: db}))
|
||||
|
|
Loading…
Reference in New Issue