Windows support PoC

Signed-off-by: Alexey Makhov <amakhov@mirantis.com>
Signed-off-by: makhov <amakhov@mirantis.com>

Author: makhov

api/bootstrap/v1beta1/k0s_types.go

@@ -122,6 +122,10 @@ type K0sWorkerConfigSpec struct {
 
 	// WorkingDir specifies the working directory where k0smotron will place its files.
 	WorkingDir string `json:"workingDir,omitempty"`
+
+	// IsWindows specifies whether the target node is Windows.
+	// +kubebuilder:validation:Optional
+	IsWindows bool `json:"isWindows,omitempty"`
 }
 
 // SecretMetadata defines metadata to be propagated to the bootstrap Secret

config/clusterapi/bootstrap/bases/bootstrap.cluster.x-k8s.io_k0sworkerconfigs.yaml

@@ -170,6 +170,9 @@ spec:
                 - variant
                 - version
                 type: object
+              isWindows:
+                description: IsWindows specifies whether the target node is Windows.
+                type: boolean
               k0sInstallDir:
                 default: /usr/local/bin
                 description: |-

config/clusterapi/bootstrap/bases/bootstrap.cluster.x-k8s.io_k0sworkerconfigtemplates.yaml

@@ -194,6 +194,10 @@ spec:
                         - variant
                         - version
                         type: object
+                      isWindows:
+                        description: IsWindows specifies whether the target node is
+                          Windows.
+                        type: boolean
                       k0sInstallDir:
                         default: /usr/local/bin
                         description: |-

config/crd/bases/bootstrap/bootstrap.cluster.x-k8s.io_k0sworkerconfigs.yaml

@@ -170,6 +170,9 @@ spec:
                 - variant
                 - version
                 type: object
+              isWindows:
+                description: IsWindows specifies whether the target node is Windows.
+                type: boolean
               k0sInstallDir:
                 default: /usr/local/bin
                 description: |-

config/crd/bases/bootstrap/bootstrap.cluster.x-k8s.io_k0sworkerconfigtemplates.yaml

@@ -194,6 +194,10 @@ spec:
                         - variant
                         - version
                         type: object
+                      isWindows:
+                        description: IsWindows specifies whether the target node is
+                          Windows.
+                        type: boolean
                       k0sInstallDir:
                         default: /usr/local/bin
                         description: |-

docs/resource-reference/bootstrap.cluster.x-k8s.io-v1beta1.md

@@ -804,6 +804,13 @@ If specified the version field is ignored and what ever version is downloaded fr
           Ignition defines the ignition configuration. If empty, k0smotron will use cloud-init.<br/>
         </td>
         <td>false</td>
+      </tr><tr>
+        <td><b>isWindows</b></td>
+        <td>boolean</td>
+        <td>
+          IsWindows specifies whether the target node is Windows.<br/>
+        </td>
+        <td>false</td>
       </tr><tr>
         <td><b>k0sInstallDir</b></td>
         <td>string</td>
@@ -1541,6 +1548,13 @@ If specified the version field is ignored and what ever version is downloaded fr
           Ignition defines the ignition configuration. If empty, k0smotron will use cloud-init.<br/>
         </td>
         <td>false</td>
+      </tr><tr>
+        <td><b>isWindows</b></td>
+        <td>boolean</td>
+        <td>
+          IsWindows specifies whether the target node is Windows.<br/>
+        </td>
+        <td>false</td>
       </tr><tr>
         <td><b>k0sInstallDir</b></td>
         <td>string</td>

internal/controller/bootstrap/common.go

