bbc/simorgh

import React from 'react';
import { EPISODE_STATUS } from '..';
import ErrorMessage from '../ErrorMessage';

const withMediaError = PageComponent => {
  const MediaErrorComponent = props => {
    const { pageData: { episodeAvailability } = null } = props;
    const mediaIsAvailable =
      episodeAvailability === EPISODE_STATUS.EPISODE_IS_AVAILABLE;

    const ErrorComponent = errorProps => (
      <ErrorMessage {...errorProps} episodeAvailability={episodeAvailability} />
    );

    return (
      <PageComponent
        {...props}
        mediaIsAvailable={mediaIsAvailable}
        MediaError={mediaIsAvailable ? () => null : ErrorComponent}
      />
    );
  };

  return MediaErrorComponent;
};

export default withMediaError;