/*
 * 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 android.content.pm.cts.shortcutmanager;

import static android.content.pm.LauncherApps.ShortcutQuery.FLAG_GET_KEY_FIELDS_ONLY;
import static android.content.pm.LauncherApps.ShortcutQuery.FLAG_MATCH_DYNAMIC;
import static android.content.pm.LauncherApps.ShortcutQuery.FLAG_MATCH_MANIFEST;
import static android.content.pm.LauncherApps.ShortcutQuery.FLAG_MATCH_PINNED;

import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.assertWith;
import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.list;
import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.retryUntil;
import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.setDefaultLauncher;

import android.graphics.BitmapFactory;
import android.graphics.drawable.Icon;
import android.test.suitebuilder.annotation.SmallTest;

@SmallTest
public class ShortcutManagerLauncherApiTest extends ShortcutManagerCtsTestsBase {
    public void testPinShortcuts() {
        runWithCaller(mPackageContext1, () -> {
            enableManifestActivity("Launcher_manifest_1", true);
            enableManifestActivity("Launcher_manifest_2", true);

            retryUntil(() -> getManager().getManifestShortcuts().size() > 0,
                    "Manifest shortcuts didn't show up");

            assertTrue(getManager().setDynamicShortcuts(list(
                    makeShortcut("s1"),
                    makeShortcut("s2"),
                    makeShortcut("s3"),
                    makeShortcut("s4"),
                    makeShortcut("s5")
            )));
            assertWith(getManager().getDynamicShortcuts())
                    .haveIds("s1", "s2", "s3", "s4", "s5")
                    .areAllDynamic()
                    .areAllEnabled();
        });
        runWithCaller(mPackageContext2, () -> {
            enableManifestActivity("Launcher_manifest_1", true);
            enableManifestActivity("Launcher_manifest_3", true);

            retryUntil(() -> getManager().getManifestShortcuts().size() > 0,
                    "Manifest shortcuts didn't show up");

            assertTrue(getManager().setDynamicShortcuts(list(
                    makeShortcut("s1"),
                    makeShortcut("s2"),
                    makeShortcut("s3"),
                    makeShortcut("s4"),
                    makeShortcut("s5")
            )));
            assertWith(getManager().getDynamicShortcuts())
                    .haveIds("s1", "s2", "s3", "s4", "s5")
                    .areAllDynamic()
                    .areAllEnabled();
        });

        setDefaultLauncher(getInstrumentation(), mLauncherContext1);

        runWithCaller(mLauncherContext1, () -> {
            getLauncherApps().pinShortcuts(
                    mPackageContext1.getPackageName(),
                    list("s1", "s2", "s3", "ms1", "ms21"), getUserHandle());
            getLauncherApps().pinShortcuts(
                    mPackageContext2.getPackageName(),
                    list("s2", "s3", "ms1", "ms31"), getUserHandle());
        });

        runWithCaller(mPackageContext1, () -> {
            getManager().removeDynamicShortcuts(list("s1", "s2"));
        });

        runWithCaller(mPackageContext2, () -> {
            enableManifestActivity("Launcher_manifest_3", false);

            retryUntil(() -> getManager().getManifestShortcuts().size() == 1,
                    "Manifest shortcuts didn't updated");

            getManager().removeDynamicShortcuts(list("s1", "s2"));
        });

        // Check the result.
        runWithCaller(mPackageContext1, () -> {
            assertWith(getManager().getDynamicShortcuts())
                    .haveIds("s3", "s4", "s5")
                    .areAllEnabled();
            assertWith(getManager().getManifestShortcuts())
                    .haveIds("ms1", "ms21", "ms22")
                    .areAllEnabled();
            assertWith(getManager().getPinnedShortcuts())
                    .haveIds("s1", "s2", "s3", "ms1", "ms21")
                    .areAllEnabled();
        });

        runWithCaller(mPackageContext2, () -> {
            assertWith(getManager().getDynamicShortcuts())
                    .haveIds("s3", "s4", "s5")
                    .areAllEnabled();
            assertWith(getManager().getManifestShortcuts())
                    .haveIds("ms1")
                    .areAllEnabled();
            assertWith(getManager().getPinnedShortcuts())
                    .haveIds("s2", "s3", "ms1", "ms31")

                    .selectByIds("s2", "s3", "ms1")
                    .areAllEnabled()

                    .revertToOriginalList()
                    .selectByIds("ms31")
                    .areAllDisabled();
        });
    }

    public void testGetShortcuts() throws Exception {

        testPinShortcuts();

        Thread.sleep(2);
        final long time1 = System.currentTimeMillis();

        runWithCaller(mPackageContext1, () -> {
            assertTrue(getManager().updateShortcuts(list(
                    makeShortcut("s3"))));

            setTargetActivityOverride("Launcher_manifest_2");

            assertTrue(getManager().updateShortcuts(list(
                    makeShortcut("s1"),
                    makeShortcut("s5"))));
        });

        Thread.sleep(2);
        final long time2 = System.currentTimeMillis();

        runWithCaller(mPackageContext1, () -> {
            assertTrue(getManager().updateShortcuts(list(
                    makeShortcutWithRank("s4", 999))));
        });

        runWithCaller(mPackageContext2, () -> {
            setTargetActivityOverride("Launcher_manifest_1");

            assertTrue(getManager().updateShortcuts(list(
                    makeShortcut("s1"))));
        });

        Thread.sleep(2);
        final long time3 = System.currentTimeMillis();

        runWithCaller(mLauncherContext1, () -> {
            assertWith(getShortcutsAsLauncher(
                    FLAG_MATCH_DYNAMIC,
                    mPackageContext1.getPackageName(),
                    null,
                    0,
                    list()))
                    .haveIds("s3", "s4", "s5")
                    .areAllNotWithKeyFieldsOnly();
            assertWith(getShortcutsAsLauncher(
                    FLAG_MATCH_PINNED,
                    mPackageContext1.getPackageName(),
                    null,
                    0,
                    list()))
                    .haveIds("s1", "s2", "s3", "ms1", "ms21")
                    .areAllNotWithKeyFieldsOnly();
            assertWith(getShortcutsAsLauncher(
                    FLAG_MATCH_MANIFEST,
                    mPackageContext1.getPackageName(),
                    null,
                    0,
                    list()))
                    .haveIds("ms1", "ms21", "ms22")
                    .areAllNotWithKeyFieldsOnly();

            assertWith(getShortcutsAsLauncher(
                    FLAG_MATCH_DYNAMIC | FLAG_GET_KEY_FIELDS_ONLY,
                    mPackageContext1.getPackageName(),
                    null,
                    0,
                    list()))
                    .haveIds("s3", "s4", "s5")
                    .areAllWithKeyFieldsOnly();

            assertWith(getShortcutsAsLauncher(
                    FLAG_MATCH_DYNAMIC | FLAG_MATCH_PINNED,
                    mPackageContext1.getPackageName(),
                    null,
                    0,
                    list()))
                    .haveIds("s3", "s4", "s5", "s1", "s2", "s3", "ms1", "ms21")
                    .areAllNotWithKeyFieldsOnly();

            assertWith(getShortcutsAsLauncher(
                    FLAG_MATCH_PINNED | FLAG_MATCH_MANIFEST,
                    mPackageContext1.getPackageName(),
                    null,
                    0,
                    list()))
                    .haveIds("s1", "s2", "s3", "ms1", "ms21", "ms1", "ms21", "ms22")
                    .areAllNotWithKeyFieldsOnly();
            assertWith(getShortcutsAsLauncher(
                    FLAG_MATCH_DYNAMIC | FLAG_MATCH_PINNED | FLAG_MATCH_MANIFEST,
                    mPackageContext1.getPackageName(),
                    null,
                    0,
                    list()))
                    .haveIds("s3", "s4", "s5", "s1", "s2", "ms1", "ms21", "ms1", "ms21", "ms22")
                    .areAllNotWithKeyFieldsOnly();
            assertWith(getShortcutsAsLauncher(
                    FLAG_MATCH_DYNAMIC | FLAG_MATCH_PINNED | FLAG_MATCH_MANIFEST
                            | FLAG_GET_KEY_FIELDS_ONLY,
                    mPackageContext1.getPackageName(),
                    null,
                    0,
                    list()))
                    .haveIds("s3", "s4", "s5", "s1", "s2", "ms1", "ms21", "ms1", "ms21", "ms22")
                    .areAllWithKeyFieldsOnly();


            // Package 2

            assertWith(getShortcutsAsLauncher(
                    FLAG_MATCH_DYNAMIC | FLAG_MATCH_PINNED | FLAG_MATCH_MANIFEST,
                    mPackageContext2.getPackageName(),
                    null,
                    0,
                    list()))
                    .haveIds("s2", "s3", "s4", "s5", "ms1", "ms31")
                    ;

            // With activity
            assertWith(getShortcutsAsLauncher(
                    FLAG_MATCH_DYNAMIC | FLAG_MATCH_PINNED | FLAG_MATCH_MANIFEST,
                    mPackageContext1.getPackageName(),
                    "Launcher_manifest_2",
                    0,
                    list()))
                    .haveIds("ms21", "ms22", "s1", "s5")
                    .areAllNotWithKeyFieldsOnly();

            // With ids
            assertWith(getShortcutsAsLauncher(
                    FLAG_MATCH_DYNAMIC | FLAG_MATCH_PINNED | FLAG_MATCH_MANIFEST,
                    mPackageContext1.getPackageName(),
                    null,
                    0,
                    list("s1", "s2", "ms1")))
                    .haveIds("s1", "s2", "ms1")
                    .areAllNotWithKeyFieldsOnly();

            // With time.
            assertWith(getShortcutsAsLauncher(
                    FLAG_MATCH_DYNAMIC | FLAG_MATCH_PINNED | FLAG_MATCH_MANIFEST,
                    mPackageContext1.getPackageName(),
                    null,
                    time2,
                    list()))
                    .haveIds("s4")
                    .areAllNotWithKeyFieldsOnly();

            // No shortcuts have changed since time3.
            assertWith(getShortcutsAsLauncher(
                    FLAG_MATCH_DYNAMIC | FLAG_MATCH_PINNED | FLAG_MATCH_MANIFEST,
                    mPackageContext1.getPackageName(),
                    null,
                    time3,
                    list()))
                    .isEmpty();
            assertWith(getShortcutsAsLauncher(
                    FLAG_MATCH_DYNAMIC | FLAG_MATCH_PINNED | FLAG_MATCH_MANIFEST,
                    mPackageContext2.getPackageName(),
                    null,
                    time3,
                    list()))
                    .isEmpty();
        });
    }

    public void testGetShortcutIcon() throws Exception {
        final Icon icon1 = Icon.createWithBitmap(BitmapFactory.decodeResource(
                getTestContext().getResources(), R.drawable.black_16x64));
        final Icon icon2 = Icon.createWithBitmap(BitmapFactory.decodeResource(
                getTestContext().getResources(), R.drawable.black_32x32));
        final Icon icon3 = loadPackageDrawableIcon(mPackageContext1, "black_64x16");
        final Icon icon4 = loadPackageDrawableIcon(mPackageContext1, "black_64x64");

        final Icon icon5 = loadPackageDrawableIcon(mPackageContext1, "black_16x16");

        runWithCaller(mPackageContext1, () -> {
            enableManifestActivity("Launcher_manifest_2", true);

            retryUntil(() -> getManager().getManifestShortcuts().size() > 0,
                    "Manifest shortcuts didn't show up");

            assertTrue(getManager().setDynamicShortcuts(list(
                    makeShortcutWithIcon("s1", icon1),
                    makeShortcutWithIcon("s2", icon2),
                    makeShortcutWithIcon("s3", icon3),
                    makeShortcutWithIcon("s4", icon4))));
        });

        setDefaultLauncher(getInstrumentation(), mLauncherContext1);

        assertIconDimensions(icon1, getIconAsLauncher(
                mLauncherContext1, mPackageContext1.getPackageName(), "s1", true));
        assertIconDimensions(icon2, getIconAsLauncher(
                mLauncherContext1, mPackageContext1.getPackageName(), "s2", true));
        assertIconDimensions(icon3, getIconAsLauncher(
                mLauncherContext1, mPackageContext1.getPackageName(), "s3", true));
        assertIconDimensions(icon4, getIconAsLauncher(
                mLauncherContext1, mPackageContext1.getPackageName(), "s4", true));

        assertIconDimensions(icon5, getIconAsLauncher(
                mLauncherContext1, mPackageContext1.getPackageName(), "ms21", true));

        assertIconDimensions(icon1, getIconAsLauncher(
                mLauncherContext1, mPackageContext1.getPackageName(), "s1", false));
        assertIconDimensions(icon2, getIconAsLauncher(
                mLauncherContext1, mPackageContext1.getPackageName(), "s2", false));
        assertIconDimensions(icon3, getIconAsLauncher(
                mLauncherContext1, mPackageContext1.getPackageName(), "s3", false));
        assertIconDimensions(icon4, getIconAsLauncher(
                mLauncherContext1, mPackageContext1.getPackageName(), "s4", false));

        assertIconDimensions(icon5, getIconAsLauncher(
                mLauncherContext1, mPackageContext1.getPackageName(), "ms21", false));
    }

    // TODO Callback test

    // TODO Launch test
}