[Android]: Hide add-to-homescreen app menu item when WebAPK is installed

chrome/android/java/src/org/chromium/chrome/browser/appmenu/AppMenuPropertiesDelegate.java

 // Copyright 2015 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.appmenu;
 
+import android.content.Context;
+import android.content.pm.ResolveInfo;
 import android.graphics.PorterDuff;
 import android.graphics.drawable.Drawable;
+import android.os.SystemClock;
 import android.view.Menu;
 import android.view.MenuItem;
 
 import org.chromium.base.ApiCompatibilityUtils;
 import org.chromium.base.CommandLine;
+import org.chromium.base.ContextUtils;
+import org.chromium.base.metrics.RecordHistogram;
 import org.chromium.chrome.R;
 import org.chromium.chrome.browser.ChromeActivity;
 import org.chromium.chrome.browser.ChromeFeatureList;
 import org.chromium.chrome.browser.ChromeSwitches;
 import org.chromium.chrome.browser.ShortcutHelper;
 import org.chromium.chrome.browser.UrlConstants;
 import org.chromium.chrome.browser.banners.AppBannerManager;
 import org.chromium.chrome.browser.bookmarks.BookmarkBridge;
 import org.chromium.chrome.browser.download.DownloadUtils;
 import org.chromium.chrome.browser.multiwindow.MultiWindowUtils;
 import org.chromium.chrome.browser.omaha.UpdateMenuItemHelper;
 import org.chromium.chrome.browser.preferences.ManagedPreferencesUtils;
 import org.chromium.chrome.browser.preferences.PrefServiceBridge;
 import org.chromium.chrome.browser.share.ShareHelper;
 import org.chromium.chrome.browser.tab.Tab;
+import org.chromium.chrome.browser.webapps.ChromeWebApkHost;
 import org.chromium.components.dom_distiller.core.DomDistillerUrlUtils;
 import org.chromium.ui.base.DeviceFormFactor;
