package main import ( "fmt" "log" "strings" "time" irc "github.com/fluffle/goirc/client" ) type Commander interface { irc.Handler fmt.Stringer Timeout(string) bool WithArgs(int) bool } type Command struct { Help string Arg string Last map[string]time.Time } var commands = make(map[string]Commander) func Register(cmd string, f Commander) { commands[cmd] = f } func (v Command) String() string { return v.Help } func (v *Command) Timeout(nick string) bool { defer func() { v.Last[nick] = time.Now() }() if v.Last == nil { v.Last = make(map[string]time.Time) } if last, ok := v.Last[nick]; ok { if to := time.Since(last); to < 5*time.Second { log.Println(nick, "timeout", to) return true } } return false } func (_ Command) WithArgs(n int) bool { return n == 1 } func Dispatch(conn *irc.Conn, line *irc.Line) { if f := strings.Fields(line.Text()); len(f) > 0 { cmd := strings.ToLower(f[0]) if c, ok := commands[cmd]; ok { if line.Public() && c.Timeout(line.Nick) { log.Println("timeout", line.Nick) conn.Kick(*room, line.Nick, "timeout") return } if c.WithArgs(len(f)) { log.Println(line.Nick, f) c.Handle(conn, line) } } } }