Display photos function

displayPhotos.js

const photos = props.data.allWordpressPost.edges;

const displayPhotos = () => {
	const photoArray = [];
	let photoRow = [];
	let count = 0;

	photos.map( photo => {
		if ( photo.node.featured_media ) {
			photoArray.push( photo );
		}
	} );

	return (
		photoArray.map( photo => {
			if ( photoRow.length === 3 ) {
				photoRow = [];
			}

			photoRow.push( photo );
			count++;

			if ( photoRow.length === 3 ) {
				return returnRow( photoRow, count );
			} else if ( photoArray.length - count === 0 ) {
				return returnRow( photoRow, count );	
			}
		} )
	)
}

const returnRow = ( photos, count ) => {
	return (
		<PhotoRow photos={ photos } key={ count } />
	)
}