taxcode/adapter/organizationdefaulttaxcodes.go

package adapter

import (
	"context"
	"fmt"

	"entgo.io/ent/dialect/sql"

	"github.com/openmeterio/openmeter/openmeter/ent/db"
	orgdefaultsdb "github.com/openmeterio/openmeter/openmeter/ent/db/organizationdefaulttaxcodes"
	"github.com/openmeterio/openmeter/openmeter/taxcode"
	"github.com/openmeterio/openmeter/pkg/framework/entutils"
	"github.com/openmeterio/openmeter/pkg/models"
)

func (a *adapter) GetOrganizationDefaultTaxCodes(ctx context.Context, input taxcode.GetOrganizationDefaultTaxCodesInput) (taxcode.OrganizationDefaultTaxCodes, error) {
	if err := input.Validate(); err != nil {
		return taxcode.OrganizationDefaultTaxCodes{}, err
	}

	return entutils.TransactingRepo(ctx, a, func(ctx context.Context, a *adapter) (taxcode.OrganizationDefaultTaxCodes, error) {
		query := a.db.OrganizationDefaultTaxCodes.Query().
			Where(orgdefaultsdb.NamespaceEQ(input.Namespace)).
			Where(orgdefaultsdb.DeletedAtIsNil())

		if input.Expand.InvoicingTaxCode {
			query = query.WithInvoicingTaxCode()
		}

		if input.Expand.CreditGrantTaxCode {
			query = query.WithCreditGrantTaxCode()
		}

		entity, err := query.Only(ctx)
		if err != nil {
			if db.IsNotFound(err) {
				return taxcode.OrganizationDefaultTaxCodes{}, taxcode.NewOrganizationDefaultTaxCodesNotFoundError(input.Namespace)
			}

			return taxcode.OrganizationDefaultTaxCodes{}, fmt.Errorf("failed to get organization default tax codes: %w", err)
		}

		return mapOrganizationDefaultTaxCodesFromEntity(entity, input.Expand)
	})
}

func (a *adapter) UpsertOrganizationDefaultTaxCodes(ctx context.Context, input taxcode.UpsertOrganizationDefaultTaxCodesInput) (taxcode.OrganizationDefaultTaxCodes, error) {
	if err := input.Validate(); err != nil {
		return taxcode.OrganizationDefaultTaxCodes{}, err
	}

	return entutils.TransactingRepo(ctx, a, func(ctx context.Context, a *adapter) (taxcode.OrganizationDefaultTaxCodes, error) {
		err := a.db.OrganizationDefaultTaxCodes.Create().
			SetNamespace(input.Namespace).
			SetInvoicingTaxCodeID(input.InvoicingTaxCodeID).
			SetCreditGrantTaxCodeID(input.CreditGrantTaxCodeID).
			OnConflict(
				sql.ConflictColumns(orgdefaultsdb.FieldNamespace),
				sql.ConflictWhere(sql.IsNull(orgdefaultsdb.FieldDeletedAt)),
			).
			UpdateNewValues().
			Exec(ctx)
		if err != nil {
			if db.IsConstraintError(err) {
				return taxcode.OrganizationDefaultTaxCodes{}, models.NewGenericConflictError(fmt.Errorf("invalid tax code reference"))
			}

			return taxcode.OrganizationDefaultTaxCodes{}, fmt.Errorf("failed to upsert organization default tax codes: %w", err)
		}

		return a.GetOrganizationDefaultTaxCodes(ctx, taxcode.GetOrganizationDefaultTaxCodesInput{
			Namespace: input.Namespace,
			Expand:    input.Expand,
		})
	})
}

func mapOrganizationDefaultTaxCodesFromEntity(entity *db.OrganizationDefaultTaxCodes, expand taxcode.OrganizationDefaultTaxCodesExpand) (taxcode.OrganizationDefaultTaxCodes, error) {
	result := taxcode.OrganizationDefaultTaxCodes{
		NamespacedID: models.NamespacedID{
			Namespace: entity.Namespace,
			ID:        entity.ID,
		},
		ManagedModel: models.ManagedModel{
			CreatedAt: entity.CreatedAt,
			UpdatedAt: entity.UpdatedAt,
			DeletedAt: entity.DeletedAt,
		},
		InvoicingTaxCodeID:   entity.InvoicingTaxCodeID,
		CreditGrantTaxCodeID: entity.CreditGrantTaxCodeID,
	}

	if expand.InvoicingTaxCode {
		invoicingEdge, err := entity.Edges.InvoicingTaxCodeOrErr()
		if err != nil {
			return taxcode.OrganizationDefaultTaxCodes{}, fmt.Errorf("failed to load invoicing_tax_code edge: %w", err)
		}

		invoicingTaxCode, err := MapTaxCodeFromEntity(invoicingEdge)
		if err != nil {
			return taxcode.OrganizationDefaultTaxCodes{}, err
		}

		result.InvoicingTaxCode = &invoicingTaxCode
	}

	if expand.CreditGrantTaxCode {
		creditGrantEdge, err := entity.Edges.CreditGrantTaxCodeOrErr()
		if err != nil {
			return taxcode.OrganizationDefaultTaxCodes{}, fmt.Errorf("failed to load credit_grant_tax_code edge: %w", err)
		}

		creditGrantTaxCode, err := MapTaxCodeFromEntity(creditGrantEdge)
		if err != nil {
			return taxcode.OrganizationDefaultTaxCodes{}, err
		}

		result.CreditGrantTaxCode = &creditGrantTaxCode
	}

	return result, nil
}