Radarr/packages/EntityFramework.4.3.0/tools/EntityFramework.psm1

486 lines
15 KiB
PowerShell

# Copyright (c) Microsoft Corporation. All rights reserved.
$InitialDatabase = '0'
$installPath = $args[0]
$knownExceptions = @(
'System.Data.Entity.Migrations.Infrastructure.MigrationsException',
'System.Data.Entity.Migrations.Infrastructure.AutomaticMigrationsDisabledException',
'System.Data.Entity.Migrations.Infrastructure.AutomaticDataLossException'
)
<#
.SYNOPSIS
Enables Code First Migrations in a project.
.DESCRIPTION
Enables Migrations by scaffolding a migrations configuration class in the project. If the
target database was created by an initializer, an initial migration will be created (unless
automatic migrations are enabled via the EnableAutomaticMigrations parameter).
.PARAMETER EnableAutomaticMigrations
Specifies whether automatic migrations will be enabled in the scaffolded migrations configuration.
If ommitted, automatic migrations will be disabled.
.PARAMETER ProjectName
Specifies the project that the scaffolded migrations configuration class will
be added to. If omitted, the default project selected in package manager
console is used.
.PARAMETER Force
Specifies that the migrations configuration be overwritten when running more
than once for a given project.
#>
function Enable-Migrations
{
[CmdletBinding(DefaultParameterSetName = 'ProjectName')]
param (
[alias("Auto")]
[switch] $EnableAutomaticMigrations,
[string] $ProjectName,
[switch] $Force
)
try
{
$commands = New-MigrationsCommandsNoConfiguration $ProjectName
$commands.EnableMigrations($EnableAutomaticMigrations, $Force)
}
catch [Exception]
{
$exception = $_.Exception
$exceptionType = $exception.GetType()
if ($exceptionType.FullName -eq 'System.Data.Entity.Migrations.Design.ToolingException')
{
if ($knownExceptions -notcontains $exception.InnerType)
{
Write-Host $exception.InnerStackTrace
}
}
elseif (!(Test-TypeInherits $exceptionType 'System.Data.Entity.Migrations.Infrastructure.MigrationsException'))
{
Write-Host $exception
}
throw $exception.Message
}
}
<#
.SYNOPSIS
Scaffolds a migration script for any pending model changes.
.DESCRIPTION
Scaffolds a new migration script and adds it to the project.
.PARAMETER Name
Specifies the name of the custom script.
.PARAMETER Force
Specifies that the migration user code be overwritten when re-scaffolding an
existing migration.
.PARAMETER ProjectName
Specifies the project that contains the migration configuration type to be
used. If ommitted, the default project selected in package manager console
is used.
.PARAMETER StartUpProjectName
Specifies the configuration file to use for named connection strings. If
omitted, the specified project's configuration file is used.
.PARAMETER ConfigurationTypeName
Specifies the migrations configuration to use. If omitted, migrations will
attempt to locate a single migrations configuration type in the target
project.
.PARAMETER ConnectionStringName
Specifies the name of a connection string to use from the application's
configuration file.
.PARAMETER ConnectionString
Specifies the the connection string to use. If omitted, the context's
default connection will be used.
.PARAMETER ConnectionProviderName
Specifies the provider invariant name of the connection string.
#>
function Add-Migration
{
[CmdletBinding(DefaultParameterSetName = 'ConnectionStringName')]
param (
[parameter(Position = 0,
Mandatory = $true)]
[string] $Name,
[switch] $Force,
[string] $ProjectName,
[string] $StartUpProjectName,
[string] $ConfigurationTypeName,
[parameter(ParameterSetName = 'ConnectionStringName')]
[string] $ConnectionStringName,
[parameter(ParameterSetName = 'ConnectionStringAndProviderName',
Mandatory = $true)]
[string] $ConnectionString,
[parameter(ParameterSetName = 'ConnectionStringAndProviderName',
Mandatory = $true)]
[string] $ConnectionProviderName)
try
{
$commands = New-MigrationsCommands $ProjectName $StartUpProjectName $ConfigurationTypeName $ConnectionStringName $ConnectionString $ConnectionProviderName
$commands.AddMigration($Name, $Force)
}
catch [Exception]
{
$exception = $_.Exception
$exceptionType = $exception.GetType()
if ($exceptionType.FullName -eq 'System.Data.Entity.Migrations.Design.ToolingException')
{
if ($knownExceptions -notcontains $exception.InnerType)
{
Write-Host $exception.InnerStackTrace
}
}
elseif (!(Test-TypeInherits $exceptionType 'System.Data.Entity.Migrations.Infrastructure.MigrationsException'))
{
Write-Host $exception
}
throw $exception.Message
}
}
<#
.SYNOPSIS
Applies any pending migrations to the database.
.DESCRIPTION
Updates the database to the current model by applying pending migrations.
.PARAMETER SourceMigration
Only valid with -Script. Specifies the name of a particular migration to use
as the update's starting point. If ommitted, the last applied migration in
the database will be used.
.PARAMETER TargetMigration
Specifies the name of a particular migration to update the database to. If
ommitted, the current model will be used.
.PARAMETER Script
Generate a SQL script rather than executing the pending changes directly.
.PARAMETER Force
Specifies that data loss is acceptable during automatic migration of the
database.
.PARAMETER ProjectName
Specifies the project that contains the migration configuration type to be
used. If ommitted, the default project selected in package manager console
is used.
.PARAMETER StartUpProjectName
Specifies the configuration file to use for named connection strings. If
omitted, the specified project's configuration file is used.
.PARAMETER ConfigurationTypeName
Specifies the migrations configuration to use. If omitted, migrations will
attempt to locate a single migrations configuration type in the target
project.
.PARAMETER ConnectionStringName
Specifies the name of a connection string to use from the application's
configuration file.
.PARAMETER ConnectionString
Specifies the the connection string to use. If omitted, the context's
default connection will be used.
.PARAMETER ConnectionProviderName
Specifies the provider invariant name of the connection string.
#>
function Update-Database
{
[CmdletBinding(DefaultParameterSetName = 'ConnectionStringName')]
param (
[string] $SourceMigration,
[string] $TargetMigration,
[switch] $Script,
[switch] $Force,
[string] $ProjectName,
[string] $StartUpProjectName,
[string] $ConfigurationTypeName,
[parameter(ParameterSetName = 'ConnectionStringName')]
[string] $ConnectionStringName,
[parameter(ParameterSetName = 'ConnectionStringAndProviderName',
Mandatory = $true)]
[string] $ConnectionString,
[parameter(ParameterSetName = 'ConnectionStringAndProviderName',
Mandatory = $true)]
[string] $ConnectionProviderName)
# TODO: If possible, convert this to a ParameterSet
if ($SourceMigration -and !$script)
{
throw '-SourceMigration can only be specified with -Script.'
}
try
{
$commands = New-MigrationsCommands $ProjectName $StartUpProjectName $ConfigurationTypeName $ConnectionStringName $ConnectionString $ConnectionProviderName
$commands.UpdateDatabase($SourceMigration, $TargetMigration, $Script, $Force)
}
catch [Exception]
{
$exception = $_.Exception
$exceptionType = $exception.GetType()
if ($exceptionType.FullName -eq 'System.Data.Entity.Migrations.Design.ToolingException')
{
if ($knownExceptions -notcontains $exception.InnerType)
{
Write-Host $exception.InnerStackTrace
}
}
elseif (!(Test-TypeInherits $exceptionType 'System.Data.Entity.Migrations.Infrastructure.MigrationsException'))
{
Write-Host $exception
}
throw $exception.Message
}
}
<#
.SYNOPSIS
Displays the migrations that have been applied to the target database.
.DESCRIPTION
Displays the migrations that have been applied to the target database.
.PARAMETER ProjectName
Specifies the project that contains the migration configuration type to be
used. If ommitted, the default project selected in package manager console
is used.
.PARAMETER StartUpProjectName
Specifies the configuration file to use for named connection strings. If
omitted, the specified project's configuration file is used.
.PARAMETER ConfigurationTypeName
Specifies the migrations configuration to use. If omitted, migrations will
attempt to locate a single migrations configuration type in the target
project.
.PARAMETER ConnectionStringName
Specifies the name of a connection string to use from the application's
configuration file.
.PARAMETER ConnectionString
Specifies the the connection string to use. If omitted, the context's
default connection will be used.
.PARAMETER ConnectionProviderName
Specifies the provider invariant name of the connection string.
#>
function Get-Migrations
{
[CmdletBinding(DefaultParameterSetName = 'ConnectionStringName')]
param (
[string] $ProjectName,
[string] $StartUpProjectName,
[string] $ConfigurationTypeName,
[parameter(ParameterSetName = 'ConnectionStringName')]
[string] $ConnectionStringName,
[parameter(ParameterSetName = 'ConnectionStringAndProviderName',
Mandatory = $true)]
[string] $ConnectionString,
[parameter(ParameterSetName = 'ConnectionStringAndProviderName',
Mandatory = $true)]
[string] $ConnectionProviderName)
try
{
$commands = New-MigrationsCommands $ProjectName $StartUpProjectName $ConfigurationTypeName $ConnectionStringName $ConnectionString $ConnectionProviderName
$commands.GetMigrations()
}
catch [Exception]
{
$exception = $_.Exception
$exceptionType = $exception.GetType()
if ($exceptionType.FullName -eq 'System.Data.Entity.Migrations.Design.ToolingException')
{
if ($knownExceptions -notcontains $exception.InnerType)
{
Write-Host $exception.InnerStackTrace
}
}
elseif (!(Test-TypeInherits $exceptionType 'System.Data.Entity.Migrations.Infrastructure.MigrationsException'))
{
Write-Host $exception
}
throw $exception.Message
}
}
function New-MigrationsCommandsNoConfiguration($ProjectName)
{
$project = Get-MigrationsProject $ProjectName
Build-Project $project
Load-EntityFramework
try
{
return New-Object 'System.Data.Entity.Migrations.MigrationsCommands' @(
$project,
$project,
$null,
$null,
$null,
$null,
$PSCmdlet )
}
catch [System.Management.Automation.MethodInvocationException]
{
throw $_.Exception.InnerException
}
}
function New-MigrationsCommands($ProjectName, $StartUpProjectName, $ConfigurationTypeName, $ConnectionStringName, $ConnectionString, $ConnectionProviderName)
{
$project = Get-MigrationsProject $ProjectName
$startUpProject = Get-MigrationsStartUpProject $StartUpProjectName
Build-Project $project
Build-Project $startUpProject
Load-EntityFramework
try
{
return New-Object 'System.Data.Entity.Migrations.MigrationsCommands' @(
$project,
$startUpProject,
$ConfigurationTypeName,
$ConnectionStringName,
$ConnectionString,
$ConnectionProviderName,
$PSCmdlet )
}
catch [System.Management.Automation.MethodInvocationException]
{
throw $_.Exception.InnerException
}
}
function Get-MigrationsProject($name)
{
if ($name)
{
return Get-SingleProject $name
}
$project = Get-Project
Write-Verbose ('Using NuGet project ''' + $project.Name + '''.')
return $project
}
function Get-MigrationsStartUpProject($name)
{
if ($name)
{
return Get-SingleProject $name
}
$startupProjectPaths = $DTE.Solution.SolutionBuild.StartupProjects
if (!$startupProjectPaths)
{
throw 'No start-up project found. Please use the -StartupProject parameter.'
}
if ($startupProjectPaths.Length -gt 1)
{
throw 'More than one start-up project found. Please use the -StartUpProject parameter.'
}
$startupProjectPath = $startupProjectPaths[0]
if (!(Split-Path -IsAbsolute $startupProjectPath))
{
$solutionPath = Split-Path $DTE.Solution.Properties.Item('Path').Value
$startupProjectPath = Join-Path $solutionPath $startupProjectPath -Resolve
}
$startupProject = $DTE.Solution.Projects | ?{
$fullName = $_.FullName
if ($fullName -and $fullName.EndsWith('\'))
{
$fullName = $fullName.Substring(0, $fullName.Length - 1)
}
return $fullName -eq $startupProjectPath
}
Write-Verbose ('Using StartUp project ''' + $startupProject.Name + '''.')
return $startupProject
}
function Get-SingleProject($name)
{
$project = Get-Project $name
if ($project -is [array])
{
throw "More than one project '$name' was found. Specify the full name of the one to use."
}
return $project
}
function Load-EntityFramework()
{
[System.AppDomain]::CurrentDomain.SetShadowCopyFiles()
[System.Reflection.Assembly]::LoadFrom((Join-Path $installPath 'lib\net40\EntityFramework.dll')) | Out-Null
[System.Reflection.Assembly]::LoadFrom((Join-Path $installPath 'tools\EntityFramework.PowerShell.dll')) | Out-Null
}
function Build-Project($project)
{
$configuration = $DTE.Solution.SolutionBuild.ActiveConfiguration.Name
$DTE.Solution.SolutionBuild.BuildProject($configuration, $project.UniqueName, $true)
if ($DTE.Solution.SolutionBuild.LastBuildInfo)
{
throw 'The project ''' + $project.Name + ''' failed to build.'
}
}
function Test-TypeInherits($type, $baseTypeName)
{
if ($type.FullName -eq $baseTypeName)
{
return $true
}
$baseType = $type.BaseType
if ($baseType)
{
return Test-TypeInherits $baseType $baseTypeName
}
return $false
}
Export-ModuleMember @( 'Enable-Migrations', 'Add-Migration', 'Update-Database', 'Get-Migrations' ) -Variable 'InitialDatabase'