Refactor canonical_alias event updating when alias removing

Signed-off-by: Colin Weill--Duflos <lieunoir@rezel.net>

clientapi/routing/directory.go

@@ -223,13 +223,41 @@ func RemoveLocalAlias(
 		}
 	}
 
-    // Check if state contains removed alias
+    var updatedCanonicalAlias *eventutil.CanonicalAlias
+    updated, resErr := getUpdatedCanonicalAliasState(req, device, queryRes.RoomID, alias, rsAPI, updatedCanonicalAlias)
+    if resErr != nil {
+        return *resErr;
+    }
+    // If the alias removed is one of the alt_aliases or the canonical one,
+    // we need to also remove it from the canonical_alias event
+    if updated {
+        resErr := updateCanonicalAlias(req, device, queryRes.RoomID, cfg, rsAPI, updatedCanonicalAlias)
+        if resErr != nil {
+            return *resErr;
+        }
+    }
+
+	return util.JSONResponse{
+		Code: http.StatusOK,
+		JSON: struct{}{},
+	}
+}
+
+func getUpdatedCanonicalAliasState(
+	req *http.Request,
+	device *api.Device,
+	roomID string,
+    alias string,
+	rsAPI roomserverAPI.RoomserverInternalAPI,
+    updatedCanonicalAlias *eventutil.CanonicalAlias,
+) ( bool, *util.JSONResponse ) {
+    updated := false
     stateTuple := gomatrixserverlib.StateKeyTuple{
         EventType: gomatrixserverlib.MRoomCanonicalAlias,
         StateKey:  "",
     }
     stateReq := roomserverAPI.QueryCurrentStateRequest {
-        RoomID:      queryRes.RoomID,
+        RoomID:      roomID,
         StateTuples: []gomatrixserverlib.StateKeyTuple{stateTuple},
     }
     stateRes := &roomserverAPI.QueryCurrentStateResponse{}
@@ -237,12 +265,13 @@ func RemoveLocalAlias(
     if err != nil {
         util.GetLogger(req.Context()).WithError(err).Error("Query state failed")
         resErr := jsonerror.InternalServerError()
-        return resErr
+        return false, &resErr
     }
 
-    inAliases := false
+    updatedCanonicalAlias = &eventutil.CanonicalAlias {
-    newAlias := ""
+        Alias: "",
-    newAltAliases := []string{""}
+        AltAliases: []string{""},
+    }
     // We try to get the current canonical_alias state, and if found compare its content
     // to the removed alias
     if canonicalAliasEvent, ok := stateRes.StateEvents[stateTuple]; ok {
@@ -254,70 +283,71 @@ func RemoveLocalAlias(
         if err != nil {
             util.GetLogger(req.Context()).WithError(err).Error("Get canonical_alias event content failed")
             resErr := jsonerror.InternalServerError()
-            return resErr
+            return false, &resErr
         }
         if alias == canonicalAliasContent.Alias {
-            inAliases = true
+            updated = true
         } else {
-            newAlias = canonicalAliasContent.Alias
+            updatedCanonicalAlias.Alias = canonicalAliasContent.Alias
         }
         for _, s := range(canonicalAliasContent.AltAliases) {
             if alias == s {
-                inAliases = true
+                updated = true
             } else {
-                newAltAliases = append(newAltAliases, s)
+                updatedCanonicalAlias.AltAliases = append(updatedCanonicalAlias.AltAliases, s)
             }
         }
     }
-    // If the alias removed is one of the alt_aliases or the canonical one,
+    return updated, nil
-    // we need to also remove it from the canonical_alias event
+}
-    if inAliases {
-        var stateKey = ""
-        // We create a new canonical_alias event with the new alias and alt_aliase
-        // May cause some auth problems
-        builder := gomatrixserverlib.EventBuilder {
-            Sender:   device.UserID,
-            RoomID:   queryRes.RoomID,
-            Type:     gomatrixserverlib.MRoomCanonicalAlias,
-            StateKey: &stateKey,
-        }
-        content := eventutil.CanonicalAlias {
-            Alias:      newAlias,
-            AltAliases: newAltAliases,
-        }
-        err := builder.SetContent(content)
-        if err != nil {
-            util.GetLogger(req.Context()).WithError(err).Error("builder.SetContent failed")
-            resErr := jsonerror.InternalServerError()
-            return resErr
-        }
 
-        evTime, err := httputil.ParseTSParam(req)
+func updateCanonicalAlias(
-        if err != nil {
+	req *http.Request,
-            return util.JSONResponse{
+	device *api.Device,
-                Code: http.StatusBadRequest,
+	roomID string,
-                JSON: jsonerror.InvalidArgumentValue(err.Error()),
+    cfg *config.ClientAPI,
-            }
+	rsAPI roomserverAPI.RoomserverInternalAPI,
-        }
+    updatedCanonicalAlias *eventutil.CanonicalAlias,
+) *util.JSONResponse {
+    var stateKey = ""
+    // We create a new canonical_alias event with the new alias and alt_aliase
+    // May cause some auth problems
+    builder := gomatrixserverlib.EventBuilder {
+        Sender:   device.UserID,
+        RoomID:   roomID,
+        Type:     gomatrixserverlib.MRoomCanonicalAlias,
+        StateKey: &stateKey,
+    }
+    err := builder.SetContent(updatedCanonicalAlias)
+    if err != nil {
+        util.GetLogger(req.Context()).WithError(err).Error("builder.SetContent failed")
+        resErr := jsonerror.InternalServerError()
+        return &resErr
+    }
 
-        // Build the event
+    evTime, err := httputil.ParseTSParam(req)
-        e, err := eventutil.QueryAndBuildEvent(req.Context(), &builder, cfg.Matrix, evTime, rsAPI, nil)
+    if err != nil {
-        if err != nil {
+        return &util.JSONResponse{
-            util.GetLogger(req.Context()).WithError(err).Errorf("failed to QueryAndBuildEvent")
+            Code: http.StatusBadRequest,
-            return jsonerror.InternalServerError()
+            JSON: jsonerror.InvalidArgumentValue(err.Error()),
-        }
-        // Send the event to the room server
-        err = roomserverAPI.SendEvents(req.Context(), rsAPI, roomserverAPI.KindNew, []*gomatrixserverlib.HeaderedEvent{e}, cfg.Matrix.ServerName, nil)
-        if  err != nil {
-            util.GetLogger(req.Context()).WithError(err).Errorf("failed to SendEvents")
-            return jsonerror.InternalServerError()
         }
     }
 
-	return util.JSONResponse{
+    // Build the event
-		Code: http.StatusOK,
+    e, err := eventutil.QueryAndBuildEvent(req.Context(), &builder, cfg.Matrix, evTime, rsAPI, nil)
-		JSON: struct{}{},
+    if err != nil {
-	}
+        util.GetLogger(req.Context()).WithError(err).Errorf("failed to QueryAndBuildEvent")
+        resErr := jsonerror.InternalServerError()
+        return &resErr
+    }
+    // Send the event to the room server
+    err = roomserverAPI.SendEvents(req.Context(), rsAPI, roomserverAPI.KindNew, []*gomatrixserverlib.HeaderedEvent{e}, cfg.Matrix.ServerName, nil)
+    if  err != nil {
+        util.GetLogger(req.Context()).WithError(err).Errorf("failed to SendEvents")
+        resErr := jsonerror.InternalServerError()
+        return &resErr
+    }
+    return nil
 }
 
 type roomVisibility struct {