136 lines
4.3 KiB
Go
136 lines
4.3 KiB
Go
package main
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"time"
|
|
|
|
"buny-jabber-bot/internal/jabber"
|
|
|
|
"github.com/eleksir/go-xmpp"
|
|
log "github.com/sirupsen/logrus"
|
|
"gopkg.in/tomb.v2"
|
|
)
|
|
|
|
// main - фактичеcки, начало и основное тело программы.
|
|
func main() {
|
|
for {
|
|
var j = jabber.Jabber{
|
|
SigChan: make(chan os.Signal, 1),
|
|
GTomb: tomb.Tomb{},
|
|
RoomsConnected: make([]string, 1),
|
|
}
|
|
|
|
log.SetFormatter(&log.TextFormatter{ //nolint:exhaustruct
|
|
DisableQuote: true,
|
|
DisableLevelTruncation: false,
|
|
DisableColors: true,
|
|
FullTimestamp: true,
|
|
TimestampFormat: "2006-01-02 15:04:05",
|
|
})
|
|
|
|
if err := j.ReadConfig(); err != nil {
|
|
log.Error(err)
|
|
|
|
os.Exit(1)
|
|
}
|
|
|
|
// no panic
|
|
switch j.C.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)
|
|
}
|
|
|
|
// Откроем лог и скормим его логгеру.
|
|
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.Fatalf("Unable to open log file %s: %s", j.C.Log, err)
|
|
}
|
|
|
|
log.SetOutput(logfile)
|
|
}
|
|
|
|
myLogLevel := log.GetLevel()
|
|
log.Warnf("Loglevel set to %v", myLogLevel)
|
|
|
|
verboseClient := false
|
|
|
|
if myLogLevel == log.TraceLevel {
|
|
verboseClient = true
|
|
}
|
|
|
|
// Самое время вгрузить белые и чёрные списки, если что - получим ошибку в лог.
|
|
if err := j.ReadWhitelist(); err != nil {
|
|
log.Error(err)
|
|
|
|
os.Exit(1)
|
|
}
|
|
|
|
if err := j.ReadBlacklist(); err != nil {
|
|
log.Error(err)
|
|
|
|
os.Exit(1)
|
|
}
|
|
|
|
// github.com/mattn/go-xmpp пишет в stdio, нам этого не надо, ловим выхлоп его в logrus с уровнем trace.
|
|
xmpp.DebugWriter = log.WithFields(log.Fields{"logger": "stdlib"}).WriterLevel(log.TraceLevel)
|
|
|
|
// Хэндлер сигналов.
|
|
j.GTomb.Go(func() error { return j.SigHandler() })
|
|
|
|
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,
|
|
}
|
|
|
|
// Враппер основной программы. Фактически на каждой серьёзной ошибке, например, сетевой, mainLoop и некоторые
|
|
// вспомогательные корутины прекращает свою работу. Здесь мы это ловим и запускаем их снова. Задача в том, чтобы
|
|
// запустить все асинхронные процессы заново, с чистого листа, но уже с распарсенным конфигом.
|
|
|
|
// Устанавливаем соединение и гребём события, посылаемые сервером - основной и вспомогательные циклы программы.
|
|
j.GTomb.Go(func() error { return j.MyLoop() })
|
|
|
|
// Ловим первый же 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: */
|