Update mautrix to version 0.6.0

Signed-off-by: Nikos Filippakis <me@nfil.dev>

clients/bot_client.go

@@ -32,7 +32,8 @@ func (botClient *BotClient) InitOlmMachine(client *mautrix.Client, nebStore *mat
 	if sdb, ok := database.GetServiceDB().(*database.ServiceDB); ok {
 		// Create an SQL crypto store based on the ServiceDB used
 		db, dialect := sdb.GetSQLDb()
-		sqlCryptoStore := crypto.NewSQLCryptoStore(db, dialect, client.DeviceID, []byte(client.DeviceID.String()+"pickle"), cryptoLogger)
+		accountID := botClient.config.UserID.String() + "-" + client.DeviceID.String()
+		sqlCryptoStore := crypto.NewSQLCryptoStore(db, dialect, accountID, client.DeviceID, []byte(client.DeviceID.String()+"pickle"), cryptoLogger)
 		// Try to create the tables if they are missing
 		if err = sqlCryptoStore.CreateTables(); err != nil {
 			return

clients/state_store.go

@@ -15,6 +15,20 @@ type NebStateStore struct {
 	Storer *mautrix.InMemoryStore
 }
 
+// GetEncryptionEvent returns the encryption event for a room.
+func (ss *NebStateStore) GetEncryptionEvent(roomID id.RoomID) *event.EncryptionEventContent {
+	room := ss.Storer.LoadRoom(roomID)
+	if room == nil {
+		return nil
+	}
+	if evts, ok := room.State[event.StateEncryption]; ok {
+		if evt, ok := evts[""]; ok {
+			return evt.Content.AsEncryption()
+		}
+	}
+	return nil
+}
+
 // IsEncrypted returns whether a room has been encrypted.
 func (ss *NebStateStore) IsEncrypted(roomID id.RoomID) bool {
 	room := ss.Storer.LoadRoom(roomID)

go.mod

@@ -51,5 +51,5 @@ require (
 	gopkg.in/alecthomas/kingpin.v2 v2.2.6 // indirect
 	gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15 // indirect
 	gopkg.in/yaml.v2 v2.3.0
-	maunium.net/go/mautrix v0.5.7
+	maunium.net/go/mautrix v0.6.0
 )

go.sum

@@ -270,5 +270,7 @@ maunium.net/go/mautrix v0.5.0-rc.1 h1:Ux2rwQfgf044lXjzCXmaY6VVJPcLioNzdN3okbOVIl
 maunium.net/go/mautrix v0.5.0-rc.1/go.mod h1:LnkFnB1yjCbb8V+upoEHDGvI/F38NHSTWYCe2RRJgSY=
 maunium.net/go/mautrix v0.5.5 h1:e0Pql1FdxoNUudx2oXo1gZHMrqIh5MC72cdXEPIrYLA=
 maunium.net/go/mautrix v0.5.5/go.mod h1:FLbMANzwqlsX2Fgm7SDe+E4I3wSa4UxJRKqS5wGkCwA=
+maunium.net/go/mautrix v0.6.0 h1:V32l4aygKk2XcH3fi8Yd0pFeSyYZJNRIvr8vdA2GtC8=
+maunium.net/go/mautrix v0.6.0/go.mod h1:Va/74MijqaS0DQ3aUqxmFO54/PMfr1LVsCOcGRHbYmo=
 maunium.net/go/mauview v0.1.1/go.mod h1:3QBUiuLct9moP1LgDhCGIg0Ovxn38Bd2sGndnUOuj4o=
 maunium.net/go/tcell v0.2.0/go.mod h1:9Apcb3lNNS6C6lCqKT9UFp7BTRzHXfWE+/tgufsAMho=

goneb_services_test.go

@@ -14,6 +14,7 @@ import (
 	"maunium.net/go/mautrix/crypto"
 	"maunium.net/go/mautrix/crypto/olm"
 	mevt "maunium.net/go/mautrix/event"
+	"maunium.net/go/mautrix/id"
 )
 
 func setupMockServer() (*http.ServeMux, *matrixTripper, *httptest.ResponseRecorder, chan string) {
@@ -164,13 +165,14 @@ func TestEncryptedRespondToEcho(t *testing.T) {
 	accountBot := olm.NewAccount()
 	signingKeyMock, identityKeyMock := accountMock.IdentityKeys()
 	signingKeyBot, identityKeyBot := accountBot.IdentityKeys()
-	ogsBot := crypto.NewOutboundGroupSession("!greatdekutree:hyrule")
+	// encryptionEvtContent := &mevt.EncryptionEventContent{Algorithm: "m.megolm.v1.aes-sha2"}
+	ogsBot := crypto.NewOutboundGroupSession("!greatdekutree:hyrule", nil)
 	ogsBot.Shared = true
 	igsMock, err := crypto.NewInboundGroupSession(identityKeyBot, signingKeyBot, "!greatdekutree:hyrule", ogsBot.Internal.Key())
 	if err != nil {
 		t.Errorf("Error creating mock IGS: %v", err)
 	}
-	ogsMock := crypto.NewOutboundGroupSession("!greatdekutree:hyrule")
+	ogsMock := crypto.NewOutboundGroupSession("!greatdekutree:hyrule", nil)
 	ogsMock.Shared = true
 	igsBot, err := crypto.NewInboundGroupSession(identityKeyMock, signingKeyMock, "!greatdekutree:hyrule", ogsMock.Internal.Key())
 	if err != nil {
@@ -279,7 +281,7 @@ func TestEncryptedRespondToEcho(t *testing.T) {
 
 	// DB is initialized, store the megolm sessions from before for the bot to be able to decrypt and encrypt
 	sqlDB, dialect := database.GetServiceDB().(*database.ServiceDB).GetSQLDb()
-	cryptoStore := crypto.NewSQLCryptoStore(sqlDB, dialect, "mastersword", []byte("masterswordpickle"), clients.CryptoMachineLogger{})
+	cryptoStore := crypto.NewSQLCryptoStore(sqlDB, dialect, "@link:hyrule-mastersword", "mastersword", []byte("masterswordpickle"), clients.CryptoMachineLogger{})
 	if err := cryptoStore.AddOutboundGroupSession(ogsBot); err != nil {
 		t.Errorf("Error storing bot OGS: %v", err)
 	}
@@ -287,6 +289,14 @@ func TestEncryptedRespondToEcho(t *testing.T) {
 	if err := cryptoStore.PutGroupSession("!greatdekutree:hyrule", identityKeyMock, igsBot.ID(), igsBot); err != nil {
 		t.Errorf("Error storing bot IGS: %v", err)
 	}
+	cryptoStore.PutDevices("@navi:hyrule", map[id.DeviceID]*crypto.DeviceIdentity{
+		"NAVI": {
+			UserID:      "@navi:hyrule",
+			DeviceID:    "NAVI",
+			IdentityKey: identityKeyMock,
+			SigningKey:  signingKeyMock,
+		},
+	})
 
 	plaintext := `{"room_id":"!greatdekutree:hyrule","type":"m.room.message","content":{"body":"!echo save zelda","msgtype":"m.text"}}`
 	ciphertext, err := ogsMock.Encrypt([]byte(plaintext))
@@ -305,7 +315,8 @@ func TestEncryptedRespondToEcho(t *testing.T) {
 						"algorithm":"m.megolm.v1.aes-sha2",
 						"sender_key":"%s",
 						"ciphertext":"%s",
-						"session_id":"%s"
+						"session_id":"%s",
+						"device_id": "NAVI"
 					},
 					"origin_server_ts": 10000,
 					"unsigned": {"age": 100},