/*
 * Copyright (C) 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.launcher3.model;

import static com.android.launcher3.icons.cache.CacheLookupFlag.DEFAULT_LOOKUP_FLAG;
import static com.android.launcher3.model.ModelUtils.WIDGET_FILTER;

import android.content.ComponentName;
import android.os.UserHandle;

import androidx.annotation.NonNull;

import com.android.launcher3.LauncherModel.ModelUpdateTask;
import com.android.launcher3.LauncherSettings;
import com.android.launcher3.icons.IconCache;
import com.android.launcher3.model.data.ItemInfo;
import com.android.launcher3.model.data.LauncherAppWidgetInfo;
import com.android.launcher3.model.data.WorkspaceItemInfo;

import java.util.ArrayList;
import java.util.HashSet;

/**
 * Handles changes due to cache updates.
 */
public class CacheDataUpdatedTask implements ModelUpdateTask {

    public static final int OP_CACHE_UPDATE = 1;
    public static final int OP_SESSION_UPDATE = 2;

    private final int mOp;

    @NonNull
    private final UserHandle mUser;

    @NonNull
    private final HashSet<String> mPackages;

    public CacheDataUpdatedTask(final int op, @NonNull final UserHandle user,
            @NonNull final HashSet<String> packages) {
        mOp = op;
        mUser = user;
        mPackages = packages;
    }

    @Override
    public void execute(@NonNull ModelTaskController taskController, @NonNull BgDataModel dataModel,
            @NonNull AllAppsList apps) {
        IconCache iconCache = taskController.getIconCache();
        ArrayList<ItemInfo> updatedItems = new ArrayList<>();

        synchronized (dataModel) {
            dataModel.forAllWorkspaceItemInfos(mUser, si -> {
                ComponentName cn = si.getTargetComponent();
                if (si.itemType == LauncherSettings.Favorites.ITEM_TYPE_APPLICATION
                        && isValidShortcut(si) && cn != null
                        && mPackages.contains(cn.getPackageName())) {
                    iconCache.getTitleAndIcon(si, si.getMatchingLookupFlag());
                    updatedItems.add(si);
                }
            });

            dataModel.itemsIdMap.stream()
                    .filter(WIDGET_FILTER)
                    .filter(item -> mUser.equals(item.user))
                    .map(item -> (LauncherAppWidgetInfo) item)
                    .filter(widget -> mPackages.contains(widget.providerName.getPackageName())
                            && widget.pendingItemInfo != null)
                    .forEach(widget -> {
                        iconCache.getTitleAndIconForApp(
                                widget.pendingItemInfo, DEFAULT_LOOKUP_FLAG);
                        updatedItems.add(widget);
                    });

            apps.updateIconsAndLabels(mPackages, mUser);
        }
        taskController.bindUpdatedWorkspaceItems(updatedItems);
        taskController.bindApplicationsIfNeeded();
    }

    public boolean isValidShortcut(@NonNull final WorkspaceItemInfo si) {
        switch (mOp) {
            case OP_CACHE_UPDATE:
                return true;
            case OP_SESSION_UPDATE:
                return si.hasPromiseIconUi();
            default:
                return false;
        }
    }
}
