Adds the missing logic for handling the preProvision extensions on Wi…

…n agents. (#3367)
Azure · Jun 26, 2018 · 91cdfe9 · 91cdfe9
1 parent c357400
commit 91cdfe9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/parts/k8s/kuberneteswindowssetup.ps1 b/parts/k8s/kuberneteswindowssetup.ps1
@@ -642,6 +642,9 @@ try
         Write-Log "Set Internet Explorer"
         Set-Explorer
 
+        Write-Log "Start preProvisioning script"
+        PREPROVISION_EXTENSION
+
         Write-Log "Setup Complete, reboot computer"
         Restart-Computer
     }

diff --git a/pkg/acsengine/template_generator.go b/pkg/acsengine/template_generator.go
@@ -598,6 +598,11 @@ func (t *TemplateGenerator) getTemplateFuncMap(cs *api.ContainerService) templat
 			if e != nil {
 				return ""
 			}
+			preprovisionCmd := ""
+			if profile.PreprovisionExtension != nil {
+				preprovisionCmd = makeAgentExtensionScriptCommands(cs, profile)
+			}
+			str = strings.Replace(str, "PREPROVISION_EXTENSION", escapeSingleLine(strings.TrimSpace(preprovisionCmd)), -1)
 			return fmt.Sprintf("\"customData\": \"[base64(concat('%s'))]\",", str)
 		},
 		"GetMasterSwarmModeCustomData": func() string {