ccchatbot/main.go

71 lines
1.5 KiB
Go
Raw Normal View History

2019-08-06 13:31:58 -05:00
package main
import (
"flag"
"os"
"os/signal"
"syscall"
"dev.sum7.eu/genofire/golang-lib/file"
"github.com/bdlm/log"
"gosrc.io/xmpp"
"gosrc.io/xmpp/stanza"
"dev.sum7.eu/ccchb/ccchatbot/runtime"
)
var config = Config{}
func main() {
configFile := "config.toml"
flag.StringVar(&configFile, "config", configFile, "path of configuration file")
flag.Parse()
if err := file.ReadTOML(configFile, &config); err != nil {
log.WithField("tip", "maybe call me with: ccchatbot --config /etc/ccchatbot.conf").Panicf("error on read config: %s", err)
}
log.SetLevel(config.LogLevel)
router := xmpp.NewRouter()
router.HandleFunc("presence", handlePresence)
router.HandleFunc("message", func(s xmpp.Sender, p stanza.Packet) {
msg, ok := p.(stanza.Message)
if !ok {
log.Errorf("ignoring wrong routed packet: %T", p)
return
}
if err := config.Schalter.HandleBotMessage(s, msg); err != nil {
log.Debugf("bot could not handle message: %s", err)
}
})
var err error
client, err := xmpp.NewClient(xmpp.Config{
Address: config.XMPP.Host,
Jid: config.XMPP.JID,
Password: config.XMPP.Password,
}, router)
if err != nil {
log.Panicf("error on startup xmpp client: %s", err)
}
cm := xmpp.NewStreamManager(client, func(s xmpp.Sender) {
config.Schalter.Run(s)
})
go func() {
err := cm.Run()
log.Panic("closed connection:", err)
}()
// Wait for system signal
sigs := make(chan os.Signal, 1)
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
sig := <-sigs
runtime.LeaveAllMUCs(client)
log.Infof("closed by receiving: %s", sig)
}