Hook screenshot trigger to test app menu

Author: lfkellogg

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

@@ -332,10 +332,10 @@ public void startFeedback(@NonNull CharSequence infoText) {
             taskExecutor,
             e -> {
               LogWrapper.getInstance().w("Failed to take screenshot for feedback", e);
-              startFeedback(infoText, null);
+              doStartFeedback(infoText, null);
             })
         .addOnSuccessListener(
-            taskExecutor, screenshotUri -> startFeedback(infoText, screenshotUri));
+            taskExecutor, screenshotUri -> doStartFeedback(infoText, screenshotUri));
   }
 
   @Override
@@ -345,6 +345,15 @@ public void startFeedback(@NonNull int infoTextResourceId, @Nullable Uri screens
 
   @Override
   public void startFeedback(@NonNull CharSequence infoText, @Nullable Uri screenshotUri) {
+    if (!feedbackInProgress.compareAndSet(/* expect= */ false, /* update= */ true)) {
+      LogWrapper.getInstance()
+          .i("Ignoring startFeedback() call because feedback is already in progress");
+      return;
+    }
+    doStartFeedback(infoText, screenshotUri);
+  }
+
+  private void doStartFeedback(CharSequence infoText, @Nullable Uri screenshotUri) {
     testerSignInManager
         .signInTester()
         .addOnFailureListener(

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

@@ -6,7 +6,11 @@ class AppDistroTestApplication : Application() {
     override fun onCreate() {
         super.onCreate()
 
-        // Default feedback triggers can also be initialized here
+        // Perform any required trigger initialization here
+        ScreenshotDetectionFeedbackTrigger.initialize(this, R.string.terms_and_conditions);
+
+        // Default feedback triggers can optionally be enabled application-wide here
 //        ShakeForFeedback.enable(this)
+//        ScreenshotDetectionFeedbackTrigger.enable()
     }
 }

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

@@ -4,8 +4,6 @@ import android.app.AlertDialog
 import android.content.Intent
 import android.os.Build
 import android.os.Bundle
-import android.os.Handler
-import android.os.HandlerThread
 import android.util.Log
 import android.view.Menu
 import android.view.MenuInflater
@@ -49,9 +47,6 @@ class MainActivity : AppCompatActivity() {
     lateinit var progressBar: ProgressBar
     lateinit var feedbackTriggerMenu: TextInputLayout
 
-    private lateinit var screenshotTriggerThread: HandlerThread
-    private lateinit var screenshotTrigger: ScreenshotDetectionFeedbackTrigger
-
     override fun onCreate(savedInstanceState: Bundle?) {
         super.onCreate(savedInstanceState)
         setContentView(R.layout.activity_main)
@@ -71,18 +66,13 @@ class MainActivity : AppCompatActivity() {
         signInStatus = findViewById(R.id.sign_in_status)
         progressBar = findViewById(R.id.progress_bar)
 
-        screenshotTriggerThread = HandlerThread("AppDistroFeedbackTrigger")
-        screenshotTriggerThread.start()
-        screenshotTrigger =
-            ScreenshotDetectionFeedbackTrigger(
-                this,
-                R.string.terms_and_conditions,
-                Handler(screenshotTriggerThread.looper)
-            )
-
         // Set up feedback trigger menu
         feedbackTriggerMenu = findViewById(R.id.feedbackTriggerMenu)
-        val items = listOf(FeedbackTrigger.NONE.label, FeedbackTrigger.SHAKE.label)
+        val items = listOf(
+            FeedbackTrigger.NONE.label,
+            FeedbackTrigger.SHAKE.label,
+            FeedbackTrigger.SCREENSHOT.label
+        )
         val adapter = ArrayAdapter(this, R.layout.list_item, items)
         val autoCompleteTextView = feedbackTriggerMenu.editText!! as AutoCompleteTextView
         autoCompleteTextView.setAdapter(adapter)
@@ -91,17 +81,28 @@ class MainActivity : AppCompatActivity() {
             // TODO: support enabling/disabling other triggers
             when(text.toString()) {
                 FeedbackTrigger.NONE.label -> {
-                    Log.i(TAG, "Disabling shake")
-                    ShakeForFeedback.disable(application)
+                    disableAllFeedbackTriggers()
                 }
                 FeedbackTrigger.SHAKE.label -> {
-                    Log.i(TAG, "Enabling shake")
+                    disableAllFeedbackTriggers()
+                    Log.i(TAG, "Enabling shake for feedback trigger")
                     ShakeForFeedback.enable(application, this)
                 }
+                FeedbackTrigger.SCREENSHOT.label -> {
+                    disableAllFeedbackTriggers()
+                    Log.i(TAG, "Enabling screenshot detection trigger")
+                    ScreenshotDetectionFeedbackTrigger.enable()
+                }
             }
         }
     }
 
+    private fun disableAllFeedbackTriggers() {
+        Log.i(TAG, "Disabling all feedback triggers")
+        ShakeForFeedback.disable(application)
+        ScreenshotDetectionFeedbackTrigger.disable()
+    }
+
     override fun onCreateOptionsMenu(menu: Menu): Boolean {
         val inflater: MenuInflater = menuInflater
         inflater.inflate(R.menu.action_menu, menu)
@@ -118,19 +119,8 @@ class MainActivity : AppCompatActivity() {
         }
     }
 
-    override fun onDestroy() {
-        super.onDestroy()
-        screenshotTriggerThread.quitSafely()
-    }
-
-    override fun onPause() {
-        super.onPause()
-        screenshotTrigger.unRegisterScreenshotObserver()
-    }
-
     override fun onResume() {
         super.onResume()
-        screenshotTrigger.registerScreenshotObserver()
 
         findViewById<TextView>(R.id.app_name).text =
             "Sample App v${BuildConfig.VERSION_NAME} (${BuildConfig.VERSION_CODE})"