notification/httpdriver/mapping.go

package httpdriver

import (
	"fmt"

	"github.com/samber/lo"

	"github.com/openmeterio/openmeter/api"
	"github.com/openmeterio/openmeter/openmeter/notification"
	"github.com/openmeterio/openmeter/pkg/convert"
	"github.com/openmeterio/openmeter/pkg/models"
)

func FromChannel(c notification.Channel) (api.NotificationChannel, error) {
	var channel api.NotificationChannel

	switch c.Type {
	case notification.ChannelTypeWebhook:
		channel = FromChannelWebhook(c)
	default:
		return channel, models.NewGenericValidationError(fmt.Errorf("invalid channel type: %s", c.Type))
	}

	return channel, nil
}

func FromChannelWebhook(c notification.Channel) api.NotificationChannelWebhook {
	return api.NotificationChannelWebhook{
		CreatedAt: c.CreatedAt,
		CustomHeaders: convert.SafeDeRef(&c.Config.WebHook.CustomHeaders, func(m map[string]string) *map[string]string {
			if len(m) > 0 {
				return &m
			}

			return nil
		}),
		Disabled:      lo.ToPtr(c.Disabled),
		Id:            c.ID,
		Name:          c.Name,
		SigningSecret: lo.ToPtr(c.Config.WebHook.SigningSecret),
		Type:          api.NotificationChannelWebhookTypeWEBHOOK,
		UpdatedAt:     c.UpdatedAt,
		Url:           c.Config.WebHook.URL,
		DeletedAt:     c.DeletedAt,
		Annotations:   lo.EmptyableToPtr(FromAnnotations(c.Annotations)),
		Metadata:      lo.EmptyableToPtr(FromMetadata(c.Metadata)),
	}
}

func FromAnnotations(a models.Annotations) api.Annotations {
	if a == nil {
		return nil
	}

	return api.Annotations(a)
}

func FromMetadata(m models.Metadata) api.Metadata {
	if m == nil {
		return nil
	}

	return api.Metadata(m)
}

func AsChannelWebhookCreateRequest(r api.NotificationChannelWebhookCreateRequest, namespace string) notification.CreateChannelInput {
	return notification.CreateChannelInput{
		NamespacedModel: models.NamespacedModel{
			Namespace: namespace,
		},
		Name:     r.Name,
		Type:     notification.ChannelType(r.Type),
		Disabled: lo.FromPtrOr(r.Disabled, notification.DefaultDisabled),
		Config: notification.ChannelConfig{
			ChannelConfigMeta: notification.ChannelConfigMeta{
				Type: notification.ChannelType(r.Type),
			},
			WebHook: notification.WebHookChannelConfig{
				CustomHeaders: lo.FromPtr(r.CustomHeaders),
				URL:           r.Url,
				SigningSecret: lo.FromPtr(r.SigningSecret),
			},
		},
		Metadata: lo.FromPtr(r.Metadata),
	}
}

func AsMetadata(m *api.Metadata) models.Metadata {
	if m == nil {
		return nil
	}

	return *m
}

func AsChannelWebhookUpdateRequest(r api.NotificationChannelWebhookCreateRequest, namespace, channelID string) notification.UpdateChannelInput {
	return notification.UpdateChannelInput{
		NamespacedID: models.NamespacedID{
			Namespace: namespace,
			ID:        channelID,
		},
		Name:     r.Name,
		Type:     notification.ChannelType(r.Type),
		Disabled: lo.FromPtrOr(r.Disabled, notification.DefaultDisabled),
		Config: notification.ChannelConfig{
			ChannelConfigMeta: notification.ChannelConfigMeta{
				Type: notification.ChannelType(r.Type),
			},
			WebHook: notification.WebHookChannelConfig{
				CustomHeaders: lo.FromPtr(r.CustomHeaders),
				URL:           r.Url,
				SigningSecret: lo.FromPtr(r.SigningSecret),
			},
		},
		Metadata: AsMetadata(r.Metadata),
	}
}

