irc-osric/bot.go

package main

import (
	"fmt"
	"github.com/thoj/go-ircevent"
	"math/rand"
	"os"
	"strconv"
	"strings"
	"time"
)

type Bot struct {
	Conn    *irc.Connection
	Nick    string
	Server  string
	Channel string
}

var (
	admins = []string{"vypr", "dami", "kirby"}
	dieopt = []string{"4", "6", "8", "10", "12", "20"}
	dunmas = ""

	rulemod = []string{"voting"}
	modeopt = []string{"adminoverride", "logging", "voting"}
	initLog = true

	votemap = make(map[string]int)

	filename = "log/log_" + time.Now().Local().Format("20060102") + "_" + time.Now().Local().Format("150405") + ".txt"
)

var dict = map[string]string{
	"hp":   "health points",
	"ap":   "armour points",
	"algn": "alignment",
	"xp":   "experience points",
	"str":  "strength",
	"con":  "constitution",
	"dex":  "dexterity",
	"wis":  "wisdom",
	"cha":  "charisma",
	"lvl":  "level",
	"hgt":  "height",
	"wgt":  "weight",
	"cls":  "class",
}

var argmap = map[string]int{
	".set":     4,
	".print":   3,
	".vote":    1,
	".d":       1,
	".mode":    1,
	".rmmode":  1,
	".dm":      1,
	".import":  1,
	".choose":  0,
	".resetdm": 0,
	".quit":    0,
}

func roll(amount int, side int) int {
	var numbers = make([]int, amount)
	var finaln = 0

	for i := 0; i < amount; i++ {
		number := rand.Intn(side) + 1
		numbers[i] = number
	}

	for i := 0; i < len(numbers); i++ {
		if i == len(numbers) {
			return finaln
		}

		finaln = finaln + numbers[i]
	}

	return finaln
}

func vote(nick string) {
	if _, j := votemap[nick]; j {
		var tmp = votemap[nick] + 1
		votemap[nick] = tmp
	} else {
		votemap[nick] = 1
	}
}

func choose() (string, int) {
	nmap := sortMapByValue(votemap)
	return nmap[0].Key, nmap[0].Value
}

// TODO: Create functions related to character import/export.

