144 lines
3.6 KiB
Go
144 lines
3.6 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/davecgh/go-spew/spew"
|
|
"github.com/eleksir/go-xmpp"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func buny(v xmpp.Presence) error {
|
|
var err error
|
|
|
|
// Если у presence-а есть JID и presence из одной из комнат, в которой мы есть и если его домен в чёрном
|
|
// списке, заносим пидора в список outcast-ов
|
|
if v.JID != "" {
|
|
// На всякий случай: себя никогда не баним, явным образом
|
|
if v.JID == talk.JID() {
|
|
return err
|
|
}
|
|
|
|
room := strings.SplitN(v.From, "/", 2)[0]
|
|
|
|
if room == "" {
|
|
log.Infof("We got empty room in repsence event, which kinda strange: %s", spew.Sdump(v))
|
|
|
|
return err
|
|
}
|
|
|
|
goodJid := strings.SplitN(v.JID, "/", 2)[0]
|
|
|
|
// Обрабатываем правила белого списка
|
|
for _, good := range whiteList.Whitelist {
|
|
// Глобальный белый список
|
|
if good.RoomName == "" {
|
|
for _, person := range good.Jid {
|
|
if person == goodJid {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
// Белый список для конкретных каналов
|
|
if good.RoomName == room {
|
|
for _, person := range good.Jid {
|
|
if person == goodJid {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
evilJid := strings.SplitN(v.JID, "/", 2)[0]
|
|
|
|
// Обрабатываем правила чёрного списка
|
|
for _, cRoom := range roomsConnected {
|
|
if cRoom == room {
|
|
for _, bEntry := range blackList.Blacklist {
|
|
|
|
// Обработаем правила гобального чёрного списка
|
|
if bEntry.RoomName == "" {
|
|
for _, jid_regexp := range bEntry.JidRe {
|
|
re, err := regexp.Compile(jid_regexp)
|
|
|
|
if err != nil {
|
|
log.Errorf("Incorrect regexp in blacklist: %s, skipping", jid_regexp)
|
|
|
|
continue
|
|
}
|
|
|
|
if re.Match([]byte(evilJid)) {
|
|
log.Infof("Hammer falls on %s", v.JID)
|
|
// https://xmpp.org/extensions/xep-0045.html#ban баним вот таким сообщением
|
|
ban := "<item affiliation='outcast' jid='" + evilJid + "'>"
|
|
ban += "<reason />"
|
|
ban += "</item>"
|
|
|
|
if id, err := talk.RawInformationQuery(
|
|
talk.JID(),
|
|
room,
|
|
"ban1",
|
|
xmpp.IQTypeSet,
|
|
"http://jabber.org/protocol/muc#admin",
|
|
ban,
|
|
); err != nil {
|
|
log.Errorf("Unable to ban user: id=%s, err=%s", id, err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
}
|
|
|
|
continue
|
|
}
|
|
|
|
// Обработаем правила конкретного канала, комнаты конференции
|
|
if bEntry.RoomName == room {
|
|
for _, jid_regexp := range bEntry.JidRe {
|
|
re, err := regexp.Compile(jid_regexp)
|
|
|
|
if err != nil {
|
|
log.Errorf("Incorrect regexp in blacklist: %s, skipping", jid_regexp)
|
|
|
|
continue
|
|
}
|
|
|
|
if re.Match([]byte(evilJid)) {
|
|
log.Infof("Hammer falls on %s", v.JID)
|
|
// https://xmpp.org/extensions/xep-0045.html#ban баним вот таким сообщением
|
|
ban := "<item affiliation='outcast' jid='" + evilJid + "'>"
|
|
ban += "<reason />"
|
|
ban += "</item>"
|
|
|
|
if id, err := talk.RawInformationQuery(
|
|
talk.JID(),
|
|
room,
|
|
"ban1",
|
|
xmpp.IQTypeSet,
|
|
"http://jabber.org/protocol/muc#admin",
|
|
ban,
|
|
); err != nil {
|
|
log.Errorf("Unable to ban user: id=%s, err=%s", id, err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
}
|
|
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return err
|
|
}
|