wip

2022-08-21 18:41:35 +03:00
commit c4b7a16e21
22 changed files with 1745 additions and 0 deletions
--- a/cmd/cmd.go
+++ b/cmd/cmd.go
@@ -0,0 +1,119 @@
+package main
+
+import (
+	"database/sql"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"github.com/getsentry/sentry-go"
+	_ "github.com/lib/pq"
+	_ "github.com/mattn/go-sqlite3"
+	"gitlab.com/etke.cc/go/logger"
+	"gitlab.com/etke.cc/linkpearl"
+	lpcfg "gitlab.com/etke.cc/linkpearl/config"
+	"gitlab.com/etke.cc/postmoogle/bot"
+	"gitlab.com/etke.cc/postmoogle/config"
+	"gitlab.com/etke.cc/postmoogle/smtp"
+	"maunium.net/go/mautrix/id"
+)
+
+var (
+	mxb *bot.Bot
+	log *logger.Logger
+)
+
+func main() {
+	quit := make(chan struct{})
+	cfg := config.New()
+	log = logger.New("postmoogle.", cfg.LogLevel)
+
+	log.Info("#############################")
+	log.Info("Postmoogle")
+	log.Info("Matrix: true")
+	log.Info("#############################")
+
+	log.Debug("starting internal components...")
+	initSentry(cfg)
+	initBot(cfg)
+	initShutdown(quit)
+	defer recovery()
+
+	smtp.NewServer(cfg.Domain, map[string]id.RoomID{}, cfg.Port)
+	log.Debug("starting matrix bot...")
+	err := mxb.Start()
+	if err != nil {
+		//nolint:gocritic
+		log.Fatal("cannot start the bot: %v", err)
+	}
+
+	<-quit
+}
+
+func initSentry(cfg *config.Config) {
+	err := sentry.Init(sentry.ClientOptions{
+		Dsn:              cfg.Sentry.DSN,
+		AttachStacktrace: true,
+		TracesSampleRate: float64(cfg.Sentry.SampleRate) / 100,
+	})
+	if err != nil {
+		log.Fatal("cannot initialize sentry: %v", err)
+	}
+}
+
+func initBot(cfg *config.Config) {
+	db, err := sql.Open(cfg.DB.Dialect, cfg.DB.DSN)
+	if err != nil {
+		log.Fatal("cannot initialize SQL database: %v", err)
+	}
+	mxlog := logger.New("matrix.", cfg.LogLevel)
+	lp, err := linkpearl.New(&lpcfg.Config{
+		Homeserver:   cfg.Homeserver,
+		Login:        cfg.Login,
+		Password:     cfg.Password,
+		DB:           db,
+		Dialect:      cfg.DB.Dialect,
+		NoEncryption: cfg.NoEncryption,
+		LPLogger:     mxlog,
+		APILogger:    logger.New("api.", cfg.LogLevel),
+		StoreLogger:  logger.New("store.", cfg.LogLevel),
+		CryptoLogger: logger.New("olm.", cfg.LogLevel),
+	})
+	if err != nil {
+		// nolint // Fatal = panic, not os.Exit()
+		log.Fatal("cannot initialize matrix bot: %v", err)
+	}
+	mxb = bot.New(lp, mxlog, cfg.Prefix, cfg.Domain)
+	log.Debug("bot has been created")
+}
+
+func initShutdown(quit chan struct{}) {
+	listener := make(chan os.Signal, 1)
+	signal.Notify(listener, os.Interrupt, syscall.SIGABRT, syscall.SIGHUP, syscall.SIGINT, syscall.SIGQUIT, syscall.SIGTERM)
+
+	go func() {
+		<-listener
+		defer close(quit)
+
+		shutdown()
+	}()
+}
+
+func shutdown() {
+	log.Info("Shutting down...")
+	mxb.Stop()
+
+	sentry.Flush(5 * time.Second)
+	log.Info("Postmoogle has been stopped")
+	os.Exit(0)
+}
+
+func recovery() {
+	defer shutdown()
+	err := recover()
+	// no problem just shutdown
+	if err == nil {
+		sentry.CurrentHub().Recover(err)
+	}
+}