department-of-veterans-affairs/vets-website

import { VaPagination } from '@department-of-veterans-affairs/component-library/dist/react-bindings';
import React, { Component } from 'react';
import { connect } from 'react-redux';
import { LocationType } from '../constants';
import { facilityLocatorRestoreCommunityCarePagination } from '../utils/featureFlagSelectors';

export class PaginationWrapper extends Component {
  shouldComponentUpdate = nextProps =>
    nextProps.results !== this.props.results ||
    nextProps.inProgress !== this.props.inProgress;

  render() {
    const {
      shouldRender,
      handlePageSelect,
      currentPage,
      totalPages,
      results,
    } = this.props;

    if (
      shouldRender &&
      currentPage &&
      totalPages > 1 &&
      results &&
      results.length > 0
    ) {
      return (
        <VaPagination
          uswds
          onPageSelect={handlePageSelect}
          page={currentPage}
          pages={totalPages}
        />
      );
    }
    return null;
  }
}

const mapStateToProps = state => {
  let shouldHidePagination = false;

  if (
    [
      LocationType.CC_PROVIDER,
      LocationType.URGENT_CARE_PHARMACIES,
      LocationType.EMERGENCY_CARE,
    ].includes(state.searchQuery.facilityType) &&
    !facilityLocatorRestoreCommunityCarePagination(state)
  ) {
    shouldHidePagination = true;
  }

  // pagination not yet supported for PPMS urgent care
  if (
    state.searchQuery.facilityType === LocationType.URGENT_CARE &&
    (!state.searchQuery.serviceType ||
      ['AllUrgentCare', 'NonVAUrgentCare'].includes(
        state.searchQuery.serviceType,
      ))
  ) {
    shouldHidePagination = true;
  }

  return {
    shouldRender: !shouldHidePagination,
  };
};

export default connect(mapStateToProps)(PaginationWrapper);