@@ -104,6 +104,10 @@ func mergeExtraArgs(configArgs []string, configOwner *bsutil.ConfigOwner, isWork
 }
 
 func getProvisioner(ignitionSpec *bootstrapv1.IgnitionSpec) provisioner.Provisioner {
+	//if isWindows {
+	//	return &provisioner.CloudBaseInitProvisioner{}
+	//}
+
 	if ignitionSpec != nil {
 		return &provisioner.IgnitionProvisioner{
 			Variant:          ignitionSpec.Variant,

internal/controller/bootstrap/worker_bootstrap_controller.go

@@ -262,11 +262,114 @@ func (r *Controller) generateBootstrapDataForWorker(ctx context.Context, log log
 		files = append(files, resolveCertsForIngress...)
 	}
 
+	commandsMap := make(map[provisioner.VarName]string)
+	var commands []string
+
+	if scope.Config.Spec.IsWindows {
+		var winFiles []provisioner.File
+		commands, winFiles = getWindowsCommands(scope)
+		files = append(files, winFiles...)
+	} else {
+		commands, commandsMap, err = getLinuxCommands(scope)
+		if err != nil {
+			return nil, fmt.Errorf("error generating linux commands: %w", err)
+		}
+	}
+
+	var (
+		customUserData string
+		vars           map[provisioner.VarName]string
+	)
+	if scope.Config.Spec.CustomUserDataRef != nil {
+		customUserData, err = resolveContentFromFile(ctx, r.Client, scope.Cluster, scope.Config.Spec.CustomUserDataRef)
+		if err != nil {
+			return nil, fmt.Errorf("error extracting the contents of the provided custom worker user data: %w", err)
+		}
+		vars = commandsMap
+	}
+
+	return scope.provisioner.ToProvisionData(&provisioner.InputProvisionData{
+		Files:          files,
+		Commands:       commands,
+		CustomUserData: customUserData,
+		Vars:           vars,
+	})
+}
+
+func getWindowsCommands(scope *Scope) ([]string, []provisioner.File) {
+	//if scope.Config.Spec.K0sInstallDir == "/usr/local/bin" {
+	//	scope.Config.Spec.K0sInstallDir = "C:\\bootstrap"
+	//}
+	k0sPath := filepath.Join(scope.Config.Spec.K0sInstallDir, "k0s.exe")
+
+	var files []provisioner.File
+	files = append(files, provisioner.File{
+		Path:        `C:\bootstrap\k0s_install.ps1`,
+		Permissions: "0644",
+		Content: fmt.Sprintf(`Write-Host "=== Checking Windows Containers feature using DISM ==="
+
+      function Is-FeatureEnabled {
+          param([string]$Name)
+          $output = dism.exe /online /Get-Features
+          return $output -match "$Name\s+:\s+Enabled"
+      }
+
+      $featureName = "Containers"
+
+      if (Is-FeatureEnabled $featureName) {
+          Write-Host "$featureName is already enabled. Nothing to do."
+      } else {
+          Write-Host "$featureName is not enabled. Installing..."
+          $result = dism.exe /online /Enable-Feature /FeatureName:$featureName /All /NoRestart
+          $exitCode = $LASTEXITCODE
+          Write-Host "DISM exit code: $exitCode"
+
+          if ($exitCode -eq 0) {
+              Write-Host "Feature installed successfully, no reboot required."
+          } elseif ($exitCode -eq 3010) {
+              Write-Host "Feature installed successfully, reboot required."
+              Restart-Computer -Force
+          } else {
+              throw "DISM failed with exit code $exitCode"
+          }
+      }
+
+      # --- Download and run k0s ---
+      Write-Host "=== Downloading k0s binary ==="
+
+      $k0sUrl = "https://get.k0sproject.io/%s/k0s-%s-amd64.exe"  # PoC supporting only amd64
+      $dest = "%s"
+
+      Invoke-WebRequest -Uri $k0sUrl -OutFile $dest -UseBasicParsing
+
+      Write-Host "=== Executing k0s to check version ==="
+      & $dest --version`, scope.Config.Spec.Version, scope.Config.Spec.Version, k0sPath),
+	})
+
+	commands := scope.Config.Spec.PreStartCommands
+	// Download and enable containers and k0s bootstrap script
+	commands = append(commands, `powershell.exe -NoProfile -NonInteractive -File "C:\bootstrap\k0s_install.ps1"`)
+	// TODO: implement ingress support for Windows
+	//commands = append(commands, ingressCommands...)
+
+	installCmdParts := []string{
+		fmt.Sprintf(`powershell.exe -NoProfile -NonInteractive -Command "& { %s install worker --token-file %s }"`, k0sPath, scope.Config.GetJoinTokenPath()),
+	}
+	installCmdParts = append(installCmdParts, mergeExtraArgs(scope.Config.Spec.Args, scope.ConfigOwner, true, scope.Config.Spec.UseSystemHostname)...)
+
+	commands = append(commands, fmt.Sprintf(`powershell.exe -NoProfile -NonInteractive -Command "& { %s }`, strings.Join(installCmdParts, " ")))
+	commands = append(commands, fmt.Sprintf(`powershell.exe -NoProfile -NonInteractive -Command "& { & %s start }`, k0sPath))
+	commands = append(commands, scope.Config.Spec.PostStartCommands...)
+
+	return commands, files
+}
+
+func getLinuxCommands(scope *Scope) ([]string, map[provisioner.VarName]string, error) {
 	commandsMap := make(map[provisioner.VarName]string)
 
 	downloadCommands, err := util.DownloadCommands(scope.Config.Spec.PreInstalledK0s, scope.Config.Spec.DownloadURL, scope.Config.Spec.Version, scope.Config.Spec.K0sInstallDir)
 	if err != nil {
-		return nil, fmt.Errorf("error generating download commands: %w", err)
+		return nil, nil, fmt.Errorf("error generating download commands: %w", err)
 	}
 	installCmd := createInstallCmd(scope)
 
@@ -288,24 +391,7 @@ func (r *Controller) generateBootstrapDataForWorker(ctx context.Context, log log
 	// https://cluster-api.sigs.k8s.io/developer/providers/contracts/bootstrap-config#sentinel-file
 	commands = append(commands, "mkdir -p /run/cluster-api && touch /run/cluster-api/bootstrap-success.complete")
 
-	var (
-		customUserData string
-		vars           map[provisioner.VarName]string
-	)
-	if scope.Config.Spec.CustomUserDataRef != nil {
-		customUserData, err = resolveContentFromFile(ctx, r.Client, scope.Cluster, scope.Config.Spec.CustomUserDataRef)
-		if err != nil {
-			return nil, fmt.Errorf("error extracting the contents of the provided custom worker user data: %w", err)
-		}
-		vars = commandsMap
-	}
-
-	return scope.provisioner.ToProvisionData(&provisioner.InputProvisionData{
-		Files:          files,
-		Commands:       commands,
-		CustomUserData: customUserData,
-		Vars:           vars,
-	})
+	return commands, commandsMap, nil
 }
 
 func (r *Controller) getK0sToken(ctx context.Context, scope *Scope) (string, error) {

internal/controller/bootstrap/worker_bootstrap_controller_test.go

@@ -126,6 +126,39 @@ func Test_createInstallCmd(t *testing.T) {
 	}
 }
 
+func Test_getWindowsCommands(t *testing.T) {
+	tests := []struct {
+		name  string
+		scope *Scope
+		want  []string
+	}{
+		{
+			name: "with default config",
+			scope: &Scope{
+				Config: &bootstrapv1.K0sWorkerConfig{
+					Spec: bootstrapv1.K0sWorkerConfigSpec{
+						IsWindows: true,
+					},
+				},
+				ConfigOwner: &bsutil.ConfigOwner{Unstructured: &unstructured.Unstructured{Object: map[string]interface{}{}}},
+			},
+			want: []string{
+				"C:\\bootstrap\\enable_containers_and_k0s.ps1",
+				"& k0s.exe install worker --token-file /etc/k0s.token --labels=k0smotron.io/machine-name= --kubelet-extra-args=\"--hostname-override=\"",
+				"& k0s.exe start",
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, _ := getWindowsCommands(tt.scope)
+			require.Equal(t, tt.want, got)
+		})
+	}
+
+}
+
 func Test_createBootstrapSecret(t *testing.T) {
 	tests := []struct {
 		name                string

internal/controller/infrastructure/ssh_provisioner.go

@@ -247,8 +247,8 @@ func (p *SSHProvisioner) Cleanup(ctx context.Context, mode RemoteMachineMode) er
 
 func (p *SSHProvisioner) uploadFile(client *rig.Client, file provisioner.File) error {
 	fsys := client.FS()
-	// Ensure base dir exists for target
-	dir := filepath.Dir(file.Path)
+	// Ensure base dir exists for target. Change Windows-style slashes to Unix-style. Windows is ok with forward slashes, but filepath.Dir is not.
+	dir := filepath.Dir(strings.Replace(file.Path, `\`, `/`, -1))
 	perms, err := file.PermissionsAsInt()
 	if err != nil {
 		return fmt.Errorf("failed to parse permissions: %w", err)