/**
 * Prevent Divi Gallery modules from using cropped image variants.
 *
 * Add to a child theme's functions.php file or a code snippets plugin.
 */

/**
 * Replace cropped WordPress image URLs with the original attachment URL.
 */
function db_restore_original_gallery_image_url( $url ) {
	if ( ! is_string( $url ) || $url === '' ) {
		return $url;
	}

	// Remove URL parameters while looking up the attachment.
	$clean_url = preg_replace( '/\?.*$/', '', $url );

	/*
	 * Convert filenames such as:
	 * image-400x284.jpg
	 *
	 * back to:
	 * image.jpg
	 */
	$possible_original = preg_replace(
		'/-\d+x\d+(?=\.[a-zA-Z0-9]{2,5}$)/',
		'',
		$clean_url
	);

	if ( $possible_original === $clean_url ) {
		return $url;
	}

	$attachment_id = attachment_url_to_postid( $possible_original );

	if ( ! $attachment_id ) {
		return $possible_original;
	}

	$original_url = wp_get_attachment_image_url( $attachment_id, 'full' );

	return $original_url ?: $possible_original;
}


/**
 * Replace cropped URLs in gallery image src and srcset attributes.
 */
function db_disable_divi_gallery_image_cropping( $html ) {
	if (
		! is_string( $html ) ||
		$html === '' ||
		strpos( $html, 'et_pb_gallery' ) === false
	) {
		return $html;
	}

	return preg_replace_callback(
		'/<img\b([^>]*)>/i',
		function ( $matches ) {
			$attributes = $matches[1];

			/*
			 * Replace the main src URL.
			 */
			$attributes = preg_replace_callback(
				'/\bsrc=(["\'])(.*?)\1/i',
				function ( $src_match ) {
					$original_url = db_restore_original_gallery_image_url(
						html_entity_decode( $src_match[2] )
					);

					return 'src=' . $src_match[1] .
						esc_url( $original_url ) .
						$src_match[1];
				},
				$attributes
			);

			/*
			 * Replace each URL in srcset.
			 */
			$attributes = preg_replace_callback(
				'/\bsrcset=(["\'])(.*?)\1/i',
				function ( $srcset_match ) {
					$sources = array_map(
						'trim',
						explode( ',', html_entity_decode( $srcset_match[2] ) )
					);

					$updated_sources = array_map(
						function ( $source ) {
							$parts = preg_split(
								'/\s+/',
								trim( $source ),
								2
							);

							$url        = $parts[0] ?? '';
							$descriptor = $parts[1] ?? '';

							$url = db_restore_original_gallery_image_url( $url );

							return trim(
								esc_url( $url ) . ' ' . $descriptor
							);
						},
						$sources
					);

					return 'srcset=' . $srcset_match[1] .
						esc_attr( implode( ', ', $updated_sources ) ) .
						$srcset_match[1];
				},
				$attributes
			);

			/*
			 * Remove width and height attributes so the image can retain
			 * its natural aspect ratio in the masonry layout.
			 */
			$attributes = preg_replace(
				'/\s+(?:width|height)=(["\']).*?\1/i',
				'',
				$attributes
			);

			return '<img' . $attributes . '>';
		},
		$html
	);
}


/**
 * Divi 4 Gallery modules.
 */
add_filter(
	'et_module_shortcode_output',
	function ( $output, $render_slug, $module ) {
		if ( $render_slug !== 'et_pb_gallery' ) {
			return $output;
		}

		return db_disable_divi_gallery_image_cropping( $output );
	},
	10,
	3
);


/**
 * Divi 5 Gallery modules.
 */
add_filter(
	'render_block_divi/gallery',
	function ( $block_content, $parsed_block, $block ) {
		return db_disable_divi_gallery_image_cropping( $block_content );
	},
	10,
	3
);

selector .et_pb_gallery_items {
	display: block !important;
	column-count: 3;
	column-gap: 15px;
	width: 100% !important;
	max-width: 100% !important;
	grid-template-columns: unset !important;
	grid-auto-rows: unset !important;
	align-items: unset !important;
	list-style: none;
	margin-left: 0;
	padding-left: 0;
}

@media only screen and (max-width: 1200px) {
	selector .et_pb_gallery_items {
		column-count: 2;
	}
}

@media only screen and (max-width: 767px) {
	selector .et_pb_gallery_items {
		column-count: 1;
	}
}

selector .et_pb_gallery_items .et_pb_gallery_item,
selector .et_pb_gallery_items > li,
selector .et_pb_gallery_items > div {
	break-inside: avoid;
	page-break-inside: avoid;
	-webkit-column-break-inside: avoid;
	-moz-column-break-inside: avoid;
	width: 100% !important;
	max-width: 100% !important;
	margin: 0 0 16px 0 !important;
	float: none !important;
	display: inline-block !important;
	flex: 0 0 auto !important;
}

selector
	.et_pb_gallery_items
	.et_pb_gallery_item
	.et_pb_gallery_image,
selector
	.et_pb_gallery_items
	.et_pb_gallery_item
	.et_pb_gallery_title {
	width: 100% !important;
	max-width: 100% !important;
}

selector
	.et_pb_gallery_items
	.et_pb_gallery_item
	.et_pb_gallery_image
	img {
	display: block;
	width: 100% !important;
	height: auto !important;
}

selector .et_pb_gallery_pagination {
	display: none !important;
}