Move migrations to storage, add Storage migration for nodes change

src/Artemis.Storage/Migrations/IProfileMigration.cs

@@ -1,11 +1,9 @@
-using LiteDB;
 using Newtonsoft.Json.Linq;
 
 namespace Artemis.Storage.Migrations;
 
-internal interface IProfileMigration
+public interface IProfileMigration
 {
     int Version { get; }
     void Migrate(JObject profileJson);
-    void Migrate(BsonDocument profileBson);
 }

src/Artemis.Storage/Migrations/IStorageMigration.cs

@@ -1,6 +1,6 @@
 using LiteDB;
 
-namespace Artemis.Storage.Migrations.Interfaces;
+namespace Artemis.Storage.Migrations;
 
 public interface IStorageMigration
 {

src/Artemis.Storage/Migrations/Profile/M0001NodeProviders.cs

@@ -1,4 +1,3 @@
-using LiteDB;
 using Newtonsoft.Json.Linq;
 
 namespace Artemis.Storage.Migrations.Profile;
@@ -21,9 +20,7 @@ internal class M0001NodeProviders : IProfileMigration
         if (folders != null)
         {
             foreach (JToken folder in folders)
-            {
                 MigrateProfileElement(folder);
-            }
         }
 
         if (layers != null)
@@ -38,12 +35,6 @@ internal class M0001NodeProviders : IProfileMigration
         }
     }
 
