import gql from 'graphql-tag'
import { graphql, compose } from 'react-apollo'

const GetReward = gql`
query GetReward($stakingAddress: String, $epochFrom: Int, $epochTo: Int) {
  getReward(stakingAddress: $stakingAddress, epochFrom: $epochFrom, epochTo: $epochTo) {
  }
}`

const withAppSyncReward = compose(
  graphql(GetReward, {
    name: 'getReward',
    options: (ownProps) => ({
      fetchPolicy: 'cache-and-network',
      variables: {
        stakingAddress: ownProps.stakingAddress,
        epochFrom: ownProps.epochFrom,
        epochTo: ownProps.epochTo
      }
    }),
    props: (props) => {
      if (props.getReward.error) console.log(props.getReward.error)
      return props
    }
  })
)

export { GetReward, withAppSyncReward }