package main

import (
	"fmt"
	"regexp"
	"strings"
	"time"

	"github.com/davecgh/go-spew/spew"
	"github.com/eleksir/go-xmpp"
	log "github.com/sirupsen/logrus"
)

// bunyPresense производит проверку по бело-чёрным спискам. Если presence пришёл от злодея (из чёрного списка), то
// отправляет его в бан.
func bunyPresense(v xmpp.Presence) error { //nolint:gocognit,gocyclo
	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 field in presence 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]
		evilNick := ""
		evilNicks := strings.SplitN(v.From, "/", 2)

		// Давайте-ка похэндлим нонсенс - когда у нас в строке нету разделителя
		if len(evilNicks) > 1 {
			evilNick = evilNicks[1]
		}

		// Обрабатываем правила чёрного списка
		for _, cRoom := range roomsConnected {
			if cRoom == room {
				for _, bEntry := range blackList.Blacklist {
					// Обработаем правила глобального чёрного списка
					if bEntry.RoomName == "" {
						for _, jidRegexp := range bEntry.JidRe {
							if jidRegexp == "" {
								continue
							}

							re, err := regexp.Compile(jidRegexp)

							if err != nil {
								log.Errorf("Incorrect regexp in global blacklist: %s, skipping", jidRegexp)

								continue
							}

							log.Debugf("Checking jid %s vs global blacklist regex %s", v.JID, jidRegexp)

							if re.MatchString(v.JID) {
								log.Warnf(
									"Hammer falls on %s (%s): jid matches with global blacklist entry: %s",
									v.From,
									evilJid,
									jidRegexp,
								)

								id, err := squash(room, evilJid, bEntry.ReasonEnable, v.Type)

								if err != nil {
									err := fmt.Errorf(
										"unable to ban user: id=%s, err=%w",
										id,
										err,
									)

									gTomb.Kill(err)
								}

								return err
							}
						}

						for _, nickRegexp := range bEntry.NickRe {
							if nickRegexp == "" {
								continue
							}

							re, err := regexp.Compile(nickRegexp)

							if err != nil {
								log.Errorf("Incorrect regexp in global blacklist: %s, skipping", nickRegexp)

								continue
							}

							log.Debugf("Checking nick %s vs global blacklist regex %s", evilNick, nickRegexp)

							if re.MatchString(evilNick) {
								log.Warnf(
									"Hammer falls on %s (%s): nick matches with global blacklist entry: %s",
									v.From,
									evilJid,
									nickRegexp,
								)

								// Баним именно jid
								id, err := squash(room, evilJid, bEntry.ReasonEnable, v.Type)

								if err != nil {
									err := fmt.Errorf(
										"unable to ban user: id=%s, err=%w",
										id,
										err,
									)

									gTomb.Kill(err)
								}

								return err
							}
						}

						continue
					}

					// Обработаем правила конкретного канала, комнаты конференции
					if bEntry.RoomName == room {
						for _, jidRegexp := range bEntry.JidRe {
							if jidRegexp == "" {
								continue
							}

							re, err := regexp.Compile(jidRegexp)

							if err != nil {
								log.Errorf("Incorrect regexp in room %s blacklist: %s, skipping", room, jidRegexp)

								continue
							}

							log.Debugf("Checking jid %s vs room %s blacklist regex %s", v.JID, room, jidRegexp)

							if re.MatchString(v.JID) {
								log.Warnf(
									"Hammer falls on %s (%s): jid matches with room blacklist entry: %s",
									v.From,
									evilJid,
									jidRegexp,
								)

								id, err := squash(room, evilJid, bEntry.ReasonEnable, v.Type)

								if err != nil {
									err := fmt.Errorf(
										"unable to ban user: id=%s, err=%w",
										id,
										err,
									)

									gTomb.Kill(err)
								}

								return err
							}
						}

						if bEntry.RoomName == room {
							for _, nickRegexp := range bEntry.NickRe {
								if nickRegexp == "" {
									continue
								}

								re, err := regexp.Compile(nickRegexp)

								if err != nil {
									log.Errorf("Incorrect regexp in room %s blacklist: %s, skipping", room, nickRegexp)

									continue
								}

								log.Debugf("Checking nick %s vs room %s blacklist regex %s", evilJid, room, nickRegexp)

								if re.MatchString(evilNick) {
									log.Warnf(
										"Hammer falls on %s (%s): nick matches with room blacklist entry: %s",
										v.From,
										evilJid,
										nickRegexp,
									)

									// Баним именно jid
									id, err := squash(room, evilJid, bEntry.ReasonEnable, v.Type)

									if err != nil {
										err := fmt.Errorf(
											"unable to ban user: id=%s, err=%w",
											id,
											err,
										)

										gTomb.Kill(err)
									}

									return err
								}
							}
						}

						continue
					}
				}
			}
		}
	}

	return err
}

