commit 2feb36fc51e085040d7978f919f0dcf9a14a2b5f
author Googler <noreply@google.com> Fri Mar 20 14:53:48 2020 -0700
committer Copybara-Service <copybara-worker@google.com> Fri Mar 20 14:54:46 2020 -0700
tree 45249e91c362df53cd5f2b9110d884a066d78f35
parent 07fb6a542a8c4b588354725415391abd1c26ecdf

Add takeScreenshot() to the stable version of ConsumerWrapper.java

PiperOrigin-RevId: 302104263

src/tools/android/java/com/google/devtools/build/android/desugar/runtime/ConsumerWrapper.java

diff --git a/src/tools/android/java/com/google/devtools/build/android/desugar/runtime/ConsumerWrapper.java b/src/tools/android/java/com/google/devtools/build/android/desugar/runtime/ConsumerWrapper.java
index c2f4f73..436ebb8 100644
--- a/src/tools/android/java/com/google/devtools/build/android/desugar/runtime/ConsumerWrapper.java
+++ b/src/tools/android/java/com/google/devtools/build/android/desugar/runtime/ConsumerWrapper.java
@@ -13,6 +13,8 @@
 // limitations under the License.
 package com.google.devtools.build.android.desugar.runtime;
 
+import android.accessibilityservice.AccessibilityService;
+import android.accessibilityservice.AccessibilityService.ScreenshotResult;
 import android.app.DirectAction;
 import android.os.Bundle;
 import android.os.CancellationSignal;
@@ -93,4 +95,15 @@
         resultExecutor,
         callback != null ? new ConsumerWrapper<List<DirectAction>>(callback) : null);
   }
+
+  public static boolean takeScreenshot(
+      AccessibilityService receiver,
+      int displayId,
+      Executor resultExecutor,
+      j$.util.function.Consumer<ScreenshotResult> callback) {
+    return receiver.takeScreenshot(
+        displayId,
+        resultExecutor,
+        callback != null ? new ConsumerWrapper<ScreenshotResult>(callback) : null);
+  }
 }