-    /// <inheritdoc />
-    public void Migrate(BsonDocument profileBson)
-    {
-        throw new System.NotImplementedException();
-    }
-
     private void MigrateProfileElement(JToken profileElement)
     {
         JArray? layerEffects = (JArray?) profileElement["LayerEffects"]?["$values"];

src/Artemis.Storage/Migrations/Storage/M0020AvaloniaReset.cs

@@ -1,9 +1,8 @@
 using System.Collections.Generic;
 using System.Linq;
-using Artemis.Storage.Migrations.Interfaces;
 using LiteDB;
 
-namespace Artemis.Storage.Migrations;
+namespace Artemis.Storage.Migrations.Storage;
 
 public class M0020AvaloniaReset : IStorageMigration
 {

src/Artemis.Storage/Migrations/Storage/M0021GradientNodes.cs

@@ -3,10 +3,9 @@ using System.Collections.Generic;
 using System.Linq;
 using Artemis.Storage.Entities.Profile;
 using Artemis.Storage.Entities.Profile.Nodes;
-using Artemis.Storage.Migrations.Interfaces;
 using LiteDB;
 
-namespace Artemis.Storage.Migrations;
+namespace Artemis.Storage.Migrations.Storage;
 
 public class M0021GradientNodes : IStorageMigration
 {

src/Artemis.Storage/Migrations/Storage/M0022TransitionNodes.cs

@@ -3,10 +3,9 @@ using Artemis.Storage.Entities.Profile;
 using Artemis.Storage.Entities.Profile.Abstract;
 using Artemis.Storage.Entities.Profile.Conditions;
 using Artemis.Storage.Entities.Profile.Nodes;
-using Artemis.Storage.Migrations.Interfaces;
 using LiteDB;
 
-namespace Artemis.Storage.Migrations;
+namespace Artemis.Storage.Migrations.Storage;
 
 public class M0022TransitionNodes : IStorageMigration
 {

src/Artemis.Storage/Migrations/Storage/M0023LayoutProviders.cs

@@ -1,8 +1,7 @@
 using System.Collections.Generic;
-using Artemis.Storage.Migrations.Interfaces;
 using LiteDB;
 
-namespace Artemis.Storage.Migrations;
+namespace Artemis.Storage.Migrations.Storage;
 
 public class M0023LayoutProviders : IStorageMigration
 {

src/Artemis.Storage/Migrations/Storage/M0024NodeProviders.cs

@@ -0,0 +1,100 @@
+using System.Collections.Generic;
+using Artemis.Storage.Entities.Profile;
+using LiteDB;
+
+namespace Artemis.Storage.Migrations.Storage;
+
+public class M0024NodeProviders : IStorageMigration
+{
+    public int UserVersion => 24;
+
+    public void Apply(LiteRepository repository)
+    {
+        List<ProfileCategoryEntity> profileCategories = repository.Query<ProfileCategoryEntity>().ToList();
+        foreach (ProfileCategoryEntity profileCategory in profileCategories)
+        {
+            foreach (ProfileConfigurationEntity profileConfigurationEntity in profileCategory.ProfileConfigurations)
+            {
+                profileConfigurationEntity.Version = 1;
+            }
+            repository.Update(profileCategory);
+        }
+
+        ILiteCollection<BsonDocument> collection = repository.Database.GetCollection("ProfileEntity");
+        foreach (BsonDocument profileBson in collection.FindAll())
+        {
+            BsonArray? folders = profileBson["Folders"]?.AsArray;
+            BsonArray? layers = profileBson["Layers"]?.AsArray;
+
+            if (folders != null)
+            {
+                foreach (BsonValue folder in folders)
+                    MigrateProfileElement(folder.AsDocument);
+            }
+
+            if (layers != null)
+            {
+                foreach (BsonValue layer in layers)
+                {
+                    MigrateProfileElement(layer.AsDocument);
+                    MigratePropertyGroup(layer.AsDocument["GeneralPropertyGroup"].AsDocument);
+                    MigratePropertyGroup(layer.AsDocument["TransformPropertyGroup"].AsDocument);
+                    MigratePropertyGroup(layer.AsDocument["LayerBrush"]?["PropertyGroup"].AsDocument);
+                }
+            }
+
+            collection.Update(profileBson);
+        }
+    }
+
+    private void MigrateProfileElement(BsonDocument profileElement)
+    {
+        BsonArray? layerEffects = profileElement["LayerEffects"]?.AsArray;
+        if (layerEffects != null)
+        {
+            foreach (BsonValue layerEffect in layerEffects)
+                MigratePropertyGroup(layerEffect.AsDocument["PropertyGroup"].AsDocument);
+        }
+
+        BsonValue? displayCondition = profileElement["DisplayCondition"];
+        if (displayCondition != null)
+            MigrateNodeScript(displayCondition.AsDocument["Script"].AsDocument);
+    }
+
+    private void MigratePropertyGroup(BsonDocument? propertyGroup)
+    {
+        if (propertyGroup == null || propertyGroup.Keys.Count == 0)
+            return;
+
+        BsonArray? properties = propertyGroup["Properties"]?.AsArray;
+        BsonArray? propertyGroups = propertyGroup["PropertyGroups"]?.AsArray;
+
+        if (properties != null)
+        {
+            foreach (BsonValue property in properties)
+                MigrateNodeScript(property.AsDocument["DataBinding"]?["NodeScript"]?.AsDocument);
+        }
+
+        if (propertyGroups != null)
+        {
+            foreach (BsonValue childPropertyGroup in propertyGroups)
+                MigratePropertyGroup(childPropertyGroup.AsDocument);
+        }
+    }
+
+    private void MigrateNodeScript(BsonDocument? nodeScript)
+    {
+        if (nodeScript == null || nodeScript.Keys.Count == 0)
+            return;
+
+        BsonArray? nodes = nodeScript["Nodes"]?.AsArray;
+        if (nodes == null)
+            return;
+
+        foreach (BsonValue node in nodes)
+        {
+            node.AsDocument["Type"] = node.AsDocument["Type"]?.AsString?.Replace("Artemis.VisualScripting.Nodes", "Artemis.Plugins.Nodes.General.Nodes");
+            node.AsDocument["ProviderId"] = "Artemis.Plugins.Nodes.General.GeneralNodesProvider-d9e1ee78";
+        }
+    }
+}

src/Artemis.Storage/StorageMigrationService.cs

@@ -1,7 +1,7 @@
 using System;
 using System.Collections.Generic;
 using System.Linq;
-using Artemis.Storage.Migrations.Interfaces;
+using Artemis.Storage.Migrations;
 using LiteDB;
 using Serilog;