bitwarden-server/util/EfShared/MigrationBuilderExtensions.cs



using System.Runtime.CompilerServices;
using Bit.Core.Utilities;
using Microsoft.EntityFrameworkCore.Migrations;

namespace Bit
{
    // This file is a manual addition to a project that it helps, a project that chooses to compile it
    // should have a projet reference to Core.csproj and a package reference to Microsoft.EntityFrameworkCore.Design
    // The reason for this is that if it belonged to it's own library you would have to add manual references to the above
    // and manage the version for the EntityFrameworkCore package. This way it also doesn't create another dll
    // To include this you can view examples in the MySqlMigrations and PostgresMigrations .csproj files. 
    // <Compile Include="..\EfShared\MigrationBuilderExtensions.cs" />

    public static class MigrationBuilderExtensions
    {
        /// <summary>
        /// Reads an embedded resource for it's SQL contents and formats it with the specified direction for easier custom migration steps
        /// </summary>
        /// <param name="migrationBuilder">The MigrationBuilder instance the sql should be applied to</param>
        /// <param name="resourceName">The file name portion of the resource name, it is assumed to be in a Scripts folder</param>
        /// <param name="dir">The direction of the migration taking place</param>
        public static void SqlResource(this MigrationBuilder migrationBuilder, string resourceName, [CallerMemberName] string dir = null)
        {
            var formattedResourceName = string.IsNullOrEmpty(dir) ? resourceName : string.Format(resourceName, dir);

            migrationBuilder.Sql(CoreHelpers.GetEmbeddedResourceContentsAsync(
                $"Scripts.{formattedResourceName}"));
        }
    }
}