Radarr/src/NzbDrone.Core/Datastore/Migration/170_fix_trakt_list_config.cs

using System;
using System.Collections.Generic;
using System.Data;
using System.Text.Json;
using Dapper;
using FluentMigrator;
using NzbDrone.Common.Extensions;
using NzbDrone.Common.Serializer;
using NzbDrone.Core.Datastore.Migration.Framework;

namespace NzbDrone.Core.Datastore.Migration
{
    [Migration(170)]
    public class fix_trakt_list_config : NzbDroneMigrationBase
    {
        private readonly JsonSerializerOptions _serializerSettings;

        public fix_trakt_list_config()
        {
            _serializerSettings = new JsonSerializerOptions
            {
                AllowTrailingCommas = true,
                IgnoreNullValues = false,
                PropertyNameCaseInsensitive = true,
                DictionaryKeyPolicy = JsonNamingPolicy.CamelCase,
                PropertyNamingPolicy = JsonNamingPolicy.CamelCase,
                WriteIndented = true
            };
        }

        protected override void MainDbUpgrade()
        {
            Execute.WithConnection(FixTraktConfig);
            Execute.WithConnection(RenameRadarrListType);
            Execute.Sql("DELETE FROM Config WHERE[KEY] IN ('TraktAuthToken', 'TraktRefreshToken', 'TraktTokenExpiry', 'NewTraktAuthToken', 'NewTraktRefreshToken', 'NewTraktTokenExpiry')");
        }

        private void RenameRadarrListType(IDbConnection conn, IDbTransaction tran)
        {
            var rows = conn.Query<ProviderDefinition169>($"SELECT Id, Implementation, ConfigContract, Settings FROM NetImport WHERE Implementation = 'RadarrLists'");

            var corrected = new List<ProviderDefinition169>();

            foreach (var row in rows)
            {
                corrected.Add(new ProviderDefinition169
                {
                    Id = row.Id,
                    Implementation = "RadarrListImport",
                    ConfigContract = "RadarrListSettings"
                });
            }

            var updateSql = "UPDATE NetImport SET Implementation = @Implementation, ConfigContract = @ConfigContract WHERE Id = @Id";
            conn.Execute(updateSql, corrected, transaction: tran);
        }

