[stripe] Handle invoice finalization

Author: easyCZ

components/usage/pkg/apiv1/billing.go

@@ -56,6 +56,20 @@ func (s *BillingService) UpdateInvoices(ctx context.Context, in *v1.UpdateInvoic
 func (s *BillingService) FinalizeInvoice(ctx context.Context, in *v1.FinalizeInvoiceRequest) (*v1.FinalizeInvoiceResponse, error) {
 	log.Infof("Finalizing invoice for invoice %q", in.GetInvoiceId())
 
+	if in.GetInvoiceId() == "" {
+		return nil, status.Errorf(codes.InvalidArgument, "Missing InvoiceID")
+	}
+
+	invoiceID := in.GetInvoiceId()
+
+	invoice, err := s.stripeClient.GetInvoice(ctx, invoiceID)
+	if err != nil {
+		return nil, status.Errorf(codes.NotFound, "Failed to get invoice with ID %s: %s", invoiceID, err.Error())
+	}
+
+	reportID := invoice.Metadata["reportId"]
+	fmt.Println(reportID)
+
 	return &v1.FinalizeInvoiceResponse{}, nil
 }
 

components/usage/pkg/stripe/stripe.go

@@ -7,6 +7,7 @@ package stripe
 import (
 	"context"
 	"encoding/json"
+	"errors"
 	"fmt"
 	"os"
 	"strings"
@@ -16,6 +17,10 @@ import (
 	"github.com/stripe/stripe-go/v72/client"
 )
 
+var (
+	NotFoundError = errors.New("not found")
+)
+
 type Client struct {
 	sc *client.API
 }
@@ -205,6 +210,22 @@ func (c *Client) GetUpcomingInvoice(ctx context.Context, customerID string) (*In
 	}, nil
 }
 
+func (c *Client) GetInvoice(ctx context.Context, invoiceID string) (*stripe.Invoice, error) {
+	if invoiceID == "" {
+		return nil, fmt.Errorf("no invoice ID specified")
+	}
+
+	invoice, err := c.sc.Invoices.Get(invoiceID, &stripe.InvoiceParams{
+		Params: stripe.Params{
+			Context: ctx,
+		},
+	})
+	if err != nil {
+		return nil, fmt.Errorf("failed to get invoice %s: %w", invoiceID, err)
+	}
+	return invoice, nil
+}
+
 // 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.