From 0cd26d4a7b3f750d03b01a5e9ee4a13520981d25 Mon Sep 17 00:00:00 2001
From: althafvly <althafvly@gmail.com>
Date: Fri, 12 Jul 2024 20:59:15 +0530
Subject: [PATCH] parental: Check for workprofile users

---
 .../java/foundation/e/parentalcontrol/MainActivity.kt | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/app/src/main/java/foundation/e/parentalcontrol/MainActivity.kt b/app/src/main/java/foundation/e/parentalcontrol/MainActivity.kt
index ab67a7d..9e7adeb 100644
--- a/app/src/main/java/foundation/e/parentalcontrol/MainActivity.kt
+++ b/app/src/main/java/foundation/e/parentalcontrol/MainActivity.kt
@@ -8,6 +8,7 @@
 package foundation.e.parentalcontrol
 
 import android.app.admin.DevicePolicyManager
+import android.app.admin.DevicePolicyManager.ACTION_PROVISION_MANAGED_PROFILE
 import android.content.ComponentName
 import android.content.Context
 import android.content.Intent
@@ -436,9 +437,12 @@ class MainActivity : ComponentActivity() {
         ) {
             val devicePolicyManager: DevicePolicyManager = dA.getDevicePolicyManager(mActivity)
             val activeAdmins = devicePolicyManager.activeAdmins
+            var showDialog by remember { mutableStateOf(false) }
 
             if (
-                !activeAdmins.isNullOrEmpty() && !activeAdmins.contains(dA.getAdminName(mActivity))
+                (!activeAdmins.isNullOrEmpty() &&
+                    !activeAdmins.contains(dA.getAdminName(mActivity))) ||
+                    !devicePolicyManager.isProvisioningAllowed(ACTION_PROVISION_MANAGED_PROFILE)
             ) {
                 AlertDialog(
                     onDismissRequest = { onExitApp() },
@@ -492,10 +496,11 @@ class MainActivity : ComponentActivity() {
                     },
                     shape = RoundedCornerShape(4.dp)
                 )
+            } else {
+                val userManager = getSystemService(Context.USER_SERVICE) as UserManager
+                showDialog = userManager.userCount > 1
             }
 
-            val userManager = getSystemService(Context.USER_SERVICE) as UserManager
-            var showDialog by remember { mutableStateOf(userManager.userCount > 1) }
             if (showDialog) {
                 AlertDialog(
                     onDismissRequest = { showDialog = false },
-- 
GitLab