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

const GetSigma = gql`
query GetSigma($poolId: String, $epochNumber: Int) {
  getSigma(poolId: $poolId, epochNumber: $epochNumber) {
    sigma
    epoch
    total_staked
    active_stake
    nonce
    d
  }
}`

const withAppSyncSigma = compose(
  graphql(GetSigma, {
    name: 'getSigma',
    options: (ownProps) => ({
      fetchPolicy: 'cache-and-network',
      variables: {
        poolId: ownProps.poolId,
        epochNumber: ownProps.epochNumber
      }
    }),
    props: (props) => {
      if (props.getSigma.error) console.log(props.getSigma.error)
      return props
    }
  })
)

export { GetSigma, withAppSyncSigma }