Webiant Logo Webiant Logo
  1. No results found.

    Try your search with a different keyword or use * as a wildcard.

DataMigration.cs

using FluentMigrator;
using LinqToDB;
using Nop.Core.Domain.Catalog;
using Nop.Core.Domain.Localization;
using Nop.Core.Domain.Logging;
using Nop.Core.Domain.Media;
using Nop.Core.Domain.Messages;
using Nop.Core.Domain.Security;
using Nop.Data.Mapping;

namespace Nop.Data.Migrations.UpgradeTo470;

[NopUpdateMigration("2023-01-01 00:00:00", "4.70", UpdateMigrationType.Data)]
public class DataMigration : Migration
{
    private readonly INopDataProvider _dataProvider;

    public DataMigration(INopDataProvider dataProvider)
    {
        _dataProvider = dataProvider;
    }

    /// 
    /// Collect the UP migration expressions
    /// 
    public override void Up()
    {
        //#5312 new activity log type
        var activityLogTypeTable = _dataProvider.GetTable();

        if (!activityLogTypeTable.Any(alt =>
                string.Compare(alt.SystemKeyword, "ImportCustomers", StringComparison.InvariantCultureIgnoreCase) ==
                0))
            _dataProvider.InsertEntity(
                new ActivityLogType
                {
                    SystemKeyword = "ImportCustomers",
                    Enabled = true,
                    Name = "Customers were imported"
                }
            );

        //6660 new activity log type for update plugin
        if (!activityLogTypeTable.Any(alt =>
                string.Compare(alt.SystemKeyword, "UpdatePlugin", StringComparison.InvariantCultureIgnoreCase) ==
                0))
            _dataProvider.InsertEntity(
                new ActivityLogType { SystemKeyword = "UpdatePlugin", Enabled = true, Name = "Update a plugin" }
            );

        //1934
        int pageIndex;
        var pageSize = 500;
        var productAttributeCombinationTableName = nameof(ProductAttributeCombination);

        var pac = Schema.Table(productAttributeCombinationTableName);
        var columnName = "PictureId";

        if (pac.Column(columnName).Exists())
        {
#pragma warning disable CS0618
            var combinationQuery =
                from c in _dataProvider.GetTable()
                join p in _dataProvider.GetTable() on c.PictureId equals p.Id
                select c;
#pragma warning restore CS0618

            pageIndex = 0;

            while (true)
            {
                var combinations = combinationQuery.ToPagedListAsync(pageIndex, pageSize).Result;

                if (!combinations.Any())
                    break;

#pragma warning disable CS0618
                foreach (var combination in combinations)
                {
                    if (!combination.PictureId.HasValue)
                        continue;

                    _dataProvider.InsertEntity(new ProductAttributeCombinationPicture
                    {
                        PictureId = combination.PictureId.Value,
                        ProductAttributeCombinationId = combination.Id
                    });

                    combination.PictureId = null;
                }
#pragma warning restore CS0618

                _dataProvider.UpdateEntitiesAsync(combinations);

                pageIndex++;
            }
        }

        var productAttributeValueTableName = nameof(ProductAttributeValue);
        var pav = Schema.Table(productAttributeValueTableName);

        if (pav.Column(columnName).Exists())
        {
#pragma warning disable CS0618
            var valueQuery =
                from c in _dataProvider.GetTable()
                join p in _dataProvider.GetTable() on c.PictureId equals p.Id
                select c;
#pragma warning restore CS0618

            pageIndex = 0;

            while (true)
            {
                var values = valueQuery.ToPagedListAsync(pageIndex, pageSize).Result;

                if (!values.Any())
                    break;

#pragma warning disable CS0618
                foreach (var value in values)
                {
                    if (!value.PictureId.HasValue)
                        continue;

                    _dataProvider.InsertEntity(new ProductAttributeValuePicture
                    {
                        PictureId = value.PictureId.Value,
                        ProductAttributeValueId = value.Id
                    });

                    value.PictureId = null;
                }
#pragma warning restore CS0618

                _dataProvider.UpdateEntitiesAsync(values);

                pageIndex++;
            }
        }

        // new permission
        if (_dataProvider.GetTable().Any(pr => string.Compare(pr.SystemName, "AccessProfiling", StringComparison.InvariantCultureIgnoreCase) == 0))
        {
            _dataProvider.BulkDeleteEntitiesAsync(pr => pr.SystemName == "AccessProfiling");
        }

        //#6890
        var productTableName = NameCompatibilityManager.GetTableName(typeof(Product));

        //remove column
        var isTelecommunicationsOrBroadcastingOrElectronicServicesColumnName = "IsTelecommunicationsOrBroadcastingOrElectronicServices";
        if (Schema.Table(productTableName).Column(isTelecommunicationsOrBroadcastingOrElectronicServicesColumnName).Exists())
            Delete.Column(isTelecommunicationsOrBroadcastingOrElectronicServicesColumnName).FromTable(productTableName);

        //New message template
        if (!_dataProvider.GetTable().Any(st => string.Compare(st.Name, MessageTemplateSystemNames.DELETE_CUSTOMER_REQUEST_STORE_OWNER_NOTIFICATION, StringComparison.InvariantCultureIgnoreCase) == 0))
        {
            var eaGeneral = _dataProvider.GetTable().FirstOrDefault() ?? throw new Exception("Default email account cannot be loaded");
            _dataProvider.InsertEntity(new MessageTemplate()
            {
                Name = MessageTemplateSystemNames.DELETE_CUSTOMER_REQUEST_STORE_OWNER_NOTIFICATION,
                Subject = "%Store.Name%. New request to delete customer (GDPR)",
                Body = $"%Customer.Email% has requested account deletion. You can consider this in the admin area.",
                IsActive = true,
                EmailAccountId = eaGeneral.Id
            });
        }

        //#7031
        var emailAccountTableName = nameof(EmailAccount);
        var credentialsColumnName = "UseDefaultCredentials";

        if (Schema.Table(emailAccountTableName).Column(credentialsColumnName).Exists())
        {
            var emailAccounts = _dataProvider.GetTable().ToList();
            foreach (var item in emailAccounts)
            {
                if (!string.IsNullOrEmpty(item.Username))
                    item.EmailAuthenticationMethod = EmailAuthenticationMethod.Login;
            }

            _dataProvider.UpdateEntities(emailAccounts);

            //remove column
            Delete.Column(credentialsColumnName).FromTable(emailAccountTableName);
        }

        //#6978
        var newsLetterSubscriptionTableName = nameof(NewsLetterSubscription);
        var languageIdColumnName = nameof(NewsLetterSubscription.LanguageId);
        if (Schema.Table(newsLetterSubscriptionTableName).Column(languageIdColumnName).Exists())
        {
            var newsLetterSubscriptions = _dataProvider.GetTable().ToList();
            var defaultLanguageId = _dataProvider.GetTable().FirstOrDefault()?.Id ?? 0;
            foreach (var newsLetterSubscription in newsLetterSubscriptions)
            {
                newsLetterSubscription.LanguageId = defaultLanguageId;
            }
            _dataProvider.UpdateEntities(newsLetterSubscriptions);
        }

    }

    public override void Down()
    {
        //add the downgrade logic if necessary 
    }
}