refactor: streamline state logic

Author: Takshil-Kunadia

packages/block-library/src/image/view.js

@@ -211,22 +211,28 @@ const { state, actions, callbacks } = store(
 					return;
 				}
 
-				// Link element to preload the image.
-				const imageMetadata = state.metadata[ imageId ];
+				// Sets the current image ID to preload.
+				state.currentImageId = imageId;
+
+				// Create link element to preload the image.
 				const imageLink = document.createElement( 'link' );
 				imageLink.rel = 'preload';
 				imageLink.as = 'image';
-				imageLink.href = imageMetadata.uploadedSrc;
+				imageLink.href = state.enlargedSrc;
 
-				// Apply srcset if available for responsive preloading
-				const srcset = imageMetadata.lightboxSrcset;
+				// Apply srcset if available for responsive preloading.
+				const srcset = state.enlargedSrcset;
 				if ( srcset ) {
 					imageLink.setAttribute( 'imagesrcset', srcset );
-					imageLink.setAttribute( 'imagesizes', '100vw' );
+					imageLink.setAttribute( 'imagesizes', state.enlargedSizes );
 				}
 
+				// Append the link element to the document head to trigger preloading.
 				document.head.appendChild( imageLink );
 				state.preloadedImageIds.add( imageId );
+
+				// Reset state after preloading the image.
+				state.currentImageId = null;
 			},
 			preloadImageWithDelay() {
 				const { imageId } = getContext();