option.go

package swt

import (
	"time"

	"github.com/golang-jwt/jwt/v5"
)

type Option func(swt *SecureWebhookToken)

// WithSubject overrides the default sub claim.
func WithSubject(s string) Option {
	return func(swt *SecureWebhookToken) {
		swt.Claims().wc().Subject = s
	}
}

// WithExpiresAt overrides the default exp claim.
func WithExpiresAt(t time.Time) Option {
	return func(swt *SecureWebhookToken) {
		swt.Claims().wc().ExpiresAt = jwt.NewNumericDate(t)
	}
}

// WithNotBefore overrides the default nbf claim.
func WithNotBefore(t time.Time) Option {
	return func(swt *SecureWebhookToken) {
		swt.Claims().wc().NotBefore = jwt.NewNumericDate(t)
	}
}

// WithID overrides the jti claim with a given id. Should be a UUID or similar unique id.
func WithID(id string) Option {
	return func(swt *SecureWebhookToken) {
		swt.Claims().wc().ID = id
	}
}

// WithAudience overrides the aud claim.
// The Audience claim is optional and is nil by default.
// It can consist of zero or more recipients containing a StringOrURI value.
func WithAudience(ids ...string) Option {
	return func(swt *SecureWebhookToken) {
		swt.Claims().wc().Audience = ids
	}
}

// WithSigningMethod allows overriding the default signing method HS256.
// Draft-02 supports both symmetrical and asymmetrical signatures.
func WithSigningMethod(method jwt.SigningMethod) Option {
	return func(swt *SecureWebhookToken) {
		swt.token.Method = method
	}
}

// WithRetryCount sets the retry count for the webhook token.
// This is metadata that indicates how many times this webhook delivery has been attempted.
// The receiver can use this to:
// - Track delivery attempts
// - Apply different handling logic for retries (e.g., idempotency checks)
// - Implement exponential backoff strategies
// Note: This library does not automatically implement retry logic - this is informational only.
// Senders should increment this count when resending failed webhooks.
func WithRetryCount(rc uint) Option {
	return func(swt *SecureWebhookToken) {
		swt.Claims().wc().Webhook.RetryCount = rc
	}
}