+import org.chromium.webapk.lib.client.WebApkValidator;
+
+import java.util.concurrent.TimeUnit;
 
 /**
  * App Menu helper that handles hiding and showing menu items based on activity state.
  */
 public class AppMenuPropertiesDelegate {
     // Indices for different levels in drawable.btn_reload_stop.
     // Used only when preparing menu and refresh reload button in menu when tab
     // page load status changes.
     private static final int RELOAD_BUTTON_LEVEL_RELOAD = 0;
     private static final int RELOAD_BUTTON_LEVEL_STOP_LOADING = 1;
@@ skipping 114 matching lines @@
             // Hide 'Add to homescreen' for the following:
             // * chrome:// pages - Android doesn't know how to direct those URLs.
             // * incognito pages - To avoid problems where users create shortcuts in incognito
             //                      mode and then open the webapp in regular mode.
             // * file:// - After API 24, file: URIs are not supported in VIEW intents and thus
             //             can not be added to the homescreen.
             // * content:// - Accessing external content URIs requires the calling app to grant
             //                access to the resource via FLAG_GRANT_READ_URI_PERMISSION, and that
             //                is not persisted when adding to the homescreen.
             // * If creating shortcuts it not supported by the current home screen.
-            MenuItem homescreenItem = menu.findItem(R.id.add_to_homescreen_id);
-            boolean homescreenItemVisible = ShortcutHelper.isAddToHomeIntentSupported()
+            boolean canShowHomeScreenMenuItem = ShortcutHelper.isAddToHomeIntentSupported()
                     && !isChromeScheme && !isFileScheme && !isContentScheme && !isIncognito;
-            if (homescreenItemVisible) {
-                homescreenItem.setTitle(AppBannerManager.getHomescreenLanguageOption());
-            }
-            homescreenItem.setVisible(homescreenItemVisible);
+            prepareAddToHomescreenMenuItem(menu, currentTab.getUrl(), canShowHomeScreenMenuItem);
 
             // Hide request desktop site on all chrome:// pages except for the NTP. Check request
             // desktop site if it's activated on this page.
             MenuItem requestItem = menu.findItem(R.id.request_desktop_site_id);
             updateRequestDesktopSiteMenuItem(requestItem, currentTab);
 
             // Only display reader mode settings menu option if the current page is in reader mode.
             menu.findItem(R.id.reader_mode_prefs_id)
                     .setVisible(DomDistillerUrlUtils.isDistilledPage(currentTab.getUrl()));
 
@@ skipping 29 matching lines @@
         // main_menu.xml contains multiple items with the same id in different groups
         // e.g.: new_incognito_tab_menu_id.
         disableEnableMenuItem(menu, R.id.new_incognito_tab_menu_id,
                 true,
                 PrefServiceBridge.getInstance().isIncognitoModeEnabled(),
                 PrefServiceBridge.getInstance().isIncognitoModeManaged());
         mActivity.prepareMenu(menu);
     }
 
     /**
+     * Sets the visibility and labels of the "Add to Home screen" and "Open WebAPK" menu items.
+     */
+    protected void prepareAddToHomescreenMenuItem(
+            Menu menu, String url, boolean canShowHomeScreenMenuItem) {
+        MenuItem homescreenItem = menu.findItem(R.id.add_to_homescreen_id);
+        MenuItem openWebApkItem = menu.findItem(R.id.open_webapk_id);
+        if (canShowHomeScreenMenuItem) {
+            Context context = ContextUtils.getApplicationContext();
+            ResolveInfo resolveInfo = null;
+            boolean openWebApkItemVisible = false;
+            if (ChromeWebApkHost.isEnabled()) {
+                long addToHomeScreenStart = SystemClock.elapsedRealtime();
+                resolveInfo = WebApkValidator.queryResolveInfo(context, url);
+                RecordHistogram.recordTimesHistogram(
+                        "Android.PrepareMenu.OpenWebApkVisibilityCheck",
+                        SystemClock.elapsedRealtime() - addToHomeScreenStart,
+                        TimeUnit.MILLISECONDS);
+
+                openWebApkItemVisible =
+                        resolveInfo != null && resolveInfo.activityInfo.packageName != null;

[nyquist, 2017/02/28 07:04:19]

The null-check of packageName seems unnecessary here. If that is something that
can happen, we would have already crashed in WebApkValidator#isValidWebApk(...):

  if (!webappPackageName.startsWith(WEBAPK_PACKAGE_PREFIX)) {
    ...

Maybe you could move this null-check of the package name there instead, and just
check if the resolveInfo is null here? At that point openWebApkItemVisible ==
resolveInfo != null, so we conceptually don't need both, but I find the variable
name very helpful so I'd keep it to ensure it's easy to understand for new
readers.

[gonzalon, 2017/02/28 16:12:23]

Great catch. Thanks!

+            }
+            if (openWebApkItemVisible) {
+                String appName = resolveInfo.loadLabel(context.getPackageManager()).toString();
+                openWebApkItem.setTitle(context.getString(R.string.menu_open_webapk, appName));
+
+                homescreenItem.setVisible(false);
+                openWebApkItem.setVisible(true);
+            } else {
+                homescreenItem.setTitle(AppBannerManager.getHomescreenLanguageOption());
+                homescreenItem.setVisible(true);
+                openWebApkItem.setVisible(false);
+            }
+        } else {
+            homescreenItem.setVisible(false);
+            openWebApkItem.setVisible(false);
+        }
+    }
+
+    /**
      * Notify the delegate that the load state changed.
      * @param isLoading Whether the page is currently loading.
      */
     public void loadingStateChanged(boolean isLoading) {
         if (mReloadMenuItem != null) {
             mReloadMenuItem.getIcon().setLevel(isLoading
                     ? RELOAD_BUTTON_LEVEL_STOP_LOADING : RELOAD_BUTTON_LEVEL_RELOAD);
             mReloadMenuItem.setTitle(isLoading
                     ? R.string.accessibility_btn_stop_loading : R.string.accessibility_btn_refresh);
         }
@@ skipping 70 matching lines @@
         String url = currentTab.getUrl();
         boolean isChromeScheme = url.startsWith(UrlConstants.CHROME_URL_PREFIX)
                 || url.startsWith(UrlConstants.CHROME_NATIVE_URL_PREFIX);
         requstMenuItem.setVisible(!isChromeScheme || currentTab.isNativePage());
         requstMenuItem.setChecked(currentTab.getUseDesktopUserAgent());
         requstMenuItem.setTitleCondensed(requstMenuItem.isChecked()
                 ? mActivity.getString(R.string.menu_request_desktop_site_on)
                 : mActivity.getString(R.string.menu_request_desktop_site_off));
     }
 }