Modified to support Prod and Dev Mode

[ealt-edge.git] / ocd / cli / ealt / cmd / init / all.go

diff --git a/ocd/cli/ealt/cmd/init/all.go b/ocd/cli/ealt/cmd/init/all.go
index 915eb5e..061c629 100644 (file)
--- a/ocd/cli/ealt/cmd/init/all.go
+++ b/ocd/cli/ealt/cmd/init/all.go
@@ -17,6 +17,7 @@ package init
 
 import (
        setup "ealt/cmd/setup"
+       "strings"
 
        "github.com/spf13/cobra"
 )
@@ -28,12 +29,21 @@ func NewAllCommand() *cobra.Command {
                Short: "Install Complete EALT Deployment Environment",
                Long:  `Install Complete EALT Deployment Environment`,
                RunE: func(cmd *cobra.Command, args []string) error {
-                       err := setup.EaltInstall("all")
+                       setupModeFlag := strings.ToLower(cmd.Flag("mode").Value.String())
+                       var err error
+                       if setupModeFlag == "dev" {
+                               err = setup.EaltInstall("all")
+                       } else if setupModeFlag == "prod" {
+                               err = setup.EaltInstall("secure")
+                       }
                        if err != nil {
                                return err
                        }
                        return nil
                },
        }
+       cmd.Flags().StringP("mode", "m", "dev", "Deployment Mode")
+       //cmd.MarkFlagRequired("mode")
+
        return cmd
 }