add dir persistent flag

Author: andfasano

cmd/node-joiner/main.go

@@ -1,8 +1,6 @@
 package main
 
 import (
-	"os"
-
 	"github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 
@@ -11,11 +9,6 @@ import (
 
 func main() {
 
-	wd, err := os.Getwd()
-	if err != nil {
-		logrus.Fatal(err)
-	}
-
 	nodesAddCmd := &cobra.Command{
 		Use:   "add-nodes",
 		Short: "Generates an ISO that could be used to boot the configured nodes to let them join an existing cluster",
@@ -24,22 +17,27 @@ func main() {
 			if err != nil {
 				return err
 			}
-			return nodejoiner.NewAddNodesCommand(wd, kubeConfig)
+			dir, err := cmd.Flags().GetString("dir")
+			if err != nil {
+				return err
+			}
+			return nodejoiner.NewAddNodesCommand(dir, kubeConfig)
 		},
 	}
 
 	nodesMonitorCmd := &cobra.Command{
 		Use:   "monitor-add-nodes",
 		Short: "Monitors the configured nodes while they are joining an existing cluster",
 		RunE: func(cmd *cobra.Command, args []string) error {
-			return nodejoiner.NewMonitorAddNodesCommand(wd)
+			return nodejoiner.NewMonitorAddNodesCommand("")
 		},
 	}
 
 	rootCmd := &cobra.Command{
 		Use: "node-joiner",
 	}
 	rootCmd.PersistentFlags().String("kubeconfig", "", "Path to the kubeconfig file.")
+	rootCmd.PersistentFlags().String("dir", ".", "assets directory")
 
 	rootCmd.AddCommand(nodesAddCmd)
 	rootCmd.AddCommand(nodesMonitorCmd)