Refactor build

pkg/install/install.go

@@ -0,0 +1,104 @@
+package install
+
+import (
+	"io/ioutil"
+	"os"
+	"os/exec"
+
+	"github.com/rancher/os/pkg/config"
+	"github.com/rancher/os/pkg/questions"
+	"sigs.k8s.io/yaml"
+)
+
+func Run(automatic bool) error {
+	cfg, err := config.ReadConfig()
+	if err != nil {
+		return err
+	}
+
+	if automatic && !cfg.Elemental.Install.Automatic {
+		return nil
+	} else if automatic {
+		cfg.Elemental.Install.Silent = true
+	}
+
+	err = Ask(&cfg)
+	if err != nil {
+		return err
+	}
+
+	tempFile, err := ioutil.TempFile("", "elemental-install")
+	if err != nil {
+		return err
+	}
+	if err := tempFile.Close(); err != nil {
+		return err
+	}
+
+	return runInstall(cfg, tempFile.Name())
+}
+
+func runInstall(cfg config.Config, output string) error {
+	installBytes, err := config.PrintInstall(cfg)
+	if err != nil {
+		return err
+	}
+
+	if !cfg.Elemental.Install.Silent {
+		val, err := questions.PromptBool("\nConfiguration\n"+"-------------\n\n"+
+			string(installBytes)+
+			"\nYour disk will be formatted and installed with the above configuration.\nContinue?", false)
+		if err != nil || !val {
+			return err
+		}
+	}
+
+	if cfg.Elemental.Install.ConfigURL == "" {
+		yip := config.YipConfig{
+			Rancherd: config.Rancherd{
+				Server: cfg.Elemental.Install.ServerURL,
+				Token:  cfg.Elemental.Install.Token,
+			},
+		}
+		if cfg.Elemental.Install.ServerURL == "" {
+			yip.Rancherd.Role = "cluster-init"
+		} else {
+			yip.Rancherd.Role = "agent"
+		}
+		if cfg.Elemental.Install.Password != "" || len(cfg.SSHAuthorizedKeys) > 0 {
+			yip.Stages = map[string][]config.Stage{
+				"initramfs": {{
+					Users: map[string]config.User{
+						"root": {
+							Name:              "root",
+							PasswordHash:      cfg.Elemental.Install.Password,
+							SSHAuthorizedKeys: cfg.SSHAuthorizedKeys,
+						},
+					}},
+				}}
+			cfg.Elemental.Install.Password = ""
+		}
+
+		data, err := yaml.Marshal(yip)
+		if err != nil {
+			return err
+		}
+
+		if err := ioutil.WriteFile(output+".yip", data, 0600); err != nil {
+			return err
+		}
+		cfg.Elemental.Install.ConfigURL = output + ".yip"
+	}
+
+	ev, err := config.ToEnv(cfg)
+	if err != nil {
+		return err
+	}
+
+	cmd := exec.Command("cos-installer")
+	cmd.Env = append(os.Environ(), ev...)
+	cmd.Stdout = os.Stdout
+	cmd.Stdin = os.Stdin
+	cmd.Stderr = os.Stderr
+	return cmd.Run()
+}