cic-custodial/internal/tasker/task/dispatch_tx.go

package task

import (
	"context"
	"encoding/json"
	"fmt"

	"github.com/celo-org/celo-blockchain/common"
	"github.com/celo-org/celo-blockchain/core/types"
	"github.com/grassrootseconomics/celoutils"
	"github.com/grassrootseconomics/cic-custodial/internal/custodial"
	"github.com/grassrootseconomics/cic-custodial/internal/events"
	"github.com/grassrootseconomics/cic-custodial/internal/store"
	"github.com/grassrootseconomics/cic-custodial/pkg/status"
	"github.com/grassrootseconomics/w3-celo-patch/module/eth"
	"github.com/hibiken/asynq"
)

type TxPayload struct {
	OtxId uint               `json:"otxId"`
	Tx    *types.Transaction `json:"tx"`
}

func DispatchTx(cu *custodial.Custodial) func(context.Context, *asynq.Task) error {
	return func(ctx context.Context, t *asynq.Task) error {
		var (
			payload        TxPayload
			dispatchStatus store.DispatchStatus
			eventPayload   events.EventPayload
			dispathchTx    common.Hash
		)

		if err := json.Unmarshal(t.Payload(), &payload); err != nil {
			return fmt.Errorf("dispatch: failed %v: %w", err, asynq.SkipRetry)
		}

		txHash := payload.Tx.Hash().Hex()

		dispatchStatus.OtxId, eventPayload.OtxId = payload.OtxId, payload.OtxId
		eventPayload.TxHash = txHash

		if err := cu.CeloProvider.Client.CallCtx(
			ctx,
			eth.SendTx(payload.Tx).Returns(&dispathchTx),
		); err != nil {
			dispatchStatus.Status = status.Unknown

			switch err.Error() {
			case celoutils.ErrGasPriceLow:
				dispatchStatus.Status = status.FailGasPrice
			case celoutils.ErrInsufficientGas:
				dispatchStatus.Status = status.FailInsufficientGas
			case celoutils.ErrNonceLow:
				dispatchStatus.Status = status.FailNonce
			}

			if err := cu.PgStore.CreateDispatchStatus(ctx, dispatchStatus); err != nil {
				return fmt.Errorf("dispatch: failed %v: %w", err, asynq.SkipRetry)
			}

			if err := cu.EventEmitter.Publish(events.DispatchFail, txHash, eventPayload); err != nil {
				return fmt.Errorf("dispatch: failed %v: %w", err, asynq.SkipRetry)
			}

			return fmt.Errorf("dispatch: failed %v: %w", err, asynq.SkipRetry)
		}

		dispatchStatus.Status = status.Successful

		if err := cu.PgStore.CreateDispatchStatus(ctx, dispatchStatus); err != nil {
			return fmt.Errorf("dispatch: failed %v: %w", err, asynq.SkipRetry)
		}

		if err := cu.EventEmitter.Publish(events.DispatchSuccess, txHash, eventPayload); err != nil {
			return fmt.Errorf("dispatch: failed %v: %w", err, asynq.SkipRetry)
		}

		return nil
	}
}