feat: if docker is not available install it (#668) (#669)

(cherry picked from commit 374dd81)

Co-authored-by: Tanmoy Sarkar <[email protected]>

Author: mergify[bot]

main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"fmt"
 	"github.com/fatih/color"
 	"github.com/moby/sys/user"
 	"github.com/swiftwave-org/swiftwave/swiftwave_service/cmd"
@@ -45,9 +46,54 @@ func main() {
 	// management node also needs docker for running postgres or registry at-least
 	_, err = exec.LookPath("docker")
 	if err != nil {
-		color.Red("Docker is not installed. Aborting.")
-		os.Exit(1)
+		color.Red("Docker is not installed.")
+		isDockerInstalled := false
+		color.Blue("Run `curl -fsSL get.docker.com | bash -` to install docker.")
+		color.Blue("Do you want to install docker now? (y/n)")
+		fmt.Print("> ")
+		var response string
+		_, err := fmt.Scanln(&response)
+		if err != nil {
+			color.Red("Error reading response. Aborting.")
+			os.Exit(1)
+		}
+		if strings.Compare(response, "y") == 0 || strings.Compare(response, "Y") == 0 {
+			color.Blue("Installing docker...")
+			// install docker
+			err = runCommand(exec.Command("bash", "-c", "curl -fsSL get.docker.com | bash -"))
+			if err != nil {
+				color.Red("Error installing docker. Aborting.")
+				os.Exit(1)
+			}
+			// enable docker service
+			err = runCommand(exec.Command("systemctl", "enable", "docker"))
+			if err != nil {
+				color.Red("Error enabling docker. Aborting.")
+				os.Exit(1)
+			}
+			// start docker service
+			err = runCommand(exec.Command("systemctl", "start", "docker"))
+			if err != nil {
+				color.Red("Error starting docker. Aborting.")
+				os.Exit(1)
+			}
+			isDockerInstalled = true
+		}
+		if !isDockerInstalled {
+			color.Red("Docker is not installed. Aborting.")
+			os.Exit(1)
+		} else {
+			color.Green("Docker is installed.")
+		}
 	}
 	// Start the command line interface
 	cmd.Execute()
 }
+
+// private function
+func runCommand(command *exec.Cmd) error {
+	command.Stdout = os.Stdout
+	command.Stderr = os.Stderr
+	command.Stdin = os.Stdin
+	return command.Run()
+}