added-schemas to exported definitions (#397)

Co-authored-by: Hunor Borondy <[email protected]>
Azure · Oct 18, 2023 · 8a29136 · 8a29136
1 parent 507bdf2
commit 8a29136
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 3 deletions.
diff --git a/Scripts/Helpers/Out-PolicyAssignmentFile.ps1 b/Scripts/Helpers/Out-PolicyAssignmentFile.ps1
@@ -57,6 +57,7 @@ function Out-PolicyAssignmentFile {
     }
 
     $assignmentDefinition = [ordered]@{
+        '$schema'       = "https://raw.githubusercontent.com/Azure/enterprise-azure-policy-as-code/main/Schemas/policy-assignment-schema.json"
         nodeName        = "/root"
         definitionEntry = $definitionEntry
     }

diff --git a/Scripts/Helpers/Out-PolicyDefinition.ps1 b/Scripts/Helpers/Out-PolicyDefinition.ps1
@@ -63,6 +63,13 @@ function Out-PolicyDefinition {
 
     # Write the content
     Remove-NullFields $Definition
-    $json = ConvertTo-Json $Definition -Depth 100
+    $outDefinition = [PSCustomObject]@{'$schema' = "https://raw.githubusercontent.com/Azure/enterprise-azure-policy-as-code/main/Schemas/policy-definition-schema.json" }
+
+    # update schema if policy set
+    if ($Definition.properties.policyDefinitions) {
+        $outDefinition.'$schema' = "https://raw.githubusercontent.com/Azure/enterprise-azure-policy-as-code/main/Schemas/policy-set-definition-schema.json" 
+    }
+    $Definition.psobject.Properties.ForEach{ ($outDefinition).psobject.Properties.Add($_, $true) }
+    $json = ConvertTo-Json $outDefinition -Depth 100
     $null = New-Item $fullPath -Force -ItemType File -Value $json
 }
diff --git a/Scripts/Helpers/Out-PolicyExemptions.ps1 b/Scripts/Helpers/Out-PolicyExemptions.ps1
@@ -125,7 +125,8 @@ function Out-PolicyExemptions {
             if (Test-Path $jsonFile) {
                 Remove-Item $jsonFile
             }
-            $outputJsonObj = @{
+            $outputJsonObj = [ordered]@{
+                '$schema'  = "https://raw.githubusercontent.com/Azure/enterprise-azure-policy-as-code/main/Schemas/policy-exemption-schema.json"
                 exemptions = $jsonArray
             }
             ConvertTo-Json $outputJsonObj -Depth 100 | Out-File $jsonFile -Force
@@ -187,7 +188,8 @@ function Out-PolicyExemptions {
             if (Test-Path $jsonFile) {
                 Remove-Item $jsonFile
             }
-            $outputJsonObj = @{
+            $outputJsonObj = [ordered]@{
+                '$schema'  = "https://raw.githubusercontent.com/Azure/enterprise-azure-policy-as-code/main/Schemas/policy-exemption-schema.json"
                 exemptions = $jsonArray
             }
             ConvertTo-Json $outputJsonObj -Depth 100 | Out-File $jsonFile -Force