Framework/Managers/ConfigurationManager.ps1
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 |
Set-StrictMode -Version Latest # # ConfigManager.ps1 # class ConfigurationManager { hidden static [AzSKConfig] GetAzSKConfigData() { return [AzSKConfig]::GetInstance([ConfigurationManager]::GetAzSKSettings().UseOnlinePolicyStore, [ConfigurationManager]::GetAzSKSettings().OnlinePolicyStoreUrl, [ConfigurationManager]::GetAzSKSettings().EnableAADAuthForOnlinePolicyStore) } hidden static [AzSKSettings] GetAzSKSettings() { return [AzSKSettings]::GetInstance() } hidden static [AzSKSettings] GetLocalAzSKSettings() { return [AzSKSettings]::GetLocalInstance() } hidden static [AzSKSettings] UpdateAzSKSettings([AzSKSettings] $localSettings) { return [AzSKSettings]::Update($localSettings) } hidden static [SVTConfig] GetSVTConfig([string] $fileName) { $usePolicyStore = [ConfigurationManager]::GetAzSKSettings().UseOnlinePolicyStore $policyStoreUrlOrFolder = [ConfigurationManager]::GetAzSKSettings().OnlinePolicyStoreUrl $useAADAuthForPolicyStore = [ConfigurationManager]::GetAzSKSettings().EnableAADAuthForOnlinePolicyStore $defaultConfigFile = [ConfigurationHelper]::LoadServerConfigFile($fileName, $usePolicyStore, $policyStoreUrlOrFolder, $useAADAuthForPolicyStore) $extendedFileName = $fileName.Replace(".json",".ext.json"); $extendedConfigFile = $null if(-not [ConfigurationHelper]::LocalPolicyEnabled) #Default/common case... i.e., not in local policy debug mode => get .ext.json file from server. { $extendedConfigFile = [ConfigurationHelper]::LoadServerFileRaw($extendedFileName, $usePolicyStore, $policyStoreUrlOrFolder, $useAADAuthForPolicyStore) } #Check if there is an .ext.json file in local org policy folder elseif ([ConfigurationHelper]::IsPolicyPresentOnServer($extendedFileName, $usePolicyStore, $policyStoreUrlOrFolder, $useAADAuthForPolicyStore)) { Write-Warning "########## Looking for .ext.json file locally..... ##########" $extendedConfigFile = [ConfigurationHelper]::LoadOfflineConfigFile($extendedFileName, <#$parseJson#> $true, $policyStoreUrlOrFolder) } $finalObject = [SVTConfig] $defaultConfigFile; if(-not [string]::IsNullOrWhiteSpace($extendedConfigFile)) { $IdPropName = "Id" $finalObject = [SVTConfig]([Helpers]::MergeObjects($defaultConfigFile,$extendedConfigFile, $IdPropName)); } return $finalObject; } hidden static [PSObject] LoadServerConfigFile([string] $fileName) { return [ConfigurationHelper]::LoadServerConfigFile($fileName, [ConfigurationManager]::GetAzSKSettings().UseOnlinePolicyStore, [ConfigurationManager]::GetAzSKSettings().OnlinePolicyStoreUrl, [ConfigurationManager]::GetAzSKSettings().EnableAADAuthForOnlinePolicyStore); } hidden static [PSObject] LoadServerFileRaw([string] $fileName) { return [ConfigurationHelper]::LoadServerFileRaw($fileName, [ConfigurationManager]::GetAzSKSettings().UseOnlinePolicyStore, [ConfigurationManager]::GetAzSKSettings().OnlinePolicyStoreUrl, [ConfigurationManager]::GetAzSKSettings().EnableAADAuthForOnlinePolicyStore); } hidden static [string] LoadExtensionFile([string] $svtClassName) { $extensionSVTClassName = $svtClassName + "Ext"; $extensionFilePath = "" #check for extension type only if we dont find the type already loaded in to the current session if(-not ($extensionSVTClassName -as [type])) { $extensionSVTClassFileName = $svtClassName + ".ext.ps1"; if(-not [ConfigurationHelper]::LocalPolicyEnabled) { try { $extensionFilePath = [ConfigurationManager]::DownloadExtFile($extensionSVTClassFileName) } catch { [EventBase]::PublishGenericException($_); } } #We are in org-policy debug mode, use local org policy folder to look for .ext.ps1 file #Check if an ext file exists for this class... elseif ([ConfigurationHelper]::IsPolicyPresentOnServer($extensionSVTClassFileName,[ConfigurationManager]::GetAzSKSettings().UseOnlinePolicyStore, [ConfigurationManager]::GetAzSKSettings().OnlinePolicyStoreUrl,[ConfigurationManager]::GetAzSKSettings().EnableAADAuthForOnlinePolicyStore)) { Write-Warning "########## Looking for .ext.ps1 file locally..... ##########" $expectedExtFolder = Join-Path ([ConfigurationManager]::GetAzSKSettings().OnlinePolicyStoreUrl) 'Config' $expectedExtFile = Join-Path $expectedExtFolder $extensionSVTClassFileName if (Test-Path $expectedExtFile) { $extensionFilePath = $expectedExtFile } else { [EventBase]::PublishGenericCustomMessage(("Could not find extension (.ext.ps1) file for [$svtClassName] in folder [$expectedExtFolder] in org-policy-debug mode."), [MessageType]::Error); } } } return $extensionFilePath } hidden static [string[]] RegisterExtListenerFiles() { $ServerConfigMetadata = [ConfigurationManager]::LoadServerConfigFile([Constants]::ServerConfigMetadataFileName) $ListenerFilePaths = @(); if($null -ne [ConfigurationHelper]::ServerConfigMetadata) { [ConfigurationHelper]::ServerConfigMetadata.OnlinePolicyList | ForEach-Object { if([Helpers]::CheckMember($_,"Name")) { if($_.Name -match "Listener.ext.ps1") { $listenerFileName = $_.Name try { $extensionFilePath = [ConfigurationManager]::DownloadExtFile($listenerFileName) # file has to be loaded here due to scope constraint $ListenerFilePaths += $extensionFilePath } catch { [EventBase]::PublishGenericException($_); } } } } } return $ListenerFilePaths; } hidden static [string] DownloadExtFile([string] $fileName) { $localExtensionsFolderPath = [Constants]::AzSKExtensionsFolderPath; $extensionFilePath = "" if(-not (Test-Path -Path $localExtensionsFolderPath)) { New-Item -ItemType Directory -Path $localExtensionsFolderPath -Force } $extensionScriptCode = [ConfigurationManager]::LoadServerFileRaw($fileName); if(-not [string]::IsNullOrWhiteSpace($extensionScriptCode)) { $extensionFilePath = Join-Path $([Constants]::AzSKExtensionsFolderPath) $fileName; Out-File -InputObject $extensionScriptCode -Force -FilePath $extensionFilePath -Encoding utf8; Set-ItemProperty -Path $extensionFilePath -Name IsReadOnly -Value $true } return $extensionFilePath } } |