Make run command use new Stripe client

Give an instance of the stripe client to the stripe billing controller.

Author: Andrew Farries

components/usage/cmd/run.go

@@ -47,11 +47,11 @@ func run() *cobra.Command {
 			var billingController controller.BillingController = &controller.NoOpBillingController{}
 
 			if apiKeyFile != "" {
-				err = stripe.Authenticate(apiKeyFile)
+				c, err := stripe.Authenticate(apiKeyFile)
 				if err != nil {
 					log.WithError(err).Fatal("Failed to initialize stripe client.")
 				}
-				billingController = &controller.StripeBillingController{}
+				billingController = controller.NewStripeBillingController(c)
 			}
 
 			ctrl, err := controller.New(schedule, controller.NewUsageReconciler(conn, billingController))

components/usage/pkg/controller/billing.go

@@ -11,16 +11,23 @@ type BillingController interface {
 }
 
 type NoOpBillingController struct{}
-type StripeBillingController struct{}
 
 func (b *NoOpBillingController) Reconcile(report []TeamUsage) {}
 
+type StripeBillingController struct {
+	sc *stripe.Client
+}
+
+func NewStripeBillingController(sc *stripe.Client) *StripeBillingController {
+	return &StripeBillingController{sc: sc}
+}
+
 func (b *StripeBillingController) Reconcile(report []TeamUsage) {
 	// Convert the usage report to sum all entries for the same team.
 	var summedReport = make(map[string]int64)
 	for _, usageEntry := range report {
 		summedReport[usageEntry.TeamID] += usageEntry.WorkspaceSeconds
 	}
 
-	stripe.UpdateUsage(summedReport)
+	b.sc.UpdateUsage(summedReport)
 }