Create entry to PWSharing through ShareHelper

chrome/android/java/src/org/chromium/chrome/browser/share/ShareHelper.java

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 package org.chromium.chrome.browser.share;
 
 import android.annotation.TargetApi;
 import android.app.Activity;
 import android.app.PendingIntent;
 import android.content.BroadcastReceiver;
 import android.content.ClipData;
 import android.content.ComponentName;
 import android.content.Context;
 import android.content.DialogInterface;
 import android.content.DialogInterface.OnDismissListener;
 import android.content.Intent;
 import android.content.IntentFilter;
 import android.content.SharedPreferences;
 import android.content.pm.ActivityInfo;
 import android.content.pm.ApplicationInfo;
 import android.content.pm.PackageManager;
 import android.content.pm.PackageManager.NameNotFoundException;
 import android.content.pm.ResolveInfo;
 import android.graphics.Bitmap;
 import android.graphics.drawable.Drawable;
 import android.net.Uri;
 import android.os.AsyncTask;
 import android.os.Build;
+import android.os.StrictMode;
 import android.support.annotation.Nullable;
 import android.support.v7.app.AlertDialog;
 import android.text.TextUtils;
 import android.util.Pair;
 import android.view.MenuItem;
 import android.view.View;
 import android.widget.AdapterView;
 import android.widget.AdapterView.OnItemClickListener;
 
+import org.chromium.base.ActivityState;
 import org.chromium.base.ApiCompatibilityUtils;
 import org.chromium.base.ApplicationState;
 import org.chromium.base.ApplicationStatus;
+import org.chromium.base.ApplicationStatus.ActivityStateListener;
 import org.chromium.base.Callback;
 import org.chromium.base.ContextUtils;
 import org.chromium.base.Log;
 import org.chromium.base.StreamUtil;
+import org.chromium.base.ThreadUtils;
 import org.chromium.base.VisibleForTesting;
 import org.chromium.base.annotations.SuppressFBWarnings;
 import org.chromium.base.metrics.RecordHistogram;
 import org.chromium.chrome.R;
 import org.chromium.components.dom_distiller.core.DomDistillerUrlUtils;
 import org.chromium.ui.UiUtils;
 
 import java.io.File;
 import java.io.FileOutputStream;
 import java.io.IOException;
 import java.util.Collections;
+import java.util.HashSet;
 import java.util.List;
+import java.util.Set;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.TimeoutException;
 
 /**
  * A helper class that helps to start an intent to share titles and URLs.
  */
 public class ShareHelper {
 
     /** Interface that receives intents for testing (to fake out actually sending them). */
@@ skipping 30 matching lines @@
      * Named "screenshot" for historical reasons as we only initially shared screenshot images.
      */
     private static final String SHARE_IMAGES_DIRECTORY_NAME = "screenshot";
 
     /** Force the use of a Chrome-specific intent chooser, not the system chooser. */
     private static boolean sForceCustomChooserForTesting;
 
     /** If non-null, will be used instead of the real activity. */
     private static FakeIntentReceiver sFakeIntentReceiverForTesting;
 
+    /** Variables for adding options dynamically to app chooser. */
+    private static Set<Activity> sPendingShareActivities =
+            Collections.synchronizedSet(new HashSet<Activity>());
+    private static ActivityStateListener sStateListener;
+    private static AsyncTask<Void, Void, Void> sStateChangeTask;
+
     private ShareHelper() {}
 
     private static void fireIntent(Activity activity, Intent intent) {
         if (sFakeIntentReceiverForTesting != null) {
             Context context = activity.getApplicationContext();
             sFakeIntentReceiverForTesting.fireIntent(context, intent);
         } else {
             activity.startActivity(intent);
         }
     }
@@ skipping 557 matching lines @@
         return intent;
     }
 
     private static ComponentName getLastShareComponentName() {
         SharedPreferences preferences = ContextUtils.getAppSharedPreferences();
         String packageName = preferences.getString(PACKAGE_NAME_KEY, null);
         String className = preferences.getString(CLASS_NAME_KEY, null);
         if (packageName == null || className == null) return null;
         return new ComponentName(packageName, className);
     }
