fix/load_tree

cmd/main.go

@@ -100,7 +100,7 @@ func buildCommands(ritchieHomePath string) *cobra.Command {
 	formulaRunner := formula.NewRunner(ritchieHomePath, envResolvers, http.DefaultClient, treeManager)
 	formulaCreator := formula.NewCreator(ritchieHomePath, treeManager)
 
-	rootCmd := cmd.NewRootCmd(workspaceManager, loginManager, sessionValidator)
+	rootCmd := cmd.NewRootCmd(workspaceManager, loginManager, repoManager, sessionValidator)
 
 	// level 1
 	autocompleteCmd := cmd.NewAutocompleteCmd()

pkg/cmd/login.go

@@ -32,19 +32,16 @@ func (l loginCmd) RunFunc() CommandRunnerFunc {
 			return err
 		}
 
-		fmt.Println("Starting login...")
 		secret := security.Passcode(s)
 		if err := l.Login(secret); err != nil {
 			return err
 		}
-		fmt.Println("Login successful!")
 
-		fmt.Println("Loading repositories...")
 		if err := l.Load(); err != nil {
 			return err
 		}
-		fmt.Println("Done.")
 
+		fmt.Println("Session created successfully!")
 		return nil
 	}
 }

pkg/cmd/root.go

@@ -3,6 +3,7 @@ package cmd
 import (
 	"errors"
 	"fmt"
+	"github.com/ZupIT/ritchie-cli/pkg/formula"
 	"github.com/ZupIT/ritchie-cli/pkg/prompt"
 	"github.com/ZupIT/ritchie-cli/pkg/security"
 	"github.com/ZupIT/ritchie-cli/pkg/session"
@@ -42,12 +43,13 @@ var (
 type rootCmd struct {
 	workspaceManager workspace.Checker
 	loginManager     security.LoginManager
+	repoLoader       formula.RepoLoader
 	sessionValidator session.Validator
 }
 
 // NewRootCmd creates the root for all ritchie commands.
-func NewRootCmd(wm workspace.Checker, l security.LoginManager, sv session.Validator) *cobra.Command {
-	o := &rootCmd{wm, l, sv}
+func NewRootCmd(wm workspace.Checker, l security.LoginManager, r formula.RepoLoader, sv session.Validator) *cobra.Command {
+	o := &rootCmd{wm, l, r, sv}
 
 	return &cobra.Command{
 		Use:               cmdUse,
@@ -59,10 +61,6 @@ func NewRootCmd(wm workspace.Checker, l security.LoginManager, sv session.Valida
 	}
 }
 
-func version() string {
-	return fmt.Sprintf(versionMsg, Version, env.Edition, BuildDate, runtime.Version())
-}
-
 func (o *rootCmd) PreRunFunc() CommandRunnerFunc {
 	return func(cmd *cobra.Command, args []string) error {
 		if err := o.workspaceManager.Check(); err != nil {
@@ -94,6 +92,10 @@ func (o *rootCmd) checkSession(commandPath string) error {
 			return err
 		}
 
+		if err := o.repoLoader.Load(); err != nil {
+			return err
+		}
+
 		fmt.Println("Session created successfully!")
 		os.Exit(0)
 	}
@@ -120,3 +122,7 @@ func sessionPrompt() (security.Passcode, error) {
 
 	return security.Passcode(passcode), nil
 }
+
+func version() string {
+	return fmt.Sprintf(versionMsg, Version, env.Edition, BuildDate, runtime.Version())
+}