From 911cfa5eef1c8e5865fb7e4480e6a84bd29684fe Mon Sep 17 00:00:00 2001
From: Devon Hudson <devonhudson@librem.one>
Date: Mon, 25 Sep 2023 20:32:23 -0600
Subject: [PATCH] Small tweaks to work with pseudoIDs

---
 internal/federation/server.go | 44 ++++++++++++++++++++++++++++++++---
 1 file changed, 41 insertions(+), 3 deletions(-)

diff --git a/internal/federation/server.go b/internal/federation/server.go
index e9159253..c82827a7 100644
--- a/internal/federation/server.go
+++ b/internal/federation/server.go
@@ -23,6 +23,7 @@ import (
 	"github.com/matrix-org/gomatrix"
 	"github.com/matrix-org/gomatrixserverlib/fclient"
 	"github.com/matrix-org/gomatrixserverlib/spec"
+	"github.com/tidwall/sjson"
 
 	"github.com/gorilla/mux"
 	"github.com/matrix-org/gomatrixserverlib"
@@ -361,17 +362,54 @@ func (s *Server) MustJoinRoom(t *testing.T, deployment *docker.Deployment, remot
 	if err != nil {
 		t.Fatalf("MustJoinRoom: invalid room version: %v", err)
 	}
+
+	var senderID spec.SenderID
+	signingKey := s.Priv
+	keyID := s.KeyID
+	origOrigin := origin
+	switch roomVer {
+	case gomatrixserverlib.RoomVersionPseudoIDs:
+		_, key, err := ed25519.GenerateKey(nil)
+		if err != nil {
+			t.Fatalf("MustJoinRoom: failed generating senderID: %v", err)
+		}
+		senderID, signingKey, err = spec.SenderIDFromPseudoIDKey(key), key, nil
+		keyID = "ed25519:1"
+		origin = spec.ServerName(senderID)
+		mapping := gomatrixserverlib.MXIDMapping{
+			UserRoomKey: senderID,
+			UserID:      userID,
+		}
+		if err = mapping.Sign(origOrigin, s.KeyID, s.Priv); err != nil {
+			t.Fatalf("MustJoinRoom: failed signing mxid_mapping: %v", err)
+		}
+
+		path := "mxid_mapping"
+		eventJSON, err := sjson.SetBytes(makeJoinResp.JoinEvent.Content, path, mapping)
+		if err != nil {
+			t.Fatalf("MustJoinRoom: failed setting mxid_mapping in content: %v", err)
+		}
+		eventJSON = gomatrixserverlib.CanonicalJSONAssumeValid(eventJSON)
+		makeJoinResp.JoinEvent.Content = eventJSON
+	default:
+		senderID = spec.SenderID(userID)
+	}
+
+	stateKey := string(senderID)
+	makeJoinResp.JoinEvent.SenderID = string(senderID)
+	makeJoinResp.JoinEvent.StateKey = &stateKey
+
 	eb := verImpl.NewEventBuilderFromProtoEvent(&makeJoinResp.JoinEvent)
-	joinEvent, err := eb.Build(time.Now(), origin, s.KeyID, s.Priv)
+	joinEvent, err := eb.Build(time.Now(), origin, keyID, signingKey)
 	if err != nil {
 		t.Fatalf("MustJoinRoom: failed to sign event: %v", err)
 	}
 	var sendJoinResp fclient.RespSendJoin
 	if len(partialState) == 0 || !partialState[0] {
 		// Default to doing a regular join.
-		sendJoinResp, err = fedClient.SendJoin(context.Background(), origin, remoteServer, joinEvent)
+		sendJoinResp, err = fedClient.SendJoin(context.Background(), origOrigin, remoteServer, joinEvent)
 	} else {
-		sendJoinResp, err = fedClient.SendJoinPartialState(context.Background(), origin, remoteServer, joinEvent)
+		sendJoinResp, err = fedClient.SendJoinPartialState(context.Background(), origOrigin, remoteServer, joinEvent)
 	}
 	if err != nil {
 		t.Fatalf("MustJoinRoom: send_join failed: %v", err)