        private void FixTraktConfig(IDbConnection conn, IDbTransaction tran)
        {
            var config = new Dictionary<string, string>();

            using (IDbCommand configCmd = conn.CreateCommand())
            {
                configCmd.Transaction = tran;
                configCmd.CommandText = @"SELECT * FROM Config";
                using (IDataReader configReader = configCmd.ExecuteReader())
                {
                    var keyIndex = configReader.GetOrdinal("Key");
                    var valueIndex = configReader.GetOrdinal("Value");

                    while (configReader.Read())
                    {
                        var key = configReader.GetString(keyIndex);
                        var value = configReader.GetString(valueIndex);

                        config.Add(key.ToLowerInvariant(), value);
                    }
                }
            }

            var rows = conn.Query<ProviderDefinition169>($"SELECT Id, Implementation, ConfigContract, Settings FROM NetImport WHERE Implementation = 'TraktImport'");

            var corrected = new List<ProviderDefinition169>();

            foreach (var row in rows)
            {
                var settings = JsonSerializer.Deserialize<TraktSettings169>(row.Settings, _serializerSettings);

                if (settings.TraktListType == (int)TraktListType169.UserCustomList)
                {
                    var newSettings = new TraktListSettings170
                    {
                        Listname = settings.Listname,
                        Username = settings.Username,
                        AuthUser = settings.Username,

                        OAuthUrl = "http://radarr.aeonlucid.com/v1/trakt/redirect",
                        RenewUri = "http://radarr.aeonlucid.com/v1/trakt/refresh",
                        ClientId = "964f67b126ade0112c4ae1f0aea3a8fb03190f71117bd83af6a0560a99bc52e6",
                        Scope = settings.Scope,
                        AccessToken = settings.AccessToken.IsNotNullOrWhiteSpace() ? settings.AccessToken : GetConfigValue(config, "TraktAuthToken", "localhost") ?? "",
                        RefreshToken = settings.RefreshToken.IsNotNullOrWhiteSpace() ? settings.RefreshToken : GetConfigValue(config, "TraktRefreshToken", "localhost") ?? "",
                        Expires = settings.Expires > DateTime.UtcNow ? settings.Expires : DateTime.UtcNow,
                        Link = settings.Link,
                        Rating = settings.Rating,
                        Certification = settings.Certification,
                        Genres = settings.Genres,
                        Years = settings.Years,
                        Limit = settings.Limit,
                        TraktAdditionalParameters = settings.TraktAdditionalParameters,
                        SignIn = settings.SignIn
                    };

                    corrected.Add(new ProviderDefinition169
                    {
                        Id = row.Id,
                        Implementation = "TraktListImport",
                        ConfigContract = "TraktListSettings",
                        Settings = JsonSerializer.Serialize(newSettings, _serializerSettings)
                    });
                }
                else if (settings.TraktListType == (int)TraktListType169.UserWatchedList || settings.TraktListType == (int)TraktListType169.UserWatchList)
                {
                    var newSettings = new TraktUserSettings170
                    {
                        TraktListType = settings.TraktListType,
                        AuthUser = settings.Username,

                        OAuthUrl = "http://radarr.aeonlucid.com/v1/trakt/redirect",
                        RenewUri = "http://radarr.aeonlucid.com/v1/trakt/refresh",
                        ClientId = "964f67b126ade0112c4ae1f0aea3a8fb03190f71117bd83af6a0560a99bc52e6",
                        Scope = settings.Scope,
                        AccessToken = settings.AccessToken.IsNotNullOrWhiteSpace() ? settings.AccessToken : GetConfigValue(config, "TraktAuthToken", "localhost") ?? "",
                        RefreshToken = settings.RefreshToken.IsNotNullOrWhiteSpace() ? settings.RefreshToken : GetConfigValue(config, "TraktRefreshToken", "localhost") ?? "",
                        Expires = settings.Expires > DateTime.UtcNow ? settings.Expires : DateTime.UtcNow,
                        Link = settings.Link,
                        Rating = settings.Rating,
                        Certification = settings.Certification,
                        Genres = settings.Genres,
                        Years = settings.Years,
                        Limit = settings.Limit,
                        TraktAdditionalParameters = settings.TraktAdditionalParameters,
                        SignIn = settings.SignIn
                    };

                    corrected.Add(new ProviderDefinition169
                    {
                        Id = row.Id,
                        Implementation = "TraktUserImport",
                        ConfigContract = "TraktUserSettings",
                        Settings = JsonSerializer.Serialize(newSettings, _serializerSettings)
                    });
                }
                else
                {
                    var newSettings = new TraktPopularSettings170
                    {
                        TraktListType = (int)Enum.Parse(typeof(TraktPopularListType170), Enum.GetName(typeof(TraktListType169), settings.TraktListType)),
                        AuthUser = settings.Username,

                        OAuthUrl = "http://radarr.aeonlucid.com/v1/trakt/redirect",
                        RenewUri = "http://radarr.aeonlucid.com/v1/trakt/refresh",
                        ClientId = "964f67b126ade0112c4ae1f0aea3a8fb03190f71117bd83af6a0560a99bc52e6",
                        Scope = settings.Scope,
                        AccessToken = settings.AccessToken.IsNotNullOrWhiteSpace() ? settings.AccessToken : GetConfigValue(config, "TraktAuthToken", "localhost") ?? "",
                        RefreshToken = settings.RefreshToken.IsNotNullOrWhiteSpace() ? settings.RefreshToken : GetConfigValue(config, "TraktRefreshToken", "localhost") ?? "",
                        Expires = settings.Expires > DateTime.UtcNow ? settings.Expires : DateTime.UtcNow,
                        Link = settings.Link,
                        Rating = settings.Rating,
                        Certification = settings.Certification,
                        Genres = settings.Genres,
                        Years = settings.Years,
                        Limit = settings.Limit,
                        TraktAdditionalParameters = settings.TraktAdditionalParameters,
                        SignIn = settings.SignIn
                    };

                    corrected.Add(new ProviderDefinition169
                    {
                        Id = row.Id,
                        Implementation = "TraktPopularImport",
                        ConfigContract = "TraktPopularSettings",
                        Settings = JsonSerializer.Serialize(newSettings, _serializerSettings)
                    });
                }
            }

            Console.WriteLine(corrected.ToJson());

            var updateSql = "UPDATE NetImport SET Implementation = @Implementation, ConfigContract = @ConfigContract, Settings = @Settings WHERE Id = @Id";
            conn.Execute(updateSql, corrected, transaction: tran);
        }

