add litho accessibility rendering to Flipper accessibility mode [1/2]

Summary:
Litho renders differently based on whether applicable accessibility services are enabled. In Flipper's accessibility mode this will be forced (with the option to turn it off) so that you don't have to be running an accessibility service to actually see what someone running an accessibility service would.

Here's an example video of what the re-rendering does (this also happens on toggle of accessibility mode, this is just the settings option to force it):

{F137856647}

Reviewed By: jknoxville

Differential Revision: D9667222

fbshipit-source-id: 292353d89f07734f1e525f795b1d7daf4130e203

android/src/main/java/com/facebook/sonar/plugins/inspector/InspectorSonarPlugin.java

@@ -32,6 +32,7 @@ import com.facebook.sonar.plugins.inspector.descriptors.utils.AccessibilityUtil;
 
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Stack;
 import javax.annotation.Nullable;
 
 public class InspectorSonarPlugin implements SonarPlugin {
@@ -44,6 +45,7 @@ public class InspectorSonarPlugin implements SonarPlugin {
   private TouchOverlayView mTouchOverlay;
   private SonarConnection mConnection;
   private @Nullable List<ExtensionCommand> mExtensionCommands;
+  private boolean mShowLithoAccessibilitySettings;
 
   /** An interface for extensions to the Inspector Sonar plugin */
   public interface ExtensionCommand {
@@ -110,6 +112,7 @@ public class InspectorSonarPlugin implements SonarPlugin {
     mApplication = wrapper;
     mScriptingEnvironment = scriptingEnvironment;
     mExtensionCommands = extensions;
+    mShowLithoAccessibilitySettings = false;
   }
 
   @Override
@@ -142,9 +145,13 @@ public class InspectorSonarPlugin implements SonarPlugin {
     connection.receive("getAXRoot", mGetAXRoot);
     connection.receive("getAXNodes", mGetAXNodes);
     connection.receive("onRequestAXFocus", mOnRequestAXFocus);
+    connection.receive("shouldShowLithoAccessibilitySettings", mShouldShowLithoAccessibilitySettings);
 
     if (mExtensionCommands != null) {
       for (ExtensionCommand extensionCommand : mExtensionCommands) {
+        if (extensionCommand.command().equals("forceLithoAXRender")) {
+          mShowLithoAccessibilitySettings = true;
+        }
         connection.receive(
             extensionCommand.command(), extensionCommand.receiver(mObjectTracker, mConnection));
       }
@@ -166,6 +173,15 @@ public class InspectorSonarPlugin implements SonarPlugin {
     mConnection = null;
   }
 
+  final SonarReceiver mShouldShowLithoAccessibilitySettings =
+          new MainThreadSonarReceiver(mConnection) {
+            @Override
+            public void onReceiveOnMainThread(SonarObject params, SonarResponder responder)
+                    throws Exception {
+              responder.success(new SonarObject.Builder().put("showLithoAccessibilitySettings", mShowLithoAccessibilitySettings).build());
+            }
+          };
+
   final SonarReceiver mGetRoot =
       new MainThreadSonarReceiver(mConnection) {
         @Override

android/src/main/java/com/facebook/sonar/plugins/litho/GenerateLithoAccessibilityRenderExtensionCommand.java

@@ -0,0 +1,84 @@
+// Copyright 2004-present Facebook. All Rights Reserved.
+
+package com.facebook.sonar.plugins.litho;
+
+import android.view.View;
+import android.view.ViewGroup;
+import com.facebook.litho.LithoView;
+import com.facebook.sonar.core.SonarConnection;
+import com.facebook.sonar.core.SonarObject;
+import com.facebook.sonar.core.SonarReceiver;
+import com.facebook.sonar.core.SonarResponder;
+import com.facebook.sonar.plugins.common.MainThreadSonarReceiver;
+import com.facebook.sonar.plugins.inspector.ApplicationWrapper;
+import com.facebook.sonar.plugins.inspector.InspectorSonarPlugin;
+import com.facebook.sonar.plugins.inspector.ObjectTracker;
+
+import java.util.Stack;
+
+public final class GenerateLithoAccessibilityRenderExtensionCommand implements InspectorSonarPlugin.ExtensionCommand {
+
+    @Override
+    public String command() {
+        return "forceLithoAXRender";
+    }
+
+    @Override
+    public SonarReceiver receiver(final ObjectTracker tracker, final SonarConnection connection) {
+        return new MainThreadSonarReceiver(connection) {
+            @Override
+            public void onReceiveOnMainThread(final SonarObject params, final SonarResponder responder)
+                    throws Exception {
+                final String applicationId = params.getString("applicationId");
+
+                // check that the application is valid
+                if (applicationId == null) {
+                    return;
+                }
+                final Object obj = tracker.get(applicationId);
+                if (obj != null && !(obj instanceof ApplicationWrapper)) {
+                    return;
+                }
+
+                final ApplicationWrapper applicationWrapper = ((ApplicationWrapper) obj);
+                final boolean forceLithoAXRender = params.getBoolean("forceLithoAXRender");
+                final boolean prevForceLithoAXRender = Boolean.getBoolean("is_accessibility_enabled");
+
+                // nothing has changed, so return
+                if (forceLithoAXRender == prevForceLithoAXRender) {
+                    return;
+                }
+
+                // change property and rerender
+                System.setProperty("is_accessibility_enabled", forceLithoAXRender + "");
+                forceRerenderAllLithoViews(forceLithoAXRender, applicationWrapper);
+            }
+        };
+    }
+
+    private void forceRerenderAllLithoViews(boolean forceLithoAXRender, ApplicationWrapper applicationWrapper) {
+
+        // iterate through tree and rerender all litho views
+        Stack<ViewGroup> lithoViewSearchStack = new Stack<>();
+        for (View root : applicationWrapper.getViewRoots()) {
+            if (root instanceof ViewGroup) {
+                lithoViewSearchStack.push((ViewGroup) root);
+            }
+        }
+
+        while (!lithoViewSearchStack.isEmpty()) {
+            ViewGroup v = lithoViewSearchStack.pop();
+            if (v instanceof LithoView) {
+                // TODO: uncomment once Litho open source updates
+//                ((LithoView) v).rerenderForAccessibility(forceLithoAXRender);
+            } else {
+                for (int i = 0; i < v.getChildCount(); i++) {
+                    View child = v.getChildAt(i);
+                    if (child instanceof ViewGroup) {
+                        lithoViewSearchStack.push((ViewGroup) child);
+                    }
+                }
+            }
+        }
+    }
+}