| /* |
| * Copyright 2016, The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package com.android.managedprovisioning.provisioning; |
| |
| import static android.app.admin.DevicePolicyManager.ACTION_PROVISION_MANAGED_PROFILE; |
| |
| import android.content.Context; |
| import android.os.UserManager; |
| |
| import com.android.internal.annotations.VisibleForTesting; |
| import com.android.managedprovisioning.R; |
| import com.android.managedprovisioning.common.ProvisionLogger; |
| import com.android.managedprovisioning.finalization.FinalizationController; |
| import com.android.managedprovisioning.model.ProvisioningParams; |
| import com.android.managedprovisioning.task.AbstractProvisioningTask; |
| import com.android.managedprovisioning.task.CopyAccountToUserTask; |
| import com.android.managedprovisioning.task.CreateManagedProfileTask; |
| import com.android.managedprovisioning.task.DeleteNonRequiredAppsTask; |
| import com.android.managedprovisioning.task.DisableInstallShortcutListenersTask; |
| import com.android.managedprovisioning.task.InstallExistingPackageTask; |
| import com.android.managedprovisioning.task.ManagedProfileSettingsTask; |
| import com.android.managedprovisioning.task.SetDevicePolicyTask; |
| import com.android.managedprovisioning.task.StartManagedProfileTask; |
| |
| /** |
| * Controller for Profile Owner provisioning. |
| */ |
| // TODO: Consider splitting this controller into one for managed profile and one for user owner |
| public class ProfileOwnerProvisioningController extends AbstractProvisioningController { |
| private final int mParentUserId; |
| |
| public ProfileOwnerProvisioningController( |
| Context context, |
| ProvisioningParams params, |
| int userId, |
| ProvisioningControllerCallback callback) { |
| this(context, params, userId, callback, new FinalizationController(context)); |
| } |
| |
| @VisibleForTesting |
| ProfileOwnerProvisioningController( |
| Context context, |
| ProvisioningParams params, |
| int userId, |
| ProvisioningControllerCallback callback, |
| FinalizationController finalizationController) { |
| super(context, params, userId, callback, finalizationController); |
| mParentUserId = userId; |
| } |
| |
| protected void setUpTasks() { |
| if (ACTION_PROVISION_MANAGED_PROFILE.equals(mParams.provisioningAction)) { |
| setUpTasksManagedProfile(); |
| } else { |
| setUpTasksManagedUser(); |
| } |
| } |
| |
| private void setUpTasksManagedProfile() { |
| addTasks( |
| new CreateManagedProfileTask(mContext, mParams, this), |
| new InstallExistingPackageTask(mParams.inferDeviceAdminPackageName(), mContext, |
| mParams, this), |
| new SetDevicePolicyTask(mContext, mParams, this), |
| new ManagedProfileSettingsTask(mContext, mParams, this), |
| new DisableInstallShortcutListenersTask(mContext, mParams, this), |
| new StartManagedProfileTask(mContext, mParams, this), |
| new CopyAccountToUserTask(mParentUserId, mContext, mParams, this)); |
| } |
| |
| private void setUpTasksManagedUser() { |
| addTasks( |
| new DeleteNonRequiredAppsTask(true /* new profile */, mContext, mParams, this), |
| new InstallExistingPackageTask(mParams.inferDeviceAdminPackageName(), mContext, |
| mParams, this), |
| new SetDevicePolicyTask(mContext, mParams, this)); |
| } |
| |
| @Override |
| public synchronized void onSuccess(AbstractProvisioningTask task) { |
| if (task instanceof CreateManagedProfileTask) { |
| // If the task was creating a managed profile, store the profile id |
| mUserId = ((CreateManagedProfileTask) task).getProfileUserId(); |
| } |
| super.onSuccess(task); |
| } |
| |
| @Override |
| protected void performCleanup() { |
| if (ACTION_PROVISION_MANAGED_PROFILE.equals(mParams.provisioningAction) |
| && mCurrentTaskIndex != 0) { |
| ProvisionLogger.logd("Removing managed profile"); |
| UserManager um = mContext.getSystemService(UserManager.class); |
| um.removeUserEvenWhenDisallowed(mUserId); |
| } |
| } |
| |
| @Override protected int getErrorTitle() { |
| return R.string.cant_set_up_profile; |
| } |
| |
| @Override |
| protected int getErrorMsgId(AbstractProvisioningTask task, int errorCode) { |
| return R.string.managed_provisioning_error_text; |
| } |
| |
| @Override |
| protected boolean getRequireFactoryReset(AbstractProvisioningTask task, int errorCode) { |
| if (mParams.isOrganizationOwnedProvisioning) { |
| // Do factory reset if any error occurs during PO provisioning in the admin |
| // integrated flow. |
| return true; |
| } else { |
| return false; |
| } |
| } |
| } |