import { connect } from 'react-redux'
import { getItem, isUpdating, isRemoving } from '../lib/stateManagement'

export default function withRecordProps(Component) {
  return connect(
    (state, ownProps) => {
      const { listId, itemId } = ownProps

      return {
        record: getItem(state, listId, itemId).toJS(),
        updating: isUpdating(state, listId, itemId),
        removing: isRemoving(state, listId, itemId)
      }
    }
  )(Component)
}