diff --git a/pkg/kt/command/config/save.go b/pkg/kt/command/config/save.go index 1e9f2f35..46ab3e05 100644 --- a/pkg/kt/command/config/save.go +++ b/pkg/kt/command/config/save.go @@ -2,23 +2,25 @@ package config import ( "fmt" - "github.com/alibaba/kt-connect/pkg/kt/util" - "github.com/rs/zerolog/log" - "github.com/spf13/cobra" "io/ioutil" "os" "strings" + + "github.com/alibaba/kt-connect/pkg/kt/util" + "github.com/rs/zerolog/log" + "github.com/spf13/cobra" ) func SaveProfile(args []string) error { if len(args) != 1 { return fmt.Errorf("must specifiy a profile name") } - profile := profileFile(args[0]) + profile := args[0] + absoluteProfile := profileFile(profile) if !profileNamePattern.MatchString(profile) { return fmt.Errorf("invalid profile name, must only contains letter, number, underline, hyphen or dot") } - if _, err := os.Stat(profile); err == nil { + if _, err := os.Stat(absoluteProfile); err == nil { var answer string fmt.Printf("Profile '%s' already exists, overwrite ? (Y/N) ", args[0]) _, err = fmt.Scanln(&answer) @@ -30,7 +32,7 @@ func SaveProfile(args []string) error { if err != nil { return fmt.Errorf("unable to read config file: %s", err) } - err = ioutil.WriteFile(profile, bytesRead, 0644) + err = ioutil.WriteFile(absoluteProfile, bytesRead, 0644) if err != nil { return fmt.Errorf("unable to create profile file: %s", err) } diff --git a/pkg/kt/command/config/save_test.go b/pkg/kt/command/config/save_test.go new file mode 100644 index 00000000..840adbf4 --- /dev/null +++ b/pkg/kt/command/config/save_test.go @@ -0,0 +1,11 @@ +package config + +import ( + "fmt" + "testing" +) + +func TestSaveProfile(t *testing.T) { + err := SaveProfile([]string{"test"}) + fmt.Println(err) +}