change init command

cmd/init.go

@@ -2,8 +2,10 @@
 package cmd
 
 import (
+	"fmt"
 	"log"
 
+	"github.com/ak1ra24/tn/internal/pkg/shell"
 	"github.com/spf13/cobra"
 )
 
@@ -12,9 +14,11 @@ var initCmd = &cobra.Command{
 	Use:   "init",
 	Short: "Generate template spec file",
 	Run: func(cmd *cobra.Command, args []string) {
-		if err := tnconfig.GenerateFile(cfgFile); err != nil {
+		tnConf, err := shell.GenerateFile()
+		if err != nil {
 			log.Fatal(err)
 		}
+		fmt.Println(tnConf)
 	},
 }
 

cmd/root.go

@@ -67,24 +67,23 @@ func initConfig() {
 	if cfgFile != "" {
 		// Use config file from the flag.
 		if !utils.Exists(cfgFile) {
-			confirmMessage := fmt.Sprintf("Do you create %s file", cfgFile)
-			isConfirmed := utils.Ask4confirm(confirmMessage)
-			if isConfirmed {
-				file, err := os.OpenFile(cfgFile, os.O_RDWR|os.O_CREATE, 0644)
-				if err != nil {
-					fmt.Println(err)
-					os.Exit(1)
-				}
-				if err := file.Close(); err != nil {
-					fmt.Println(err)
-					os.Exit(1)
-				}
-			} else {
-				err := fmt.Errorf("%s is not Found", cfgFile)
-				fmt.Println(err)
-				os.Exit(1)
-			}
-
+			// confirmMessage := fmt.Sprintf("Do you create %s file", cfgFile)
+			// isConfirmed := utils.Ask4confirm(confirmMessage)
+			// if isConfirmed {
+			// 	file, err := os.OpenFile(cfgFile, os.O_RDWR|os.O_CREATE, 0644)
+			// 	if err != nil {
+			// 		fmt.Println(err)
+			// 		os.Exit(1)
+			// 	}
+			// 	if err := file.Close(); err != nil {
+			// 		fmt.Println(err)
+			// 		os.Exit(1)
+			// 	}
+			// } else {
+			err := fmt.Errorf("%s is not Found", cfgFile)
+			fmt.Println(err)
+			os.Exit(1)
+			// }
 		}
 
 		viper.SetConfigFile(cfgFile)

internal/pkg/shell/shell.go

@@ -2,13 +2,12 @@ package shell
 
 import (
 	"fmt"
-	"io/ioutil"
 	"strings"
 
 	l "github.com/sirupsen/logrus"
+	"gopkg.in/yaml.v2"
 
 	"github.com/ak1ra24/tn/internal/pkg/utils"
-	"gopkg.in/yaml.v2"
 )
 
 var log = l.New()
@@ -185,7 +184,7 @@ func (tnconfig *Tn) Exec(nodeName string, Cmds []string) string {
 }
 
 // GenerateFile Generate tinet template config file
-func (tnconfig *Tn) GenerateFile(cfgFile string) error {
+func GenerateFile() (string, error) {
 	precmd := PreCmd{
 		Cmds: []Cmd{
 			Cmd{
@@ -259,7 +258,7 @@ func (tnconfig *Tn) GenerateFile(cfgFile string) error {
 		},
 	}
 
-	tnconfig = &Tn{
+	tnconfig := &Tn{
 		PreCmd:      []PreCmd{precmd},
 		PreInit:     []PreInit{preinit},
 		PostInit:    []PostInit{postinit},
@@ -272,15 +271,16 @@ func (tnconfig *Tn) GenerateFile(cfgFile string) error {
 
 	data, err := yaml.Marshal(tnconfig)
 	if err != nil {
-		return err
-	}
-
-	err = ioutil.WriteFile(cfgFile, data, 0644)
-	if err != nil {
-		return err
+		return "", err
 	}
-
-	return nil
+	//
+	// err = ioutil.WriteFile(cfgFile, data, 0644)
+	// if err != nil {
+	// 	return err
+	// }
+
+	// return nil
+	return string(data), nil
 }
 
 // DockerPs Show docker ps

internal/pkg/shell/shell_test.go

@@ -238,35 +238,6 @@ func TestTn_Exec(t *testing.T) {
 	}
 }
 
-func TestTn_GenerateFile(t *testing.T) {
-	type args struct {
-		cfgFile string
-	}
-	tests := []struct {
-		name     string
-		tnconfig *Tn
-		args     args
-		wantErr  bool
-	}{
-		// TODO: Add test cases.
-		// {
-		// 	name:     "tinet config template generate",
-		// 	tnconfig: &Tn{},
-		// 	args: args{
-		// 		cfgFile: "spec.yaml",
-		// 	},
-		// 	wantErr: true,
-		// },
-	}
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			if err := tt.tnconfig.GenerateFile(tt.args.cfgFile); (err != nil) != tt.wantErr {
-				t.Errorf("Tn.GenerateFile() error = %v, wantErr %v", err, tt.wantErr)
-			}
-		})
-	}
-}
-
 func TestDockerPs(t *testing.T) {
 	type args struct {
 		all bool
@@ -878,3 +849,25 @@ func TestGetContainerPid(t *testing.T) {
 		})
 	}
 }
+
+func TestGenerateFile(t *testing.T) {
+	tests := []struct {
+		name    string
+		want    string
+		wantErr bool
+	}{
+		// TODO: Add test cases.
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := GenerateFile()
+			if (err != nil) != tt.wantErr {
+				t.Errorf("GenerateFile() error = %v, wantErr %v", err, tt.wantErr)
+				return
+			}
+			if got != tt.want {
+				t.Errorf("GenerateFile() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}