1 /*
2  * Copyright (C) 2018 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 package com.android.settings.enterprise;
18 
19 import android.app.Activity;
20 import android.app.admin.DevicePolicyManager;
21 import android.content.DialogInterface;
22 import android.content.Intent;
23 import android.os.Bundle;
24 import android.os.UserHandle;
25 
26 import com.android.settingslib.RestrictedLockUtils;
27 import com.android.settingslib.RestrictedLockUtils.EnforcedAdmin;
28 
29 public class ActionDisabledByAdminDialog extends Activity
30         implements DialogInterface.OnDismissListener {
31 
32     private ActionDisabledByAdminDialogHelper mDialogHelper;
33 
34     @Override
onCreate(Bundle savedInstanceState)35     protected void onCreate(Bundle savedInstanceState) {
36         super.onCreate(savedInstanceState);
37         final RestrictedLockUtils.EnforcedAdmin enforcedAdmin =
38                 getAdminDetailsFromIntent(getIntent());
39         final String restriction = getRestrictionFromIntent(getIntent());
40         mDialogHelper = new ActionDisabledByAdminDialogHelper(this, restriction);
41         mDialogHelper.prepareDialogBuilder(restriction, enforcedAdmin)
42                 .setOnDismissListener(this)
43                 .show();
44     }
45 
46     @Override
onNewIntent(Intent intent)47     public void onNewIntent(Intent intent) {
48         super.onNewIntent(intent);
49         final EnforcedAdmin admin = getAdminDetailsFromIntent(intent);
50         final String restriction = getRestrictionFromIntent(intent);
51         mDialogHelper.updateDialog(restriction, admin);
52     }
53 
54     @androidx.annotation.VisibleForTesting
getAdminDetailsFromIntent(Intent intent)55     EnforcedAdmin getAdminDetailsFromIntent(Intent intent) {
56         final EnforcedAdmin admin = new EnforcedAdmin(null, UserHandle.of(UserHandle.myUserId()));
57         if (intent == null) {
58             return admin;
59         }
60         admin.component = intent.getParcelableExtra(DevicePolicyManager.EXTRA_DEVICE_ADMIN);
61         int userId = intent.getIntExtra(Intent.EXTRA_USER_ID, UserHandle.myUserId());
62 
63         Bundle adminDetails = null;
64         if (admin.component == null) {
65             DevicePolicyManager devicePolicyManager = getSystemService(DevicePolicyManager.class);
66             adminDetails = devicePolicyManager.getEnforcingAdminAndUserDetails(userId,
67                     getRestrictionFromIntent(intent));
68             if (adminDetails != null) {
69                 admin.component = adminDetails.getParcelable(
70                         DevicePolicyManager.EXTRA_DEVICE_ADMIN);
71             }
72         }
73 
74         if (intent.hasExtra(Intent.EXTRA_USER)) {
75             admin.user = intent.getParcelableExtra(Intent.EXTRA_USER);
76         } else {
77             if (adminDetails != null) {
78                 userId = adminDetails.getInt(Intent.EXTRA_USER_ID, UserHandle.myUserId());
79             }
80             if (userId == UserHandle.USER_NULL) {
81                 admin.user = null;
82             } else {
83                 admin.user = UserHandle.of(userId);
84             }
85         }
86         return admin;
87     }
88 
89     @androidx.annotation.VisibleForTesting
getRestrictionFromIntent(Intent intent)90     String getRestrictionFromIntent(Intent intent) {
91         if (intent == null) return null;
92         return intent.getStringExtra(DevicePolicyManager.EXTRA_RESTRICTION);
93     }
94 
95     @Override
onDismiss(DialogInterface dialog)96     public void onDismiss(DialogInterface dialog) {
97         finish();
98     }
99 }
100