commerce/api/billing/setup_intents.go

package billing

import (
	"github.com/zap-proto/zip"

	"github.com/hanzoai/commerce/billing/engine"
	"github.com/hanzoai/commerce/datastore"
	"github.com/hanzoai/commerce/log"
	"github.com/hanzoai/commerce/middleware"
	"github.com/hanzoai/commerce/models/setupintent"
	"github.com/hanzoai/commerce/util/json/http"
)

type createSetupIntentRequest struct {
	CustomerId      string `json:"customerId"`
	PaymentMethodId string `json:"paymentMethodId,omitempty"`
	Usage           string `json:"usage,omitempty"`
}

// CreateSetupIntent creates a new setup intent for saving a payment method.
//
//	POST /v1/billing/setup-intents
func CreateSetupIntent(c *zip.Ctx) error {
	org := middleware.GetOrganization(c)
	db := datastore.New(org.Namespaced(c.Context()))

	var req createSetupIntentRequest
	if err := c.Bind(&req); err != nil {
		return http.Fail(c, 400, "invalid request body", err)
	}

	si, err := engine.CreateSetupIntent(db, engine.CreateSetupIntentParams{
		CustomerId:      req.CustomerId,
		PaymentMethodId: req.PaymentMethodId,
		Usage:           req.Usage,
	})
	if err != nil {
		log.Error("Failed to create setup intent: %v", err, c)
		return http.Fail(c, 400, err.Error(), err)
	}

	return c.JSON(201, setupIntentResponse(si))
}

// GetSetupIntent retrieves a setup intent by ID.
//
//	GET /v1/billing/setup-intents/:id
func GetSetupIntent(c *zip.Ctx) error {
	org := middleware.GetOrganization(c)
	db := datastore.New(org.Namespaced(c.Context()))

	si := setupintent.New(db)
	if err := si.GetById(c.Param("id")); err != nil {
		return http.Fail(c, 404, "setup intent not found", err)
	}

	return c.JSON(200, setupIntentResponse(si))
}

type confirmSetupIntentRequest struct {
	PaymentMethodId string `json:"paymentMethodId,omitempty"`
}

// ConfirmSetupIntent confirms a setup intent, saving the payment method.
//
//	POST /v1/billing/setup-intents/:id/confirm
func ConfirmSetupIntent(c *zip.Ctx) error {
	org := middleware.GetOrganization(c)
	db := datastore.New(org.Namespaced(c.Context()))

	si := setupintent.New(db)
	if err := si.GetById(c.Param("id")); err != nil {
		return http.Fail(c, 404, "setup intent not found", err)
	}

	var req confirmSetupIntentRequest
	_ = c.Bind(&req)

	if err := engine.ConfirmSetupIntent(c.Context(), db, si, req.PaymentMethodId, nil); err != nil {
		log.Error("Failed to confirm setup intent: %v", err, c)
		return http.Fail(c, 400, err.Error(), err)
	}

	return c.JSON(200, setupIntentResponse(si))
}

type cancelSetupIntentRequest struct {
	CancellationReason string `json:"cancellationReason,omitempty"`
}

// CancelSetupIntent cancels a setup intent.
//
//	POST /v1/billing/setup-intents/:id/cancel
func CancelSetupIntent(c *zip.Ctx) error {
	org := middleware.GetOrganization(c)
	db := datastore.New(org.Namespaced(c.Context()))

	si := setupintent.New(db)
	if err := si.GetById(c.Param("id")); err != nil {
		return http.Fail(c, 404, "setup intent not found", err)
	}

	var req cancelSetupIntentRequest
	_ = c.Bind(&req)

	if err := engine.CancelSetupIntent(si, req.CancellationReason); err != nil {
		log.Error("Failed to cancel setup intent: %v", err, c)
		return http.Fail(c, 400, err.Error(), err)
	}

	return c.JSON(200, setupIntentResponse(si))
}

func setupIntentResponse(si *setupintent.SetupIntent) map[string]interface{} {
	resp := map[string]interface{}{
		"id":         si.Id(),
		"customerId": si.CustomerId,
		"status":     si.Status,
		"usage":      si.Usage,
		"created":    si.Created,
	}
	if si.PaymentMethodId != "" {
		resp["paymentMethodId"] = si.PaymentMethodId
	}
	if si.ProviderRef != "" {
		resp["providerRef"] = si.ProviderRef
	}
	if si.LastError != "" {
		resp["lastError"] = si.LastError
	}
	if si.Metadata != nil {
		resp["metadata"] = si.Metadata
	}
	return resp
}