client

var Events = map[Event]string{
	EventOpen:                          "ClientEventOpen",
	EventPaymentChannelErrored:         "ClientEventPaymentChannelErrored",
	EventDealProposed:                  "ClientEventDealProposed",
	EventAllocateLaneErrored:           "ClientEventAllocateLaneErrored",
	EventPaymentChannelCreateInitiated: "ClientEventPaymentChannelCreateInitiated",
	EventPaymentChannelReady:           "ClientEventPaymentChannelReady",
	EventPaymentChannelAddingFunds:     "ClientEventPaymentChannelAddingFunds",
	EventPaymentChannelAddFundsErrored: "ClientEventPaymentChannelAddFundsErrored",
	EventPaymentChannelSkip:            "ClientEventPaymentChannelSkip",
	EventWriteDealProposalErrored:      "ClientEventWriteDealProposalErrored",
	EventDealRejected:                  "ClientEventDealRejected",
	EventDealNotFound:                  "ClientEventDealNotFound",
	EventDealAccepted:                  "ClientEventDealAccepted",
	EventProviderCancelled:             "ClientEventProviderCancelled",
	EventUnknownResponseReceived:       "ClientEventUnknownResponseReceived",
	EventLastPaymentRequested:          "ClientEventLastPaymentRequested",
	EventAllBlocksReceived:             "ClientEventAllBlocksReceived",
	EventPaymentRequested:              "ClientEventPaymentRequested",
	EventUnsealPaymentRequested:        "ClientEventUnsealPaymentRequested",
	EventBlocksReceived:                "ClientEventBlocksReceived",
	EventSendFunds:                     "ClientEventSendFunds",
	EventFundsExpended:                 "ClientEventFundsExpended",
	EventBadPaymentRequested:           "ClientEventBadPaymentRequested",
	EventCreateVoucherFailed:           "ClientEventCreateVoucherFailed",
	EventWriteDealPaymentErrored:       "ClientEventWriteDealPaymentErrored",
	EventPaymentSent:                   "ClientEventPaymentSent",
	EventPaymentNotSent:                "ClientEventPaymentNotSent",
	EventDataTransferError:             "ClientEventDataTransferError",
	EventComplete:                      "ClientEventComplete",
	EventCancelComplete:                "ClientEventCancelComplete",
	EventEarlyTermination:              "ClientEventEarlyTermination",
	EventCompleteVerified:              "ClientEventCompleteVerified",
	EventLaneAllocated:                 "ClientEventLaneAllocated",
	EventVoucherShortfall:              "ClientEventVoucherShortfall",
	EventRecheckFunds:                  "ClientEventRecheckFunds",
	EventCancel:                        "ClientEventCancel",
	EventWaitForLastBlocks:             "ClientEventWaitForLastBlocks",
	EventProviderErrored:               "ClientEventProviderErrored",
}