func AsRuleBalanceThresholdCreateRequest(r api.NotificationRuleBalanceThresholdCreateRequest, namespace string) notification.CreateRuleInput {
	return notification.CreateRuleInput{
		NamespacedModel: models.NamespacedModel{
			Namespace: namespace,
		},
		Name:     r.Name,
		Type:     notification.EventType(r.Type),
		Disabled: lo.FromPtrOr(r.Disabled, notification.DefaultDisabled),
		Config: notification.RuleConfig{
			RuleConfigMeta: notification.RuleConfigMeta{
				Type: notification.EventType(r.Type),
			},
			BalanceThreshold: &notification.BalanceThresholdRuleConfig{
				Features:   lo.FromPtr(r.Features),
				Thresholds: r.Thresholds,
			},
		},
		Channels: r.Channels,
		Metadata: AsMetadata(r.Metadata),
	}
}

func AsRuleBalanceThresholdUpdateRequest(r api.NotificationRuleBalanceThresholdCreateRequest, namespace, ruleID string) notification.UpdateRuleInput {
	return notification.UpdateRuleInput{
		NamespacedID: models.NamespacedID{
			Namespace: namespace,
			ID:        ruleID,
		},
		Name:     r.Name,
		Type:     notification.EventType(r.Type),
		Disabled: lo.FromPtrOr(r.Disabled, notification.DefaultDisabled),
		Config: notification.RuleConfig{
			RuleConfigMeta: notification.RuleConfigMeta{
				Type: notification.EventType(r.Type),
			},
			BalanceThreshold: &notification.BalanceThresholdRuleConfig{
				Features:   lo.FromPtr(r.Features),
				Thresholds: r.Thresholds,
			},
		},
		Channels: r.Channels,
		Metadata: AsMetadata(r.Metadata),
	}
}

func AsRuleEntitlementResetCreateRequest(r api.NotificationRuleEntitlementResetCreateRequest, namespace string) notification.CreateRuleInput {
	return notification.CreateRuleInput{
		NamespacedModel: models.NamespacedModel{
			Namespace: namespace,
		},
		Name:     r.Name,
		Type:     notification.EventType(r.Type),
		Disabled: lo.FromPtrOr(r.Disabled, notification.DefaultDisabled),
		Config: notification.RuleConfig{
			RuleConfigMeta: notification.RuleConfigMeta{
				Type: notification.EventType(r.Type),
			},
			EntitlementReset: &notification.EntitlementResetRuleConfig{
				Features: lo.FromPtr(r.Features),
			},
		},
		Channels: r.Channels,
		Metadata: AsMetadata(r.Metadata),
	}
}

func AsRuleEntitlementResetUpdateRequest(r api.NotificationRuleEntitlementResetCreateRequest, namespace, ruleID string) notification.UpdateRuleInput {
	return notification.UpdateRuleInput{
		NamespacedID: models.NamespacedID{
			Namespace: namespace,
			ID:        ruleID,
		},
		Name:     r.Name,
		Type:     notification.EventType(r.Type),
		Disabled: lo.FromPtrOr(r.Disabled, notification.DefaultDisabled),
		Config: notification.RuleConfig{
			RuleConfigMeta: notification.RuleConfigMeta{
				Type: notification.EventType(r.Type),
			},
			EntitlementReset: &notification.EntitlementResetRuleConfig{
				Features: lo.FromPtr(r.Features),
			},
		},
		Channels: r.Channels,
		Metadata: AsMetadata(r.Metadata),
	}
}

func AsRuleInvoiceCreatedCreateRequest(r api.NotificationRuleInvoiceCreatedCreateRequest, namespace string) notification.CreateRuleInput {
	return notification.CreateRuleInput{
		NamespacedModel: models.NamespacedModel{
			Namespace: namespace,
		},
		Name:     r.Name,
		Type:     notification.EventType(r.Type),
		Disabled: lo.FromPtrOr(r.Disabled, notification.DefaultDisabled),
		Config: notification.RuleConfig{
			RuleConfigMeta: notification.RuleConfigMeta{
				Type: notification.EventType(r.Type),
			},
			Invoice: &notification.InvoiceRuleConfig{},
		},
		Channels: r.Channels,
		Metadata: AsMetadata(r.Metadata),
	}
}

