Private/Import-ADTConfig.ps1
|
#----------------------------------------------------------------------------- # # MARK: Import-ADTConfig # #----------------------------------------------------------------------------- function Import-ADTConfig { [CmdletBinding()] param ( [Parameter(Mandatory = $true)] [ValidateScript({ if ([System.String]::IsNullOrWhiteSpace($_)) { $PSCmdlet.ThrowTerminatingError((New-ADTValidateScriptErrorRecord -ParameterName BaseDirectory -ProvidedValue $_ -ExceptionMessage 'The specified input is null or empty.')) } if (![System.IO.Directory]::Exists($_)) { $PSCmdlet.ThrowTerminatingError((New-ADTValidateScriptErrorRecord -ParameterName BaseDirectory -ProvidedValue $_ -ExceptionMessage 'The specified directory does not exist.')) } return $_ })] [System.String]$BaseDirectory ) # Internal filter to process asset file paths. filter Update-ADTAssetFilePath { # Go recursive if we've received a hashtable, otherwise just update the values. foreach ($asset in $($_.GetEnumerator())) { # Re-process if this is a hashtable. if ($asset.Value -is [System.Collections.Hashtable]) { $asset.Value | & $MyInvocation.MyCommand; continue } # Skip if the path is fully qualified. if ([System.IO.Path]::IsPathRooted($asset.Value)) { continue } # Get the asset's full path based on the supplied BaseDirectory. # Fall back to the module's path if the asset is unable to be found. if ([System.IO.File]::Exists("$BaseDirectory\$($_.($asset.Key))")) { $_.($asset.Key) = (Get-Item -LiteralPath "$BaseDirectory\$($_.($asset.Key))").FullName } else { $_.($asset.Key) = (Get-Item -LiteralPath "$($BaseDirectory -replace '^.+\\', "$Script:PSScriptRoot\")\$($_.($asset.Key))").FullName } } } # Internal filter to expand variables. filter Expand-ADTVariablesInConfig { # Go recursive if we've received a hashtable, otherwise just update the values. foreach ($section in $($_.GetEnumerator())) { if ($section.Value -is [System.Collections.Hashtable]) { $section.Value | & $MyInvocation.MyCommand } elseif ($section.Value -is [System.String]) { $_.($section.Key) = $ExecutionContext.InvokeCommand.ExpandString($section.Value) } } } # Import the config from disk. $config = Import-ADTModuleDataFile @PSBoundParameters -FileName config.psd1 # Place restrictions on non-ConsoleHost targets. if ($Host.Name.Equals('Windows PowerShell ISE Host')) { $config.UI.DialogStyle = 'Classic' } # Confirm the specified dialog type is valid. if (($config.UI.DialogStyle -ne 'Classic') -and (Test-ADTNonNativeCaller)) { $config.UI.DialogStyle = if ($config.UI.ContainsKey('DialogStyleCompatMode')) { $config.UI.DialogStyleCompatMode } else { 'Classic' } } if (!$Script:Dialogs.Contains($config.UI.DialogStyle)) { $naerParams = @{ Exception = [System.NotSupportedException]::new("The specified dialog style [$($config.UI.DialogStyle)] is not supported. Valid styles are ['$($Script:Dialogs.Keys -join "', '")'].") Category = [System.Management.Automation.ErrorCategory]::InvalidData ErrorId = 'DialogStyleNotSupported' TargetObject = $config RecommendedAction = "Please review the supplied configuration file and try again." } $PSCmdlet.ThrowTerminatingError((New-ADTErrorRecord @naerParams)) } # Expand out environment variables and asset file paths. ($adtEnv = Get-ADTEnvironment).GetEnumerator() | & { process { New-Variable -Name $_.Name -Value $_.Value -Option Constant } end { $config | Expand-ADTVariablesInConfig } } $config.Assets | Update-ADTAssetFilePath # Process the classic assets by grabbing the bytes of each image asset, storing them into a memory stream, then as an image for WinForms to use. $Script:Dialogs.Classic.Assets.Logo = [System.Drawing.Image]::FromStream([System.IO.MemoryStream]::new([System.IO.File]::ReadAllBytes($config.Assets.Logo))) $Script:Dialogs.Classic.Assets.Icon = [PSADT.Shared.Utility]::ConvertImageToIcon($Script:Dialogs.Classic.Assets.Logo) $Script:Dialogs.Classic.Assets.Banner = [System.Drawing.Image]::FromStream([System.IO.MemoryStream]::new([System.IO.File]::ReadAllBytes($config.Assets.Banner))) $Script:Dialogs.Classic.BannerHeight = [System.Math]::Ceiling($Script:Dialogs.Classic.Width * ($Script:Dialogs.Classic.Assets.Banner.Height / $Script:Dialogs.Classic.Assets.Banner.Width)) # Set the app's AUMID so it doesn't just say "Windows PowerShell". if ($config.UI.BalloonNotifications -and ![PSADT.LibraryInterfaces.Shell32]::SetCurrentProcessExplicitAppUserModelID($config.UI.BalloonTitle)) { $regKey = "$(if ($adtEnv.IsAdmin) { 'HKEY_CLASSES_ROOT' } else { 'HKEY_CURRENT_USER\Software\Classes' })\AppUserModelId\$($config.UI.BalloonTitle)" [Microsoft.Win32.Registry]::SetValue($regKey, 'DisplayName', $config.UI.BalloonTitle, [Microsoft.Win32.RegistryValueKind]::String) [Microsoft.Win32.Registry]::SetValue($regKey, 'IconUri', $config.Assets.Logo, [Microsoft.Win32.RegistryValueKind]::ExpandString) } # Change paths to user accessible ones if user isn't an admin. if (!$adtEnv.IsAdmin) { if ($config.Toolkit.TempPathNoAdminRights) { $config.Toolkit.TempPath = $config.Toolkit.TempPathNoAdminRights } if ($config.Toolkit.RegPathNoAdminRights) { $config.Toolkit.RegPath = $config.Toolkit.RegPathNoAdminRights } if ($config.Toolkit.LogPathNoAdminRights) { $config.Toolkit.LogPath = $config.Toolkit.LogPathNoAdminRights } if ($config.MSI.LogPathNoAdminRights) { $config.MSI.LogPath = $config.MSI.LogPathNoAdminRights } } # Append the toolkit's name onto the temporary path. $config.Toolkit.TempPath = [System.IO.Path]::Combine($config.Toolkit.TempPath, $adtEnv.appDeployToolkitName) # Finally, return the config for usage within module. return $config } |