2024-03-10 17:21:27 -05:00
|
|
|
// package main is the main package for the LapisDeploy program
|
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"log"
|
2024-03-10 21:22:05 -05:00
|
|
|
"time"
|
2024-03-10 22:05:15 -05:00
|
|
|
"strings"
|
|
|
|
"fmt"
|
2024-03-10 17:21:27 -05:00
|
|
|
|
|
|
|
"maunium.net/go/mautrix"
|
|
|
|
"maunium.net/go/mautrix/event"
|
2024-03-10 21:22:05 -05:00
|
|
|
"maunium.net/go/mautrix/id"
|
|
|
|
"maunium.net/go/mautrix/format"
|
2024-03-10 17:21:27 -05:00
|
|
|
)
|
|
|
|
|
2024-03-10 22:05:15 -05:00
|
|
|
// MatrixMessage stores information about a matrix message.
|
2024-03-10 21:22:05 -05:00
|
|
|
type MatrixMessage struct {
|
|
|
|
text string
|
|
|
|
}
|
|
|
|
|
2024-03-10 22:05:15 -05:00
|
|
|
// MatrixCommand stores information about a matrix chat command
|
|
|
|
type MatrixCommand struct {
|
|
|
|
cmd string
|
|
|
|
callback func(ctx context.Context, evt *event.Event) string
|
|
|
|
description string
|
|
|
|
}
|
|
|
|
|
|
|
|
// Output queue
|
2024-03-10 21:22:05 -05:00
|
|
|
var MatrixOut []MatrixMessage
|
|
|
|
|
2024-03-10 22:05:15 -05:00
|
|
|
// ChatCommands
|
|
|
|
var Chatcommands []MatrixCommand
|
|
|
|
|
2024-03-10 21:22:05 -05:00
|
|
|
// sendMessage sends a message through Matrix
|
|
|
|
func sendMessage(msg MatrixMessage) {
|
|
|
|
MatrixOut = append(MatrixOut, msg)
|
|
|
|
}
|
|
|
|
|
2024-03-10 22:05:15 -05:00
|
|
|
// registerChatCommand
|
|
|
|
func registerChatCommand(cmd string, description string, callback func(ctx context.Context, evt *event.Event) string) {
|
|
|
|
Chatcommands = append(Chatcommands, MatrixCommand{
|
|
|
|
cmd: cmd,
|
|
|
|
callback: callback,
|
|
|
|
description: description,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2024-03-10 21:22:05 -05:00
|
|
|
// initMatrix starts a Matrix bot.
|
2024-03-10 17:21:27 -05:00
|
|
|
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())
|
|
|
|
}
|
|
|
|
}
|
2024-03-10 21:22:05 -05:00
|
|
|
})
|
2024-03-10 22:05:15 -05:00
|
|
|
syncer.OnEventType(event.EventMessage, func(ctx context.Context, evt *event.Event) {
|
|
|
|
if evt.RoomID.String() == configuration.matrix.room_id {
|
2024-03-10 22:40:45 -05:00
|
|
|
prefixes := []string{"!", configuration.matrix.username+" "}
|
|
|
|
for _,prefix := range prefixes {
|
|
|
|
if strings.HasPrefix(evt.Content.AsMessage().Body, prefix) {
|
|
|
|
found_command := false
|
|
|
|
for _, command := range Chatcommands {
|
|
|
|
if command.cmd == strings.TrimPrefix(evt.Content.AsMessage().Body, prefix) {
|
|
|
|
found_command = true
|
|
|
|
out := command.callback(ctx, evt)
|
|
|
|
content := format.RenderMarkdown(out, true, false)
|
|
|
|
content.SetReply(evt)
|
|
|
|
client.SendMessageEvent(ctx, evt.RoomID, event.EventMessage, content)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if !found_command {
|
|
|
|
client.SendText(ctx, evt.RoomID, "🔴 Command not found")
|
|
|
|
}
|
|
|
|
break
|
2024-03-10 22:05:15 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
})
|
|
|
|
|
|
|
|
registerChatCommand("help", "Show the help dialouge", func(ctx context.Context, evt *event.Event) string {
|
|
|
|
text := ""
|
|
|
|
for _,cmd := range Chatcommands {
|
|
|
|
text = fmt.Sprintf("%s- %s\n %s\n", text, cmd.cmd, cmd.description)
|
|
|
|
}
|
|
|
|
return text
|
|
|
|
})
|
2024-03-10 21:22:05 -05:00
|
|
|
|
2024-03-10 22:40:45 -05:00
|
|
|
registerChatCommand("sites", "Display all available sites", func(ctx context.Context, evt *event.Event) string {
|
|
|
|
text := "⚪️ Getting sites...\n"
|
|
|
|
sites, err := getAllSites()
|
|
|
|
if err.code != 0 {
|
|
|
|
return text+"🔴 Failed to get sites!"
|
|
|
|
}
|
|
|
|
for _, site := range sites {
|
|
|
|
text = fmt.Sprintf("%s- %s\n", text, site)
|
|
|
|
}
|
|
|
|
return text
|
|
|
|
})
|
|
|
|
|
2024-03-10 21:22:05 -05:00
|
|
|
go func () {
|
2024-03-10 22:52:07 -05:00
|
|
|
for range time.Tick(time.Second * 6) {
|
2024-03-10 21:22:05 -05:00
|
|
|
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{}
|
|
|
|
}
|
|
|
|
}()
|
2024-03-10 17:21:27 -05:00
|
|
|
|
|
|
|
log.Print("[Matrix] Running!")
|
|
|
|
err = client.Sync()
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
}
|