mirror of
https://github.com/Sonarr/Sonarr.git
synced 2024-12-16 11:37:58 +02:00
486 lines
15 KiB
PowerShell
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'
|