Artemis/src/Artemis.Storage/Migrations/M4ProfileSegments.cs

using Artemis.Storage.Entities.Profile;
using Artemis.Storage.Migrations.Interfaces;
using LiteDB;
using System;
using System.Linq;

namespace Artemis.Storage.Migrations
{
    public class M4ProfileSegmentsMigration : IStorageMigration
    {
        public int UserVersion => 4;

        public void Apply(LiteRepository repository)
        {
            var profiles = repository.Query<ProfileEntity>().ToList();
            foreach (var profileEntity in profiles)
            {
                foreach (var folder in profileEntity.Folders.Where(f => f.MainSegmentLength == TimeSpan.Zero))
                {
                    if (folder.PropertyEntities.Any(p => p.KeyframeEntities.Any()))
                        folder.MainSegmentLength = folder.PropertyEntities.Where(p => p.KeyframeEntities.Any()).Max(p => p.KeyframeEntities.Max(k => k.Position));
                    if (folder.MainSegmentLength == TimeSpan.Zero)
                        folder.MainSegmentLength = TimeSpan.FromSeconds(5);

                    folder.DisplayContinuously = true;
                }

                foreach (var layer in profileEntity.Layers.Where(l => l.MainSegmentLength == TimeSpan.Zero))
                {
                    if (layer.PropertyEntities.Any(p => p.KeyframeEntities.Any()))
                        layer.MainSegmentLength = layer.PropertyEntities.Where(p => p.KeyframeEntities.Any()).Max(p => p.KeyframeEntities.Max(k => k.Position));
                    if (layer.MainSegmentLength == TimeSpan.Zero)
                        layer.MainSegmentLength = TimeSpan.FromSeconds(5);

                    layer.DisplayContinuously = true;
                }

                repository.Update(profileEntity);
            }
        }
    }
}