func AsRuleInvoiceCreatedUpdateRequest(r api.NotificationRuleInvoiceCreatedCreateRequest, namespace, ruleID string) notification.UpdateRuleInput {
	return notification.UpdateRuleInput{
		NamespacedID: models.NamespacedID{
			Namespace: namespace,
			ID:        ruleID,
		},
		Type:     notification.EventType(r.Type),
		Name:     r.Name,
		Disabled: lo.FromPtrOr(r.Disabled, notification.DefaultDisabled),
		Config: notification.RuleConfig{
			RuleConfigMeta: notification.RuleConfigMeta{
				Type: notification.EventType(r.Type),
			},
			Invoice: &notification.InvoiceRuleConfig{},
		},
		Channels: r.Channels,
		Metadata: AsMetadata(r.Metadata),
	}
}

func AsRuleInvoiceUpdatedCreateRequest(r api.NotificationRuleInvoiceUpdatedCreateRequest, namespace string) notification.CreateRuleInput {
	return notification.CreateRuleInput{
		NamespacedModel: models.NamespacedModel{
			Namespace: namespace,
		},
		Name:     r.Name,
		Type:     notification.EventType(r.Type),
		Disabled: lo.FromPtrOr(r.Disabled, notification.DefaultDisabled),
		Config: notification.RuleConfig{
			RuleConfigMeta: notification.RuleConfigMeta{
				Type: notification.EventType(r.Type),
			},
			Invoice: &notification.InvoiceRuleConfig{},
		},
		Channels: r.Channels,
		Metadata: AsMetadata(r.Metadata),
	}
}

func AsRuleInvoiceUpdatedUpdateRequest(r api.NotificationRuleInvoiceUpdatedCreateRequest, namespace, ruleID string) notification.UpdateRuleInput {
	return notification.UpdateRuleInput{
		NamespacedID: models.NamespacedID{
			Namespace: namespace,
			ID:        ruleID,
		},
		Type:     notification.EventType(r.Type),
		Name:     r.Name,
		Disabled: lo.FromPtrOr(r.Disabled, notification.DefaultDisabled),
		Config: notification.RuleConfig{
			RuleConfigMeta: notification.RuleConfigMeta{
				Type: notification.EventType(r.Type),
			},
			Invoice: &notification.InvoiceRuleConfig{},
		},
		Channels: r.Channels,
		Metadata: AsMetadata(r.Metadata),
	}
}

func FromRule(r notification.Rule) (api.NotificationRule, error) {
	var (
		rule api.NotificationRule
		err  error
	)

	switch r.Type {
	case notification.EventTypeBalanceThreshold:
		err = rule.FromNotificationRuleBalanceThreshold(FromRuleBalanceThreshold(r))
		if err != nil {
			return rule, fmt.Errorf("failed to cast notification rule with type: %s: %w", r.Type, err)
		}
	case notification.EventTypeEntitlementReset:
		err = rule.FromNotificationRuleEntitlementReset(FromRuleEntitlementReset(r))
		if err != nil {
			return rule, fmt.Errorf("failed to cast notification rule with type: %s: %w", r.Type, err)
		}
	case notification.EventTypeInvoiceCreated:
		err = rule.FromNotificationRuleInvoiceCreated(FromRuleInvoiceCreated(r))
		if err != nil {
			return rule, fmt.Errorf("failed to cast notification rule with type: %s: %w", r.Type, err)
		}
	case notification.EventTypeInvoiceUpdated:
		err = rule.FromNotificationRuleInvoiceUpdated(FromRuleInvoiceUpdated(r))
		if err != nil {
			return rule, fmt.Errorf("failed to cast notification rule with type: %s: %w", r.Type, err)
		}
	default:
		return rule, models.NewGenericValidationError(fmt.Errorf("invalid rule type: %s", r.Type))
	}

	return rule, nil
}

