provider

var Events = map[Event]string{
	EventOpen:                   "ProviderEventOpen",
	EventDealNotFound:           "ProviderEventDealNotFound",
	EventDealRejected:           "ProviderEventDealRejected",
	EventDealAccepted:           "ProviderEventDealAccepted",
	EventBlockSent:              "ProviderEventBlockSent",
	EventBlocksCompleted:        "ProviderEventBlocksCompleted",
	EventPaymentRequested:       "ProviderEventPaymentRequested",
	EventSaveVoucherFailed:      "ProviderEventSaveVoucherFailed",
	EventPartialPaymentReceived: "ProviderEventPartialPaymentReceived",
	EventPaymentReceived:        "ProviderEventPaymentReceived",
	EventComplete:               "ProviderEventComplete",
	EventUnsealError:            "ProviderEventUnsealError",
	EventUnsealComplete:         "ProviderEventUnsealComplete",
	EventDataTransferError:      "ProviderEventDataTransferError",
	EventCancelComplete:         "ProviderEventCancelComplete",
	EventCleanupComplete:        "ProviderEventCleanupComplete",
	EventMultiStoreError:        "ProviderEventMultiStoreError",
	EventClientCancelled:        "ProviderEventClientCancelled",
}

var FSMEvents = fsm.Events{

	fsm.Event(EventOpen).
		From(deal.StatusNew).ToNoChange(),

	fsm.Event(EventDealAccepted).
		From(deal.StatusNew).To(deal.StatusOngoing),

	fsm.Event(EventBlockSent).
		FromMany(deal.StatusOngoing).ToNoChange().
		Action(func(ds *deal.ProviderState, totalSent uint64) error {
			ds.TotalSent = totalSent
			return nil
		}),
	fsm.Event(EventBlocksCompleted).
		FromMany(deal.StatusOngoing).To(deal.StatusBlocksComplete),

	fsm.Event(EventPaymentRequested).
		From(deal.StatusOngoing).To(deal.StatusFundsNeeded).
		From(deal.StatusFundsNeeded).ToJustRecord().
		From(deal.StatusBlocksComplete).To(deal.StatusFundsNeededLastPayment).
		Action(func(ds *deal.ProviderState, totalSent uint64) error {
			ds.TotalSent = totalSent
			return nil
		}),

	fsm.Event(EventSaveVoucherFailed).
		FromMany(deal.StatusFundsNeeded, deal.StatusFundsNeededLastPayment).To(deal.StatusFailing).
		Action(recordError),
	fsm.Event(EventPartialPaymentReceived).
		FromMany(deal.StatusFundsNeeded, deal.StatusFundsNeededLastPayment).ToNoChange().
		Action(func(ds *deal.ProviderState, fundsReceived abi.TokenAmount, ch address.Address) error {
			ds.FundsReceived = big.Add(ds.FundsReceived, fundsReceived)
			ds.PayCh = &ch
			return nil
		}),
	fsm.Event(EventPaymentReceived).
		From(deal.StatusFundsNeeded).To(deal.StatusOngoing).
		From(deal.StatusFundsNeededLastPayment).To(deal.StatusFinalizing).
		FromMany(deal.StatusBlocksComplete, deal.StatusOngoing, deal.StatusFinalizing).ToJustRecord().
		Action(func(ds *deal.ProviderState, fundsReceived abi.TokenAmount, ch address.Address) error {
			ds.FundsReceived = big.Add(ds.FundsReceived, fundsReceived)
			ds.CurrentInterval = ds.NextInterval()
			ds.PayCh = &ch
			return nil
		}),

	fsm.Event(EventComplete).FromMany(deal.StatusBlocksComplete, deal.StatusFinalizing).To(deal.StatusCompleting),
	fsm.Event(EventCleanupComplete).From(deal.StatusCompleting).To(deal.StatusCompleted),

	fsm.Event(EventCancelComplete).
		From(deal.StatusCancelling).To(deal.StatusCancelled).
		From(deal.StatusFailing).To(deal.StatusErrored),

	fsm.Event(EventDataTransferError).
		FromAny().To(deal.StatusErrored).
		Action(recordError),

	fsm.Event(EventMultiStoreError).
		FromAny().To(deal.StatusErrored).
		Action(recordError),

	fsm.Event(EventClientCancelled).
		From(deal.StatusFailing).ToJustRecord().
		From(deal.StatusCancelling).ToJustRecord().
		FromAny().To(deal.StatusCancelling).Action(
		func(ds *deal.ProviderState) error {
			if ds.Status != deal.StatusFailing {
				ds.Message = "Client cancelled retrieval"
			}
			return nil
		},
	),
}

var FinalityStates = []fsm.StateKey{
	deal.StatusErrored,
	deal.StatusCompleted,
	deal.StatusCancelled,
}

var StateEntryFuncs = fsm.StateEntryFuncs{

	deal.StatusFailing:    CancelDeal,
	deal.StatusCancelling: CancelDeal,
	deal.StatusCompleting: CleanupDeal,
}