|
|
@@ -58,6 +58,52 @@ func (c *ListPlansHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
|
c.WriteResult(w, r, plan)
|
|
|
}
|
|
|
|
|
|
+// ListCreditsHandler is a handler for getting available credits
|
|
|
+type ListCreditsHandler struct {
|
|
|
+ handlers.PorterHandlerWriter
|
|
|
+}
|
|
|
+
|
|
|
+// NewListCreditsHandler will create a new ListCreditsHandler
|
|
|
+func NewListCreditsHandler(
|
|
|
+ config *config.Config,
|
|
|
+ writer shared.ResultWriter,
|
|
|
+) *ListCreditsHandler {
|
|
|
+ return &ListCreditsHandler{
|
|
|
+ PorterHandlerWriter: handlers.NewDefaultPorterHandler(config, nil, writer),
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func (c *ListCreditsHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
|
+ ctx, span := telemetry.NewSpan(r.Context(), "serve-list-credits")
|
|
|
+ defer span.End()
|
|
|
+
|
|
|
+ proj, _ := ctx.Value(types.ProjectScope).(*models.Project)
|
|
|
+
|
|
|
+ if !c.Config().BillingManager.MetronomeConfigLoaded || !proj.GetFeatureFlag(models.MetronomeEnabled, c.Config().LaunchDarklyClient) {
|
|
|
+ c.WriteResult(w, r, "")
|
|
|
+
|
|
|
+ telemetry.WithAttributes(span,
|
|
|
+ telemetry.AttributeKV{Key: "metronome-config-exists", Value: c.Config().BillingManager.MetronomeConfigLoaded},
|
|
|
+ telemetry.AttributeKV{Key: "metronome-enabled", Value: proj.GetFeatureFlag(models.MetronomeEnabled, c.Config().LaunchDarklyClient)},
|
|
|
+ )
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ credits, err := c.Config().BillingManager.MetronomeClient.ListCustomerCredits(ctx, proj.UsageID)
|
|
|
+ if err != nil {
|
|
|
+ err := telemetry.Error(ctx, span, err, "error listing credits")
|
|
|
+ c.HandleAPIError(w, r, apierrors.NewErrInternal(err))
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ telemetry.WithAttributes(span,
|
|
|
+ telemetry.AttributeKV{Key: "metronome-enabled", Value: true},
|
|
|
+ telemetry.AttributeKV{Key: "usage-id", Value: proj.UsageID},
|
|
|
+ )
|
|
|
+
|
|
|
+ c.WriteResult(w, r, credits)
|
|
|
+}
|
|
|
+
|
|
|
// ListCustomerUsageHandler returns customer usage aggregations like CPU and RAM hours.
|
|
|
type ListCustomerUsageHandler struct {
|
|
|
handlers.PorterHandlerReadWriter
|