109 lines
2.4 KiB
Go
109 lines
2.4 KiB
Go
package xmpp
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
xmpp "github.com/mattn/go-xmpp"
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/genofire/logmania/bot"
|
|
"github.com/genofire/logmania/lib"
|
|
"github.com/genofire/logmania/notify"
|
|
configNotify "github.com/genofire/logmania/notify/config"
|
|
)
|
|
|
|
var logger = log.WithField("notify", "xmpp")
|
|
|
|
type Notifier struct {
|
|
notify.Notifier
|
|
client *xmpp.Client
|
|
state *configNotify.NotifyState
|
|
formatter *log.TextFormatter
|
|
}
|
|
|
|
func Init(config *lib.NotifyConfig, state *configNotify.NotifyState, bot *bot.Bot) notify.Notifier {
|
|
options := xmpp.Options{
|
|
Host: config.XMPP.Host,
|
|
User: config.XMPP.Username,
|
|
Password: config.XMPP.Password,
|
|
NoTLS: config.XMPP.NoTLS,
|
|
Debug: config.XMPP.Debug,
|
|
Session: config.XMPP.Session,
|
|
Status: config.XMPP.Status,
|
|
StatusMessage: config.XMPP.StatusMessage,
|
|
}
|
|
client, err := options.NewClient()
|
|
if err != nil {
|
|
logger.Error(err)
|
|
return nil
|
|
}
|
|
go func() {
|
|
for {
|
|
chat, err := client.Recv()
|
|
if err != nil {
|
|
logger.Warn(err)
|
|
}
|
|
switch v := chat.(type) {
|
|
case xmpp.Chat:
|
|
bot.Handle(func(answer string) {
|
|
client.SendHtml(xmpp.Chat{Remote: v.Remote, Type: "chat", Text: answer})
|
|
}, fmt.Sprintf("xmpp:%s", strings.Split(v.Remote, "/")[0]), v.Text)
|
|
}
|
|
}
|
|
}()
|
|
logger.Info("startup")
|
|
return &Notifier{
|
|
client: client,
|
|
state: state,
|
|
formatter: &log.TextFormatter{
|
|
DisableTimestamp: true,
|
|
},
|
|
}
|
|
}
|
|
|
|
func (n *Notifier) Fire(e *log.Entry) error {
|
|
to := n.state.SendTo(e)
|
|
if to == nil {
|
|
return errors.New("no reciever found")
|
|
}
|
|
text, err := n.formatter.Format(e)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, toAddr := range to {
|
|
to := strings.TrimPrefix(toAddr, "xmpp:")
|
|
if strings.Contains(toAddr, "conference") || strings.Contains(toAddr, "irc") {
|
|
n.client.JoinMUCNoHistory(to, "logmania")
|
|
_, err = n.client.SendHtml(xmpp.Chat{Remote: to, Type: "groupchat", Text: string(text)})
|
|
if err != nil {
|
|
logger.Error("xmpp to ", to, " error:", err)
|
|
}
|
|
} else {
|
|
_, err := n.client.SendHtml(xmpp.Chat{Remote: to, Type: "chat", Text: string(text)})
|
|
if err != nil {
|
|
logger.Error("xmpp to ", to, " error:", err)
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (n *Notifier) Levels() []log.Level {
|
|
return []log.Level{
|
|
log.DebugLevel,
|
|
log.InfoLevel,
|
|
log.WarnLevel,
|
|
log.ErrorLevel,
|
|
log.FatalLevel,
|
|
log.PanicLevel,
|
|
}
|
|
}
|
|
|
|
func (n *Notifier) Close() {}
|
|
|
|
func init() {
|
|
notify.AddNotifier(Init)
|
|
}
|