// modify executes the requested changes
func (c *Command) modify() bool {
effective := false
cnf := configuration.Config
for _, n := range c.toStart.Commands {
if _, ok := plugins.Commands[n]; ok {
cnf.Commands = append(cnf.Commands, n)
effective = true
}
}
for _, n := range c.toStart.Middlewares {
if _, ok := plugins.Middlewares[n]; ok {
cnf.Middlewares = append(cnf.Middlewares, n)
effective = true
}
}
for _, n := range c.toStop.Commands {
if _, ok := plugins.Commands[n]; ok && utils.StringInSlice(n, cnf.Commands) {
cnf.Commands, _ = utils.RemoveStringInSlice(n, cnf.Commands)
plugins.Commands[n].Stop()
effective = true
}
}
for _, n := range c.toStop.Middlewares {
if _, ok := plugins.Middlewares[n]; ok && utils.StringInSlice(n, cnf.Middlewares) {
cnf.Middlewares, _ = utils.RemoveStringInSlice(n, cnf.Middlewares)
plugins.Middlewares[n].Stop()
effective = true
}
}
plugins.Start()
return effective
}
func (c *Command) processArgs(ib *irc.Connection, to string) {
cnf := configuration.Config
if len(c.args) == 0 {
c.Start()
return
}
switch c.args[0] {
case "save":
if err := c.save(); err != nil {
ib.Privmsg(to, "Error while saving configuration. Consult the logs.")
} else {
ib.Privmsg(to, "Configuration saved")
}
c.Start()
return
case "admins":
if len(c.args) == 1 {
ib.Privmsgf(to, "Admins : %v", cnf.Admins)
} else {
if c.admins() {
ib.Privmsg(to, "Admins configuration changed")
}
}
c.Start()
return
case "reset":
// Can be done in a smarter way, we should copy the slices from configuration
// and stop/start only those different from the default config
plugins.Stop()
configuration.Load()
plugins.Start()
ib.Privmsg(to, "Configuration was reseted")
return
case "plugins":
if len(c.args) == 1 {
list := list()
ib.Privmsgf(to, "Commands : %s", list[0])
ib.Privmsgf(to, "Middlewares : %s", list[1])
c.Start()
return
}
if c.plugins() {
ib.Privmsg(to, "Commands configuration changed")
}
}
c.Start()
}
func main() {
var err error
cnf := configuration.Config
// Argument parsing
configuration.ConfPath = flag.String("c", "conf.yml", "Local path to configuration file.")
flag.Parse()
// Load the configuration of the bot
configuration.Load()
// Storage initialization
if err := database.BotStorage.Open(); err != nil {
log.Fatalf("something went wrong with the databse : %v", err)
}
plugins.Start()
// Defer all Close/Stop methods
defer database.BotStorage.Close()
defer plugins.Stop()
// Bot initialization
ib := irc.IRC(cnf.BotName, cnf.BotName)
if cnf.TLS {
ib.UseTLS = true
if cnf.InsecureTLS {
ib.TLSConfig = &tls.Config{InsecureSkipVerify: true}
}
}
if err = ib.Connect(cnf.Server); err != nil {
log.Fatal(err)
}
// Callback on 'Connected' event
ib.AddCallback("001", func(e *irc.Event) {
ib.Join(cnf.Channel)
})
// Callback on 'Message' event
ib.AddCallback("PRIVMSG", func(e *irc.Event) {
from := e.Nick
to := e.Arguments[0]
m := e.Message()
for _, c := range plugins.Middlewares {
c.Get(ib, from, to, m)
}
if strings.HasPrefix(m, cnf.CommandCharacter) {
if len(m) > 1 {
splitted := strings.Fields(m[1:])
command := splitted[0]
args := splitted[1:]
if p, ok := plugins.Commands[command]; ok {
p.Get(ib, from, to, args)
}
}
}
})
ib.Loop()
}