Sonarr/src/NzbDrone.Core/Datastore/Migration/204_add_release_hash.cs

using System.Collections.Generic;
using System.Data;
using System.IO;
using Dapper;
using FluentMigrator;
using NzbDrone.Common.Extensions;
using NzbDrone.Core.Datastore.Migration.Framework;
using NzbDrone.Core.Parser.Model;

namespace NzbDrone.Core.Datastore.Migration
{
    [Migration(204)]
    public class add_add_release_hash : NzbDroneMigrationBase
    {
        protected override void MainDbUpgrade()
        {
            Alter.Table("EpisodeFiles").AddColumn("ReleaseHash").AsString().Nullable();

            Execute.WithConnection(UpdateEpisodeFiles);
        }

        private void UpdateEpisodeFiles(IDbConnection conn, IDbTransaction tran)
        {
            var updates = new List<object>();

            using (var cmd = conn.CreateCommand())
            {
                cmd.Transaction = tran;
                cmd.CommandText = "SELECT \"Id\", \"SceneName\", \"RelativePath\", \"OriginalFilePath\" FROM \"EpisodeFiles\"";

                using var reader = cmd.ExecuteReader();
                while (reader.Read())
                {
                    var id = reader.GetInt32(0);
                    var sceneName = reader[1] as string;
                    var relativePath = reader[2] as string;
                    var originalFilePath = reader[3] as string;

                    ParsedEpisodeInfo parsedEpisodeInfo = null;

                    var originalTitle = sceneName;

                    if (originalTitle.IsNullOrWhiteSpace() && originalFilePath.IsNotNullOrWhiteSpace())
                    {
                        originalTitle = Path.GetFileNameWithoutExtension(originalFilePath);
                    }

                    if (originalTitle.IsNotNullOrWhiteSpace())
                    {
                        parsedEpisodeInfo = Parser.Parser.ParseTitle(originalTitle);
                    }

                    if (parsedEpisodeInfo == null || parsedEpisodeInfo.ReleaseHash.IsNullOrWhiteSpace())
                    {
                        parsedEpisodeInfo = Parser.Parser.ParseTitle(Path.GetFileNameWithoutExtension(relativePath));
                    }

                    if (parsedEpisodeInfo != null && parsedEpisodeInfo.ReleaseHash.IsNotNullOrWhiteSpace())
                    {
                        updates.Add(new
                        {
                            Id = id,
                            ReleaseHash = parsedEpisodeInfo.ReleaseHash
                        });
                    }
                }
            }

            if (updates.Count > 0)
            {
                var updateEpisodeFilesSql = "UPDATE \"EpisodeFiles\" SET \"ReleaseHash\" = @ReleaseHash WHERE \"Id\" = @Id";
                conn.Execute(updateEpisodeFilesSql, updates, transaction: tran);
            }
        }
    }
}