Working with Go's Style Guide
This commit is contained in:
Elliott Pardee
parent
216cb5d62d
commit
3266cc0f7e
244
bot.go
244
bot.go
@ -1,168 +1,178 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"github.com/thoj/go-ircevent"
|
||||
"strings"
|
||||
"fmt"
|
||||
"os"
|
||||
"fmt"
|
||||
"github.com/thoj/go-ircevent"
|
||||
"os"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type Bot struct {
|
||||
Conn *irc.Connection
|
||||
Nick string
|
||||
Server string
|
||||
Channel string
|
||||
Conn *irc.Connection
|
||||
Nick string
|
||||
Server string
|
||||
Channel string
|
||||
}
|
||||
|
||||
var (
|
||||
admins = []string{"vypr"}
|
||||
dunmas = ""
|
||||
admins = []string{"vypr"}
|
||||
dunmas = ""
|
||||
|
||||
rulemod = make([]string, len(modeopt))
|
||||
modeopt = []string{"adminoverride", "saves", "logging"}
|
||||
rulemod = make([]string, len(modeopt))
|
||||
modeopt = []string{"adminoverride", "saves", "logging"}
|
||||
|
||||
charmap = make(map[string]map[string]map[string]string)
|
||||
monsmap = make(map[string]string)
|
||||
charmap = make(map[string]map[string]map[string]string)
|
||||
monsmap = make(map[string]string)
|
||||
)
|
||||
|
||||
var dict = map[string]string{
|
||||
"hp": "health points",
|
||||
"ap": "armour points",
|
||||
"algn": "alignment",
|
||||
"xp": "experience points",
|
||||
"str": "strength",
|
||||
"dex": "dexterity",
|
||||
"wis": "wisdom",
|
||||
"cha": "charisma",
|
||||
"lvl": "level",
|
||||
"hgt": "height",
|
||||
"wgt": "weight",
|
||||
"cls": "class",
|
||||
"hp": "health points",
|
||||
"ap": "armour points",
|
||||
"algn": "alignment",
|
||||
"xp": "experience points",
|
||||
"str": "strength",
|
||||
"dex": "dexterity",
|
||||
"wis": "wisdom",
|
||||
"cha": "charisma",
|
||||
"lvl": "level",
|
||||
"hgt": "height",
|
||||
"wgt": "weight",
|
||||
"cls": "class",
|
||||
}
|
||||
|
||||
var argmap = map[string]int{
|
||||
".set": 4,
|
||||
".print": 3,
|
||||
".mode": 1,
|
||||
".rmmode": 1,
|
||||
".dm": 1,
|
||||
".resetdm": 1,
|
||||
".quit": 0,
|
||||
".set": 4,
|
||||
".print": 3,
|
||||
".mode": 1,
|
||||
".rmmode": 1,
|
||||
".dm": 1,
|
||||
".resetdm": 1,
|
||||
".quit": 0,
|
||||
}
|
||||
|
||||
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, }, }, }
|
||||
charmap = map[string]map[string]map[string]string{
|
||||
nick: map[string]map[string]string{
|
||||
cat: map[string]string{
|
||||
item: val,
|
||||
},
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func (b *Bot) Command(nick string, msg string) {
|
||||
var args = make([]string, len(strings.Split(msg, " ")) - 1)
|
||||
var command = ""
|
||||
var args = make([]string, len(strings.Split(msg, " "))-1)
|
||||
var command = ""
|
||||
|
||||
for i, j := range strings.Split(msg, " ") {
|
||||
if j != " " && i !=0 {
|
||||
args[i - 1] = strings.Split(msg, " ")[i]
|
||||
}
|
||||
}
|
||||
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
|
||||
}
|
||||
}
|
||||
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 }
|
||||
if argmap[strings.Split(msg, " ")[0]] != len(args) {
|
||||
return
|
||||
}
|
||||
|
||||
// TODO: Check if mode is enabled and if command can be applied.
|
||||
// TODO: Check if mode is enabled and if command can be applied.
|
||||
|
||||
switch command {
|
||||
case ".set":
|
||||
if nick == dunmas {
|
||||
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])
|
||||
} else if stringInSlice(nick, admins) && !stringInSlice("adminoverride", 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])
|
||||
b.Say(nick + " used override, it's super effective!")
|
||||
}
|
||||
break
|
||||
switch command {
|
||||
case ".set":
|
||||
if nick == dunmas {
|
||||
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])
|
||||
} else if stringInSlice(nick, admins) && !stringInSlice("adminoverride", 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])
|
||||
b.Say(nick + " used override, it's super effective!")
|
||||
}
|
||||
break
|
||||
|
||||
case ".print":
|
||||
fmt.Println("[cmd] print - " + args[0] + "'s " + args[2] + " in " + args[1])
|
||||
b.Say(args[0] + "'s " + args[2] + " is set to " + charmap[args[0]][args[1]][args[2]])
|
||||
break
|
||||
case ".print":
|
||||
fmt.Println("[cmd] print - " + args[0] + "'s " + args[2] + " in " + args[1])
|
||||
b.Say(args[0] + "'s " + args[2] + " is set to " + charmap[args[0]][args[1]][args[2]])
|
||||
break
|
||||
|
||||
case ".mode":
|
||||
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 ".mode":
|
||||
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 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 ".rmmode":
|
||||
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 len(dunmas) == 0 {
|
||||
dunmas = args[0]
|
||||
fmt.Println("[cmd] dm - " + dunmas)
|
||||
b.Say("dm is now set to " + dunmas)
|
||||
} else {
|
||||
b.Say("dm has already been set, the current DM is " + dunmas)
|
||||
}
|
||||
break
|
||||
case ".dm":
|
||||
if len(dunmas) == 0 {
|
||||
dunmas = args[0]
|
||||
fmt.Println("[cmd] dm - " + dunmas)
|
||||
b.Say("dm is now set to " + dunmas)
|
||||
} else {
|
||||
b.Say("dm has already been set, the current DM is " + dunmas)
|
||||
}
|
||||
break
|
||||
|
||||
case ".resetdm":
|
||||
if nick == dunmas || stringInSlice(nick, admins) {
|
||||
dunmas = ""
|
||||
fmt.Println("[cmd] resetdm")
|
||||
b.Say("dm has been reset")
|
||||
}
|
||||
break
|
||||
case ".resetdm":
|
||||
if nick == dunmas || stringInSlice(nick, admins) {
|
||||
dunmas = ""
|
||||
fmt.Println("[cmd] resetdm")
|
||||
b.Say("dm has been reset")
|
||||
}
|
||||
break
|
||||
|
||||
case ".quit":
|
||||
if stringInSlice(nick, admins) {
|
||||
fmt.Println("[cmd] shutdown from " + nick)
|
||||
os.Exit(1)
|
||||
}
|
||||
break
|
||||
}
|
||||
case ".quit":
|
||||
if stringInSlice(nick, admins) {
|
||||
fmt.Println("[cmd] shutdown from " + nick)
|
||||
os.Exit(1)
|
||||
}
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
func (b *Bot) Say(msg string) {
|
||||
b.Conn.Privmsg(b.Channel, msg)
|
||||
b.Conn.Privmsg(b.Channel, msg)
|
||||
}
|
||||
|
||||
func (b *Bot) Listen() {
|
||||
err := b.Conn.Connect(b.Server)
|
||||
err := b.Conn.Connect(b.Server)
|
||||
|
||||
if err != nil { panic(err) }
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
b.Conn.AddCallback("001", func(e *irc.Event) {
|
||||
b.Conn.Join(b.Channel)
|
||||
})
|
||||
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.AddCallback("PRIVMSG", func(e *irc.Event) {
|
||||
b.Command(e.Nick, e.Message())
|
||||
})
|
||||
|
||||
b.Conn.Loop()
|
||||
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}
|
||||
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()
|
||||
b := NewBot("irc.iotek.org:6667", "#d20", "bot")
|
||||
b.Listen()
|
||||
}
|
||||
|
||||
24
slices.go
24
slices.go
@ -1,20 +1,22 @@
|
||||
package main
|
||||
|
||||
func stringInSlice(a string, list []string) bool {
|
||||
for _, b := range list {
|
||||
if b == a { return true}
|
||||
}
|
||||
for _, b := range list {
|
||||
if b == a {
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
return false
|
||||
}
|
||||
|
||||
func removeItemInSlice(a string, list []string) bool {
|
||||
for c, b := range list {
|
||||
if b == a {
|
||||
list[c] = ""
|
||||
return true
|
||||
}
|
||||
}
|
||||
for c, b := range list {
|
||||
if b == a {
|
||||
list[c] = ""
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
return false
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user