samples/ShortcutSample/src/com/example/android/shortcutsample/Main.java

/*
 * 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.shortcutsample;

import android.app.Activity;
import android.app.AlertDialog;
import android.app.ListActivity;
import android.content.Context;
import android.content.Intent;
import android.content.pm.ShortcutInfo;
import android.content.pm.ShortcutManager;
import android.os.AsyncTask;
import android.os.Bundle;
import android.support.v4.content.pm.ShortcutManagerCompat;
import android.util.ArraySet;
import android.util.Log;
import android.view.LayoutInflater;
import android.view.View;
import android.view.View.OnClickListener;
import android.view.ViewGroup;
import android.view.inputmethod.EditorInfo;
import android.widget.BaseAdapter;
import android.widget.Button;
import android.widget.EditText;
import android.widget.TextView;

import java.util.ArrayList;
import java.util.List;
import java.util.Set;

public class Main extends ListActivity implements OnClickListener {
    static final String TAG = "ShortcutSample";

    private static final String ID_ADD_WEBSITE = "add_website";

    private static final String ACTION_ADD_WEBSITE =
            "com.example.android.shortcutsample.ADD_WEBSITE";

    private MyAdapter mAdapter;

    private ShortcutHelper mHelper;
    private ShortcutManager mShortcutManager;

    // @GuardedBy("sVisibleInstances")
    private static final Set<Main> sVisibleInstances = new ArraySet<>();

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

        setContentView(R.layout.main);

        mHelper = new ShortcutHelper(this);
        mShortcutManager = getSystemService(ShortcutManager.class);

        mHelper.maybeRestoreAllDynamicShortcuts();

        mHelper.refreshShortcuts(/*force=*/ false);

        if (ACTION_ADD_WEBSITE.equals(getIntent().getAction())) {
            // Invoked via the manifest shortcut.
            addWebSite(/* forPin=*/ false, /* forResult= */ false);
        } else if (Intent.ACTION_CREATE_SHORTCUT.equals(getIntent().getAction())) {
            addWebSite(/* forPin=*/ true, /* forResult= */ true);
        }

        mAdapter = new MyAdapter(this.getApplicationContext());
        setListAdapter(mAdapter);
    }

    @Override
    protected void onStart() {
        super.onStart();
        refreshList();
        synchronized (sVisibleInstances) {
            sVisibleInstances.add(this);
        }
        findViewById(R.id.request_new_pin_shortcut).setVisibility(
                ShortcutManagerCompat.isRequestPinShortcutSupported(this)
                        ? View.VISIBLE : View.GONE);
    }

    @Override
    protected void onStop() {
        synchronized (sVisibleInstances) {
            sVisibleInstances.remove(this);
        }
        super.onStop();
    }

    /**
     * Handle the add button.
     */
    public void onAddPressed(View v) {
        addWebSite(/* forPin=*/ false, /* forResult= */ false);
    }

    /**
     * Handle the add button.
     */
    public void onRequestNewPinPressed(View v) {
        addWebSite(/* forPin=*/ true, /* forResult= */ false);
    }

    private void addWebSite(boolean forPin, boolean forResult) {
        Log.i(TAG, "addWebSite forPin=" + forPin);

        // This is important.  This allows the launcher to build a prediction model.
        mHelper.reportShortcutUsed(ID_ADD_WEBSITE);

        final EditText editUri = new EditText(this);

        editUri.setHint("http://www.android.com/");
        editUri.setInputType(EditorInfo.TYPE_TEXT_VARIATION_URI);

        new AlertDialog.Builder(this)
                .setTitle(forPin ? "Create pin shortcut for website" : "Add new website")
                .setMessage("Type URL of a website")
                .setView(editUri)
                .setPositiveButton("Add", (dialog, whichButton) -> {
                    final String url = editUri.getText().toString().trim();
                    if (url.length() > 0) {
                        addUriAsync(url, forPin, forResult);
                    }
                })
                .setOnCancelListener((dialog) -> {
                    if (forResult) {
                        setResult(Activity.RESULT_CANCELED);
                        finish();
                    }
                })
                .show();
    }

    private void addUriAsync(String uri, boolean forPin, boolean forResult) {
        if (forResult) {
            new AsyncTask<Void, Void, ShortcutInfo>() {
                @Override
                protected ShortcutInfo doInBackground(Void... params) {
                    return mHelper.createShortcutForUrl(uri);
                }

                @Override
                protected void onPostExecute(ShortcutInfo shortcut) {
                    setResult(Activity.RESULT_OK,
                            mShortcutManager.createShortcutResultIntent(shortcut));
                    finish();
                }
            }.execute();
        } else {
            new AsyncTask<Void, Void, Void>() {
                @Override
                protected Void doInBackground(Void... params) {
                    mHelper.addWebSiteShortcut(uri, forPin);
                    return null;
                }

                @Override
                protected void onPostExecute(Void aVoid) {
                    refreshList();
                }
            }.execute();
        }
    }

    private void refreshList() {
        mAdapter.setShortcuts(mHelper.getShortcuts());
    }

    @Override
    public void onClick(View v) {
        final ShortcutInfo shortcut = (ShortcutInfo) ((View) v.getParent()).getTag();

        switch (v.getId()) {
            case R.id.disable:
                if (shortcut.isEnabled()) {
                    mHelper.disableShortcut(shortcut);
                } else {
                    mHelper.enableShortcut(shortcut);
                }
                refreshList();
                break;
            case R.id.remove:
                mHelper.removeShortcut(shortcut);
                refreshList();
                break;
            case R.id.request_pin:
                // This is an update case, so just pass the ID.
                mHelper.requestPinShortcut(shortcut.getId());
                refreshList();
                break;
        }
    }

    private static final List<ShortcutInfo> EMPTY_LIST = new ArrayList<>();

    private String getType(ShortcutInfo shortcut) {
        final StringBuilder sb = new StringBuilder();
        String sep = "";
        if (shortcut.isDeclaredInManifest()) {
            sb.append(sep);
            sb.append("Manifest");
            sep = ", ";
        }
        if (shortcut.isDynamic()) {
            sb.append(sep);
            sb.append("Dynamic");
            sep = ", ";
        }
        if (shortcut.isPinned()) {
            sb.append(sep);
            sb.append("Pinned");
            sep = ", ";
        }
        if (!shortcut.isEnabled()) {
            sb.append(sep);
            sb.append("Disabled");
            sep = ", ";
        }
        return sb.toString();
    }

    private class MyAdapter extends BaseAdapter {
        private final Context mContext;
        private final LayoutInflater mInflater;
        private List<ShortcutInfo> mList = EMPTY_LIST;

        public MyAdapter(Context context) {
            mContext = context;
            mInflater = mContext.getSystemService(LayoutInflater.class);
        }

        @Override
        public int getCount() {
            return mList.size();
        }

        @Override
        public Object getItem(int position) {
            return mList.get(position);
        }

        @Override
        public long getItemId(int position) {
            return position;
        }

        @Override
        public boolean hasStableIds() {
            return false;
        }

        @Override
        public boolean areAllItemsEnabled() {
            return true;
        }

        @Override
        public boolean isEnabled(int position) {
            return true;
        }

        public void setShortcuts(List<ShortcutInfo> list) {
            mList = list;
            notifyDataSetChanged();
        }

        @Override
        public View getView(int position, View convertView, ViewGroup parent) {
            final View view;
            if (convertView != null) {
                view = convertView;
            } else {
                view = mInflater.inflate(R.layout.list_item, null);
            }

            bindView(view, position, mList.get(position));

            return view;
        }

        public void bindView(View view, int position, ShortcutInfo shortcut) {
            view.setTag(shortcut);

            final TextView line1 = (TextView) view.findViewById(R.id.line1);
            final TextView line2 = (TextView) view.findViewById(R.id.line2);

            line1.setText(shortcut.getLongLabel());
            line2.setText(getType(shortcut));

            final Button remove = (Button) view.findViewById(R.id.remove);
            final Button disable = (Button) view.findViewById(R.id.disable);
            final Button requestPin = (Button) view.findViewById(R.id.request_pin);

            disable.setText(
                    shortcut.isEnabled() ? R.string.disable_shortcut : R.string.enable_shortcut);

            remove.setVisibility(shortcut.isImmutable() ? View.GONE : View.VISIBLE);
            disable.setVisibility(shortcut.isImmutable() ? View.GONE : View.VISIBLE);
            requestPin.setVisibility(
                    ShortcutManagerCompat.isRequestPinShortcutSupported(mContext)
                            ? View.VISIBLE : View.GONE);

            remove.setOnClickListener(Main.this);
            disable.setOnClickListener(Main.this);
            requestPin.setOnClickListener(Main.this);
        }
    }

    public static void refreshAllInstances() {
        synchronized (sVisibleInstances) {
            for (Main instance : sVisibleInstances) {
                instance.refreshList();
            }
        }
    }
}