Query Stripe API to find customers for each teamId

For each teamId in the usage report, query the Stripe API to find Stripe
Customer records that correspond to those teamIds.

Author: Andrew Farries

components/usage/pkg/controller/reconciler.go

@@ -10,6 +10,7 @@ import (
 	"fmt"
 	"github.com/gitpod-io/gitpod/common-go/log"
 	"github.com/gitpod-io/gitpod/usage/pkg/db"
+	"github.com/gitpod-io/gitpod/usage/pkg/stripe"
 	"github.com/google/uuid"
 	"gorm.io/gorm"
 	"io/ioutil"
@@ -125,9 +126,26 @@ func (u *UsageReconciler) ReconcileTimeRange(ctx context.Context, from, to time.
 	}
 	status.Report = report
 
+	submitUsageReport(status.Report)
+
 	return status, nil
 }
 
+func submitUsageReport(report []TeamUsage) {
+	var teamIdSet = make(map[string]bool)
+
+	// Convert the usage report into a set of teamIds occurring in the report
+	for _, usageEntry := range report {
+		teamIdSet[usageEntry.TeamID] = true
+	}
+	teamIds := make([]string, 0, len(teamIdSet))
+	for k := range teamIdSet {
+		teamIds = append(teamIds, k)
+	}
+
+	stripe.FindCustomersForTeamIds(teamIds)
+}
+
 func generateUsageReport(teams []teamWithWorkspaces, maxStopTime time.Time) ([]TeamUsage, error) {
 	var report []TeamUsage
 	for _, team := range teams {

components/usage/pkg/stripe/stripe.go

@@ -10,7 +10,9 @@ import (
 	"os"
 	"strings"
 
+	"github.com/gitpod-io/gitpod/common-go/log"
 	"github.com/stripe/stripe-go/v72"
+	"github.com/stripe/stripe-go/v72/customer"
 )
 
 type stripeKeys struct {
@@ -34,6 +36,21 @@ func Authenticate(apiKeyFile string) error {
 	return nil
 }
 
+// FindCustomersForTeamIds queries the stripe API to find all customers with a teamId in `teamIds`.
+func FindCustomersForTeamIds(teamIds []string) {
+	queries := queriesForCustomersWithTeamIds(teamIds)
+
+	for _, query := range queries {
+		log.Infof("about to make query %q", query)
+		params := &stripe.CustomerSearchParams{SearchParams: stripe.SearchParams{Query: query}}
+		iter := customer.Search(params)
+		for iter.Next() {
+			customer := iter.Customer()
+			log.Infof("found customer %q for teamId %q", customer.Name, customer.Metadata["teamId"])
+		}
+	}
+}
+
 // queriesForCustomersWithTeamIds constructs Stripe query strings to find the Stripe Customer for each teamId
 // It returns multiple queries, each being a big disjunction of subclauses so that we can process multiple teamIds in one query.
 // `clausesPerQuery` is a limit enforced by the Stripe API.