mirror of
https://github.com/hanzoai/commerce.git
synced 2026-10-30 15:08:42 +00:00
zeekay
1943f1c093
Multi-tenant commerce and billing: orders, subscriptions, invoices, plans, usage metering and a credit ledger. One Go binary. MIT OR Apache-2.0. Claude-Session: https://claude.ai/code/session_01SAtaDFm8VDiWrhYHNNtxMw
80 lines
2.1 KiB
Go
80 lines
2.1 KiB
Go
package billing
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/zap-proto/zip"
|
|
|
|
"github.com/hanzoai/commerce/datastore"
|
|
"github.com/hanzoai/commerce/middleware"
|
|
"github.com/hanzoai/commerce/models/paymentmethod"
|
|
"github.com/hanzoai/commerce/models/transaction/util"
|
|
"github.com/hanzoai/commerce/models/types/currency"
|
|
"github.com/hanzoai/commerce/util/json/http"
|
|
)
|
|
|
|
// GetBillingStatus returns a unified billing status for a user.
|
|
// Used by the bot gateway billing-gate to decide whether to allow LLM requests.
|
|
//
|
|
// GET /v1/billing/status?user=<userId>
|
|
//
|
|
// Response:
|
|
//
|
|
// {
|
|
// "user": "alice",
|
|
// "hasPaymentMethod": true,
|
|
// "creditBalance": 500, // cents
|
|
// "tier": "developer"
|
|
// }
|
|
func GetBillingStatus(c *zip.Ctx) error {
|
|
org := middleware.GetOrganization(c)
|
|
ctx := org.Namespaced(c.Context())
|
|
db := datastore.New(ctx)
|
|
|
|
user := strings.TrimSpace(c.Query("user"))
|
|
if user == "" {
|
|
// Fall back to the authenticated IAM user from context
|
|
iamEmail, _ := c.Locals("iam_email").(string)
|
|
iamUserID, _ := c.Locals("iam_user_id").(string)
|
|
if email := strings.TrimSpace(iamEmail); email != "" {
|
|
user = email
|
|
} else if sub := strings.TrimSpace(iamUserID); sub != "" {
|
|
user = sub
|
|
}
|
|
}
|
|
if user == "" {
|
|
return http.Fail(c, 400, "user query parameter is required", nil)
|
|
}
|
|
|
|
// 1. Check whether the user has at least one active payment method.
|
|
rootKey := db.NewKey("synckey", "", 1, nil)
|
|
iter := paymentmethod.Query(db).Ancestor(rootKey).
|
|
Filter("CustomerId=", user).
|
|
Limit(1).
|
|
Run()
|
|
|
|
hasPaymentMethod := false
|
|
probe := paymentmethod.New(db)
|
|
if _, err := iter.Next(probe); err == nil {
|
|
hasPaymentMethod = true
|
|
}
|
|
|
|
// 2. Get available credit balance (USD).
|
|
var creditBalance currency.Cents
|
|
datas, err := util.GetTransactionsByCurrency(ctx, user, "iam-user", currency.USD, org.TestMode())
|
|
if err == nil {
|
|
if data, ok := datas.Data[currency.USD]; ok {
|
|
avail := data.Balance - data.Holds
|
|
if avail > 0 {
|
|
creditBalance = avail
|
|
}
|
|
}
|
|
}
|
|
|
|
return c.JSON(200, map[string]any{
|
|
"user": user,
|
|
"hasPaymentMethod": hasPaymentMethod,
|
|
"creditBalance": creditBalance,
|
|
})
|
|
}
|