Support for maximum profile time off.

Bug: 143517719
Test: manual
Change-Id: I492b9713325aa19f6f67300ba60412c0d0691e6a

Author: Pavel Grafov

app/src/main/java/com/afwsamples/testdpc/policy/PolicyManagementFragment.java

@@ -107,6 +107,7 @@
 import com.afwsamples.testdpc.common.UserArrayAdapter;
 import com.afwsamples.testdpc.common.Util;
 import com.afwsamples.testdpc.common.preference.CustomConstraint;
+import com.afwsamples.testdpc.common.preference.DpcEditTextPreference;
 import com.afwsamples.testdpc.common.preference.DpcPreference;
 import com.afwsamples.testdpc.common.preference.DpcPreferenceBase;
 import com.afwsamples.testdpc.common.preference.DpcPreferenceHelper;
@@ -395,6 +396,7 @@ public class PolicyManagementFragment extends BaseSearchablePolicyPreferenceFrag
     private static final String SET_LOCATION_ENABLED_KEY = "set_location_enabled";
     private static final String SET_LOCATION_MODE_KEY = "set_location_mode";
     private static final String SUSPEND_PERSONAL_APPS_KEY = "suspend_personal_apps";
+    private static final String PROFILE_MAX_TIME_OFF_KEY = "profile_max_time_off";
 
     private static final String BATTERY_PLUGGED_ANY = Integer.toString(
             BatteryManager.BATTERY_PLUGGED_AC |
@@ -478,6 +480,7 @@ public class PolicyManagementFragment extends BaseSearchablePolicyPreferenceFrag
 
     private DpcPreference mUserRestrictionsParentPreference;
     private DpcSwitchPreference mSuspendPersonalApps;
+    private DpcEditTextPreference mProfileMaxTimeOff;
 
     private GetAccessibilityServicesTask mGetAccessibilityServicesTask = null;
     private GetInputMethodsTask mGetInputMethodsTask = null;
@@ -734,6 +737,12 @@ public void onCreatePreferences(Bundle savedInstanceState, String rootKey) {
         mSuspendPersonalApps.setCustomConstraint(
                 this::validateProfileOwnerOfOrganizationOwnedDevice);
 
+        mProfileMaxTimeOff = (DpcEditTextPreference) findPreference(PROFILE_MAX_TIME_OFF_KEY);
+        mProfileMaxTimeOff.setOnPreferenceChangeListener(this);
+        mProfileMaxTimeOff.setCustomConstraint(
+                this::validateProfileOwnerOfOrganizationOwnedDevice);
+        maybeUpdateProfileMaxTimeOff();
+
         onCreateSetNewPasswordWithComplexityPreference();
         constrainSpecialCasePreferences();
 
@@ -756,6 +765,14 @@ public void onCreatePreferences(Bundle savedInstanceState, String rootKey) {
         reloadPersonalAppsSuspendedUi();
     }
 
+    private void maybeUpdateProfileMaxTimeOff() {
+        if (mProfileMaxTimeOff.isEnabled()) {
+            final String currentValueAsString = Long.toString(getManagedProfileMaximumTimeOff());
+            mProfileMaxTimeOff.setText(currentValueAsString);
+            mProfileMaxTimeOff.setSummary(currentValueAsString);
+        }
+    }
+
     @TargetApi(Util.R_VERSION_CODE)
     private void reloadPersonalAppsSuspendedUi() {
         // TODO: nuke it when R sdk is available
@@ -773,7 +790,7 @@ int getPersonalAppsSuspensionReasons() {
                     "getPersonalAppsSuspendedReasons", new Class<?>[]{ComponentName.class},
                     mAdminComponentName);
         } catch (ReflectionIsTemporaryException e) {
-            Log.e(TAG, "Error invoking getPersonalAppsSuspendedReasons", e);
+            logAndShowToast("Error invoking getPersonalAppsSuspendedReasons", e);
             return 0;
         }
     }
@@ -785,10 +802,38 @@ void setPersonalAppsSuspended(boolean suspended) {
                     new Class<?>[]{ComponentName.class, boolean.class},
                     mAdminComponentName, suspended);
         } catch (ReflectionIsTemporaryException e) {
-            Log.e(TAG, "Error invoking setPersonalAppsSuspended", e);
+            logAndShowToast("Error invoking setPersonalAppsSuspended", e);
+        }
+    }
+
+    //TODO: nuke it when R sdk is available.
+    public long getManagedProfileMaximumTimeOff() {
+        try {
+            return (Long) ReflectionUtil.invoke(mDevicePolicyManager,
+                    "getManagedProfileMaximumTimeOff", new Class<?>[]{ComponentName.class},
+                    mAdminComponentName);
+        } catch (ReflectionIsTemporaryException e) {
+            logAndShowToast("Error invoking getManagedProfileMaximumTimeOff", e);
+            return 0;
+        }
+    }
+
+    //TODO: nuke it when R sdk is available.
+    public void setManagedProfileMaximumTimeOff(long timeoutMs) {
+        try {
+            ReflectionUtil.invoke(mDevicePolicyManager, "setManagedProfileMaximumTimeOff",
+                    new Class<?>[]{ComponentName.class, long.class},
+                    mAdminComponentName, timeoutMs);
+        } catch (ReflectionIsTemporaryException e) {
+            logAndShowToast("Error invoking setManagedProfileMaximumTimeOff", e);
         }
     }
 
+    private void logAndShowToast(String message, Exception e) {
+        Log.e(TAG, message, e);
+        showToast(message + ": " + e.getMessage());
+    }
+
     private void onCreateSetNewPasswordWithComplexityPreference() {
         ListPreference complexityPref =
             (ListPreference) findPreference(SET_NEW_PASSWORD_WITH_COMPLEXITY);
@@ -1488,6 +1533,10 @@ public boolean onPreferenceChange(Preference preference, Object newValue) {
                 setPersonalAppsSuspended((Boolean) newValue);
                 reloadPersonalAppsSuspendedUi();
                 return true;
+            case PROFILE_MAX_TIME_OFF_KEY:
+                setManagedProfileMaximumTimeOff(Long.parseLong((String) newValue));
+                maybeUpdateProfileMaxTimeOff();
+                return true;
         }
         return false;
     }

app/src/main/res/values/strings.xml

@@ -616,6 +616,9 @@
     <!-- Suspension of personal apps -->
     <string name="suspend_personal_apps">Suspend personal apps</string>
 
+    <!-- Maximum time work profile is allowed to be off -->
+    <string name="profile_max_time_off">Work profile max time off (seconds)</string>
+
     <!-- Enterprise security logging -->
     <string name="enable_security_logging">Enable security logging</string>
     <string name="request_security_logs">Request security logs</string>

app/src/main/res/xml/device_policy_header.xml

@@ -739,6 +739,14 @@
             android:title="@string/suspend_personal_apps"
             testdpc:admin="profileOwner"
             testdpc:minSdkVersion="R" />
+        <com.afwsamples.testdpc.common.preference.DpcEditTextPreference
+            android:defaultValue="0"
+            android:dialogTitle="@string/profile_max_time_off"
+            android:inputType="number"
+            android:key="profile_max_time_off"
+            android:title="@string/profile_max_time_off"
+            testdpc:admin="profileOwner"
+            testdpc:minSdkVersion="R" />
     </PreferenceCategory>
 
     <PreferenceCategory android:title="@string/transfer_ownership">