customer/adapter/entitymapping.go

package adapter

import (
	"errors"
	"slices"

	"github.com/samber/lo"
	"github.com/samber/mo"

	"github.com/openmeterio/openmeter/openmeter/customer"
	"github.com/openmeterio/openmeter/openmeter/ent/db"
	"github.com/openmeterio/openmeter/pkg/models"
)

func CustomerFromDBEntity(e db.Customer, expands customer.Expands) (*customer.Customer, error) {
	subjectKeys, err := subjectKeysFromDBEntity(e)
	if err != nil {
		return nil, err
	}

	var metadata *models.Metadata

	if len(e.Metadata) > 0 {
		metadata = lo.ToPtr(models.NewMetadata(e.Metadata))
	}

	var annotations *models.Annotations

	if len(e.Annotations) > 0 {
		annotations = &e.Annotations
	}

	result := &customer.Customer{
		ManagedResource: models.NewManagedResource(models.ManagedResourceInput{
			ID:          e.ID,
			Namespace:   e.Namespace,
			CreatedAt:   e.CreatedAt,
			UpdatedAt:   e.UpdatedAt,
			DeletedAt:   e.DeletedAt,
			Name:        e.Name,
			Description: e.Description,
		}),
		PrimaryEmail: e.PrimaryEmail,
		Currency:     e.Currency,
		Metadata:     metadata,
		Annotation:   annotations,
	}

	// Only set UsageAttribution if there are subject keys
	if len(subjectKeys) > 0 {
		result.UsageAttribution = &customer.CustomerUsageAttribution{
			SubjectKeys: subjectKeys,
		}
	}

	if slices.Contains(expands, customer.ExpandSubscriptions) {
		activeSubscriptionIDs, err := resolveActiveSubscriptionIDs(e)
		if err != nil {
			return nil, err
		}

		result.ActiveSubscriptionIDs = mo.Some(activeSubscriptionIDs)
	}

	if e.Key != "" {
		result.Key = &e.Key
	}

	if e.BillingAddressCity != nil || e.BillingAddressCountry != nil || e.BillingAddressLine1 != nil || e.BillingAddressLine2 != nil || e.BillingAddressPhoneNumber != nil || e.BillingAddressPostalCode != nil || e.BillingAddressState != nil {
		result.BillingAddress = &models.Address{
			City:        e.BillingAddressCity,
			Country:     e.BillingAddressCountry,
			Line1:       e.BillingAddressLine1,
			Line2:       e.BillingAddressLine2,
			PhoneNumber: e.BillingAddressPhoneNumber,
			PostalCode:  e.BillingAddressPostalCode,
			State:       e.BillingAddressState,
		}
	}

	return result, nil
}

func resolveActiveSubscriptionIDs(e db.Customer) ([]string, error) {
	subscriptions, err := e.Edges.SubscriptionOrErr()
	if err != nil {
		if db.IsNotLoaded(err) {
			return nil, errors.New("subscriptions must be loaded for customer")
		}

		return nil, err
	}

	subscriptionIDs := lo.FilterMap(subscriptions, func(item *db.Subscription, _ int) (string, bool) {
		if item == nil {
			return "", false
		}

		return item.ID, true
	})

	return subscriptionIDs, nil
}

func subjectKeysFromDBEntity(customerEntity db.Customer) ([]string, error) {
	subjectEntities, err := customerEntity.Edges.SubjectsOrErr()
	if err != nil {
		if db.IsNotLoaded(err) {
			return nil, errors.New("subjects must be loaded for customer")
		}

		return nil, err
	}

	subjectKeys := lo.FilterMap(subjectEntities, func(item *db.CustomerSubjects, _ int) (string, bool) {
		if item == nil {
			return "", false
		}

		return item.SubjectKey, true
	})

	// Sort the subject keys to make sure the order is consistent
	slices.Sort(subjectKeys)

	return subjectKeys, nil
}