Add brev redeem command for coupon code redemption (#259)

* Add brev redeem command for coupon code redemption

- Add RedeemCouponCode method to organization store
- Create new redeem command with --org flag support
- Wire command into CLI command tree
- Include completion handler for org names
- Display success message with credits amount and duration

Co-Authored-By: Alec Fong <alecsanf@usc.edu>

* Fix string formatting in redeem output

---------

Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Co-authored-by: Alec Fong <alecsanf@usc.edu>
Co-authored-by: Colin McNaughton <cloin@users.noreply.github.com>

Author: 3 people

pkg/cmd/cmd.go

@@ -30,6 +30,7 @@ import (
 	"github.com/brevdev/brev-cli/pkg/cmd/profile"
 	"github.com/brevdev/brev-cli/pkg/cmd/proxy"
 	"github.com/brevdev/brev-cli/pkg/cmd/recreate"
+	"github.com/brevdev/brev-cli/pkg/cmd/redeem"
 	"github.com/brevdev/brev-cli/pkg/cmd/refresh"
 	"github.com/brevdev/brev-cli/pkg/cmd/reset"
 	"github.com/brevdev/brev-cli/pkg/cmd/runtasks"
@@ -238,6 +239,7 @@ func createCmdTree(cmd *cobra.Command, t *terminal.Terminal, loginCmdStore *stor
 	cmd.AddCommand(ls.NewCmdLs(t, loginCmdStore, noLoginCmdStore))
 	cmd.AddCommand(org.NewCmdOrg(t, loginCmdStore, noLoginCmdStore))
 	cmd.AddCommand(invite.NewCmdInvite(t, loginCmdStore, noLoginCmdStore))
+	cmd.AddCommand(redeem.NewCmdRedeem(t, loginCmdStore, noLoginCmdStore))
 	cmd.AddCommand(portforward.NewCmdPortForwardSSH(loginCmdStore, t))
 	cmd.AddCommand(login.NewCmdLogin(t, noLoginCmdStore, loginAuth))
 	cmd.AddCommand(logout.NewCmdLogout(loginAuth, noLoginCmdStore))

pkg/cmd/redeem/redeem.go

@@ -0,0 +1,108 @@
+package redeem
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/brevdev/brev-cli/pkg/cmd/cmderrors"
+	"github.com/brevdev/brev-cli/pkg/cmd/completions"
+	"github.com/brevdev/brev-cli/pkg/cmdcontext"
+	"github.com/brevdev/brev-cli/pkg/entity"
+	breverrors "github.com/brevdev/brev-cli/pkg/errors"
+	"github.com/brevdev/brev-cli/pkg/store"
+	"github.com/brevdev/brev-cli/pkg/terminal"
+
+	"github.com/spf13/cobra"
+)
+
+type RedeemStore interface {
+	GetActiveOrganizationOrDefault() (*entity.Organization, error)
+	GetOrganizations(options *store.GetOrganizationsOptions) ([]entity.Organization, error)
+	RedeemCouponCode(organizationID string, code string) (*store.RedeemCouponCodeResponse, error)
+	completions.CompletionStore
+}
+
+func NewCmdRedeem(t *terminal.Terminal, redeemStore RedeemStore, noRedeemStore RedeemStore) *cobra.Command {
+	var orgFlag string
+
+	cmd := &cobra.Command{
+		Annotations:           map[string]string{"housekeeping": ""},
+		Use:                   "redeem <code>",
+		DisableFlagsInUseLine: true,
+		Short:                 "Redeem a coupon code for credits",
+		Long:                  "Redeem a coupon code to add credits to your active organization",
+		Example: `
+  brev redeem ABC123XYZ
+  brev redeem ABC123XYZ --org myorg
+		`,
+		PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
+			err := cmdcontext.InvokeParentPersistentPreRun(cmd, args)
+			if err != nil {
+				return breverrors.WrapAndTrace(err)
+			}
+
+			return nil
+		},
+		Args: cmderrors.TransformToValidationError(cobra.ExactArgs(1)),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			err := RunRedeem(t, redeemStore, args[0], orgFlag)
+			if err != nil {
+				return breverrors.WrapAndTrace(err)
+			}
+			return nil
+		},
+	}
+
+	cmd.Flags().StringVarP(&orgFlag, "org", "o", "", "organization (will override active org)")
+	err := cmd.RegisterFlagCompletionFunc("org", completions.GetOrgsNameCompletionHandler(noRedeemStore, t))
+	if err != nil {
+		breverrors.GetDefaultErrorReporter().ReportError(breverrors.WrapAndTrace(err))
+		fmt.Print(breverrors.WrapAndTrace(err))
+	}
+
+	return cmd
+}
+
+func RunRedeem(t *terminal.Terminal, redeemStore RedeemStore, code string, orgFlag string) error {
+	startTime := time.Now()
+
+	var org *entity.Organization
+	if orgFlag != "" {
+		orgs, err := redeemStore.GetOrganizations(&store.GetOrganizationsOptions{Name: orgFlag})
+		if err != nil {
+			return breverrors.WrapAndTrace(err)
+		}
+		if len(orgs) == 0 {
+			return fmt.Errorf("no org found with name %s", orgFlag)
+		} else if len(orgs) > 1 {
+			return fmt.Errorf("more than one org found with name %s", orgFlag)
+		}
+
+		org = &orgs[0]
+	} else {
+		currOrg, err := redeemStore.GetActiveOrganizationOrDefault()
+		if err != nil {
+			return breverrors.WrapAndTrace(err)
+		}
+		if currOrg == nil {
+			return fmt.Errorf("no orgs exist")
+		}
+		org = currOrg
+	}
+
+	result, err := redeemStore.RedeemCouponCode(org.ID, code)
+	if err != nil {
+		return breverrors.WrapAndTrace(err)
+	}
+
+	duration := time.Since(startTime)
+
+	t.Vprint(t.Green(fmt.Sprintf("✓ Successfully redeemed coupon code: %s\n", code)))
+	if result.Data.Transaction.AmountUSD != "" {
+		t.Vprintf("  Credits added: $%s\n", result.Data.Transaction.AmountUSD)
+	}
+	t.Vprintf("  Organization: %s\n", org.Name)
+	t.Vprintf("  Duration: %v\n", duration.Round(time.Millisecond))
+
+	return nil
+}

pkg/store/organization.go

@@ -213,3 +213,37 @@ func GetDefaultOrNilOrg(orgs []entity.Organization) *entity.Organization {
 		return nil
 	}
 }
+
+type RedeemCouponCodeRequest struct {
+	Code string `json:"Code"`
+}
+
+type RedeemCouponCodeResponse struct {
+	Data struct {
+		Transaction struct {
+			AmountUSD string `json:"amount_usd"`
+		} `json:"transaction"`
+	} `json:"data"`
+}
+
+func (s AuthHTTPStore) RedeemCouponCode(organizationID string, code string) (*RedeemCouponCodeResponse, error) {
+	var result RedeemCouponCodeResponse
+	path := orgPath + "/" + organizationID + "/credits/code/redeem"
+	req := RedeemCouponCodeRequest{
+		Code: code,
+	}
+
+	res, err := s.authHTTPClient.restyClient.R().
+		SetHeader("Content-Type", "application/json").
+		SetResult(&result).
+		SetBody(req).
+		Post(path)
+	if err != nil {
+		return nil, breverrors.WrapAndTrace(err)
+	}
+	if res.IsError() {
+		return nil, NewHTTPResponseError(res)
+	}
+
+	return &result, nil
+}