var FSMEvents = fsm.Events{
	fsm.Event(EventOpen).
		From(deal.StatusNew).ToNoChange(),

	fsm.Event(EventWriteDealProposalErrored).
		FromAny().To(deal.StatusErrored).
		Action(func(ds *deal.ClientState, err error) error {
			ds.Message = fmt.Errorf("proposing deal: %w", err).Error()
			return nil
		}),
	fsm.Event(EventDealProposed).
		From(deal.StatusNew).To(deal.StatusWaitForAcceptance).
		Action(func(ds *deal.ClientState, channelID datatransfer.ChannelID) error {
			ds.ChannelID = channelID
			ds.Message = ""
			return nil
		}),

	fsm.Event(EventDealRejected).
		From(deal.StatusWaitForAcceptance).To(deal.StatusRejected).
		Action(func(deal *deal.ClientState, message string) error {
			deal.Message = fmt.Sprintf("deal rejected: %s", message)
			return nil
		}),
	fsm.Event(EventDealNotFound).
		From(deal.StatusWaitForAcceptance).To(deal.StatusDealNotFound).
		Action(func(ds *deal.ClientState, message string) error {
			ds.Message = fmt.Sprintf("deal not found: %s", message)
			return nil
		}),
	fsm.Event(EventDealAccepted).
		From(deal.StatusWaitForAcceptance).To(deal.StatusAccepted).
		From(deal.StatusClientWaitingForLastBlocks).ToNoChange(),
	fsm.Event(EventUnknownResponseReceived).
		FromAny().To(deal.StatusFailing).
		Action(func(ds *deal.ClientState, status deal.Status) error {
			ds.Message = fmt.Sprintf("Unexpected deal response status: %s", deal.Statuses[status])
			return nil
		}),

	fsm.Event(EventPaymentChannelSkip).
		From(deal.StatusAccepted).To(deal.StatusOngoing),

	fsm.Event(EventPaymentChannelErrored).
		FromMany(deal.StatusAccepted, deal.StatusPaymentChannelCreating, deal.StatusPaymentChannelAddingFunds).To(deal.StatusFailing).
		Action(func(ds *deal.ClientState, err error) error {
			ds.Message = fmt.Errorf("error from payment channel: %w", err).Error()
			return nil
		}),
	fsm.Event(EventPaymentChannelCreateInitiated).
		From(deal.StatusAccepted).To(deal.StatusPaymentChannelCreating).
		Action(func(ds *deal.ClientState, msgCID cid.Cid) error {
			ds.WaitMsgCID = &msgCID
			return nil
		}),
	fsm.Event(EventPaymentChannelAddingFunds).
		From(deal.StatusAccepted).To(deal.StatusPaymentChannelAddingInitialFunds).
		From(deal.StatusCheckFunds).To(deal.StatusPaymentChannelAddingFunds).
		Action(func(ds *deal.ClientState, msgCID cid.Cid, payCh address.Address) error {
			ds.WaitMsgCID = &msgCID
			if ds.PaymentInfo == nil {
				ds.PaymentInfo = &deal.PaymentInfo{
					PayCh: payCh,
				}
			}
			return nil
		}),
	fsm.Event(EventPaymentChannelReady).
		FromMany(deal.StatusPaymentChannelCreating, deal.StatusPaymentChannelAddingInitialFunds, deal.StatusAccepted).
		To(deal.StatusPaymentChannelAllocatingLane).
		From(deal.StatusPaymentChannelAddingFunds).To(deal.StatusOngoing).
		From(deal.StatusCheckFunds).To(deal.StatusOngoing).
		Action(func(ds *deal.ClientState, payCh address.Address) error {
			if ds.PaymentInfo == nil {
				ds.PaymentInfo = &deal.PaymentInfo{
					PayCh: payCh,
				}
			}
			ds.WaitMsgCID = nil

			ds.Message = ""
			return nil
		}),
	fsm.Event(EventAllocateLaneErrored).
		FromMany(deal.StatusPaymentChannelAllocatingLane).
		To(deal.StatusFailing).
		Action(func(ds *deal.ClientState, err error) error {
			ds.Message = fmt.Errorf("allocating payment lane: %w", err).Error()
			return nil
		}),

	fsm.Event(EventLaneAllocated).
		From(deal.StatusPaymentChannelAllocatingLane).To(deal.StatusOngoing).
		Action(func(ds *deal.ClientState, lane uint64) error {
			ds.PaymentInfo.Lane = lane
			return nil
		}),

	fsm.Event(EventProviderErrored).
		FromAny().To(deal.StatusErrored).
		Action(func(ds *deal.ClientState, msg string) error {
			ds.Message = msg
			return nil
		}),

	fsm.Event(EventDataTransferError).
		FromAny().To(deal.StatusErrored).
		Action(func(ds *deal.ClientState, err error) error {
			ds.Message = fmt.Sprintf("error generated by data transfer: %s", err.Error())
			return nil
		}),

	fsm.Event(EventLastPaymentRequested).
		FromMany(
			deal.StatusOngoing,
			deal.StatusFundsNeededLastPayment,
			deal.StatusFundsNeeded).To(deal.StatusFundsNeededLastPayment).
		From(deal.StatusSendFunds).To(deal.StatusOngoing).
		From(deal.StatusCheckComplete).ToNoChange().
		From(deal.StatusBlocksComplete).To(deal.StatusSendFundsLastPayment).
		FromMany(
			paymentChannelCreationStates...).ToJustRecord().
		Action(func(ds *deal.ClientState, paymentOwed abi.TokenAmount) error {
			ds.PaymentRequested = big.Add(ds.PaymentRequested, paymentOwed)
			ds.LastPaymentRequested = true
			return nil
		}),
	fsm.Event(EventPaymentRequested).
		FromMany(
			deal.StatusOngoing,
			deal.StatusBlocksComplete,
			deal.StatusFundsNeededLastPayment,
			deal.StatusFundsNeeded).To(deal.StatusFundsNeeded).
		From(deal.StatusSendFunds).To(deal.StatusOngoing).
		From(deal.StatusCheckComplete).ToNoChange().
		FromMany(
			paymentChannelCreationStates...).ToJustRecord().
		Action(func(ds *deal.ClientState, paymentOwed abi.TokenAmount) error {
			ds.PaymentRequested = big.Add(ds.PaymentRequested, paymentOwed)
			return nil
		}),

	fsm.Event(EventUnsealPaymentRequested).
		From(deal.StatusWaitForAcceptance).To(deal.StatusAccepted).
		Action(func(ds *deal.ClientState, paymentOwed abi.TokenAmount) error {
			ds.PaymentRequested = big.Add(ds.PaymentRequested, paymentOwed)
			return nil
		}),

	fsm.Event(EventAllBlocksReceived).
		FromMany(
			deal.StatusOngoing,
			deal.StatusBlocksComplete,
		).To(deal.StatusBlocksComplete).
		FromMany(paymentChannelCreationStates...).ToJustRecord().
		FromMany(deal.StatusSendFunds, deal.StatusSendFundsLastPayment).To(deal.StatusOngoing).
		From(deal.StatusFundsNeeded).ToNoChange().
		From(deal.StatusFundsNeededLastPayment).To(deal.StatusSendFundsLastPayment).
		FromMany(
			deal.StatusClientWaitingForLastBlocks,

			deal.StatusCheckComplete,
		).To(deal.StatusCompleted).
		Action(func(ds *deal.ClientState) error {
			ds.AllBlocksReceived = true
			return nil
		}),
	fsm.Event(EventBlocksReceived).
		FromMany(deal.StatusOngoing,
			deal.StatusFundsNeeded,
			deal.StatusFundsNeededLastPayment,
			deal.StatusCheckComplete,
			deal.StatusClientWaitingForLastBlocks).ToNoChange().
		FromMany(deal.StatusSendFunds, deal.StatusSendFundsLastPayment).To(deal.StatusOngoing).
		FromMany(paymentChannelCreationStates...).ToJustRecord().
		Action(recordReceived),

	fsm.Event(EventSendFunds).
		FromMany(deal.StatusSendFunds, deal.StatusSendFundsLastPayment).To(deal.StatusOngoing).
		From(deal.StatusFundsNeeded).To(deal.StatusSendFunds).
		From(deal.StatusFundsNeededLastPayment).To(deal.StatusSendFundsLastPayment),

	fsm.Event(EventFundsExpended).
		From(deal.StatusCheckFunds).To(deal.StatusInsufficientFunds).
		Action(func(ds *deal.ClientState, shortfall abi.TokenAmount) error {
			ds.Message = fmt.Sprintf("not enough current or pending funds in payment channel, shortfall of %s", shortfall.String())
			ds.VoucherShortfall = shortfall
			return nil
		}),
	fsm.Event(EventBadPaymentRequested).
		FromMany(deal.StatusSendFunds, deal.StatusSendFundsLastPayment).To(deal.StatusFailing).
		Action(func(ds *deal.ClientState, message string) error {
			ds.Message = message
			return nil
		}),
	fsm.Event(EventCreateVoucherFailed).
		FromMany(deal.StatusSendFunds, deal.StatusSendFundsLastPayment).To(deal.StatusFailing).
		Action(func(ds *deal.ClientState, err error) error {
			ds.Message = fmt.Errorf("creating payment voucher: %w", err).Error()
			return nil
		}),
	fsm.Event(EventVoucherShortfall).
		FromMany(deal.StatusSendFunds, deal.StatusSendFundsLastPayment).To(deal.StatusCheckFunds).
		Action(func(ds *deal.ClientState, shortfall abi.TokenAmount) error {
			return nil
		}),

	fsm.Event(EventWriteDealPaymentErrored).
		FromAny().To(deal.StatusErrored).
		Action(func(ds *deal.ClientState, err error) error {
			ds.Message = fmt.Errorf("writing deal payment: %w", err).Error()
			return nil
		}),

	fsm.Event(EventPaymentNotSent).
		From(deal.StatusOngoing).ToJustRecord().
		From(deal.StatusSendFunds).To(deal.StatusOngoing).
		From(deal.StatusSendFundsLastPayment).To(deal.StatusFinalizing),
	fsm.Event(EventPaymentSent).
		From(deal.StatusOngoing).ToJustRecord().
		From(deal.StatusBlocksComplete).To(deal.StatusCheckComplete).
		From(deal.StatusCheckComplete).ToNoChange().
		FromMany(
			deal.StatusFundsNeeded,
			deal.StatusFundsNeededLastPayment,
			deal.StatusSendFunds).To(deal.StatusOngoing).
		From(deal.StatusSendFundsLastPayment).To(deal.StatusFinalizing).
		Action(func(state *deal.ClientState, voucherAmt abi.TokenAmount) error {

			sentAmt := big.Sub(voucherAmt, state.FundsSpent)
			state.PaymentRequested = big.Sub(state.PaymentRequested, sentAmt)

			state.FundsSpent = voucherAmt

			if state.UnsealPrice.GreaterThanEqual(state.FundsSpent) {
				state.UnsealFundsPaid = state.FundsSpent
				return nil
			}

			state.UnsealFundsPaid = state.UnsealPrice

			if state.PricePerByte.IsZero() {
				return nil
			}

			paidSoFarForTransfer := big.Sub(state.FundsSpent, state.UnsealFundsPaid)
			state.BytesPaidFor = big.Div(paidSoFarForTransfer, state.PricePerByte).Uint64()

			if state.BytesPaidFor >= state.CurrentInterval {
				state.CurrentInterval = state.NextInterval()
			}

			return nil
		}),

	fsm.Event(EventComplete).
		FromMany(
			deal.StatusSendFunds,
			deal.StatusSendFundsLastPayment,
			deal.StatusFundsNeeded,
			deal.StatusFundsNeededLastPayment,
			deal.StatusBlocksComplete,
			deal.StatusWaitForAcceptance,
			deal.StatusAccepted,
			deal.StatusOngoing).To(deal.StatusCheckComplete).
		From(deal.StatusFinalizing).To(deal.StatusCompleted),
	fsm.Event(EventCompleteVerified).
		From(deal.StatusCheckComplete).To(deal.StatusCompleted),
	fsm.Event(EventEarlyTermination).
		From(deal.StatusCheckComplete).To(deal.StatusErrored).
		Action(func(state *deal.ClientState) error {
			state.Message = "Provider sent complete status without sending all data"
			return nil
		}),

	fsm.Event(EventWaitForLastBlocks).
		From(deal.StatusCheckComplete).To(deal.StatusClientWaitingForLastBlocks).
		From(deal.StatusCompleted).ToJustRecord(),

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

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

	fsm.Event(EventCancel).FromAny().To(deal.StatusCancelling).Action(func(ds *deal.ClientState) error {
		ds.Message = "Client cancelled retrieval"
		return nil
	}),

	fsm.Event(EventRecheckFunds).From(deal.StatusInsufficientFunds).To(deal.StatusCheckFunds),
}

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

var StateEntryFuncs = fsm.StateEntryFuncs{
	deal.StatusNew:                              ProposeDeal,
	deal.StatusAccepted:                         SetupPaymentChannelStart,
	deal.StatusPaymentChannelCreating:           WaitPaymentChannelReady,
	deal.StatusPaymentChannelAddingInitialFunds: WaitPaymentChannelReady,
	deal.StatusPaymentChannelAllocatingLane:     AllocateLane,
	deal.StatusOngoing:                          Ongoing,
	deal.StatusFundsNeeded:                      ProcessPaymentRequested,
	deal.StatusFundsNeededLastPayment:           ProcessPaymentRequested,
	deal.StatusSendFunds:                        SendFunds,
	deal.StatusSendFundsLastPayment:             SendFunds,
	deal.StatusCheckFunds:                       CheckFunds,
	deal.StatusPaymentChannelAddingFunds:        WaitPaymentChannelReady,
	deal.StatusFailing:                          CancelDeal,
	deal.StatusCancelling:                       CancelDeal,
	deal.StatusCheckComplete:                    CheckComplete,
}