// bunyChat производит проверку сообщений участников чата по списку забаненных фраз и в случае нахождения запрещённого
// шаблона банит участника чата.
func bunyChat(v xmpp.Chat) error {
	var (
		room = (strings.SplitN(v.Remote, "/", 2))[0]
		// nick = (strings.SplitN(v.Remote, "/", 2))[1]
		err error
	)

	// Действовать мы можем только в рамках тех комнат, где явно присуствуем.
	for _, cRoom := range roomsConnected {
		if cRoom == room {
			// Перебирём правила чёрных списков.
			for _, bEntry := range blackList.Blacklist {
				// Обработаем правила глобального чёрного списка
				if bEntry.RoomName == "" {
					for _, phraseRegexp := range bEntry.PhraseRe {
						if phraseRegexp == "" {
							continue
						}

						re, err := regexp.Compile(phraseRegexp)

						if err != nil {
							log.Errorf("Incorrect regexp in room %s blacklist: %s, skipping", room, phraseRegexp)

							continue
						}

						log.Debugf("Checking phrase %s vs room %s blacklist regex %s", v.Text, room, phraseRegexp)

						if re.MatchString(v.Text) {
							realJID := getRealJIDfromNick(v.Remote)

							log.Warnf(
								"Hammer falls on %s (%s): phrase matches with global blacklist entry: %s vs %s",
								v.Remote,
								realJID,
								v.Text,
								phraseRegexp,
							)

							if id, err := squash(room, realJID, bEntry.ReasonEnable, v.Type); err != nil {
								err := fmt.Errorf(
									"unable to ban user: id=%s, err=%w",
									id,
									err,
								)

								gTomb.Kill(err)
							}

							return err
						}
					}
				}

				// Обработаем правила чёрного списка для конкретной комнаты
				if bEntry.RoomName == room {
					for _, phraseRegexp := range bEntry.PhraseRe {
						if phraseRegexp == "" {
							continue
						}

						re, err := regexp.Compile(phraseRegexp)

						if err != nil {
							log.Errorf("Incorrect regexp in room %s blacklist: %s, skipping", room, phraseRegexp)

							continue
						}

						log.Debugf("Checking phrase %s vs room %s blacklist regex %s", v.Text, room, phraseRegexp)

						if re.MatchString(v.Text) {
							realJID := getRealJIDfromNick(v.Remote)

							log.Warnf(
								"Hammer falls on %s (%s): phrase matches with room blacklist entry: %s vs %s",
								v.Remote,
								realJID,
								v.Text,
								phraseRegexp,
							)

							if id, err := squash(room, realJID, bEntry.ReasonEnable, v.Type); err != nil {
								err := fmt.Errorf(
									"unable to ban user: id=%s, err=%w",
									id,
									err,
								)

								gTomb.Kill(err)
							}

							return err
						}
					}
				}
			}

			// Если включено, проверяем фразу на КАПС.
			for _, channel := range config.Jabber.Channels {
				if channel.Name == room && channel.AllCaps.Enabled {
					// Нормализуем строку и вырежем из неё пробелы
					normPhrase := strings.ReplaceAll(nString(v.Text), " ", "")

					// Проверяем согласно тому, что длина фразы более чем сколько-то символов
					if len(normPhrase) >= channel.AllCaps.MinLength {
						normPhraseUpper := strings.ReplaceAll(nStringUpper(v.Text), " ", "")

						if normPhrase == normPhraseUpper {
							realJID := getRealJIDfromNick(v.Remote)
							id, err := squash(room, realJID, false, v.Type)

							if err != nil {
								err := fmt.Errorf(
									"unable to ban user: id=%s, err=%w",
									id,
									err,
								)

								gTomb.Kill(err)
							}
						}
					}
				}
			}

			break
		}
	}

	return err
}

// squash банит указанный jid в указанной комнате.
// reasonEnable указывает, надо ли писать дату автобана в банлисте в поле reason (это единственная причина, в которую
// умеет бот).
func squash(room, jid string, reasonEnable bool, vType string) (string, error) {
	var (
		id  string
		err error
	)

	if config.Jabber.BanPhrasesEnable {
		phrase := randomPhrase(config.Jabber.BanPhrases)

		if _, err = talk.Send(
			xmpp.Chat{ //nolint:exhaustruct
				Remote: room,
				Text:   phrase,
				Type:   vType,
			},
		); err != nil {
			err = fmt.Errorf("unable to send phrase to room %s: %w", room, err)

			// Здесь возвращаем nil, т.к. за нас ошибку залоггирует код выше
			return id, err
		}
	}

	// https://xmpp.org/extensions/xep-0045.html#ban баним вот таким сообщением
	ban := "<item affiliation='outcast' jid='" + jid + "'>"

	if reasonEnable {
		var t = time.Now()
		ban += fmt.Sprintf(
			"<reason>autoban at %04d.%02d.%02d %02d:%02d:%02d</reason>",
			t.Year(),
			t.Month(),
			t.Day(),
			t.Hour(),
			t.Minute(),
			t.Second(),
		)
	} else {
		ban += "<reason />"
	}

	ban += "</item>"

	// Выжидаем некоторое время перед баном. А то можно настолько рано забанить, что сервер не внесёт злодея в банлист
	// комнаты и пришлёт affiliation: none вместо affiliation: outcast.
	if config.Jabber.BanDelay > 0 {
		time.Sleep(time.Duration(config.Jabber.BanDelay) * time.Millisecond)
	}

	if id, err = talk.RawInformationQuery(
		talk.JID(),
		room,
		"ban1",
		xmpp.IQTypeSet,
		"http://jabber.org/protocol/muc#admin",
		ban,
	); err != nil {
		err = fmt.Errorf(
			"unable to ban user: id=%s, err=%w",
			id,
			err,
		)
	}

	return id, err
}

/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */