preserve styleProperties when FF is disabled

Author: dario-daza

dotCMS/src/main/java/com/dotmarketing/factories/MultiTreeAPIImpl.java

@@ -126,6 +126,8 @@ public class MultiTreeAPIImpl implements MultiTreeAPI {
     private static final String SELECT_CHILD_BY_PARENT_RELATION_PERSONALIZATION_VARIANT_LANGUAGE =
             SELECT_CHILD_BY_PARENT_RELATION_PERSONALIZATION_VARIANT + " AND child IN (SELECT DISTINCT identifier FROM contentlet, multi_tree " +
                     "WHERE multi_tree.child = contentlet.identifier AND multi_tree.parent1 = ? AND language_id = ?)";
+    private static final String SELECT_NOT_EMPTY_CONTENTLET_STYLES_BY_PARENTS_AND_RELATIONS =
+            SELECT_ALL + "WHERE parent1 = ? AND style_properties IS NOT NULL";
 
     @WrapInTransaction
     @Override
@@ -661,7 +663,7 @@ public void overridesMultitreesByPersonalization(final String pageId,
      * @param multiTrees {@link List} of {@link MultiTree} to safe
      * @param languageIdOpt {@link Optional} {@link Long}   optional language, if present will deletes only the contentlets that have a version on this language.
      *                                        Since it is by identifier, when deleting for instance in spanish, will remove the english and any other lang version too.
-     * @throws DotDataException 
+     * @throws DotDataException If there is an issue retrieving data from the DB.
      */
     @Override
     @WrapInTransaction
@@ -683,6 +685,13 @@ public void overridesMultitreesByPersonalization(final String pageId,
         Logger.debug(MultiTreeAPIImpl.class, ()->String.format("Saving page's content: %s", multiTrees));
         Set<String> originalContentletIds = new HashSet<>();
         final DotConnect db = new DotConnect();
+
+        // preserve style properties if feature flag is disabled and there are style properties to preserve
+        final boolean isStyleEditorEnabled = Config.getBooleanProperty("FEATURE_FLAG_UVE_STYLE_EDITOR", false);
+        if (!isStyleEditorEnabled) {
+            preserveStylesBeforeSaving(pageId, multiTrees);
+        }
+
         if (languageIdOpt.isPresent()) {
             if (DbConnectionFactory.isMySql()) {
                 deleteMultiTreeToMySQL(pageId, personalization, languageIdOpt, variantId);
@@ -1570,6 +1579,55 @@ private Set<String> getOriginalContentlets(final String sqlQuery, final List<Obj
         return contentletData.stream().map(dataMap -> dataMap.get("child").toString()).collect(Collectors.toSet());
     }
 
+    /**
+     * Restores the style properties for a list of MultiTree objects by looking up
+     * their existing values in the database.
+     * * This prevents style data loss when overwriting existing records. It matches
+     * records based on the composite key of Container + Contentlet.
+     *
+     * @param pageId      The identifier of the page being processed.
+     * @param multiTrees  The list of MultiTree objects to be saved.
+     * NOTE: This list is modified in-place.
+     * The same list of multiTrees will be enriched with their original styles.
+     * @throws DotDataException If there is an issue retrieving data from the DB.
+     */
+    private void preserveStylesBeforeSaving(final String pageId, List<MultiTree> multiTrees) throws DotDataException {
+        // Gets existing multiTrees by Page from DB
+        final List<MultiTree> multiTreesFromDB = fetchStylesFromDB(pageId);
+
+        if (multiTreesFromDB.isEmpty()) {
+            return;
+        }
+
+        // Create a "Lookup Map" from the DB multiTrees.
+        // Key: Unique combination of Container + Contentlet
+        // Value: Contentlet styleProperties of the Contentlet
+        final Map<String, Map<String, Object>> dbStyleMap = multiTreesFromDB.stream()
+                .collect(Collectors.toMap(
+                        mt -> mt.getContainer() + "_" + mt.getContentlet(),
+                        MultiTree::getStyleProperties,
+                        // In case of duplicates, keep last entrance value (shouldn't happen)
+                        (existing, replacement) -> replacement
+                ));
+
+        // Update the multiTrees list to preserve existing styleProperties
+        for (MultiTree multiTree : multiTrees) {
+            String key = multiTree.getContainer() + "_" + multiTree.getContentlet();
+
+            // If this relationship already existed in DB, preserves the old styles
+            if (dbStyleMap.containsKey(key)) {
+                multiTree.setStyleProperties(dbStyleMap.get(key));
+            }
+        }
+    }
+
+    protected List<MultiTree> fetchStylesFromDB(String pageId) throws DotDataException {
+        final DotConnect db = new DotConnect()
+                .setSQL(SELECT_NOT_EMPTY_CONTENTLET_STYLES_BY_PARENTS_AND_RELATIONS)
+                .addParam(pageId);
+        return TransformerLocator.createMultiTreeTransformer(db.loadObjectResults()).asList();
+    }
+
     /**
      * Takes the list of Contentlet IDs present in an HTML Page before any change is made, and the list of
      * {@link MultiTree} objects representing the updated Contentlets. Then, compares both lists and determines what