select first context when there is only one (#28)

Signed-off-by: l1b0k <libokang.dev@gmail.com>
sunny0826 · Jan 4, 2021 · af34722 · af34722
1 parent 25a00c0
commit af34722
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/cmd/add.go b/cmd/add.go
@@ -46,6 +46,11 @@ func (ac *AddCommand) runAdd(cmd *cobra.Command, args []string) error {
 		return err
 	}
 	outConfig := appendConfig(oldConfig, newConfig)
+	if len(outConfig.Contexts) == 1 {
+		for k := range outConfig.Contexts {
+			outConfig.CurrentContext = k
+		}
+	}
 	cover := BoolUI(fmt.Sprintf("Are you sure you want to add 「%s」 to the 「%s」context?", newName, cfgFile))
 	confirm, err := strconv.ParseBool(cover)
 	if err != nil {