Merge pull request #338 from Kentico/develop

Release 2.1.0

Author: fialafilip

KVA/Migration.Tool.Source/Handlers/MigrateCategoriesCommandHandler.cs

@@ -16,13 +16,15 @@
 using Migration.Tool.Common.Abstractions;
 using Migration.Tool.Common.Builders;
 using Migration.Tool.Common.Helpers;
+using Migration.Tool.Common.Model;
 using Migration.Tool.KXP.Api;
 using Migration.Tool.Source.Mappers;
 using Migration.Tool.Source.Model;
 using Migration.Tool.Source.Providers;
 using Migration.Tool.Source.Services;
 
 using Newtonsoft.Json;
+using Newtonsoft.Json.Linq;
 
 namespace Migration.Tool.Source.Handlers;
 
@@ -34,7 +36,8 @@ public class MigrateCategoriesCommandHandler(
     IUmtMapper<TagModelSource> tagModelMapper,
     SpoiledGuidContext spoiledGuidContext,
     KxpClassFacade kxpClassFacade,
-    ClassMappingProvider classMappingProvider
+    ClassMappingProvider classMappingProvider,
+    DeferredTreeNodesService deferredTreeNodesService
 ) : IRequestHandler<MigrateCategoriesCommand, CommandResult>
 {
     public async Task<CommandResult> Handle(MigrateCategoriesCommand request, CancellationToken cancellationToken)
@@ -69,6 +72,8 @@ FROM View_CMS_Tree_Joined [TJ]
             var skippedClasses = new List<int>();
             var schemaGuid = Guid.Empty;
             string categoryFieldName = "Category_Legacy";
+            string widgetCategoryFieldName = "Category";
+
             foreach (var classWithCategoryUsage in classesWithCategories)
             {
                 var targetDataClass = DataClassInfoProvider.ProviderObject.Get(classWithCategoryUsage.ClassGuid);
@@ -171,6 +176,42 @@ FROM View_CMS_Tree_Joined [TJ]
                                 }
                             }
 
+                            if (deferredTreeNodesService.WidgetizedDocuments.TryGetValue(dwc.DocumentGUID, out var widget))
+                            {
+                                var widgetHostPageCommonData = ContentItemCommonDataInfo.Provider.Get()
+                                    .WhereEquals(nameof(ContentItemCommonDataInfo.ContentItemCommonDataGUID), widget.ContentItemCommonDataGuid)
+                                    .FirstOrDefault();
+
+                                if (widgetHostPageCommonData.ContentItemCommonDataVisualBuilderWidgets is not null)
+                                {
+                                    var visualBuilderWidgets = JsonConvert.DeserializeObject<EditableAreasConfiguration>(widgetHostPageCommonData.ContentItemCommonDataVisualBuilderWidgets);
+                                    foreach (var area in visualBuilderWidgets.EditableAreas)
+                                    {
+                                        foreach (var section in area.Sections)
+                                        {
+                                            foreach (var zone in section.Zones)
+                                            {
+                                                foreach (var w in zone.Widgets)
+                                                {
+                                                    foreach (var variant in w.Variants)
+                                                    {
+                                                        if (variant.Identifier.Equals(widget.WidgetVariantGuid))
+                                                        {
+                                                            var array = variant.Properties.ContainsKey(widgetCategoryFieldName)
+                                                                ? variant.Properties.Value<JArray>(widgetCategoryFieldName)
+                                                                : [];
+                                                            array.Add(new JObject { { "identifier", tag.TagGUID.ToString() } });
+                                                            variant.Properties[widgetCategoryFieldName] = array;
+                                                        }
+                                                    }
+                                                }
+                                            }
+                                        }
+                                    }
+                                    widgetHostPageCommonData.ContentItemCommonDataVisualBuilderWidgets = JsonConvert.SerializeObject(visualBuilderWidgets);
+                                    ContentItemCommonDataInfo.Provider.Set(widgetHostPageCommonData);
+                                }
+                            }
                             var commonData = ContentItemCommonDataInfo.Provider.Get()
                                 .WhereEquals(nameof(ContentItemCommonDataInfo.ContentItemCommonDataGUID), dwc.DocumentGUID)
                                 .FirstOrDefault();
@@ -187,15 +228,13 @@ FROM View_CMS_Tree_Joined [TJ]
 
                             foreach (var infoWithTag in infosWithTag)
                             {
-                                List<TagReference> tagReferences = [];
+                                HashSet<Guid> tagReferences = [];
                                 if (infoWithTag[categoryFieldName] is string jsonTags)
                                 {
-                                    tagReferences = JsonConvert.DeserializeObject<List<TagReference>>(jsonTags) ?? [];
+                                    tagReferences = new HashSet<Guid>(JsonConvert.DeserializeObject<List<TagReference>>(jsonTags)?.Select(x => x.Identifier) ?? []);
                                 }
-
-                                tagReferences.Add(new TagReference { Identifier = tag.TagGUID });
-
-                                infoWithTag[categoryFieldName] = JsonConvert.SerializeObject(tagReferences);
+                                tagReferences.Add(tag.TagGUID);
+                                infoWithTag[categoryFieldName] = JsonConvert.SerializeObject(tagReferences.Select(x => new TagReference { Identifier = x }).ToList());
 
                                 ContentItemCommonDataInfo.Provider.Set(infoWithTag);
                             }