Android: Initiate image prefetching on ImageShadowNode layout

Summary:
This diff introduces a code path to trigger image prefetching from `ImageShadowNode::layout`.
Changelog: [Internal]

Differential Revision: D66454087

Author: dmytrorykun

packages/react-native/ReactAndroid/api/ReactAndroid.api

@@ -2638,6 +2638,7 @@ public class com/facebook/react/fabric/FabricUIManager : com/facebook/react/brid
 	public fun receiveEvent (IILjava/lang/String;ZLcom/facebook/react/bridge/WritableMap;IZ)V
 	public fun receiveEvent (ILjava/lang/String;Lcom/facebook/react/bridge/WritableMap;)V
 	public fun removeUIManagerEventListener (Lcom/facebook/react/bridge/UIManagerListener;)V
+	public fun requestImage (IILcom/facebook/react/common/mapbuffer/ReadableMapBuffer;)V
 	public fun resolveCustomDirectEventName (Ljava/lang/String;)Ljava/lang/String;
 	public fun resolveView (I)Landroid/view/View;
 	public fun sendAccessibilityEvent (II)V

packages/react-native/ReactAndroid/src/main/java/com/facebook/react/fabric/FabricUIManager.java

@@ -83,6 +83,7 @@
 import com.facebook.react.uimanager.events.FabricEventDispatcher;
 import com.facebook.react.uimanager.events.RCTEventEmitter;
 import com.facebook.react.uimanager.events.SynchronousEventReceiver;
+import com.facebook.react.views.image.ReactImageManager;
 import com.facebook.react.views.text.TextLayoutManager;
 import java.util.ArrayList;
 import java.util.HashMap;
@@ -872,6 +873,15 @@ public void runGuarded() {
     }
   }
 
+  /**
+   * This method initiates preloading of an image specified by ImageSource. It can later be consumed
+   * by an ImageView.
+   */
+  public void requestImage(int surfaceId, int reactTag, ReadableMapBuffer params) {
+    mMountingManager.experimental_prefetchResource(
+        mReactApplicationContext, ReactImageManager.REACT_CLASS, surfaceId, reactTag, params);
+  }
+
   public void setBinding(FabricUIManagerBinding binding) {
     mBinding = binding;
   }
@@ -954,7 +964,6 @@ public void receiveEvent(
    * @param reactTag
    * @param eventName
    * @param canCoalesceEvent
-   * @param customCoalesceKey
    * @param params
    * @param eventCategory
    */

packages/react-native/ReactCommon/react/renderer/imagemanager/platform/android/ImageFetcher.cpp

@@ -0,0 +1,39 @@
+/*
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+#include "ImageFetcher.h"
+#include <react/renderer/imagemanager/conversions.h>
+
+namespace facebook::react {
+
+ImageFetcher::ImageFetcher(const ContextContainer::Shared& contextContainer) {
+  fabricUIManager_ =
+      contextContainer->at<jni::global_ref<jobject>>("FabricUIManager");
+  requestImage_ =
+      fabricUIManager_->getClass()
+          ->getMethod<void(SurfaceId, Tag, JReadableMapBuffer::javaobject)>(
+              "requestImage");
+}
+
+ImageRequest ImageFetcher::requestImage(
+    const ImageSource& imageSource,
+    const ImageRequestParams& imageRequestParams,
+    SurfaceId surfaceId,
+    Tag tag) const {
+  auto serializedImageRequest =
+      serializeImageRequest(imageSource, imageRequestParams);
+
+  auto readableMapBuffer =
+      JReadableMapBuffer::createWithContents(std::move(serializedImageRequest));
+
+  requestImage_(fabricUIManager_, surfaceId, tag, readableMapBuffer.get());
+
+  auto telemetry = std::make_shared<ImageTelemetry>(surfaceId);
+
+  return {imageSource, telemetry};
+}
+} // namespace facebook::react

packages/react-native/ReactCommon/react/renderer/imagemanager/platform/android/ImageFetcher.h

@@ -0,0 +1,41 @@
+/*
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+#pragma once
+
+#include <fbjni/fbjni.h>
+
+#include <react/common/mapbuffer/JReadableMapBuffer.h>
+#include <react/jni/ReadableNativeMap.h>
+#include <react/renderer/imagemanager/ImageRequest.h>
+#include <react/renderer/imagemanager/ImageRequestParams.h>
+#include <react/utils/ContextContainer.h>
+
+namespace facebook::react {
+
+class ImageFetcher {
+ public:
+  ImageFetcher(const ContextContainer::Shared& contextContainer);
+  ImageFetcher(const ImageFetcher&) = delete;
+  ImageFetcher& operator=(const ImageFetcher&) = delete;
+  ImageFetcher(ImageFetcher&&) = default;
+  ~ImageFetcher() = default;
+
+  void discardImageRequest(const std::string& imageRequestCacheKey) const;
+
+  ImageRequest requestImage(
+      const ImageSource& imageSource,
+      const ImageRequestParams& imageRequestParams,
+      SurfaceId surfaceId,
+      Tag tag) const;
+
+ private:
+  jni::global_ref<jobject> fabricUIManager_;
+  jni::JMethod<void(SurfaceId, Tag, JReadableMapBuffer::javaobject)>
+      requestImage_;
+};
+} // namespace facebook::react

packages/react-native/ReactCommon/react/renderer/imagemanager/platform/android/ImageManager.cpp

@@ -7,16 +7,18 @@
 
 #include "ImageManager.h"
 
+#include <react/featureflags/ReactNativeFeatureFlags.h>
+#include "ImageFetcher.h"
+
 namespace facebook::react {
 
-ImageManager::ImageManager(
-    const ContextContainer::Shared& /*contextContainer*/) {
-  // Silence unused-private-field warning.
-  (void)self_;
-  // Not implemented.
-}
+ImageManager::ImageManager(const ContextContainer::Shared& contextContainer)
+    : self_(new ImageFetcher(contextContainer)) {}
 
-ImageManager::~ImageManager() = default;
+ImageManager::~ImageManager() {
+  // NOLINT(cppcoreguidelines-no-malloc)
+  free(self_);
+}
 
 ImageRequest ImageManager::requestImage(
     const ImageSource& imageSource,
@@ -26,10 +28,16 @@ ImageRequest ImageManager::requestImage(
 
 ImageRequest ImageManager::requestImage(
     const ImageSource& imageSource,
-    SurfaceId /*surfaceId*/,
-    const ImageRequestParams& /*imageRequestParams*/,
-    Tag /* tag */) const {
-  return {imageSource, nullptr, {}};
+    SurfaceId surfaceId,
+    const ImageRequestParams& imageRequestParams,
+    Tag tag) const {
+  if (ReactNativeFeatureFlags::enableImagePrefetchingAndroid()) {
+    // NOLINT(cppcoreguidelines-pro-type-cstyle-cast)
+    return ((ImageFetcher*)self_)
+        ->requestImage(imageSource, imageRequestParams, surfaceId, tag);
+  } else {
+    return {imageSource, nullptr, {}};
+  }
 }
 
 } // namespace facebook::react