mirror-nourybot-matrix/cmd/nourybot/main.go

package main

import (
	"context"
	"errors"
	"flag"
	"fmt"
	"log"
	"os"
	"sync"
	"time"

	"github.com/chzyer/readline"
	"github.com/joho/godotenv"
	"github.com/lyx0/nourybot-matrix/internal/common"
	_ "github.com/mattn/go-sqlite3"
	"github.com/rs/zerolog"
	"maunium.net/go/mautrix"
	"maunium.net/go/mautrix/crypto/cryptohelper"
	"maunium.net/go/mautrix/event"
	"maunium.net/go/mautrix/id"
)

type config struct {
	matrixHomeserver string
	matrixUser       string
	matrixPass       string
	database         string
}

type Application struct {
	Mc  *mautrix.Client
	Log zerolog.Logger
}

var debug = flag.Bool("debug", false, "Enable debug logs")

func main() {
	flag.Parse()

	err := godotenv.Load()
	if err != nil {
		log.Fatal("Error loading .env")
	}

	var cfg config
	cfg.matrixHomeserver = os.Getenv("MATRIX_HOMESERVER")
	cfg.matrixUser = os.Getenv("MATRIX_USERNAME")
	cfg.matrixPass = os.Getenv("MATRIX_PASSWORD")
	cfg.database = os.Getenv("SQLITE_DATABASE")

	client, err := mautrix.NewClient(cfg.matrixHomeserver, "", "")
	if err != nil {
		panic(err)
	}

	rl, err := readline.New("[no room]> ")
	if err != nil {
		panic(err)
	}
	defer rl.Close()
	log := zerolog.New(zerolog.NewConsoleWriter(func(w *zerolog.ConsoleWriter) {
		w.Out = rl.Stdout()
		w.TimeFormat = time.Stamp
	})).With().Timestamp().Logger()
	if !*debug {
		log = log.Level(zerolog.InfoLevel)
	}
	client.Log = log

	var lastRoomID id.RoomID

	common.StartTime()

	app := &Application{
		Mc:  client,
		Log: log,
	}

	syncer := app.Mc.Syncer.(*mautrix.DefaultSyncer)
	syncer.OnEventType(event.EventMessage, func(source mautrix.EventSource, evt *event.Event) {
		lastRoomID = evt.RoomID
		rl.SetPrompt(fmt.Sprintf("%s> ", lastRoomID))

		// Filter out messages from the bot user account.
		if evt.Sender.String() != fmt.Sprintf("@%s:%s", cfg.matrixUser, cfg.matrixHomeserver) {
			app.Log.Info().
				Str("sender", evt.Sender.String()).
				Str("type", evt.Type.String()).
				Str("id", evt.ID.String()).
				Str("body", evt.Content.AsMessage().Body).
				Msg("Received message")

			if evt.Content.AsMessage().Body[:1] == "!" {
				app.ParseCommand(evt)
				return
			} else {
				app.ParseEvent(evt)
				return
			}
		}
	})
	syncer.OnEventType(event.StateMember, func(source mautrix.EventSource, evt *event.Event) {
		if evt.GetStateKey() == app.Mc.UserID.String() && evt.Content.AsMember().Membership == event.MembershipInvite {
			_, err := app.Mc.JoinRoomByID(evt.RoomID)
			if err == nil {
				lastRoomID = evt.RoomID
				rl.SetPrompt(fmt.Sprintf("%s> ", lastRoomID))
				app.Log.Info().
					Str("room_id", evt.RoomID.String()).
					Str("inviter", evt.Sender.String()).
					Msg("Joined room after invite")
			} else {
				app.Log.Error().Err(err).
					Str("room_id", evt.RoomID.String()).
					Str("inviter", evt.Sender.String()).
					Msg("Failed to join room after invite")
			}
		}
	})

	cryptoHelper, err := cryptohelper.NewCryptoHelper(app.Mc, []byte("meow"), cfg.database)
	if err != nil {
		panic(err)
	}

	cryptoHelper.LoginAs = &mautrix.ReqLogin{
		Type:       mautrix.AuthTypePassword,
		Identifier: mautrix.UserIdentifier{Type: mautrix.IdentifierTypeUser, User: cfg.matrixUser},
		Password:   cfg.matrixPass,
	}

	err = cryptoHelper.Init()
	if err != nil {
		panic(err)
	}

	app.Mc.Crypto = cryptoHelper

	app.Log.Info().Msg("Now running")
	syncCtx, cancelSync := context.WithCancel(context.Background())
	var syncStopWait sync.WaitGroup
	syncStopWait.Add(1)

	go func() {
		err = app.Mc.SyncWithContext(syncCtx)
		defer syncStopWait.Done()
		if err != nil && !errors.Is(err, context.Canceled) {
			panic(err)
		}
	}()

	for {
		line, err := rl.Readline()
		if err != nil {
			break
		}
		if lastRoomID == "" {
			app.Log.Error().Msg("Wait for an incoming message before sending messages")
			continue
		}
		resp, err := app.Mc.SendText(lastRoomID, line)
		if err != nil {
			app.Log.Error().Err(err).Msg("Failed to send event")
		} else {
			app.Log.Info().Str("event_id", resp.EventID.String()).Msg("Event sent")
		}
	}
	cancelSync()
	syncStopWait.Wait()
	err = cryptoHelper.Close()
	if err != nil {
		app.Log.Error().Err(err).Msg("Error closing database")
	}
}