add autostop cmd

Author: r33drichards

pkg/cmd/autostop/autostop.go

@@ -1,6 +1,9 @@
 package autostop
 
 import (
+	"github.com/hashicorp/go-multierror"
+	"github.com/samber/lo"
+	"github.com/samber/mo"
 	"github.com/spf13/cobra"
 
 	"github.com/brevdev/brev-cli/pkg/entity"
@@ -22,24 +25,23 @@ func NewCmdautostop(t *terminal.Terminal, store autostopStore) *cobra.Command {
 		Long:                  long,
 		Example:               example,
 		RunE: func(cmd *cobra.Command, args []string) error {
-			err := Runautostop(
-				runAutostopArgs{
-					t:     t,
-					args:  args,
-					store: store,
-				},
-			)
-			if err != nil {
-				return breverrors.WrapAndTrace(err)
-			}
-			return nil
+			return breverrors.WrapAndTrace(
+				Runautostop(
+					runAutostopArgs{
+						t:     t,
+						args:  args,
+						store: store,
+					},
+				))
 		},
 	}
 	return cmd
 }
 
 type autostopStore interface {
 	AutoStopWorkspace(workspaceID string) (*entity.Workspace, error)
+	GetActiveOrganizationOrDefault() (*entity.Organization, error)
+	GetWorkspaceByNameOrID(orgID string, nameOrID string) ([]entity.Workspace, error)
 }
 
 type runAutostopArgs struct {
@@ -49,6 +51,28 @@ type runAutostopArgs struct {
 }
 
 func Runautostop(args runAutostopArgs) error {
-	_, err := args.store.AutoStopWorkspace(args.args[0])
-	return breverrors.WrapAndTrace(err)
+	org, err := args.store.GetActiveOrganizationOrDefault()
+	if err != nil {
+		return breverrors.WrapAndTrace(err)
+	}
+	envs, err := args.store.GetWorkspaceByNameOrID(org.ID, args.args[0])
+	if err != nil {
+		return breverrors.WrapAndTrace(err)
+	}
+
+	asResults := lo.Map(
+		envs,
+		func(env entity.Workspace, _ int) mo.Result[*entity.Workspace] {
+			return mo.TupleToResult(args.store.AutoStopWorkspace(env.ID))
+		},
+	)
+	return breverrors.WrapAndTrace(
+		lo.Reduce(
+			asResults,
+			func(acc error, res mo.Result[*entity.Workspace], _ int) error {
+				return multierror.Append(acc, res.Error())
+			},
+			nil,
+		),
+	)
 }