+
+    /**
+     * Dynamically add Chrome activities to Share chooser.
+     *
+     * @param activity The activity that will be triggering the share action.
+     * @param options  List of classes to add to the chooser.
+     * @param callback The callback to be triggered after the options have been enabled.  This
+     *                 may or may not be synchronous depending on whether this will require
+     *                 interacting with the Android framework.
+     */
+    public static void enableOptions(final Activity activity,
+            final List<Class> options, final Runnable callback) {
+        Log.v(TAG, "enableOptions Called");
+        ThreadUtils.assertOnUiThread();
+
+        if (sStateListener == null) {
+            sStateListener = new ActivityStateListener() {
+                @Override
+                public void onActivityStateChange(Activity activity, int newState) {
+                    if (newState == ActivityState.PAUSED) return;
+                    unregisterActivity(activity, options);
+                }
+            };
+        }
+        ApplicationStatus.registerStateListenerForAllActivities(sStateListener);
+        boolean wasEmpty = sPendingShareActivities.isEmpty();
+        sPendingShareActivities.add(activity);
+
+        waitForPendingStateChangeTask();
+        if (wasEmpty) {
+            sStateChangeTask = new AsyncTask<Void, Void, Void>() {
+                @Override
+                protected Void doInBackground(Void... params) {
+                    if (sPendingShareActivities.isEmpty()) return null;
+
+                    for (int i = 0; i < options.size(); i++) {
+                        Log.v(TAG, "enableOptions: " + options.get(i).getName());

[cco3, 2017/02/02 00:58:58]

remove extra logging.

[iankc, 2017/02/02 01:52:24]

Done.

+                        activity.getPackageManager().setComponentEnabledSetting(
+                                new ComponentName(activity, options.get(i)),
+                                PackageManager.COMPONENT_ENABLED_STATE_ENABLED,
+                                PackageManager.DONT_KILL_APP);
+                    }
+                    return null;
+                }
+
+                @Override
+                protected void onPostExecute(Void result) {
+                    if (sStateChangeTask == this) {
+                        sStateChangeTask = null;
+                    } else {
+                        waitForPendingStateChangeTask();
+                    }
+                    callback.run();
+                }
+            }.executeOnExecutor(AsyncTask.THREAD_POOL_EXECUTOR);
+        } else {
+            callback.run();
+        }
+    }
+
+    public static void unregisterActivity(final Activity activity,
+            final List<Class> options) {
+        Log.v(TAG, "unregisterActivity Called");
+        ThreadUtils.assertOnUiThread();
+
+        sPendingShareActivities.remove(activity);
+        if (!sPendingShareActivities.isEmpty()) return;
+        ApplicationStatus.unregisterActivityStateListener(sStateListener);
+
+        waitForPendingStateChangeTask();
+        sStateChangeTask = new AsyncTask<Void, Void, Void>() {
+            @Override
+            protected Void doInBackground(Void... params) {
+                if (!sPendingShareActivities.isEmpty()) return null;
+
+                for (int i = 0; i < options.size(); i++) {
+                        Log.v(TAG, "unregisterActivity: " + options.get(i).getName());
+                        activity.getPackageManager().setComponentEnabledSetting(
+                                new ComponentName(activity, options.get(i)),
+                                PackageManager.COMPONENT_ENABLED_STATE_DISABLED,
+                                PackageManager.DONT_KILL_APP);
+                    }
+                return null;
+            }
+
+            @Override
+            protected void onPostExecute(Void result) {
+                if (sStateChangeTask == this) sStateChangeTask = null;
+            }
+        }.executeOnExecutor(AsyncTask.THREAD_POOL_EXECUTOR);
+    }
+
+    /**
+     * Waits for any pending state change operations to be completed.
+     *
+     * This will avoid timing issues described here: crbug.com/649453.
+     */
+    public static void waitForPendingStateChangeTask() {
+        ThreadUtils.assertOnUiThread();
+
+        if (sStateChangeTask == null) return;
+        StrictMode.ThreadPolicy oldPolicy = StrictMode.allowThreadDiskReads();
+        try {
+            sStateChangeTask.get();
+            sStateChangeTask = null;
+        } catch (InterruptedException | ExecutionException e) {
+            Log.e(TAG, "Share Helper state change task did not complete as expected");
+        } finally {
+            StrictMode.setThreadPolicy(oldPolicy);
+        }
+    }
 }