config: make sftp could be configured dynamically

- add `harvester.install.enableSftp` to config `sftp` dynamically Signed-off-by: Vicente Cheng <[email protected]>
harvester · Dec 8, 2023 · 4c75112 · 4c75112
1 parent 35f62da
commit 4c75112
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/pkg/config/config.go b/pkg/config/config.go
@@ -147,6 +147,7 @@ type Install struct {
 	TTY       string `json:"tty,omitempty"`
 	ForceGPT  bool   `json:"forceGpt,omitempty"`
 	Role      string `json:"role,omitempty"`
+	EnableSFTP bool   `json:"enableSftp,omitempty"`
 
 	// Following options are not cOS installer flag
 	ForceMBR bool   `json:"forceMbr,omitempty"`

diff --git a/pkg/config/cos.go b/pkg/config/cos.go
@@ -205,6 +205,12 @@ func ConvertToCOS(config *HarvesterConfig) (*yipSchema.YipConfig, error) {
 		},
 	}
 
+	// Add SFTP related config to after install stage
+	if config.Install.EnableSFTP {
+		config.OS.AfterInstallChrootCommands = append(config.OS.AfterInstallChrootCommands, "mkdir -p /etc/ssh/sshd_config.d")
+		config.OS.AfterInstallChrootCommands = append(config.OS.AfterInstallChrootCommands, "echo 'Subsystem	sftp	/usr/lib/ssh/sftp-server' > /etc/ssh/sshd_config.d/sftp.conf")
+	}
+
 	// Add after-install-chroot stage
 	if len(config.OS.AfterInstallChrootCommands) > 0 {
 		afterInstallChroot := yipSchema.Stage{}