refactoring list branch (#98)

define ListBranches as Refs method.

cmd/branch.go

@@ -9,7 +9,6 @@ import (
 	"path/filepath"
 
 	"github.com/JunNishimura/Goit/internal/store"
-	"github.com/fatih/color"
 	"github.com/spf13/cobra"
 )
 
@@ -18,16 +17,6 @@ var (
 	deleteOption string = ""
 )
 
-func listBranch(client *store.Client) {
-	for _, branch := range client.Refs.Heads {
-		if branch.Name == client.Head.Reference {
-			color.Green("* %s", branch.Name)
-		} else {
-			fmt.Println(branch.Name)
-		}
-	}
-}
-
 func renameBranch(client *store.Client, newName string) error {
 	// check if new name is not used for other branches
 	for _, branch := range client.Refs.Heads {
@@ -119,7 +108,7 @@ var branchCmd = &cobra.Command{
 
 		// list branches
 		if isList {
-			listBranch(client)
+			client.Refs.ListBranches(client.Head.Reference)
 		}
 
 		// rename current branch

internal/store/refs.go

@@ -7,6 +7,7 @@ import (
 	"sort"
 
 	"github.com/JunNishimura/Goit/internal/sha"
+	"github.com/fatih/color"
 )
 
 const (
@@ -80,6 +81,16 @@ func (r *Refs) GetBranch(name string) (*branch, error) {
 	return nil, fmt.Errorf("fail to find '%s' branch", name)
 }
 
+func (r *Refs) ListBranches(headBranchName string) {
+	for _, b := range r.Heads {
+		if b.Name == headBranchName {
+			color.Green("* %s", b.Name)
+		} else {
+			fmt.Println(b.Name)
+		}
+	}
+}
+
 func (r *Refs) AddBranch(rootGoitPath, newBranchName string, newBranchHash sha.SHA1) error {
 	// check if branch already exists
 	n := r.getBranchPos(newBranchName)