Umbraco-CMS/src/Umbraco.Infrastructure/Migrations/MigrationContext.cs

using System;
using System.Collections.Generic;
using Microsoft.Extensions.Logging;
using Umbraco.Cms.Infrastructure.Persistence;

namespace Umbraco.Cms.Infrastructure.Migrations
{
    /// <summary>
    /// Implements <see cref="IMigrationContext"/>.
    /// </summary>
    internal class MigrationContext : IMigrationContext
    {
        /// <summary>
        /// Initializes a new instance of the <see cref="MigrationContext"/> class.
        /// </summary>
        public MigrationContext(IUmbracoDatabase database, ILogger<MigrationContext> logger)
        {
            Database = database ?? throw new ArgumentNullException(nameof(database));
            Logger = logger ?? throw new ArgumentNullException(nameof(logger));
        }

        /// <inheritdoc />
        public ILogger<IMigrationContext> Logger { get; }

        /// <inheritdoc />
        public IUmbracoDatabase Database { get; }

        /// <inheritdoc />
        public ISqlContext SqlContext => Database.SqlContext;

        /// <inheritdoc />
        public int Index { get; set; }

        /// <inheritdoc />
        public bool BuildingExpression { get; set; }

        // this is only internally exposed
        public List<Type> PostMigrations { get; } = new List<Type>();

        /// <inheritdoc />
        public void AddPostMigration<TMigration>()
            where TMigration : MigrationBase
        {
            // just adding - will be de-duplicated when executing
            PostMigrations.Add(typeof(TMigration));
        }
    }
}