package main

import (
	"crypto/tls"
	"errors"
	"fmt"
	"io"
	"net"
	"os"
	"os/signal"
	"time"

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

func init() {
	log.SetFormatter(&log.TextFormatter{ //nolint:exhaustruct
		DisableQuote:           true,
		DisableLevelTruncation: false,
		DisableColors:          true,
		FullTimestamp:          true,
		TimestampFormat:        "2006-01-02 15:04:05",
	})

	readConfig()

	// no panic
	switch config.Loglevel {
	case "fatal":
		log.SetLevel(log.FatalLevel)
	case "error":
		log.SetLevel(log.ErrorLevel)
	case "warn":
		log.SetLevel(log.WarnLevel)
	case "info":
		log.SetLevel(log.InfoLevel)
	case "debug":
		log.SetLevel(log.DebugLevel)
	case "trace":
		log.SetLevel(log.TraceLevel)
	default:
		log.SetLevel(log.InfoLevel)
	}

	//  Пока про capabilities мы ничего не знаем
	serverCapsQueried = false

	// Мы пока что не пинговали сервер.
	serverPingTimestampRx = 0
	serverPingTimestampTx = 0
}

func main() {
	// Откроем лог и скормим его логгеру
	if config.Log != "" {
		logfile, err := os.OpenFile(config.Log, os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0644)

		if err != nil {
			log.Fatalf("Unable to open log file %s: %s", config.Log, err)
		}

		log.SetOutput(logfile)
	}

	myLogLevel := log.GetLevel()
	log.Warnf("Loglevel set to %v", myLogLevel)

	verboseClient := false

	if myLogLevel == log.TraceLevel {
		verboseClient = true
	}

	// Самое время вгрузить белые и чёрные списки, если что - получим ошибку в лог.
	readWhitelist()
	readBlacklist()

	// github.com/mattn/go-xmpp пишет в stdio, нам этого не надо, ловим выхлоп его в logrus с уровнем trace
	xmpp.DebugWriter = log.WithFields(log.Fields{"logger": "stdlib"}).WriterLevel(log.TraceLevel)

	options = &xmpp.Options{ //nolint:exhaustruct
		Host:     fmt.Sprintf("%s:%d", config.Jabber.Server, config.Jabber.Port),
		User:     config.Jabber.User,
		Password: config.Jabber.Password,
		Resource: config.Jabber.Resource,
		NoTLS:    !config.Jabber.Ssl,
		StartTLS: config.Jabber.StartTLS,
		TLSConfig: &tls.Config{ //nolint:exhaustruct
			ServerName:         config.Jabber.Server,
			InsecureSkipVerify: !config.Jabber.SslVerify, //nolint:gosec
		},
		InsecureAllowUnencryptedAuth: config.Jabber.InsecureAllowUnencryptedAuth,
		Debug:                        verboseClient,
		Session:                      false,
		Status:                       "xa",
		StatusMessage:                "echo bot is starting up",
		DialTimeout:                  10 * time.Second,
	}

	go sigHandler()
	signal.Notify(sigChan, os.Interrupt)
	// Установим коннект
	establishConnection()

	serverPingTimestampRx = time.Now().Unix() // Считаем, что если коннект запустился, то первый пинг успешен

	// Тыкаем сервер палочкой, проверяем, что коннект жив и переустанавливаем его, если он не жив
	go probeServerLiveness()

	// Тыкаем muc-и палочкой, проверяем, что они живы и пере-заходим в них, если пинги пропали
	go probeMUCLiveness()

	// Гребём сообщения...
	for {
		// Стриггерилось завершение работы приложения, или соединение не установлено (порвалось, например)
		// грести не надо
		if shutdown {
			break
		}

		if !isConnected {
			continue
		}

		chat, err := talk.Recv()

		if err != nil {
			log.Errorf("Unable to get events from server: %s", err)

			switch {
			// Стрим не читается, он закрылся с той стороны во время чтения
			case errors.Is(err, io.EOF):
				os.Exit(1)

			// Пытаемся читать закрытый сокет
			case errors.Is(err, net.ErrClosed):
				os.Exit(1)

			// Не смогли записать в сокет
			case errors.Is(err, net.ErrWriteToConnected):
				os.Exit(1)

			// Не сетевая проблема
			default:
				// Это уже что-то странное.
				// Вероятно, ошибка парсинга xml. Собственно, баг сервера, тут мы ничего поделать не можем

				os.Exit(1)
			}
		}

		parseEvent(chat)
	}
}

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