Add some cats

pkg/connector/backfill.go

@@ -3,6 +3,7 @@ package connector
 import (
 	"context"
 	"fmt"
+	"math"
 	"math/rand"
 	"time"
 
@@ -33,7 +34,7 @@ func (dc *DummyClient) FetchMessages(ctx context.Context, fetchParams bridgev2.F
 	}
 
 	for i := 0; i < fetchParams.Count; i++ {
-		sender := stablePortalUserIDByIndex(fetchParams.Portal.ID, rand.Intn(dc.Connector.Config.Automation.Portals.Members))
+		sender := stablePortalUserIDByIndex(fetchParams.Portal.ID, rand.Intn((int)(math.Max(1, (float64)(dc.Connector.Config.Automation.Portals.Members)))))
 		_, err := dc.UserLogin.Bridge.GetGhostByID(ctx, sender)
 		if err != nil {
 			return nil, fmt.Errorf("failed to get ghost by id: %w", err)

pkg/connector/commands.go

@@ -1,7 +1,14 @@
 package connector
 
 import (
+	"context"
+	"encoding/json"
+	"io"
+	"math/rand"
+	"net/http"
 	"strconv"
+	"strings"
+	"time"
 
 	"maunium.net/go/mautrix/bridgev2/commands"
 	"maunium.net/go/mautrix/event"
@@ -11,6 +18,8 @@ var AllCommands = []commands.CommandHandler{
 	NewRoomCommand,
 	GhostsCommand,
 	MessagesCommand,
+	FileCommand,
+	CatCommand,
 }
 
 var DummyHelpsection = commands.HelpSection{
@@ -116,3 +125,203 @@ var MessagesCommand = &commands.FullHandler{
 	},
 	RequiresLogin: true,
 }
+
+var FileCommand = &commands.FullHandler{
+	Func: func(e *commands.Event) {
+		e.Reply("Generating file event in this room")
+
+		var mediaData []byte
+		mediaData = []byte("Test text file")
+		mediaName := "test.txt"
+		mediaMime := "text/plain"
+
+		url, fi, err := e.Bot.UploadMedia(e.Ctx, e.RoomID, mediaData, mediaName, mediaMime)
+		if err != nil {
+			e.Reply(err.Error())
+			return
+		}
+
+		content := event.Content{
+			Parsed: &event.MessageEventContent{
+				MsgType: event.MsgFile,
+				URL:     url,
+				Body:    "Caption for file " + mediaName,
+				Info: &event.FileInfo{
+					Size:     len(mediaData),
+					MimeType: mediaMime,
+				},
+				File: fi,
+			},
+		}
+
+		resp, err := e.Bot.SendMessage(e.Ctx, e.RoomID, event.EventMessage, &content, nil)
+		if err != nil {
+			e.Reply(err.Error())
+			return
+		}
+
+		e.Reply(resp.EventID.String())
+	},
+	Name: "file",
+	Help: commands.HelpMeta{
+		Description: "Create boring file events in room",
+		Section:     DummyHelpsection,
+	},
+}
+
+var catpions []string = []string{
+	"You’ve cat to be kitten me!",
+	"I’m feline fine!",
+	"That was a total cat-astrophe.",
+	"Cut the cat-itude!",
+	"Let's go to Pawbucks and grab a few cat-puccinos.",
+	"Meow you doing?",
+	"My cat got fined for littering.",
+	"You’re the cat’s paw-jamas!",
+	"Meow you’re talking!",
+	"I’m just kitten around.",
+	"My cat sure is purr-suasive.",
+	"How do you like me meow?",
+	"Want a meowtini? Shaken, not purred, of course.",
+	"Turn up the mewsic and let’s get this pawty started!",
+	"Stop stressing meowt!",
+	"These puns are just hiss-terical!",
+	"My cat is often confused. You could say he gets pretty purr-plexed.",
+	"Now wait just a meowment…",
+	"My purr-fect cat thinks everyone is in-fur-rior to them!",
+	"Let me put my thinking cat on.",
+	"You're going to be my feline friend fur-ever!",
+	"Don’t fur-get to buy more catnip!",
+	"Looking good, feline good.",
+	"My cat is going down in hiss-story as the best cat ever.",
+	"Stop fighting! Just hiss and make up.",
+	"Press paws and live in the meow.",
+	"I love you, meow and forever.",
+	"It was meant to be. You could say it was kitten in the stars.",
+	"My cat has quite the purr-sonality!",
+	"You're as purr-ty as a picture!",
+	"With the right catitude, anything is pawsible!",
+	"I can tell you have a secret. It’s kitten all over your face!",
+	"I’m litter-ly in love with you.",
+	"Happy purr-thday!",
+	"I’m a total cat purr-son.",
+	"Pass the paw-pcorn, please.",
+}
+
+var catClient http.Client = http.Client{
+	Timeout: 10 * time.Second,
+}
+
+type catDesc struct {
+	ID     string `json:"id"`
+	URL    string `json:"url"`
+	Width  int    `json:"width"`
+	Height int    `json:"height"`
+}
+
+func searchCat(ctx context.Context) (catDesc, error) {
+	// this is only for individual testing so we're not using the authenticated API
+	req, err := http.NewRequestWithContext(ctx, http.MethodGet, "https://api.thecatapi.com/v1/images/search", nil)
+	if err != nil {
+		return catDesc{}, err
+	}
+
+	resp, err := catClient.Do(req)
+	if err != nil {
+		return catDesc{}, err
+	}
+
+	var catDescs []catDesc
+	err = json.NewDecoder(resp.Body).Decode(&catDescs)
+	if len(catDescs) > 0 {
+		return catDescs[0], err
+	}
+
+	return catDesc{}, err
+}
+
+func getCat(ctx context.Context, url string) (string, []byte, error) {
+	req, err := http.NewRequestWithContext(ctx, http.MethodGet, url, nil)
+	if err != nil {
+		return "", nil, err
+	}
+
+	resp, err := catClient.Do(req)
+	if err != nil {
+		return "", nil, err
+	}
+
+	data, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return "", nil, err
+	}
+
+	return resp.Header.Get("Content-Type"), data, nil
+}
+
+var CatCommand = &commands.FullHandler{
+	Func: func(e *commands.Event) {
+		e.Log.Debug().Msg("Searching for cat")
+
+		catDesc, err := searchCat(e.Ctx)
+		if err != nil {
+			e.Reply(err.Error())
+			return
+		}
+
+		mediaMime, mediaData, err := getCat(e.Ctx, catDesc.URL)
+		if err != nil {
+			e.Reply(err.Error())
+			return
+		}
+
+		mediaName := catDesc.ID
+		if strings.Contains(mediaMime, "png") {
+			mediaName = mediaName + ".png"
+		} else if strings.Contains(mediaMime, "jp") {
+			mediaName = mediaName + ".jpg"
+		} else if strings.Contains(mediaMime, "gif") {
+			mediaName = mediaName + ".gif"
+		} else if !strings.HasPrefix(mediaMime, "image/") {
+			e.Reply("Failed to get a cat: %s", mediaMime)
+			return
+		}
+
+		e.Log.Debug().Msg("Uploading cat")
+		url, fi, err := e.Bot.UploadMedia(e.Ctx, e.RoomID, mediaData, mediaName, mediaMime)
+		if err != nil {
+			e.Reply(err.Error())
+			return
+		}
+
+		catpion := catpions[rand.Intn(len(catpions)-1)]
+		content := event.Content{
+			Parsed: &event.MessageEventContent{
+				MsgType: event.MsgImage,
+				URL:     url,
+				Body:    catpion,
+				Info: &event.FileInfo{
+					Size:     len(mediaData),
+					Width:    catDesc.Width,
+					Height:   catDesc.Height,
+					MimeType: mediaMime,
+				},
+				File: fi,
+			},
+		}
+
+		e.Log.Debug().Msg("Sending cat")
+
+		e.Reply(catpion)
+		_, err = e.Bot.SendMessage(e.Ctx, e.RoomID, event.EventMessage, &content, nil)
+		if err != nil {
+			e.Reply(err.Error())
+			return
+		}
+	},
+	Name: "cat",
+	Help: commands.HelpMeta{
+		Description: "You know if you need one",
+		Section:     DummyHelpsection,
+	},
+}