logmania/notify/xmpp/main.go

46 lines
1.1 KiB
Go

package xmpp
import (
"github.com/genofire/logmania/database"
"github.com/genofire/logmania/lib"
"github.com/genofire/logmania/log"
"github.com/genofire/logmania/notify"
xmpp "github.com/mattn/go-xmpp"
)
type Notifier struct {
notify.Notifier
client *xmpp.Client
}
func NotifyInit(config *lib.NotifyConfig) 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 {
return nil
}
return &Notifier{client: client}
}
func (n *Notifier) Send(e *database.Entry) {
users := database.UserByApplication(e.ApplicationID)
for _, user := range users {
if user.NotifyXMPP && user.NotifyAfterLoglevel <= log.LogLevel(e.Level) {
n.client.SendHtml(xmpp.Chat{Remote: user.XMPP, Type: "chat", Text: FormatEntry(e)})
}
}
}
func init() {
notify.AddNotifier(NotifyInit)
}