Some fixes to the test app's screenshot detection trigger (#4145)

* WIP - Screenshot feedback trigger

* Get screenshot trigger working

* Some fixes and refactors

* Format java

* Resolve TODOs in SDK

* Update api.txt

* Address PR feedback and fix tests

* Address Kai's feedback

* Request different permission on 33+ and some more feedback

* Wait for pending screenshots

Co-authored-by: Kai Bolay <[email protected]>

Author: lfkellogg

firebase-appdistribution/src/main/java/com/google/firebase/appdistribution/impl/FeedbackActivity.java

@@ -86,7 +86,7 @@ private Bitmap readThumbnail() {
       thumbnail =
           ImageUtils.readScaledImage(
               getContentResolver(), screenshotUri, THUMBNAIL_WIDTH, THUMBNAIL_HEIGHT);
-    } catch (IOException e) {
+    } catch (IOException | SecurityException e) {
       LogWrapper.getInstance()
           .e(TAG, "Could not read screenshot image from URI: " + screenshotUri, e);
       return null;

firebase-appdistribution/src/main/java/com/google/firebase/appdistribution/impl/ImageUtils.java

@@ -53,6 +53,8 @@ static ImageSize read(InputStream inputStream) {
    *
    * @return the image, or null if it could not be decoded
    * @throws IllegalArgumentException if target height or width are less than or equal to zero
+   * @throws SecurityException if the app does not have access to the image
+   * @throws IOException if the image can't be read
    */
   @Nullable
   public static Bitmap readScaledImage(

firebase-appdistribution/test-app/src/main/java/com/googletest/firebase/appdistribution/testapp/ScreenshotDetectionFeedbackTrigger.java

@@ -45,10 +45,8 @@ class ScreenshotDetectionFeedbackTrigger extends ContentObserver {
   private static final boolean SHOULD_CHECK_IF_PENDING = Build.VERSION.SDK_INT >= 29;
   private static final String[] PROJECTION =
       SHOULD_CHECK_IF_PENDING
-          ? new String[] {MediaStore.Images.Media.DATA}
-          : new String[] {
-            MediaStore.Images.Media.DATA, android.provider.MediaStore.MediaColumns.IS_PENDING
-          };
+          ? new String[] {MediaStore.Images.Media.DATA, MediaStore.MediaColumns.IS_PENDING}
+          : new String[] {MediaStore.Images.Media.DATA};
   private final Set<Uri> seenImages = new HashSet<>();
 
   private final AppCompatActivity activity;
@@ -58,6 +56,8 @@ class ScreenshotDetectionFeedbackTrigger extends ContentObserver {
 
   private Uri currentUri;
 
+  private boolean hasRequestedPermission = false;
+
   /**
    * Creates a FeedbackTriggers instance for an activity.
    *
@@ -102,13 +102,25 @@ void unRegisterScreenshotObserver() {
   @Override
   public void onChange(boolean selfChange, Uri uri) {
     if (!uri.toString().matches(String.format("%s/[0-9]+", Media.EXTERNAL_CONTENT_URI))
-        || !seenImages.add(uri)) {
+        || seenImages.contains(uri)) {
       return;
     }
 
     if (ContextCompat.checkSelfPermission(activity, PERMISSION_TO_REQUEST) == PERMISSION_GRANTED) {
       maybeStartFeedbackForScreenshot(uri);
-    } else if (activity.shouldShowRequestPermissionRationale(PERMISSION_TO_REQUEST)) {
+    } else if (hasRequestedPermission) {
+      Log.i(
+          TAG,
+          "We've already request permission. Not requesting again for the life of the activity.");
+    } else {
+      // Set an in memory flag so we don't ask them again right away
+      hasRequestedPermission = true;
+      requestReadPermission(uri);
+    }
+  }
+
+  private void requestReadPermission(Uri uri) {
+    if (activity.shouldShowRequestPermissionRationale(PERMISSION_TO_REQUEST)) {
       Log.i(TAG, "Showing customer rationale for requesting permission.");
       new AlertDialog.Builder(activity)
           .setMessage(
@@ -122,7 +134,7 @@ public void onChange(boolean selfChange, Uri uri) {
               })
           .show();
     } else {
-      Log.i(TAG, "Does not have permission. Launching request.");
+      Log.i(TAG, "Launching request for permission without rationale.");
       currentUri = uri;
       requestPermissionLauncher.launch(PERMISSION_TO_REQUEST);
     }
@@ -133,8 +145,13 @@ private void maybeStartFeedbackForScreenshot(Uri uri) {
     try {
       cursor = activity.getContentResolver().query(uri, PROJECTION, null, null, null);
       if (cursor != null && cursor.moveToFirst()) {
-        // TODO: check if it's pending
-        // (http://google3/lens/screenshots/demo/java/com/google/android/lensonscreenshots/ScreenshotDetector.java?l=184)
+        if (SHOULD_CHECK_IF_PENDING
+            && cursor.getInt(cursor.getColumnIndexOrThrow(MediaStore.MediaColumns.IS_PENDING))
+                == 1) {
+          Log.i(TAG, "Ignoring pending image: " + uri);
+          return;
+        }
+        seenImages.add(uri);
         String path = cursor.getString(cursor.getColumnIndexOrThrow(MediaStore.Images.Media.DATA));
         Log.i(TAG, "Path: " + path);
         if (path.toLowerCase().contains("screenshot")) {