From c83f0be5389f0aadd263e0635ba0bdc5e7e0dab4 Mon Sep 17 00:00:00 2001
From: Elliott Pardee <me@elliottpardee.me>
Date: Sun, 11 Jan 2015 10:52:06 -0500
Subject: [PATCH] now using @karlmcg's regbot codebase

---
 bot.go | 69 ++++++++++++++++++++++++++++++++++------------------------
 1 file changed, 40 insertions(+), 29 deletions(-)

diff --git a/bot.go b/bot.go
index 666d3d7..4b1783e 100755
--- a/bot.go
+++ b/bot.go
@@ -1,20 +1,20 @@
 package main
 
 import (
-
     "github.com/thoj/go-ircevent"
     "strings"
-    "strconv"
     "fmt"
     "os"
 )
 
-var (
-    server = "irc.iotek.org"
-    port = 6667
-    channel = "#d20"
-    nickname = "bot"
+type Bot struct {
+    Conn    *irc.Connection
+    Nick    string
+    Server  string
+    Channel string
+}
 
+var (
     admins = []string{"vypr"}
     dunmas = ""
 
@@ -40,6 +40,10 @@ var dict = map[string]string{
     "cls": "class",
 }
 
+func (b *Bot) Say(msg string) {
+    b.Conn.Privmsg(b.Channel, msg)
+}
+
 func stringInSlice(a string, list []string) bool {
     // thanks stackoverflow
     for _, b := range list {
@@ -89,7 +93,7 @@ func fillCharmap(nick string, cat string, item string, val string) {
     charmap = map[string]map[string]map[string]string { nick: map[string]map[string]string{ cat: map[string]string{ item: val, }, }, }
 }
 
-func handleMessage(nick string, msg string, conn *irc.Connection) {
+func (b *Bot) Command(nick string, msg string) {
     var args = findArguments(msg)
 
     // TODO: Check if mode is enabled and if command can be applied.
@@ -101,58 +105,65 @@ func handleMessage(nick string, msg string, conn *irc.Connection) {
         } else if stringInSlice(nick, admins) && !stringInSlice("nocharoverride", rulemod) {
             fillCharmap(args[0], args[1], args[2], args[3])
             fmt.Println("[cmd] set - " + args[0] + "'s " + args[2] + " in " + args[1] + " is set to " + args[3] + ".")
-            conn.Privmsg(channel, nick + " used override, it's super effective!")
+            b.Say(nick + " used override, it's super effective!")
         }
     } else if strings.HasPrefix(msg, ".print") && len(args) == 3 {
         fmt.Println("[cmd] print - " + args[0] + "'s " + args[1] + " in " + args[3] + ".")
-        conn.Privmsg(channel, args[0] + "'s " + args[0] + " is set to " + charmap[args[0]][args[1]][args[2]] + ".")
+        b.Say(args[0] + "'s " + args[0] + " is set to " + charmap[args[0]][args[1]][args[2]] + ".")
     } else if strings.HasPrefix(msg, ".mode") && len(args) == 1 {
         if stringInSlice(args[0], rulemod) {
             fmt.Println("[cmd] mode - change to " + args[0] + " failed, already set to true")
-            conn.Privmsg(channel, args[0] + " is already set to true.")
+            b.Say(args[0] + " is already set to true.")
         } else {
             fmt.Println("[cmd] mode - " + args[0])
-            conn.Privmsg(channel, args[0] + " is now enabled.")
+            b.Say(args[0] + " is now enabled.")
         }
     } else if strings.HasPrefix(msg, ".rmmode") && len(args) == 1 {
         if removeItemInSlice(args[0], rulemod) {
             fmt.Println("[cmd] rmmode - " + args[0])
-            conn.Privmsg(channel, args[0] + " has been removed from the list of modes.")
+            b.Say(args[0] + " has been removed from the list of modes.")
         } else {
-            conn.Privmsg(channel, args[0] + " isn't in the list of modes.")
+            b.Say(args[0] + " isn't in the list of modes.")
         }
     } else if strings.HasPrefix(msg, ".dm") && len(args) == 1 {
         if len(dunmas) == 0 {
             dunmas = args[0]
             fmt.Println("[cmd] dm - " + dunmas)
-            conn.Privmsg(channel, "dm is now set to " + dunmas)
+            b.Say("dm is now set to " + dunmas)
         } else {
-            conn.Privmsg(channel, "dm has already been set, the current DM is " + dunmas)
+            b.Say("dm has already been set, the current DM is " + dunmas)
         }
     } else if msg == ".resetdm" && (nick == dunmas || stringInSlice(nick, admins)) {
         dunmas = ""
         fmt.Println("[cmd] resetdm")
-        conn.Privmsg(channel, "dm has been reset")
+        b.Say("dm has been reset")
     } else if msg == ".quit" && stringInSlice(nick, admins) {
         fmt.Println("[cmd] shutdown from " + nick)
         os.Exit(1)
     }
 }
 
-func main() {
-    conn := irc.IRC(nickname, nickname)
+func (b *Bot) Listen() {
+    err := b.Conn.Connect(b.Server)
 
-    err := conn.Connect(server + ":" + strconv.Itoa(port))
-    if err != nil {
-        fmt.Print("[err] connection failed - ")
-        fmt.Println(err)
-    }
+    if err != nil { panic(err) }
 
-    conn.AddCallback("001", func(e *irc.Event) { conn.Join(channel) })
-
-    conn.AddCallback("PRIVMSG", func(e *irc.Event) {
-        handleMessage(e.Nick, e.Message(), conn)
+    b.Conn.AddCallback("001", func(e *irc.Event) {
+        b.Conn.Join(b.Channel)
     })
 
-    conn.Loop()
+    b.Conn.AddCallback("PRIVMSG", func(e *irc.Event) {
+        b.Command(e.Nick, e.Message())
+    })
+
+    b.Conn.Loop()
+}
+
+func NewBot(server string, channel string, nick string) *Bot {
+    return &Bot{Conn: irc.IRC(nick, nick), Server: server, Channel: channel, Nick: nick}
+}
+
+func main() {
+    b := NewBot("irc.iotek.org:6667", "#d20", "bot")
+    b.Listen()
 }