Added commands to set / get user location.

Test: adb shell dumpsys activity service --user 0 com.afwsamples.testdpc is-location-enabled
Test: adb shell dumpsys activity service --user 0 com.afwsamples.testdpc set-location-enabled false

Bug: 171350084
Bug: 185523465

Change-Id: I98d770a3fdeb2f9ab0d75cd0f34e0dee090e4ae4

Author: the-felipeal

app/src/main/java/com/afwsamples/testdpc/DevicePolicyManagerGateway.java

@@ -402,6 +402,17 @@ void setPermissionGrantState(String packageName, String permission, int grantSta
      */
     int getPermissionGrantState(String packageName, String permission);
 
+    /**
+     * See {@link android.app.admin.DevicePolicyManager#setLocationEnabled(ComponentName, boolean)}.
+     */
+    void setLocationEnabled(boolean enabled, @NonNull Consumer<Void> onSuccess,
+            @NonNull Consumer<Exception> onError);
+
+    /**
+     * See {@link android.location.LocationManager#isLocationEnabled()}.
+     */
+    boolean isLocationEnabled();
+
     /**
      * Used on error callbacks to indicate a {@link android.app.admin.DevicePolicyManager} method
      * call failed.

app/src/main/java/com/afwsamples/testdpc/DevicePolicyManagerGatewayImpl.java

@@ -24,6 +24,7 @@
 import android.content.pm.PackageManager;
 import android.content.pm.PackageManager.NameNotFoundException;
 import android.graphics.Bitmap;
+import android.location.LocationManager;
 import android.os.Build;
 import android.os.Bundle;
 import android.os.PersistableBundle;
@@ -54,19 +55,22 @@ public final class DevicePolicyManagerGatewayImpl implements DevicePolicyManager
     private final UserManager mUserManager;
     private final ComponentName mAdminComponentName;
     private final PackageManager mPackageManager;
+    private final LocationManager mLocationManager;
 
     public DevicePolicyManagerGatewayImpl(@NonNull Context context) {
         this(context.getSystemService(DevicePolicyManager.class),
                 context.getSystemService(UserManager.class),
                 context.getPackageManager(),
+                context.getSystemService(LocationManager.class),
                 DeviceAdminReceiver.getComponentName(context));
     }
 
     public DevicePolicyManagerGatewayImpl(@NonNull DevicePolicyManager dpm, @NonNull UserManager um,
-            @NonNull PackageManager pm, @NonNull ComponentName admin) {
+            @NonNull PackageManager pm, @NonNull LocationManager lm, @NonNull ComponentName admin) {
         mDevicePolicyManager = dpm;
         mUserManager = um;
         mPackageManager = pm;
+        mLocationManager = lm;
         mAdminComponentName = admin;
 
         Log.d(TAG, "constructor: admin=" + mAdminComponentName + ", dpm=" + dpm);
@@ -78,7 +82,8 @@ public static DevicePolicyManagerGatewayImpl forParentProfile(@NonNull Context c
                 .getParentProfileInstance(admin);
         UserManager um = context.getSystemService(UserManager.class);
         PackageManager pm = context.getPackageManager();
-        return new DevicePolicyManagerGatewayImpl(dpm, um, pm, admin);
+        LocationManager lm = context.getSystemService(LocationManager.class);
+        return new DevicePolicyManagerGatewayImpl(dpm, um, pm, lm, admin);
     }
 
     @Override
@@ -792,6 +797,23 @@ public int getPermissionGrantState(String packageName, String permission) {
         return grantState;
     }
 
+    @Override
+    public void setLocationEnabled(boolean enabled, Consumer<Void> onSuccess,
+            Consumer<Exception> onError) {
+        Log.d(TAG, "setLocationEnabled(" + enabled + ")");
+        try {
+            mDevicePolicyManager.setLocationEnabled(mAdminComponentName, enabled);
+            onSuccess.accept(null);
+        } catch (Exception e) {
+            onError.accept(e);
+        }
+    }
+
+    @Override
+    public boolean isLocationEnabled() {
+        return mLocationManager.isLocationEnabled();
+    }
+
     @Override
     public String toString() {
         return "DevicePolicyManagerGatewayImpl[" + mAdminComponentName + "]";

app/src/main/java/com/afwsamples/testdpc/ShellCommand.java

@@ -97,6 +97,8 @@ final class ShellCommand {
     private static final String CMD_GET_APP_RESTRICTIONS = "get-app-restrictions";
     private static final String CMD_SET_PERMISSION_GRANT_STATE = "set-permission-grant-state";
     private static final String CMD_GET_PERMISSION_GRANT_STATE = "get-permission-grant-state";
+    private static final String CMD_SET_LOCATION_ENABLED = "set-location-enabled";
+    private static final String CMD_IS_LOCATION_ENABLED = "is-location-enabled";
 
     // Commands for APIs added on Android S
     private static final String CMD_SET_PERMITTED_INPUT_METHODS_PARENT =
@@ -277,6 +279,12 @@ public void run() {
             case CMD_GET_PERMISSION_GRANT_STATE:
                 execute(() -> getPermissionGrantState());
                 break;
+            case CMD_SET_LOCATION_ENABLED:
+                execute(() -> setLocationEnabled());
+                break;
+            case CMD_IS_LOCATION_ENABLED:
+                execute(() -> isLocationEnabled());
+                break;
             default:
                 mWriter.printf("Invalid command: %s\n\n", cmd);
                 showUsage();
@@ -382,6 +390,10 @@ private void showUsage() {
                 CMD_SET_PERMISSION_GRANT_STATE);
         mWriter.printf("\t%s <PKG> <PERMISSION> - get the grant state for the given permission / "
                 + "package\n", CMD_GET_PERMISSION_GRANT_STATE);
+        mWriter.printf("\t%s <true|false> - set location enabled for the user\n",
+                CMD_SET_LOCATION_ENABLED);
+        mWriter.printf("\t%s - get whether location is enabled for the user\n",
+                CMD_IS_LOCATION_ENABLED);
 
         // Separator for S / pre-S commands - do NOT remove line to avoid cherry-pick conflicts
 
@@ -891,6 +903,20 @@ private void getPermissionGrantState() {
                 Util.grantStateToString(grantState));
     }
 
+    private void setLocationEnabled() {
+        // TODO(b/171350084): check args
+        boolean enabled = Boolean.parseBoolean(mArgs[1]);
+
+        mDevicePolicyManagerGateway.setLocationEnabled(enabled,
+                (v) -> onSuccess("Set location enabled to %b", enabled),
+                (e) -> onError(e, "Error setting location enabled to %b", enabled));
+    }
+
+    private void isLocationEnabled() {
+        boolean enabled = mDevicePolicyManagerGateway.isLocationEnabled();
+        mWriter.printf("Location enabled: %b\n", enabled);
+    }
+
     private static String permittedToString(boolean permitted) {
         return permitted ? "PERMITTED" : "NOT PERMITTED";
     }

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

@@ -535,22 +535,22 @@ public class PolicyManagementFragment extends BaseSearchablePolicyPreferenceFrag
 
     @Override
     public void onCreate(Bundle savedInstanceState) {
+        Context context = getActivity();
         if (isDelegatedApp() || isCredentialManagerApp()) {
             mAdminComponentName = null;
         } else {
-            mAdminComponentName = DeviceAdminReceiver.getComponentName(getActivity());
+            mAdminComponentName = DeviceAdminReceiver.getComponentName(context);
         }
-        mDevicePolicyManager = (DevicePolicyManager) getActivity().getSystemService(
-                Context.DEVICE_POLICY_SERVICE);
-        mUserManager = (UserManager) getActivity().getSystemService(Context.USER_SERVICE);
-        mPackageManager = getActivity().getPackageManager();
+        mDevicePolicyManager = context.getSystemService(DevicePolicyManager.class);
+        mUserManager = context.getSystemService(UserManager.class);
+        mPackageManager = context.getPackageManager();
         mDevicePolicyManagerGateway = new DevicePolicyManagerGatewayImpl(mDevicePolicyManager,
-                mUserManager, mPackageManager, mAdminComponentName);
+                mUserManager, mPackageManager, context.getSystemService(LocationManager.class),
+                mAdminComponentName);
         mIsProfileOwner = mDevicePolicyManagerGateway.isProfileOwnerApp();
-        mTelephonyManager = (TelephonyManager) getActivity()
-                .getSystemService(Context.TELEPHONY_SERVICE);
-        mAccountManager = AccountManager.get(getActivity());
-        mPackageName = getActivity().getPackageName();
+        mTelephonyManager = context.getSystemService(TelephonyManager.class);
+        mAccountManager = AccountManager.get(context);
+        mPackageName = context.getPackageName();
 
         mImageUri = getStorageUri("image.jpg");
         mVideoUri = getStorageUri("video.mp4");