buny-jabber-bot/cmd/buny-jabber-bot/main.go

// Package main содержит основную функцию для запуска логики бота.
package main

import (
	"crypto/tls"
	"fmt"
	"log/slog"
	"os"
	"os/signal"
	"regexp"
	"time"

	_ "embed"

	"buny-jabber-bot/internal/jabber"
	"buny-jabber-bot/internal/log"

	"github.com/xmppo/go-xmpp"
	"gopkg.in/tomb.v2"
)

//go:embed version
var version string //nolint: grouper

// main - фактичеcки, начало и основное тело программы.
func main() {
	var (
		err error
	)

	for {
		var (
			logfile = os.Stderr
			j       = jabber.Jabber{ //nolint:exhaustruct
				SigChan:        make(chan os.Signal, 1),
				GTomb:          tomb.Tomb{},
				RoomsConnected: make([]string, 1),
			}
		)

		if err := j.ReadConfig(); err != nil {
			log.Errorf("%s", err)

			os.Exit(1)
		}

		j.C.Version = version

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

			if err != nil {
				log.Errorf("Unable to open log file %s: %s", j.C.Log, err)
				os.Exit(1)
			}
		}

		log.Init(j.C.Loglevel, logfile)

		j.C.ExeName, err = os.Executable()

		if err != nil {
			log.Fatalf("Unable to find my executable: %s", err)
		}

		// handle pricky windows case. But honestly i do not expect it runs on windows.
		if regexp.MustCompile(".[Ee][Xx][Ee]$").MatchString(j.C.ExeName) {
			j.C.ExeName = j.C.ExeName[:len(j.C.ExeName)-4]

			if j.C.ExeName == "" {
				j.C.ExeName = "buny-jabber-bot"
			}
		}

		log.Infof("Service %s v%s starting", j.C.ExeName, j.C.Version)

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

		verboseClient := false

		if log.LogLevel == slog.LevelDebug {
			if j.C.Debug.Lib {
				verboseClient = true
			}
		}

		// Самое время вгрузить белые и чёрные списки, если что - получим ошибку в лог.
		if err := j.ReadWhitelist(); err != nil {
			log.Error(err.Error())

			os.Exit(1)
		}

		if err := j.ReadBlacklist(); err != nil {
			log.Error(err.Error())

			os.Exit(1)
		}

		// Хэндлер сигналов.
		j.GTomb.Go(func() error { return j.SigHandler() }) //nolint: gocritic

		signal.Notify(j.SigChan, os.Interrupt)

		j.Options = &xmpp.Options{ //nolint:exhaustruct
			Host:     fmt.Sprintf("%s:%d", j.C.Jabber.Server, j.C.Jabber.Port),
			User:     j.C.Jabber.User,
			Password: j.C.Jabber.Password,
			Resource: j.C.Jabber.Resource,
			NoTLS:    !j.C.Jabber.Ssl,
			StartTLS: j.C.Jabber.StartTLS,
			TLSConfig: &tls.Config{ //nolint:exhaustruct
				ServerName:         j.C.Jabber.Server,
				InsecureSkipVerify: !j.C.Jabber.SslVerify, //nolint:gosec
			},
			InsecureAllowUnencryptedAuth: j.C.Jabber.InsecureAllowUnencryptedAuth,
			Debug:                        verboseClient,
			Session:                      false,
			Status:                       "xa",
			StatusMessage:                jabber.RandomPhrase(j.C.Jabber.StartupStatus),
			DialTimeout:                  time.Duration(j.C.Jabber.ConnectionTimeout) * time.Second,
			DebugWriter:                  log.Writer,
			PeriodicServerPings:          false,
			ReportSoftwareVersion:        true,
			SoftwareName:                 "Buny Jabber Bot",
			SoftwareVersion:              version,
			ReportSoftwareOS:             true,
		}

		// Враппер основной программы. Фактически на каждой серьёзной ошибке, например, сетевой, mainLoop и некоторые
		// вспомогательные корутины прекращает свою работу. Здесь мы это ловим и запускаем их снова. Задача в том, чтобы
		// запустить все асинхронные процессы заново, с чистого листа, но уже с распарсенным конфигом.

		// Устанавливаем соединение и гребём события, посылаемые сервером - основной и вспомогательные циклы программы.
		j.GTomb.Go(func() error { return j.MyLoop() }) //nolint: gocritic

		// Ловим первый же kill и не дождаемся остальных, хотя формально надо бы.
		<-j.GTomb.Dying()

		// Если у нас wire error, то вызов .Close() повлечёт за собой ошибку. А если у нас не wire error, то по ходу мы
		// получим утечку сокетов.

		time.Sleep(time.Duration(j.C.Jabber.ReconnectDelay) * time.Second)
	}
}

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