        private T GetConfigValue<T>(Dictionary<string, string> config, string key, T defaultValue)
        {
            key = key.ToLowerInvariant();

            if (config.ContainsKey(key))
            {
                return (T)Convert.ChangeType(config[key], typeof(T));
            }

            return defaultValue;
        }
    }

    public class ProviderDefinition169 : ModelBase
    {
        public string Implementation { get; set; }
        public string ConfigContract { get; set; }
        public string Settings { get; set; }
    }

    public class TraktBaseSettings170
    {
        public string OAuthUrl { get; set; }
        public string RenewUri { get; set; }
        public string ClientId { get; set; }
        public string Scope { get; set; }
        public string AuthUser { get; set; }
        public string AccessToken { get; set; }
        public string RefreshToken { get; set; }
        public DateTime Expires { get; set; }
        public string Link { get; set; }
        public string Rating { get; set; }
        public string Certification { get; set; }
        public string Genres { get; set; }
        public string Years { get; set; }
        public int Limit { get; set; }
        public string TraktAdditionalParameters { get; set; }
        public string SignIn { get; set; }
    }

    public class TraktListSettings170 : TraktBaseSettings170
    {
        public string Username { get; set; }
        public string Listname { get; set; }
    }

    public class TraktPopularSettings170 : TraktBaseSettings170
    {
        public int TraktListType { get; set; }
    }

    public class TraktUserSettings170 : TraktBaseSettings170
    {
        public int TraktListType { get; set; }
    }

    public class TraktSettings169
    {
        public string OAuthUrl { get; set; }
        public string RenewUri { get; set; }
        public string ClientId { get; set; }
        public virtual string Scope { get; set; }
        public string AccessToken { get; set; }
        public string RefreshToken { get; set; }
        public DateTime Expires { get; set; }
        public string Link { get; set; }
        public int TraktListType { get; set; }
        public string Username { get; set; }
        public string Listname { get; set; }
        public string Rating { get; set; }
        public string Certification { get; set; }
        public string Genres { get; set; }
        public string Years { get; set; }
        public int Limit { get; set; }
        public string TraktAdditionalParameters { get; set; }
        public string SignIn { get; set; }
    }

    public enum TraktListType169
    {
        UserWatchList = 0,
        UserWatchedList = 1,
        UserCustomList = 2,

        Trending = 3,
        Popular = 4,
        Anticipated = 5,
        BoxOffice = 6,

        TopWatchedByWeek = 7,
        TopWatchedByMonth = 8,
        TopWatchedByYear = 9,
        TopWatchedByAllTime = 10
    }

    public enum TraktUserListType170
    {
        UserWatchList = 0,
        UserWatchedList = 1,
        UserCollectionList = 2,
    }

    public enum TraktPopularListType170
    {
        Trending = 0,
        Popular = 1,
        Anticipated = 2,
        BoxOffice = 3,

        TopWatchedByWeek = 4,
        TopWatchedByMonth = 5,
        TopWatchedByYear = 6,
        TopWatchedByAllTime = 7
    }
}