Examples with FLAG_GET_KEY_FIELDS_ONLY android.content.pm.LauncherApps.ShortcutQuery.FLAG_GET_KEY_FIELDS_ONLY used on opensource projects

Example 1 with FLAG_GET_KEY_FIELDS_ONLY

use of android.content.pm.LauncherApps.ShortcutQuery.FLAG_GET_KEY_FIELDS_ONLY in project android_packages_apps_Launcher3 by crdroidandroid.

the class BgDataModel method updateShortcutPinnedState.

/**
 * Updates the deep shortucts state in system to match out internal model, pinning any missing
 * shortcuts and unpinning any extra shortcuts.
 */
public synchronized void updateShortcutPinnedState(Context context, UserHandle user) {
    if (GO_DISABLE_WIDGETS) {
        return;
    }
    // Collect all system shortcuts
    QueryResult result = new ShortcutRequest(context, user).query(PINNED | FLAG_GET_KEY_FIELDS_ONLY);
    if (!result.wasSuccess()) {
        return;
    }
    // Map of packageName to shortcutIds that are currently in the system
    Map<String, Set<String>> systemMap = result.stream().collect(groupingBy(ShortcutInfo::getPackage, mapping(ShortcutInfo::getId, Collectors.toSet())));
    // Collect all model shortcuts
    Stream.Builder<WorkspaceItemInfo> itemStream = Stream.builder();
    forAllWorkspaceItemInfos(user, itemStream::accept);
    // Map of packageName to shortcutIds that are currently in our model
    Map<String, Set<String>> modelMap = Stream.concat(// Model shortcuts
    itemStream.build().filter(wi -> wi.itemType == Favorites.ITEM_TYPE_DEEP_SHORTCUT).map(ShortcutKey::fromItemInfo), // Pending shortcuts
    ItemInstallQueue.INSTANCE.get(context).getPendingShortcuts(user)).collect(groupingBy(ShortcutKey::getPackageName, mapping(ShortcutKey::getId, Collectors.toSet())));
    // Check for diff
    for (Map.Entry<String, Set<String>> entry : modelMap.entrySet()) {
        Set<String> modelShortcuts = entry.getValue();
        Set<String> systemShortcuts = systemMap.remove(entry.getKey());
        if (systemShortcuts == null) {
            systemShortcuts = Collections.emptySet();
        }
        // Do not use .equals as it can vary based on the type of set
        if (systemShortcuts.size() != modelShortcuts.size() || !systemShortcuts.containsAll(modelShortcuts)) {
            // Update system state for this package
            try {
                context.getSystemService(LauncherApps.class).pinShortcuts(entry.getKey(), new ArrayList<>(modelShortcuts), user);
            } catch (SecurityException | IllegalStateException e) {
                Log.w(TAG, "Failed to pin shortcut", e);
            }
        }
    }
    // If there are any extra pinned shortcuts, remove them
    systemMap.keySet().forEach(packageName -> {
        // Update system state
        try {
            context.getSystemService(LauncherApps.class).pinShortcuts(packageName, Collections.emptyList(), user);
        } catch (SecurityException | IllegalStateException e) {
            Log.w(TAG, "Failed to unpin shortcut", e);
        }
    });
}