snapshot of https://github.com/ory/kratos.git (master, 2026-09-15; blobs > 131072B omitted)

courier/courier.go

@@ -0,0 +1,108 @@
+// Copyright © 2023 Ory Corp
+// SPDX-License-Identifier: Apache-2.0
+
+package courier
+
+import (
+	"context"
+	"time"
+
+	"github.com/cenkalti/backoff"
+	"github.com/gofrs/uuid"
+	"github.com/pkg/errors"
+
+	"github.com/ory/x/httpx"
+	"github.com/ory/x/jsonnetsecure"
+	"github.com/ory/x/logrusx"
+	"github.com/ory/x/otelx"
+
+	"github.com/ory/kratos/courier/template"
+	"github.com/ory/kratos/driver/config"
+)
+
+type (
+	Dependencies interface {
+		PersistenceProvider
+		otelx.Provider
+		logrusx.Provider
+		ConfigProvider
+		httpx.ClientProvider
+		jsonnetsecure.VMProvider
+	}
+
+	Courier interface {
+		Work(ctx context.Context) error
+		QueueEmail(ctx context.Context, t EmailTemplate) (uuid.UUID, error)
+		QueueSMS(ctx context.Context, t SMSTemplate) (uuid.UUID, error)
+		DispatchQueue(ctx context.Context) error
+		DispatchMessage(ctx context.Context, msg Message) error
+		UseBackoff(b backoff.BackOff)
+		FailOnDispatchError()
+	}
+
+	Provider interface {
+		Courier(ctx context.Context) (Courier, error)
+	}
+
+	ConfigProvider interface {
+		CourierConfig() config.CourierConfigs
+	}
+
+	courier struct {
+		deps                        Dependencies
+		failOnDispatchError         bool
+		backoff                     backoff.BackOff
+		newEmailTemplateFromMessage func(d template.Dependencies, msg Message) (EmailTemplate, error)
+	}
+)
+
+func NewCourier(ctx context.Context, deps Dependencies) (Courier, error) {
+	return NewCourierWithCustomTemplates(ctx, deps, NewEmailTemplateFromMessage)
+}
+
+func NewCourierWithCustomTemplates(_ context.Context, deps Dependencies, newEmailTemplateFromMessage func(d template.Dependencies, msg Message) (EmailTemplate, error)) (Courier, error) {
+	return &courier{
+		deps:                        deps,
+		backoff:                     backoff.NewExponentialBackOff(),
+		newEmailTemplateFromMessage: newEmailTemplateFromMessage,
+	}, nil
+}
+
+func (c *courier) FailOnDispatchError() {
+	c.failOnDispatchError = true
+}
+
+func (c *courier) Work(ctx context.Context) error {
+	errChan := make(chan error)
+	defer close(errChan)
+
+	go c.watchMessages(ctx, errChan)
+
+	select {
+	case <-ctx.Done():
+		if errors.Is(ctx.Err(), context.Canceled) {
+			return nil
+		}
+		return ctx.Err()
+	case err := <-errChan:
+		return errors.WithStack(err)
+	}
+}
+
+func (c *courier) UseBackoff(b backoff.BackOff) {
+	c.backoff = b
+}
+
+func (c *courier) watchMessages(ctx context.Context, errChan chan error) {
+	wait := c.deps.CourierConfig().CourierWorkerPullWait(ctx)
+	c.backoff.Reset()
+	for {
+		if err := backoff.Retry(func() error {
+			return c.DispatchQueue(ctx)
+		}, c.backoff); err != nil {
+			errChan <- errors.WithStack(err)
+			return
+		}
+		time.Sleep(wait)
+	}
+}