Add setNativeSelectedPosition command to ReactPickerManager

Summary:
Fabric doesn't support setNativeProps, so we have to use commands instead to set the value of the native component.

Changelog: [Android] [Added] - Add setNativeSelectedPosition command to ReactPickerManager

Reviewed By: mdvacca

Differential Revision: D18007791

fbshipit-source-id: dfa26792205189a336ab43b1c51f43f8f57c8e72

Author: makovkastar

Libraries/Components/Picker/AndroidDialogPickerNativeComponent.js

@@ -10,6 +10,9 @@
 
 'use strict';
 
+import * as React from 'react';
+
+import codegenNativeCommands from 'react-native/Libraries/Utilities/codegenNativeCommands';
 import {requireNativeComponent} from 'react-native';
 
 import type {
@@ -46,8 +49,19 @@ type NativeProps = $ReadOnly<{|
   onSelect?: DirectEventHandler<PickerItemSelectEvent>,
 |}>;
 
-const AndroidDialogPickerNativeComponent: HostComponent<NativeProps> = requireNativeComponent<NativeProps>(
-  'AndroidDialogPicker',
-);
+type NativeType = HostComponent<NativeProps>;
+
+interface NativeCommands {
+  +setNativeSelectedPosition: (
+    viewRef: React.ElementRef<NativeType>,
+    index: number,
+  ) => void;
+}
 
-module.exports = AndroidDialogPickerNativeComponent;
+export const Commands: NativeCommands = codegenNativeCommands<NativeCommands>({
+  supportedCommands: ['setNativeSelectedPosition'],
+});
+
+export default (requireNativeComponent<NativeProps>(
+  'AndroidDialogPicker',
+): NativeType);

Libraries/Components/Picker/AndroidDropdownPickerNativeComponent.js

@@ -10,6 +10,9 @@
 
 'use strict';
 
+import * as React from 'react';
+
+import codegenNativeCommands from 'react-native/Libraries/Utilities/codegenNativeCommands';
 import {requireNativeComponent} from 'react-native';
 
 import type {
@@ -46,8 +49,19 @@ type NativeProps = $ReadOnly<{|
   onSelect?: DirectEventHandler<PickerItemSelectEvent>,
 |}>;
 
-const AndroidDropdownPickerNativeComponent: HostComponent<NativeProps> = requireNativeComponent<NativeProps>(
-  'AndroidDropdownPicker',
-);
+type NativeType = HostComponent<NativeProps>;
+
+interface NativeCommands {
+  +setNativeSelectedPosition: (
+    viewRef: React.ElementRef<NativeType>,
+    index: number,
+  ) => void;
+}
 
-module.exports = AndroidDropdownPickerNativeComponent;
+export const Commands: NativeCommands = codegenNativeCommands<NativeCommands>({
+  supportedCommands: ['setNativeSelectedPosition'],
+});
+
+export default (requireNativeComponent<NativeProps>(
+  'AndroidDropdownPicker',
+): NativeType);

ReactAndroid/src/main/java/com/facebook/react/views/picker/ReactPicker.java

@@ -123,6 +123,15 @@ public OnSelectListener getOnSelectListener() {
     mStagedSelection = selection;
   }
 
+  /** Will set the "selection" value immediately as opposed to {@link #setStagedSelection(int)} */
+  /* package */ void setImmediateSelection(int selection) {
+    if (selection != getSelectedItemPosition()) {
+      setOnItemSelectedListener(null);
+      setSelection(selection, false);
+      setOnItemSelectedListener(mItemSelectedListener);
+    }
+  }
+
   /* package */ void setStagedPrimaryTextColor(@Nullable Integer primaryColor) {
     mStagedPrimaryTextColor = primaryColor;
   }

ReactAndroid/src/main/java/com/facebook/react/views/picker/ReactPickerManager.java

@@ -8,6 +8,7 @@
 package com.facebook.react.views.picker;
 
 import android.widget.Spinner;
+import androidx.annotation.NonNull;
 import androidx.annotation.Nullable;
 import com.facebook.react.bridge.ReadableArray;
 import com.facebook.react.uimanager.SimpleViewManager;
@@ -66,6 +67,18 @@ protected void addEventEmitters(final ThemedReactContext reactContext, final Rea
             picker, reactContext.getNativeModule(UIManagerModule.class).getEventDispatcher()));
   }
 
+  @Override
+  public void receiveCommand(
+      @NonNull ReactPicker view, String commandId, @Nullable ReadableArray args) {
+    switch (commandId) {
+      case "setNativeSelectedPosition":
+        if (args != null) {
+          view.setImmediateSelection(args.getInt(0));
+          break;
+        }
+    }
+  }
+
   private static class PickerEventEmitter implements ReactPicker.OnSelectListener {
 
     private final ReactPicker mReactPicker;