func FromRuleBalanceThreshold(r notification.Rule) api.NotificationRuleBalanceThreshold {
	channels := make([]api.NotificationChannelMeta, 0, len(r.Channels))
	for _, channel := range r.Channels {
		channels = append(channels, api.NotificationChannelMeta{
			Id:   channel.ID,
			Type: api.NotificationChannelType(channel.Type),
		})
	}

	return api.NotificationRuleBalanceThreshold{
		Channels:  channels,
		CreatedAt: r.CreatedAt,
		Disabled:  lo.ToPtr(r.Disabled),
		Features: convert.SafeDeRef(&r.Config.BalanceThreshold.Features, func(featureIDs []string) *[]notification.FeatureMeta {
			var features []notification.FeatureMeta
			for _, id := range featureIDs {
				features = append(features, notification.FeatureMeta{
					Id: id,
				})
			}

			if len(features) == 0 {
				return nil
			}

			return &features
		}),
		Id:         r.ID,
		Name:       r.Name,
		Thresholds: r.Config.BalanceThreshold.Thresholds,
		Type:       api.NotificationRuleBalanceThresholdTypeEntitlementsBalanceThreshold,
		UpdatedAt:  r.UpdatedAt,
		DeletedAt:  r.DeletedAt,

		Annotations: lo.EmptyableToPtr(api.Annotations(r.Annotations)),
		Metadata:    lo.EmptyableToPtr(api.Metadata(r.Metadata)),
	}
}

func FromRuleEntitlementReset(r notification.Rule) api.NotificationRuleEntitlementReset {
	channels := make([]api.NotificationChannelMeta, 0, len(r.Channels))
	for _, channel := range r.Channels {
		channels = append(channels, api.NotificationChannelMeta{
			Id:   channel.ID,
			Type: api.NotificationChannelType(channel.Type),
		})
	}

	return api.NotificationRuleEntitlementReset{
		Channels:  channels,
		CreatedAt: r.CreatedAt,
		Disabled:  lo.ToPtr(r.Disabled),
		Features: convert.SafeDeRef(&r.Config.EntitlementReset.Features, func(featureIDs []string) *[]notification.FeatureMeta {
			var features []notification.FeatureMeta
			for _, id := range featureIDs {
				features = append(features, notification.FeatureMeta{
					Id: id,
				})
			}

			if len(features) == 0 {
				return nil
			}

			return &features
		}),
		Id:        r.ID,
		Name:      r.Name,
		Type:      api.NotificationRuleEntitlementResetTypeEntitlementsReset,
		UpdatedAt: r.UpdatedAt,
		DeletedAt: r.DeletedAt,

		Annotations: lo.EmptyableToPtr(api.Annotations(r.Annotations)),
		Metadata:    lo.EmptyableToPtr(api.Metadata(r.Metadata)),
	}
}

func FromRuleInvoiceCreated(r notification.Rule) api.NotificationRuleInvoiceCreated {
	channels := make([]api.NotificationChannelMeta, 0, len(r.Channels))
	for _, channel := range r.Channels {
		channels = append(channels, api.NotificationChannelMeta{
			Id:   channel.ID,
			Type: api.NotificationChannelType(channel.Type),
		})
	}

	return api.NotificationRuleInvoiceCreated{
		Channels:  channels,
		CreatedAt: r.CreatedAt,
		Disabled:  lo.ToPtr(r.Disabled),
		Id:        r.ID,
		Name:      r.Name,
		Type:      api.NotificationRuleInvoiceCreatedTypeInvoiceCreated,
		UpdatedAt: r.UpdatedAt,
		DeletedAt: r.DeletedAt,

		Annotations: lo.EmptyableToPtr(api.Annotations(r.Annotations)),
		Metadata:    lo.EmptyableToPtr(api.Metadata(r.Metadata)),
	}
}

func FromRuleInvoiceUpdated(r notification.Rule) api.NotificationRuleInvoiceUpdated {
	channels := make([]api.NotificationChannelMeta, 0, len(r.Channels))
	for _, channel := range r.Channels {
		channels = append(channels, api.NotificationChannelMeta{
			Id:   channel.ID,
			Type: api.NotificationChannelType(channel.Type),
		})
	}

	return api.NotificationRuleInvoiceUpdated{
		Channels:  channels,
		CreatedAt: r.CreatedAt,
		Disabled:  lo.ToPtr(r.Disabled),
		Id:        r.ID,
		Name:      r.Name,
		Type:      api.NotificationRuleInvoiceUpdatedTypeInvoiceUpdated,
		UpdatedAt: r.UpdatedAt,
		DeletedAt: r.DeletedAt,

		Annotations: lo.EmptyableToPtr(api.Annotations(r.Annotations)),
		Metadata:    lo.EmptyableToPtr(api.Metadata(r.Metadata)),
	}
}

