Enable profile anyway after certain timeout

Fix: 29214877

Change-Id: Ia6c102ce05dfdf2e79748697f6fc58d4298571db

Author: tony-mak

app/src/main/AndroidManifest.xml

@@ -103,6 +103,9 @@
             <intent-filter>
                 <action android:name="com.google.android.work.action.FIRST_ACCOUNT_READY"/>
             </intent-filter>
+            <intent-filter>
+                <action android:name="com.afwsamples.testdpc.FIRST_ACCOUNT_READY_TIMEOUT"/>
+            </intent-filter>
         </receiver>
 
         <provider

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

@@ -59,13 +59,21 @@ public class EnableProfileActivity extends Activity implements NavigationBar.Nav
     public static final String EXTRA_ENABLE_PROFILE_NOW = "enable_profile_now";
     private static final IntentFilter sIntentFilter =
             new IntentFilter(FIRST_ACCOUNT_READY_PROCESSED_ACTION);
+    private static final long WAIT_FOR_FIRST_ACCOUNT_READY_TIMEOUT = 60 * 1000;
 
     @Override
     protected void onCreate(Bundle savedInstanceState) {
         super.onCreate(savedInstanceState);
         mEnableProfileNow = getIntent().getBooleanExtra(EXTRA_ENABLE_PROFILE_NOW, false);
-        if (savedInstanceState == null && mEnableProfileNow) {
-            ProvisioningUtil.enableProfile(this);
+        if (savedInstanceState == null) {
+            if (mEnableProfileNow) {
+                ProvisioningUtil.enableProfile(this);
+            } else {
+                // Set up an alarm to enable profile in case we do not receive first account
+                // broadcast for whatever reason.
+                FirstAccountReadyBroadcastReceiver.scheduleFirstAccountReadyTimeoutAlarm(
+                        this, WAIT_FOR_FIRST_ACCOUNT_READY_TIMEOUT);
+            }
         }
         setContentView(R.layout.enable_profile_activity);
         mSetupWizardLayout = (SetupWizardLayout) findViewById(R.id.setup_wizard_layout);

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

@@ -16,12 +16,14 @@
 
 package com.afwsamples.testdpc;
 
-import android.app.admin.DevicePolicyManager;
+import android.app.AlarmManager;
+import android.app.PendingIntent;
 import android.content.BroadcastReceiver;
 import android.content.ComponentName;
 import android.content.Context;
 import android.content.Intent;
 import android.content.pm.PackageManager;
+import android.os.SystemClock;
 import android.util.Log;
 
 import com.afwsamples.testdpc.provision.CheckInState;
@@ -40,12 +42,19 @@ public class FirstAccountReadyBroadcastReceiver extends BroadcastReceiver {
     private static final String FIRST_ACCOUNT_READY_ACTION =
             "com.google.android.work.action.FIRST_ACCOUNT_READY";
 
+    public static final String FIRST_ACCOUNT_READY_TIMEOUT_ACTION =
+            "com.afwsamples.testdpc.FIRST_ACCOUNT_READY_TIMEOUT";
+
     public void onReceive(Context context, Intent intent) {
-        Log.d(TAG, "Received: " + intent.getAction());
-        if (FIRST_ACCOUNT_READY_ACTION.equals(intent.getAction())) {
+        final String action = intent.getAction();
+        Log.d(TAG, "Received: " + action);
+        if (FIRST_ACCOUNT_READY_ACTION.equals(action) ||
+                FIRST_ACCOUNT_READY_TIMEOUT_ACTION.equals(action)) {
             CheckInState checkInState = new CheckInState(context);
-            checkInState.setFirstAccountReady();
-            ProvisioningUtil.enableProfile(context);
+            if (!checkInState.isFirstAccountReady()) {
+                checkInState.setFirstAccountReady();
+                ProvisioningUtil.enableProfile(context);
+            }
             // This receiver is disabled in ProvisioningUtil.enableProfile, no more code should
             // be put after it.
         }
@@ -59,4 +68,26 @@ public static void setEnabled(Context context, boolean enabled) {
                 DONT_KILL_APP
         );
     }
+
+    /**
+     * Enable profile anyway if we cannot receive the broadcast after certain amount time.
+     */
+    public static void scheduleFirstAccountReadyTimeoutAlarm(Context context, long timeout) {
+        AlarmManager alarmManager = (AlarmManager) context.getSystemService(Context.ALARM_SERVICE);
+        alarmManager.set(AlarmManager.ELAPSED_REALTIME_WAKEUP,
+                SystemClock.elapsedRealtime() + timeout,
+                createFirstAccountReadyTimeoutPendingIntent(context));
+    }
+
+    public static void cancelFirstAccountReadyTimeoutAlarm(Context context) {
+        AlarmManager alarmManager = (AlarmManager) context.getSystemService(Context.ALARM_SERVICE);
+        alarmManager.cancel(createFirstAccountReadyTimeoutPendingIntent(context));
+    }
+
+    private static PendingIntent createFirstAccountReadyTimeoutPendingIntent(Context context) {
+        Intent intent = new Intent(context, FirstAccountReadyBroadcastReceiver.class);
+        intent.setAction(FirstAccountReadyBroadcastReceiver.FIRST_ACCOUNT_READY_TIMEOUT_ACTION);
+        return PendingIntent.getBroadcast(
+                context, 0, intent, PendingIntent.FLAG_UPDATE_CURRENT);
+    }
 }

app/src/main/java/com/afwsamples/testdpc/provision/ProvisioningUtil.java

@@ -26,6 +26,7 @@
 
 public class ProvisioningUtil {
     public static void enableProfile(Context context) {
+        FirstAccountReadyBroadcastReceiver.cancelFirstAccountReadyTimeoutAlarm(context);
         DevicePolicyManager manager = (DevicePolicyManager) context.getSystemService(
                 Context.DEVICE_POLICY_SERVICE);
         ComponentName componentName = DeviceAdminReceiver.getComponentName(context);