Rearrange pseudoID power_levels user logic

This commit is contained in:
Devon Hudson 2023-09-26 11:14:49 -06:00
parent 1d992aea75
commit 3df55333b7
No known key found for this signature in database
GPG key ID: CD06B18E77F6A628

View file

@ -347,10 +347,15 @@ func updatePowerLevelEvent(userIDForSender spec.UserIDForSender, se gomatrixserv
return se, nil
}
newEv := se.JSON()
usersField := gjson.GetBytes(se.JSON(), "content.users")
if usersField.Exists() {
pls, err := gomatrixserverlib.NewPowerLevelContentFromEvent(se)
if err != nil {
return nil, err
}
newPls := make(map[string]int64)
var userID *spec.UserID
for user, level := range pls.Users {
@ -363,9 +368,8 @@ func updatePowerLevelEvent(userIDForSender spec.UserIDForSender, se gomatrixserv
}
newPls[user] = level
}
var newPlBytes []byte
newEv := se.JSON()
if len(pls.Users) > 0 {
newPlBytes, err = json.Marshal(newPls)
if err != nil {
return nil, err
@ -377,26 +381,27 @@ func updatePowerLevelEvent(userIDForSender spec.UserIDForSender, se gomatrixserv
}
// do the same for prev content
prevUsersField := gjson.GetBytes(se.JSON(), "unsigned.prev_content.users")
if prevUsersField.Exists() {
prevContent := gjson.GetBytes(se.JSON(), "unsigned.prev_content")
if !prevContent.Exists() {
var evNew gomatrixserverlib.PDU
evNew, err = gomatrixserverlib.MustGetRoomVersion(se.Version()).NewEventFromTrustedJSON(newEv, false)
evNew, err := gomatrixserverlib.MustGetRoomVersion(se.Version()).NewEventFromTrustedJSON(newEv, false)
if err != nil {
return nil, err
}
return evNew, err
}
pls = gomatrixserverlib.PowerLevelContent{}
err = json.Unmarshal([]byte(prevContent.Raw), &pls)
pls := gomatrixserverlib.PowerLevelContent{}
err := json.Unmarshal([]byte(prevContent.Raw), &pls)
if err != nil {
return nil, err
}
newPls = make(map[string]int64)
newPls := make(map[string]int64)
for user, level := range pls.Users {
if eventFormat != FormatSyncFederation {
userID, err = userIDForSender(se.RoomID(), spec.SenderID(user))
userID, err := userIDForSender(se.RoomID(), spec.SenderID(user))
if err != nil {
return nil, err
}
@ -404,7 +409,8 @@ func updatePowerLevelEvent(userIDForSender spec.UserIDForSender, se gomatrixserv
}
newPls[user] = level
}
if len(pls.Users) > 0 {
var newPlBytes []byte
newPlBytes, err = json.Marshal(newPls)
if err != nil {
return nil, err
@ -415,8 +421,7 @@ func updatePowerLevelEvent(userIDForSender spec.UserIDForSender, se gomatrixserv
}
}
var evNew gomatrixserverlib.PDU
evNew, err = gomatrixserverlib.MustGetRoomVersion(se.Version()).NewEventFromTrustedJSONWithEventID(se.EventID(), newEv, false)
evNew, err := gomatrixserverlib.MustGetRoomVersion(se.Version()).NewEventFromTrustedJSONWithEventID(se.EventID(), newEv, false)
if err != nil {
return nil, err
}