X-Git-Url: https://gerrit.akraino.org/r/gitweb?a=blobdiff_plain;f=ocd%2Fcli%2Fealt%2Fcmd%2Finit%2Fedge.go;h=3b16f9380d9a1eae38e6e6c389941efd9af910a2;hb=24126aeee45624fe8cc31ec8017e57eb2594391b;hp=8cce554ce351c55b53af027b42fed0fceef5955d;hpb=3a7f37fe5f9c475eb38829bba26270fa6903ff52;p=ealt-edge.git

diff --git a/ocd/cli/ealt/cmd/init/edge.go b/ocd/cli/ealt/cmd/init/edge.go
index 8cce554..3b16f93 100644
--- a/ocd/cli/ealt/cmd/init/edge.go
+++ b/ocd/cli/ealt/cmd/init/edge.go
@@ -17,6 +17,7 @@ package init
 
 import (
 	"fmt"
+	"strings"
 
 	setup "ealt/cmd/setup"
 
@@ -30,12 +31,19 @@ func NewEdgeCommand() *cobra.Command {
 		Long:  `Command to Install Edge Node only For Example : ealt init edge`,
 		RunE: func(cmd *cobra.Command, args []string) error {
 			fmt.Println("Installation of Edge components")
-			err := setup.EaltInstall("edge")
+			setupModeFlag := strings.ToLower(cmd.Flag("mode").Value.String())
+			var err error
+			if setupModeFlag == "dev" {
+				err = setup.EaltInstall("edge")
+			} else if setupModeFlag == "prod" {
+				err = setup.EaltInstall("ssledge")
+			}
 			if err != nil {
 				return err
 			}
 			return nil
 		},
 	}
+	cmd.Flags().StringP("mode", "m", "dev", "Deployment Mode")
 	return cmd
 }