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

import static com.android.launcher3.model.data.ItemInfoWithIcon.FLAG_DISABLED_LOCKED_USER;

import android.content.Context;
import android.content.pm.ShortcutInfo;
import android.os.UserHandle;

import androidx.annotation.NonNull;

import com.android.launcher3.LauncherModel.ModelUpdateTask;
import com.android.launcher3.LauncherSettings;
import com.android.launcher3.model.AllAppsList;
import com.android.launcher3.model.BgDataModel;
import com.android.launcher3.model.ModelTaskController;
import com.android.launcher3.model.data.ItemInfo;
import com.android.launcher3.shortcuts.ShortcutKey;
import com.android.launcher3.shortcuts.ShortcutRequest;
import com.android.launcher3.shortcuts.ShortcutRequest.QueryResult;
import com.android.launcher3.util.ItemInfoMatcher;

import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;

/**
 * Task to handle changing of lock state of the user
 */
public class UserLockStateChangedTask implements ModelUpdateTask {

    @NonNull
    private final UserHandle mUser;
    private boolean mIsUserUnlocked;

    public UserLockStateChangedTask(@NonNull final UserHandle user, final boolean isUserUnlocked) {
        mUser = user;
        mIsUserUnlocked = isUserUnlocked;
    }

    @Override
    public void execute(@NonNull ModelTaskController taskController, @NonNull BgDataModel dataModel,
            @NonNull AllAppsList apps) {
        Context context = taskController.getContext();

        HashMap<ShortcutKey, ShortcutInfo> pinnedShortcuts = new HashMap<>();
        if (mIsUserUnlocked) {
            QueryResult shortcuts = new ShortcutRequest(context, mUser)
                    .query(ShortcutRequest.PINNED);
            if (shortcuts.wasSuccess()) {
                for (ShortcutInfo shortcut : shortcuts) {
                    pinnedShortcuts.put(ShortcutKey.fromInfo(shortcut), shortcut);
                }
            } else {
                // Shortcut manager can fail due to some race condition when the lock state
                // changes too frequently. For the purpose of the update,
                // consider it as still locked.
                mIsUserUnlocked = false;
            }
        }

        // Update the workspace to reflect the changes to updated shortcuts residing on it.
        List<ItemInfo> updatedItemInfos;
        HashSet<ShortcutKey> removedKeys = new HashSet<>();

        synchronized (dataModel) {
            updatedItemInfos = dataModel.updateAndCollectWorkspaceItemInfos(mUser, si -> {
                if (si.itemType == LauncherSettings.Favorites.ITEM_TYPE_DEEP_SHORTCUT) {
                    if (mIsUserUnlocked) {
                        ShortcutKey key = ShortcutKey.fromItemInfo(si);
                        ShortcutInfo shortcut = pinnedShortcuts.get(key);
                        // We couldn't verify the shortcut during loader. If its no longer available
                        // (probably due to clear data), delete the workspace item as well
                        if (shortcut == null) {
                            removedKeys.add(key);
                            return false;
                        }
                        si.runtimeStatusFlags &= ~FLAG_DISABLED_LOCKED_USER;
                        si.updateFromDeepShortcutInfo(shortcut, context);
                        taskController.getIconCache().getShortcutIcon(si, shortcut);
                    } else {
                        si.runtimeStatusFlags |= FLAG_DISABLED_LOCKED_USER;
                    }
                    return true;
                }
                return false;
            });
        }
        taskController.bindUpdatedWorkspaceItems(updatedItemInfos);
        if (!removedKeys.isEmpty()) {
            taskController.deleteAndBindComponentsRemoved(
                    ItemInfoMatcher.ofShortcutKeys(removedKeys),
                    "removed during unlock because it's no longer available"
                            + " (possibly due to clear data)");
        }

        // Remove shortcut id map for that user and update the shourtcuts if unlocked
        dataModel.updateDeepShortcutCounts(
                mIsUserUnlocked
                        ? new ShortcutRequest(context, mUser).query(ShortcutRequest.ALL)
                        : Collections.emptyList(),
                key -> key.user.equals(mUser));
    }
}