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

using System;
using System.Collections.Generic;
using Umbraco.Core.Logging;
using Umbraco.Core.Persistence;

namespace Umbraco.Core.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 logger)
        {
            Database = database ?? throw new ArgumentNullException(nameof(database));
            Logger = logger ?? throw new ArgumentNullException(nameof(logger));
        }

        /// <inheritdoc />
        public ILogger 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 : IMigration
        {
            // just adding - will be de-duplicated when executing
            PostMigrations.Add(typeof(TMigration));
        }
    }
}