func FromEvent(e notification.Event) (api.NotificationEvent, error) {
	var (
		err  error
		rule api.NotificationRule
	)

	rule, err = FromRule(e.Rule)
	if err != nil {
		return api.NotificationEvent{}, fmt.Errorf("failed to cast notification rule: %w", err)
	}

	// Populate ChannelMeta in EventDeliveryStatus from Even.Rule.Channels as we only store Channel.ID in database
	// for EventDeliveryStatus objects.
	channelsByID := make(map[string]notification.Channel, len(e.Rule.Channels))
	for _, channel := range e.Rule.Channels {
		channelsByID[channel.ID] = channel
	}

	deliveryStatuses := make([]api.NotificationEventDeliveryStatus, 0, len(e.DeliveryStatus))
	for _, deliveryStatus := range e.DeliveryStatus {
		status := api.NotificationEventDeliveryStatus{
			Annotations: lo.EmptyableToPtr(api.Annotations(deliveryStatus.Annotations)),
			Attempts:    AsEventDeliveryAttempts(deliveryStatus.Attempts),
			Channel: api.NotificationChannelMeta{
				Id: deliveryStatus.ChannelID,
			},
			NextAttempt: deliveryStatus.NextAttempt,
			Reason:      deliveryStatus.Reason,
			State:       api.NotificationEventDeliveryStatusState(deliveryStatus.State),
			UpdatedAt:   deliveryStatus.UpdatedAt,
		}
		if channel, ok := channelsByID[deliveryStatus.ChannelID]; ok {
			status.Channel = api.NotificationChannelMeta{
				Id:   deliveryStatus.ChannelID,
				Type: api.NotificationChannelType(channel.Type),
			}
		}

		deliveryStatuses = append(deliveryStatuses, status)
	}

	event := api.NotificationEvent{
		CreatedAt:      e.CreatedAt,
		DeliveryStatus: deliveryStatuses,
		Id:             e.ID,
		Rule:           rule,
		Annotations:    lo.EmptyableToPtr(api.Annotations(e.Annotations)),
	}

	event.Type, err = FromEventType(e.Type)
	if err != nil {
		return event, fmt.Errorf("failed to cast notification event type: %w", err)
	}

	switch e.Type {
	case notification.EventTypeBalanceThreshold:
		payload, err := FromEventAsBalanceThresholdPayload(e)
		if err != nil {
			return event, fmt.Errorf("failed to cast notification event payload: %w", err)
		}

		err = event.Payload.FromNotificationEventBalanceThresholdPayload(payload)
		if err != nil {
			return event, fmt.Errorf("failed to cast notification event payload: %w", err)
		}
	case notification.EventTypeEntitlementReset:
		payload, err := FromEventAsEntitlementResetPayload(e)
		if err != nil {
			return event, fmt.Errorf("failed to cast notification event payload: %w", err)
		}

		err = event.Payload.FromNotificationEventResetPayload(payload)
		if err != nil {
			return event, fmt.Errorf("failed to cast notification event payload: %w", err)
		}
	case notification.EventTypeInvoiceCreated:
		payload, err := FromEventAsInvoiceCreatedPayload(e)
		if err != nil {
			return event, fmt.Errorf("failed to cast notification event payload: %w", err)
		}

		err = event.Payload.FromNotificationEventInvoiceCreatedPayload(payload)
		if err != nil {
			return event, fmt.Errorf("failed to cast notification event payload: %w", err)
		}
	case notification.EventTypeInvoiceUpdated:
		payload, err := FromEventAsInvoiceUpdatedPayload(e)
		if err != nil {
			return event, fmt.Errorf("failed to cast notification event payload: %w", err)
		}

		err = event.Payload.FromNotificationEventInvoiceUpdatedPayload(payload)
		if err != nil {
			return event, fmt.Errorf("failed to cast notification event payload: %w", err)
		}
	default:
		return event, models.NewGenericValidationError(fmt.Errorf("invalid event payload type: %s", e.Type))
	}

	return event, nil
}

