LapisDeploy/matrix.go

82 lines
2.3 KiB
Go

// package main is the main package for the LapisDeploy program
package main
import (
"context"
"log"
"time"
"maunium.net/go/mautrix"
"maunium.net/go/mautrix/event"
"maunium.net/go/mautrix/id"
"maunium.net/go/mautrix/format"
)
type MatrixMessage struct {
text string
}
// I/O queues
var MatrixIn []MatrixMessage
var MatrixOut []MatrixMessage
// sendMessage sends a message through Matrix
func sendMessage(msg MatrixMessage) {
MatrixOut = append(MatrixOut, msg)
}
// initMatrix starts a Matrix bot.
func initMatrix() {
client, err := mautrix.NewClient(configuration.matrix.homeserver, "", "")
if err != nil {
log.Fatal(err)
}
resp, err := client.Login(context.Background(), &mautrix.ReqLogin{
Type: "m.login.password",
Identifier: mautrix.UserIdentifier{Type: mautrix.IdentifierTypeUser, User: configuration.matrix.username},
Password: configuration.matrix.password,
StoreCredentials: true,
})
if err != nil {
log.Panic(err)
}
log.Printf("[Matrix] Logged in as %s", resp.UserID)
syncer := client.Syncer.(*mautrix.DefaultSyncer)
syncer.OnEventType(event.StateMember, func(ctx context.Context, evt *event.Event) {
if evt.GetStateKey() == client.UserID.String() && evt.Content.AsMember().Membership == event.MembershipInvite {
if evt.RoomID.String() == configuration.matrix.room_id {
_, err := client.JoinRoomByID(ctx, evt.RoomID)
if err == nil {
log.Printf("[Matrix] Joined room '%s' (Invited by '%s')", evt.RoomID.String(), evt.Sender.String())
} else {
log.Printf("[Matrix] Failed to join room '%s' after invite (Invited by '%s')", evt.RoomID.String(), evt.Sender.String())
}
} else {
log.Printf("[Matrix] Rejected invite to room '%s' (Invited by '%s')", evt.RoomID.String(), evt.Sender.String())
}
}
})
go func () {
for range time.Tick(time.Second * 3) {
log.Printf("Scanning for queued messages...")
for _, msg := range MatrixOut {
_, err := client.SendMessageEvent(context.Background(), id.RoomID(configuration.matrix.room_id), event.EventMessage, format.RenderMarkdown(msg.text, true, false))
if err != nil {
log.Printf("[Matrix] ERROR : %s", err)
continue
}
log.Print("[Matrix] Sent queued message successfully.")
}
MatrixOut = []MatrixMessage{}
}
}()
log.Print("[Matrix] Running!")
err = client.Sync()
if err != nil {
log.Fatal(err)
}
}