func (b *Bot) Command(nick string, msg string) {
	var args = make([]string, len(strings.Split(msg, " "))-1)
	var command = ""

	if stringInSlice(modeopt[2], rulemod) {
		b.Log(nick+": "+msg, initLog)
		initLog = false
	}

	for i, j := range strings.Split(msg, " ") {
		if j != " " && i != 0 {
			args[i-1] = strings.Split(msg, " ")[i]
		}
	}

	for i := range argmap {
		if i == strings.Split(msg, " ")[0] {
			command = strings.Split(msg, " ")[0]
			break
		}
	}

	if argmap[strings.Split(msg, " ")[0]] != len(args) {
		return
	}

	switch command {
	case ".import":
		if importChar(args[0]) {
			b.Say("importing " + args[0] + " successful")
			fmt.Println("[cmd] import " + args[0])
		}

	case ".set":
		if nick == dunmas {
			if len(args) == 5 {
				if setChar(args[0], args[1], args[2], args[3], args[4]) {
					fmt.Println("[cmd] set")
				}
			} else {
				if setChar(args[0], args[1], "nil", args[2], args[3]) {
					fmt.Println("[cmd] set")
				}
			}
		} else if stringInSlice(nick, admins) && stringInSlice(modeopt[0], rulemod) {
			if len(args) == 5 {
				if setChar(args[0], args[1], args[2], args[3], args[4]) {
					fmt.Println("[cmd] set")
					b.Say(nick + " used override, it's super effective!")
				}
			} else {
				if setChar(args[0], args[1], "nil", args[2], args[3]) {
					fmt.Println("[cmd] set")
					b.Say(nick + " used override, it's super effective!")
				}
			}
		}
	case ".print":
		if len(args) == 4 {
			b.Say(args[0] + "[" + args[3] + "] = " + printChar(args[0], args[1], args[2], args[3]))
			fmt.Println("[cmd] print")
		} else {
			b.Say(args[0] + "[" + args[2] + "] = " + printChar(args[0], args[1], "nil", args[2]))
			fmt.Println("[cmd] print")
		}
		break

	case ".vote":
		if stringInSlice(modeopt[3], rulemod) {
			vote(args[0])
			fmt.Println("[cmd] vote - " + args[0])
		} else {
			b.Say(nick + " - the dm has already been chosen")
			b.Say(nick + " - have the dm or an admin use .resetdm if necessary")
		}
		break

	case ".d":
		amount, _ := strconv.Atoi(strings.Split(args[0], "d")[0])
		side, _ := strconv.Atoi(strings.Split(args[0], "d")[1])

		if stringInSlice(strconv.Itoa(side), dieopt) && len(strings.Split(args[0], "d")) < 3 && amount > 0 && amount <= 20 {
			fmt.Println("[cmd] rolling " + args[0])
			b.Say(nick + " rolled a " + strconv.Itoa(roll(amount, side)))
		}
		break

	case ".mode":
		if nick == dunmas || stringInSlice(nick, admins) {
			if stringInSlice(args[0], rulemod) {
				b.Say(args[0] + " is already set to true")
			} else if stringInSlice(args[0], modeopt) {
				fmt.Println("[cmd] mode - " + args[0])
				b.Say(args[0] + " is now enabled")
				rulemod = append(rulemod, args[0])
			}
		}
		break

	case ".rmmode":
		if nick == dunmas || stringInSlice(nick, admins) {
			if removeItemInSlice(args[0], rulemod) {
				fmt.Println("[cmd] rmmode - " + args[0])
				b.Say(args[0] + " has been removed from the list of modes")
			} else {
				b.Say(args[0] + " isn't in the list of modes")
			}
		}
		break

	case ".dm":
		if stringInSlice(nick, admins) && stringInSlice(modeopt[0], rulemod) {
			dunmas = args[0]
			fmt.Println("[cmd] dm - " + dunmas)
			b.Say("dm is now set to " + dunmas)
		}
		break

	case ".choose":
		var val = 0

		if stringInSlice(modeopt[3], rulemod) {
			dunmas, val = choose()
			fmt.Println("[cmd] choosing " + dunmas + " as dm")
			b.Say("the dm is now " + dunmas + " after " + strconv.Itoa(val) + " vote(s)")

			if removeItemInSlice(modeopt[3], rulemod) {
				b.Say("voting is now disabled")
			}
		}
		break

	case ".resetdm":
		if nick == dunmas && len(dunmas) > 0 {
			dunmas = ""
			fmt.Println("[cmd] resetdm")
			b.Say("dm has been reset")

			if stringInSlice(modeopt[3], rulemod) {
				rulemod = append(rulemod, modeopt[3])
				b.Say("voting is now enabled")
			}
		} else if stringInSlice(nick, admins) && stringInSlice(modeopt[0], rulemod) && len(dunmas) > 0 {
			dunmas = ""
			fmt.Println("[cmd] resetdm")
			b.Say("dm has been reset")
			b.Say(nick + " used override, it's super effective!")

			if stringInSlice(modeopt[3], rulemod) {
				rulemod = append(rulemod, modeopt[3])
				b.Say("voting is now enabled")
			}
		}
		break

	case ".quit":
		if stringInSlice(nick, admins) {
			fmt.Println("[cmd] shutdown from " + nick)
			os.Exit(1)
		}
		break
	}
}

func (b *Bot) Log(line string, initLog bool) {
	if initLog {
		os.Create(filename)

		file, fileerr := os.OpenFile(filename, os.O_RDWR|os.O_APPEND, 0660)

		if fileerr != nil {
			panic(fileerr)
		}

		file.WriteString("IRC-OSRIC by Elliott Pardee (vypr)\n")
		file.WriteString("----------------------------------\n\n")
		file.WriteString(line + "\n")
	} else {
		file, fileerr := os.OpenFile(filename, os.O_RDWR|os.O_APPEND, 0660)

		if fileerr != nil {
			panic(fileerr)
		}

		file.WriteString(line + "\n")
	}
}

func (b *Bot) Say(msg string) {
	if stringInSlice(modeopt[2], rulemod) {
		b.Log("bot: "+msg, initLog)
	}

	b.Conn.Privmsg(b.Channel, msg)
}

func (b *Bot) Listen() {
	err := b.Conn.Connect(b.Server)

	if err != nil {
		panic(err)
	}

	b.Conn.AddCallback("001", func(e *irc.Event) {
		b.Conn.Join(b.Channel)
	})

	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()
}