func AsEventDeliveryAttempts(attempts []notification.EventDeliveryAttempt) []api.NotificationEventDeliveryAttempt {
	result := make([]api.NotificationEventDeliveryAttempt, 0, len(attempts))

	notification.SortEventDeliveryAttemptsInDescOrder(attempts)

	for _, attempt := range attempts {
		result = append(result, api.NotificationEventDeliveryAttempt{
			Response: api.EventDeliveryAttemptResponse{
				Body:       attempt.Response.Body,
				StatusCode: attempt.Response.StatusCode,
				Url:        attempt.Response.URL,
				DurationMs: int(attempt.Response.Duration.Milliseconds()),
			},
			State:     api.NotificationEventDeliveryStatusState(attempt.State),
			Timestamp: attempt.Timestamp,
		})
	}

	return result
}

func FromEventType(t notification.EventType) (api.NotificationEventType, error) {
	switch t {
	case notification.EventTypeBalanceThreshold:
		return api.NotificationEventTypeEntitlementsBalanceThreshold, nil
	case notification.EventTypeEntitlementReset:
		return api.NotificationEventTypeEntitlementsReset, nil
	case notification.EventTypeInvoiceCreated:
		return api.NotificationEventTypeInvoiceCreated, nil
	case notification.EventTypeInvoiceUpdated:
		return api.NotificationEventTypeInvoiceUpdated, nil
	default:
		return "", fmt.Errorf("invalid notification event type: %s", t)
	}
}

func FromEventAsBalanceThresholdPayload(e notification.Event) (api.NotificationEventBalanceThresholdPayload, error) {
	if e.Payload.BalanceThreshold == nil {
		return api.NotificationEventBalanceThresholdPayload{}, fmt.Errorf("balance threshold is nil")
	}

	return api.NotificationEventBalanceThresholdPayload{
		Id:        e.ID,
		Timestamp: e.CreatedAt,
		Type:      api.NotificationEventBalanceThresholdPayloadTypeEntitlementsBalanceThreshold,
		Data: api.NotificationEventBalanceThresholdPayloadData{
			Value:       e.Payload.BalanceThreshold.Value,
			Entitlement: e.Payload.BalanceThreshold.Entitlement,
			Feature:     e.Payload.BalanceThreshold.Feature,
			Subject:     e.Payload.BalanceThreshold.Subject,
			Customer:    lo.ToPtr(e.Payload.BalanceThreshold.Customer),
			Threshold:   e.Payload.BalanceThreshold.Threshold,
		},
	}, nil
}

func FromEventAsEntitlementResetPayload(e notification.Event) (api.NotificationEventResetPayload, error) {
	if e.Payload.EntitlementReset == nil {
		return api.NotificationEventResetPayload{}, fmt.Errorf("entitlement reset is nil")
	}

	return api.NotificationEventResetPayload{
		Id:        e.ID,
		Timestamp: e.CreatedAt,
		Type:      api.NotificationEventResetPayloadTypeEntitlementsReset,
		Data: api.NotificationEventEntitlementValuePayloadBase{
			Value:       e.Payload.EntitlementReset.Value,
			Entitlement: e.Payload.EntitlementReset.Entitlement,
			Feature:     e.Payload.EntitlementReset.Feature,
			Subject:     e.Payload.EntitlementReset.Subject,
			Customer:    lo.ToPtr(e.Payload.EntitlementReset.Customer),
		},
	}, nil
}

func FromEventAsInvoiceCreatedPayload(e notification.Event) (api.NotificationEventInvoiceCreatedPayload, error) {
	if e.Payload.Invoice == nil {
		return api.NotificationEventInvoiceCreatedPayload{}, fmt.Errorf("invoice is nil")
	}

	return api.NotificationEventInvoiceCreatedPayload{
		Id:        e.ID,
		Timestamp: e.CreatedAt,
		Data:      e.Payload.Invoice.Invoice,
		Type:      api.NotificationEventInvoiceCreatedPayloadTypeInvoiceCreated,
	}, nil
}

func FromEventAsInvoiceUpdatedPayload(e notification.Event) (api.NotificationEventInvoiceUpdatedPayload, error) {
	if e.Payload.Invoice == nil {
		return api.NotificationEventInvoiceUpdatedPayload{}, fmt.Errorf("invoice is nil")
	}

	return api.NotificationEventInvoiceUpdatedPayload{
		Id:        e.ID,
		Timestamp: e.CreatedAt,
		Data:      e.Payload.Invoice.Invoice,
		Type:      api.NotificationEventInvoiceUpdatedPayloadTypeInvoiceUpdated,
	}, nil
}