Add preliminary support for NickServ authentication

(remove everything else)

internaltriggers.go

@@ -0,0 +1,112 @@
+package hbot
+
+import (
+	"fmt"
+	"strings"
+)
+
+// A trigger to respond to the servers ping pong messages.
+// If PingPong messages are not responded to, the server assumes the
+// client has timed out and will close the connection.
+var pingPong = Trigger{
+	Condition: func(bot *Bot, m *Message) bool {
+		return m.Command == "PING"
+	},
+	Action: func(bot *Bot, m *Message) {
+		bot.Send("PONG :" + m.Trailing())
+	},
+}
+
+var joinChannels = Trigger{
+	Condition: func(bot *Bot, m *Message) bool {
+		return m.Command == "001" || m.Command == "372"
+	},
+	Action: func(bot *Bot, m *Message) {
+		bot.joinOnce.Do(func() {
+			for _, channel := range bot.config.Channels {
+				splitchan := strings.SplitN(channel, ":", 2)
+				bot.config.Logger.Verbosef("Joining %q", channel)
+				if len(splitchan) == 2 {
+					channel = splitchan[0]
+					password := splitchan[1]
+					bot.Send(fmt.Sprintf("JOIN %s %s", channel, password))
+				} else {
+					bot.Send(fmt.Sprintf("JOIN %s", channel))
+				}
+			}
+			// Fire Joined
+			select {
+			case <-bot.joined:
+			default:
+				close(bot.joined)
+			}
+		})
+	},
+}
+
+// Get bot's prefix by catching its own join
+var getPrefix = Trigger{
+	Condition: func(bot *Bot, m *Message) bool {
+		return m.Command == "JOIN" && m.Prefix.Name == bot.Nick()
+	},
+	Action: func(bot *Bot, m *Message) {
+		bot.mu.Lock()
+		bot.config.Nick = m.Prefix.Name
+		bot.prefix = m.Prefix
+		bot.mu.Unlock()
+	},
+}
+
+// Track nick changes internally so we can adjust the bot's prefix
+var setNick = Trigger{
+	Condition: func(bot *Bot, m *Message) bool {
+		return m.Command == "NICK" && m.Prefix.Name == bot.Nick()
+	},
+	Action: func(bot *Bot, m *Message) {
+		to := m.Param(0)
+		bot.mu.Lock()
+		bot.config.Nick = to
+		bot.prefix.Name = to
+		bot.mu.Unlock()
+		bot.config.Logger.Verbosef("Nick changed to %q", to)
+	},
+}
+
+// Throw errors on invalid nick changes
+var nickError = Trigger{
+	Condition: func(bot *Bot, m *Message) bool {
+		return m.Command == "436" || m.Command == "433" ||
+			m.Command == "432" || m.Command == "431" || m.Command == "400"
+	},
+	Action: func(bot *Bot, m *Message) {
+		bot.config.Logger.Errorf("Nick change error %q: %q", m.Command, m.Params)
+	},
+}
+
+var saslFail = Trigger{
+	Condition: func(bot *Bot, m *Message) bool {
+		return m.Command == "904" || m.Command == "905" ||
+			m.Command == "906" || m.Command == "907"
+	},
+	Action: func(bot *Bot, m *Message) {
+		bot.config.Logger.Errorf("SASL error %q: %q", m.Command, m.Params)
+	},
+}
+
+var saslSuccess = Trigger{
+	Condition: func(bot *Bot, m *Message) bool {
+		return m.Command == "900" || m.Command == "903"
+	},
+	Action: func(bot *Bot, m *Message) {
+		bot.config.Logger.Verbosef("SASL success: %q", m.Trailing())
+	},
+}
+
+var errorLogger = Trigger{
+	Condition: func(bot *Bot, m *Message) bool {
+		return m.Command == "ERROR"
+	},
+	Action: func(bot *Bot, m *Message) {
+		bot.config.Logger.Errorf("Server error: %q", m.Trailing())
+	},
+}