diff options
Diffstat (limited to 'wat')
| -rw-r--r-- | wat/bot.go | 15 | 
1 files changed, 14 insertions, 1 deletions
| @@ -20,6 +20,7 @@ type WatBot struct {  type WatConfig struct {  	AdminHosts        []string  	IgnoredHosts      []string +	AutoJoinChannels  []string  	PermittedChannels []string  } @@ -36,12 +37,24 @@ func CleanNick(nick string) string {  	return string(nick[0]) + "\u200c" + nick[1:]  } +func PrefixChannel(channel string) string { +	// there could theoretically be other channel prefixes .. +	if channel[0] != '#' && channel[0] != '!' { +		channel = "#" + channel +	} +	return channel +} +  func (w *WatBot) HandleIrcMsg(c *irc.Client, m *irc.Message) {  	switch cmd := m.Command; cmd {  	case "PING":  		w.write("PONG", m.Params[0])  	case "PRIVMSG":  		w.Msg(m) +	case "001": +		for _, channel := range w.c.AutoJoinChannels { +			w.write("JOIN", PrefixChannel(channel)) +		}  	}  } @@ -68,7 +81,7 @@ func (w *WatBot) CanRespond(m *irc.Message) bool {  	// if !strings.Contains(m.Prefix.Host, "") {  	// 	return false  	// } -	if !w.Allowed(m.Params[0], w.c.PermittedChannels) { +	if !w.Allowed(PrefixChannel(m.Params[0]), w.c.PermittedChannels) {  		return false  	}  	return true | 
