Merge edit and edit-env

cmd/edit.go

@@ -8,9 +8,9 @@ import (
 
 // editCmd represents the edit command
 var editCmd = &cobra.Command{
-	Use:   "edit workspace",
+	Use:   "edit workspace [environment]",
 	Short: "Edit a workspace",
-	Args:  cobra.ExactArgs(1),
+	Args:  cobra.RangeArgs(1, 2),
 	ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
 		w, err := workspace.NewWorkspaceManager()
 		if err != nil {
@@ -24,6 +24,12 @@ var editCmd = &cobra.Command{
 				return []string{}, cobra.ShellCompDirectiveNoFileComp
 			}
 			return workspaces, cobra.ShellCompDirectiveNoFileComp
+		case 1:
+			envs, err := c.FindEnvs(args[0], toComplete)
+			if err != nil {
+				return []string{}, cobra.ShellCompDirectiveNoFileComp
+			}
+			return envs, cobra.ShellCompDirectiveNoFileComp
 		}
 		return []string{}, cobra.ShellCompDirectiveNoFileComp
 	},
@@ -32,7 +38,13 @@ var editCmd = &cobra.Command{
 		if err != nil {
 			return err
 		}
-		return w.Edit(args[0])
+		switch len(args) {
+		case 1:
+			w.Edit(args[0])
+		case 2:
+			w.EditEnv(args[0], args[1])
+		}
+		return nil
 	},
 }
 

cmd/internal/completion/completion.go

@@ -41,3 +41,17 @@ func (c Completion) FindCommands(workspace string, toComplete string) ([]string,
 	}
 	return cs, nil
 }
+
+func (c Completion) FindEnvs(workspace string, toComplete string) ([]string, error) {
+	w, err := c.workspaceManager.Get(workspace)
+	if err != nil {
+		return []string{}, err
+	}
+	envs := []string{}
+	for _, env := range w.Envs {
+		if strings.HasPrefix(env, toComplete) {
+			envs = append(envs, env)
+		}
+	}
+	return envs, nil
+}