From 23abc02d69065037083b0fc80aff317e0203df28 Mon Sep 17 00:00:00 2001
From: Chirayu Desai <chirayudesai1@gmail.com>
Date: Fri, 20 Jan 2023 23:20:04 +0530
Subject: [PATCH] Disable screenshots on more recovery code screens

* Basically try to cover all screens that may have a recovery code

Fixes: #506
Change-Id: If0143b71596ff470a04982d8f87b7a622d362309
---
 .../seedvault/ui/recoverycode/RecoveryCodeInputFragment.kt    | 3 +++
 .../seedvault/ui/recoverycode/RecoveryCodeOutputFragment.kt   | 4 ++++
 2 files changed, 7 insertions(+)

diff --git a/app/src/main/java/com/stevesoltys/seedvault/ui/recoverycode/RecoveryCodeInputFragment.kt b/app/src/main/java/com/stevesoltys/seedvault/ui/recoverycode/RecoveryCodeInputFragment.kt
index a0d507a6..76ee2d6f 100644
--- a/app/src/main/java/com/stevesoltys/seedvault/ui/recoverycode/RecoveryCodeInputFragment.kt
+++ b/app/src/main/java/com/stevesoltys/seedvault/ui/recoverycode/RecoveryCodeInputFragment.kt
@@ -14,6 +14,7 @@ import android.view.View.GONE
 import android.view.View.OnFocusChangeListener
 import android.view.View.VISIBLE
 import android.view.ViewGroup
+import android.view.WindowManager.LayoutParams.FLAG_SECURE
 import android.widget.ArrayAdapter
 import android.widget.AutoCompleteTextView
 import android.widget.Button
@@ -70,6 +71,8 @@ class RecoveryCodeInputFragment : Fragment() {
     ): View {
         val v: View = inflater.inflate(R.layout.fragment_recovery_code_input, container, false)
 
+        if (!isDebugBuild()) getActivity()?.window?.addFlags(FLAG_SECURE)
+
         introText = v.findViewById(R.id.introText)
         doneButton = v.findViewById(R.id.doneButton)
         newCodeButton = v.findViewById(R.id.newCodeButton)
diff --git a/app/src/main/java/com/stevesoltys/seedvault/ui/recoverycode/RecoveryCodeOutputFragment.kt b/app/src/main/java/com/stevesoltys/seedvault/ui/recoverycode/RecoveryCodeOutputFragment.kt
index 9cb65d47..012ce09f 100644
--- a/app/src/main/java/com/stevesoltys/seedvault/ui/recoverycode/RecoveryCodeOutputFragment.kt
+++ b/app/src/main/java/com/stevesoltys/seedvault/ui/recoverycode/RecoveryCodeOutputFragment.kt
@@ -5,11 +5,13 @@ import android.os.Bundle
 import android.view.LayoutInflater
 import android.view.View
 import android.view.ViewGroup
+import android.view.WindowManager.LayoutParams.FLAG_SECURE
 import android.widget.Button
 import androidx.fragment.app.Fragment
 import androidx.recyclerview.widget.GridLayoutManager
 import androidx.recyclerview.widget.RecyclerView
 import com.stevesoltys.seedvault.R
+import com.stevesoltys.seedvault.isDebugBuild
 import org.koin.androidx.viewmodel.ext.android.sharedViewModel
 
 class RecoveryCodeOutputFragment : Fragment() {
@@ -26,6 +28,8 @@ class RecoveryCodeOutputFragment : Fragment() {
     ): View {
         val v: View = inflater.inflate(R.layout.fragment_recovery_code_output, container, false)
 
+        if (!isDebugBuild()) getActivity()?.window?.addFlags(FLAG_SECURE)
+
         wordList = v.findViewById(R.id.wordList)
         confirmCodeButton = v.findViewById(R.id.confirmCodeButton)