diff --git a/app/src/main/java/org/thoughtcrime/securesms/components/FullScreenDialogFragment.java b/app/src/main/java/org/thoughtcrime/securesms/components/FullScreenDialogFragment.java index 036667433b..b2dfdd0a34 100644 --- a/app/src/main/java/org/thoughtcrime/securesms/components/FullScreenDialogFragment.java +++ b/app/src/main/java/org/thoughtcrime/securesms/components/FullScreenDialogFragment.java @@ -13,6 +13,7 @@ import androidx.appcompat.widget.Toolbar; import androidx.fragment.app.DialogFragment; import org.thoughtcrime.securesms.R; +import org.thoughtcrime.securesms.util.WindowUtil; /** * Base dialog fragment for rendering as a full screen dialog with animation @@ -43,6 +44,12 @@ public abstract class FullScreenDialogFragment extends DialogFragment { return view; } + @Override + public void onResume() { + super.onResume(); + WindowUtil.initializeScreenshotSecurity(requireContext(), requireDialog().getWindow()); + } + protected void onNavigateUp() { dismissAllowingStateLoss(); } diff --git a/app/src/main/java/org/thoughtcrime/securesms/compose/ComposeFullScreenDialogFragment.kt b/app/src/main/java/org/thoughtcrime/securesms/compose/ComposeFullScreenDialogFragment.kt index 8e97b64887..db0f8518f8 100644 --- a/app/src/main/java/org/thoughtcrime/securesms/compose/ComposeFullScreenDialogFragment.kt +++ b/app/src/main/java/org/thoughtcrime/securesms/compose/ComposeFullScreenDialogFragment.kt @@ -12,6 +12,7 @@ import androidx.fragment.app.DialogFragment import org.signal.core.ui.compose.theme.SignalTheme import org.thoughtcrime.securesms.R import org.thoughtcrime.securesms.util.DynamicTheme +import org.thoughtcrime.securesms.util.WindowUtil /** * Generic ComposeFragment which can be subclassed to build UI with compose. @@ -36,6 +37,11 @@ abstract class ComposeFullScreenDialogFragment : DialogFragment() { } } + override fun onResume() { + super.onResume() + WindowUtil.initializeScreenshotSecurity(requireContext(), requireDialog().window!!) + } + @Composable abstract fun DialogContent() }