/*
 * 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.example.android.pm.shortcutdemo;

import android.app.Activity;
import android.app.Notification;
import android.app.Notification.Action;
import android.app.Notification.Builder;
import android.app.NotificationManager;
import android.app.PendingIntent;
import android.app.RemoteInput;
import android.content.ComponentName;
import android.content.Context;
import android.content.Intent;
import android.content.pm.ShortcutInfo;
import android.content.pm.ShortcutManager;
import android.graphics.BitmapFactory;
import android.graphics.drawable.Icon;
import android.net.Uri;
import android.os.Bundle;
import androidx.core.content.pm.ShortcutInfoCompat;
import androidx.core.content.pm.ShortcutManagerCompat;
import androidx.core.graphics.drawable.IconCompat;
import android.text.format.Time;
import android.util.ArrayMap;
import android.util.Log;
import android.util.Pair;
import android.view.View;
import android.widget.ListView;
import android.widget.Toast;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.Random;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.function.BooleanSupplier;

public class ShortcutPublisher extends Activity {
    public static final String TAG = "ShortcutDemo";

    private static final String SETUP_SHORTCUT_ID = "setup";

    private ShortcutManager mShortcutManager;

    private ListView mList;
    private MyAdapter mAdapter;

    private static final Random sRandom = new Random();

    private static final AtomicInteger sSequenceNumber = new AtomicInteger();

    private ComponentName mMyActivity;

    @Override
    public void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);

        setContentView(R.layout.main);

        mShortcutManager = getSystemService(ShortcutManager.class);

        mMyActivity = getIntent().getComponent();
        if (mMyActivity == null) {
            mMyActivity = new ComponentName(this, ShortcutPublisher.class);
        }

        mList = (ListView) findViewById(android.R.id.list);
        mAdapter = new MyAdapter(this);
        mList.setAdapter(mAdapter);

        Log.d(TAG, "intent=" + getIntent());
        Log.d(TAG, "extras=" + getIntent().getExtras());
    }

    @Override
    protected void onResume() {
        super.onResume();

        refreshList();
    }

    @Override
    protected void onDestroy() {
        super.onDestroy();
    }

    private List<ShortcutInfo> getAllShortcuts() {
        final Map<String, ShortcutInfo> map = new ArrayMap<>();
        for (ShortcutInfo si : mShortcutManager.getManifestShortcuts()) {
            if (!Objects.equals(si.getActivity(), mMyActivity)) continue;
            if (!map.containsKey(si.getId())) {
                map.put(si.getId(), si);
            }
        }
        for (ShortcutInfo si : mShortcutManager.getDynamicShortcuts()) {
            if (!Objects.equals(si.getActivity(), mMyActivity)) continue;
            if (!map.containsKey(si.getId())) {
                map.put(si.getId(), si);
            }
        }
        for (ShortcutInfo si : mShortcutManager.getPinnedShortcuts()) {
            if (!Objects.equals(si.getActivity(), mMyActivity)) continue;
            if (!map.containsKey(si.getId())) {
                map.put(si.getId(), si);
            }
        }
        return new ArrayList<>(map.values());
    }

    private void refreshList() {
        final List<ShortcutInfo> list = getAllShortcuts();
        Collections.sort(list, mShortcutComparator);
        mAdapter.setShortcuts(list);
    }

    private final Comparator<ShortcutInfo> mShortcutComparator =
            (ShortcutInfo s1, ShortcutInfo s2) -> {
                int ret = 0;
                ret = (s1.isDeclaredInManifest() ? 0 : 1) - (s2.isDeclaredInManifest() ? 0 : 1);
                if (ret != 0) return ret;

                ret = (s1.isDynamic() ? 0 : 1) - (s2.isDynamic() ? 0 : 1);
                if (ret != 0) return ret;

                ret = s1.getId().compareTo(s2.getId());
                if (ret != 0) return ret;

                return 0;
            };

    private void dumpCurrentShortcuts() {
        Log.d(TAG, "Dynamic shortcuts:");
        for (ShortcutInfo si : mShortcutManager.getDynamicShortcuts()) {
            Log.d(TAG, "  " + si.toString());
        }
        Log.d(TAG, "Pinned shortcuts:");
        for (ShortcutInfo si : mShortcutManager.getPinnedShortcuts()) {
            Log.d(TAG, "  " + si.toString());
        }
    }

    public static void showToast(Context context, String message) {
        Toast.makeText(context, message, Toast.LENGTH_SHORT).show();
    }

    private static void showThrottledToast(Context context) {
        showToast(context,
                "Throttled, use \"adb shell cmd shortcut reset-throttling\" to reset counters");
    }

    public static void callApi(Context context, BooleanSupplier call) {
        try {
            if (!call.getAsBoolean()) {
                showThrottledToast(context);
            }
        } catch (RuntimeException r) {
            Log.w(TAG, r.getMessage(), r);
            showToast(context, r.getMessage());
        }
    }

    private static List<Pair<String, String>> sIntentList = Arrays.asList(
            Pair.create("Google Search", "http://www.google.com"),
            Pair.create("Google Mail", "http://mail.google.com"),
            Pair.create("Google Maps", "http://maps.google.com"),
            Pair.create("Google Drive", "http://drive.google.com"),
            Pair.create("Google Photos", "http://photos.google.com"),
            Pair.create("Google Hangouts", "http://hangouts.google.com"),
            Pair.create("Google+", "http://plus.google.com")
    );

    public static ShortcutInfo.Builder addRandomIntents(Context context, ShortcutInfo.Builder b) {
        final int i = sRandom.nextInt(sIntentList.size());
        b.setShortLabel(sIntentList.get(i).first);
        b.setIntent(new Intent(Intent.ACTION_VIEW, Uri.parse(sIntentList.get(i).second)));

        if (sRandom.nextBoolean()) {
            b.setIcon(Icon.createWithResource(context, R.drawable.icon2));
        } else {
            b.setIcon(Icon.createWithBitmap(BitmapFactory.decodeResource(context.getResources(),
                    R.drawable.icon_large_2)));
        }
        return b;
    }

    public static ShortcutInfoCompat.Builder addRandomIntents(Context context,
            ShortcutInfoCompat.Builder b) {
        final int i = sRandom.nextInt(sIntentList.size());
        b.setShortLabel(sIntentList.get(i).first);
        b.setIntent(new Intent(Intent.ACTION_VIEW, Uri.parse(sIntentList.get(i).second)));

        if (sRandom.nextBoolean()) {
            b.setIcon(IconCompat.createWithResource(context, R.drawable.icon2));
        } else {
            b.setIcon(IconCompat.createWithBitmap(
                    BitmapFactory.decodeResource(context.getResources(), R.drawable.icon_large_2)));
        }
        return b;
    }

    public void onPublishPressed(View view) {
        dumpCurrentShortcuts();
        final Icon icon2 = Icon.createWithBitmap(BitmapFactory.decodeResource(getResources(),
                R.drawable.icon_large_2));
        final Icon icon3 = Icon.createWithResource(this, R.drawable.icon_large_3);

        final Intent intent2 = new Intent(Intent.ACTION_VIEW);
        intent2.setClass(this, ShortcutPublisher.class);

        final Intent intent3 = new Intent(Intent.ACTION_VIEW);
        intent3.setClass(this, ShortcutPublisher.class);
        intent3.putExtra("str", "str-value");
        intent3.putExtra("nest", new Bundle());
        intent3.getBundleExtra("nest").putInt("int", 123);

        final ShortcutInfo si1 = addRandomIntents(this, new ShortcutInfo.Builder(this, "shortcut1"))
                .setActivity(mMyActivity)
                .build();

        final ShortcutInfo si2 = new ShortcutInfo.Builder(this, SETUP_SHORTCUT_ID)
                .setActivity(mMyActivity)
                .setShortLabel("Shortcut Demo Main")
                .setIcon(icon2)
                .setIntent(intent2)
                .build();

        final ShortcutInfo si3 = new ShortcutInfo.Builder(this, "shortcut3")
                .setActivity(mMyActivity)
                .setShortLabel("Shortcut Demo Main with extras")
                .setIcon(icon3)
                .setIntent(intent3)
                .build();

        callApi(this, () -> mShortcutManager.setDynamicShortcuts(Arrays.asList(si1, si2, si3)));
        refreshList();
    }

    public void onDeleteAllPressed(View view) {
        callApi(this, () -> {
            mShortcutManager.removeAllDynamicShortcuts();
            return true;
        });
        refreshList();
    }

    static String formatTime(long time) {
        Time tobj = new Time();
        tobj.set(time);
        return tobj.format("%Y-%m-%d %H:%M:%S");
    }

    public void onAddPressed(View view) {
        final ShortcutInfo si = addRandomIntents(this, new ShortcutInfo.Builder(this,
                    "shortcut-" + formatTime(System.currentTimeMillis()) + "-"
                        + sSequenceNumber.getAndIncrement()))
                .setActivity(mMyActivity)
                .build();
        callApi(this, () -> mShortcutManager.addDynamicShortcuts(Arrays.asList(si)));
        refreshList();
    }

    public void onUpdatePressed(View view) {
        final List updateList = new ArrayList<>();

        for (ShortcutInfo si : getAllShortcuts()) {
            if (SETUP_SHORTCUT_ID.equals(si.getId())) continue;
            if (si.isImmutable()) continue;
            if (!Objects.equals(si.getActivity(), mMyActivity)) continue;
            updateList.add(addRandomIntents(this, new ShortcutInfo.Builder(this, si.getId()))
                    .build());
        }
        callApi(this, () -> mShortcutManager.updateShortcuts(updateList));
        refreshList();
    }

    void launch(ShortcutInfo si) {
        startActivity(si.getIntent());
    }

    void deleteDynamic(ShortcutInfo si) {
        mShortcutManager.removeDynamicShortcuts(Arrays.asList(si.getId()));
        refreshList();
    }

    public void onShowNotificationPressed(View v) {
        final PendingIntent receiverIntent =
                PendingIntent.getBroadcast(this, 0,
                        new Intent().setComponent(new ComponentName(this, ShortcutReceiver.class)),
                        PendingIntent.FLAG_UPDATE_CURRENT);
        final RemoteInput ri = new RemoteInput.Builder("result").setLabel("Remote input").build();

        final Notification.Builder nb = new Builder(this)
                .setContentText("Test")
                .setContentTitle(getPackageName())
                .setSmallIcon(R.drawable.icon_large_2)
                .addAction(new Action.Builder(0, "Remote input", receiverIntent)
                        .addRemoteInput(ri)
                        .build());
        getSystemService(NotificationManager.class).notify(0, nb.build());
    }

    // event handler
    public void onRequestPinPressed(View v) {
        requestPinShortcut(this);
    }

    public static void requestPinShortcut(Context context) {
        if (!ShortcutManagerCompat.isRequestPinShortcutSupported(context)) {
            showToast(context, "requestPinShortcut() is not supported by launcher");
            return;
        }
        final ShortcutInfoCompat si = addRandomIntents(
                context, new ShortcutInfoCompat.Builder(context,
                        "shortcut-" + System.currentTimeMillis()))
                .build();
        final PendingIntent resultIntent = PendingIntent.getBroadcast(context, 0,
                new Intent(context, RequestPinShortcutResultReceiver.class),
                PendingIntent.FLAG_UPDATE_CURRENT);

        ShortcutPublisher.callApi(context, () -> {
            ShortcutManagerCompat.requestPinShortcut(context,
                    si, resultIntent.getIntentSender());
            return true;
        });
    }

    class MyAdapter extends ShortcutAdapter {
        public MyAdapter(Context context) {
            super(context);
        }

        @Override
        protected int getLayoutId() {
            return R.layout.list_item;
        }

        @Override
        protected int getText1Id() {
            return R.id.line1;
        }

        @Override
        protected int getText2Id() {
            return R.id.line2;
        }

        @Override
        protected int getImageId() {
            return R.id.image;
        }

        @Override
        protected int getLaunchId() {
            return R.id.launch;
        }

        @Override
        protected int getAction2Id() {
            return R.id.action2;
        }

        @Override
        protected boolean showLaunch(ShortcutInfo si) {
            return true;
        }

        @Override
        protected boolean showAction2(ShortcutInfo si) {
            return si.isDynamic(); // TODO Need disable too.
        }

        @Override
        protected String getAction2Text(ShortcutInfo si) {
            return "Delete";
        }

        @Override
        protected void onLaunchClicked(ShortcutInfo si) {
            launch(si);
        }

        @Override
        protected void onAction2Clicked(ShortcutInfo si) {
            deleteDynamic(si);
        }
    }
}