android/quicksearchbox/SearchActivity.java

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

import com.android.quicksearchbox.ui.SearchSourceSelector;
import com.android.quicksearchbox.ui.SuggestionClickListener;
import com.android.quicksearchbox.ui.SuggestionViewFactory;
import com.android.quicksearchbox.ui.SuggestionsAdapter;
import com.android.quicksearchbox.ui.SuggestionsView;

import android.app.Activity;
import android.app.SearchManager;
import android.content.ComponentName;
import android.content.Intent;
import android.database.DataSetObserver;
import android.graphics.Rect;
import android.graphics.drawable.Animatable;
import android.graphics.drawable.Drawable;
import android.os.Bundle;
import android.text.Editable;
import android.text.TextUtils;
import android.text.TextWatcher;
import android.util.Log;
import android.view.KeyEvent;
import android.view.Menu;
import android.view.View;
import android.view.View.OnFocusChangeListener;
import android.view.inputmethod.InputMethodManager;
import android.widget.EditText;
import android.widget.ImageButton;

// TODO: restore user query when using dpad to move up from top suggestion
// TODO: nicer progress animation. one per source?
// TODO: handle long clicks
// TODO: support action keys
// TODO: support IME search action
// TODO: allow typing everywhere in the UI

// TODO: don't show new results until there is at least one, or it's done
// TODO: add timeout for source queries
// TODO: group refreshes that happen close to each other.
// TODO: use queryAfterZeroResults()

// TODO: use permission to get extended Genie suggestions
// TODO: support intent extras for source (e.g. launcher widget)
// TODO: support intent extras for initial state, e.g. query, selection
// TODO: make Config server-side configurable
// TODO: log impressions
// TODO: use source ranking

/**
 * The main activity for Quick Search Box. Shows the search UI.
 *
 */
public class SearchActivity extends Activity {

    private static final boolean DBG = true;
    private static final String TAG = "QSB.SearchActivity";

    // TODO: This is hidden in SearchManager
    public final static String INTENT_ACTION_SEARCH_SETTINGS
            = "android.search.action.SEARCH_SETTINGS";

    public static final String EXTRA_KEY_SEARCH_SOURCE
            = "search_source";

    // Keys for the saved instance state.
    private static final String INSTANCE_KEY_SOURCE = "source";
    private static final String INSTANCE_KEY_USER_QUERY = "query";

    protected SuggestionsAdapter mSuggestionsAdapter;

    protected EditText mQueryTextView;

    protected SuggestionsView mSuggestionsView;

    protected ImageButton mSearchGoButton;
    protected ImageButton mVoiceSearchButton;
    protected SearchSourceSelector mSourceSelector;

    private Launcher mLauncher;

    private Source mSource;
    private boolean mUpdateSuggestions;
    private String mUserQuery;
    private boolean mSelectAll;

    /** Called when the activity is first created. */
    @Override
    public void onCreate(Bundle savedInstanceState) {
        if (DBG) Log.d(TAG, "onCreate()");
        super.onCreate(savedInstanceState);

        setContentView(R.layout.search_bar);

        Config config = getConfig();
        SuggestionViewFactory viewFactory = getSuggestionViewFactory();
        mSuggestionsAdapter = new SuggestionsAdapter(viewFactory);
        mSuggestionsAdapter
                .setInitialSourceResultWaitMillis(config.getInitialSourceResultWaitMillis());
        mSuggestionsAdapter
                .setSourceResultPublishDelayMillis(config.getSourceResultPublishDelayMillis());

        mQueryTextView = (EditText) findViewById(R.id.search_src_text);
        mSuggestionsView = (SuggestionsView) findViewById(R.id.suggestions);
        mSuggestionsView.setSuggestionClickListener(new ClickHandler());
        mSuggestionsView.setInteractionListener(new InputMethodCloser());
        mSuggestionsView.setOnKeyListener(new SuggestionsViewKeyListener());

        mSearchGoButton = (ImageButton) findViewById(R.id.search_go_btn);
        mVoiceSearchButton = (ImageButton) findViewById(R.id.search_voice_btn);
        mSourceSelector = new SearchSourceSelector(findViewById(R.id.search_source_selector));

        mLauncher = new Launcher(this);
        // TODO: should this check for voice search in the current source?
        mVoiceSearchButton.setVisibility(
                mLauncher.isVoiceSearchAvailable() ? View.VISIBLE : View.GONE);

        mQueryTextView.addTextChangedListener(new SearchTextWatcher());
        mQueryTextView.setOnKeyListener(new QueryTextViewKeyListener());
        mQueryTextView.setOnFocusChangeListener(new SuggestListFocusListener());

        mSearchGoButton.setOnClickListener(new SearchGoButtonClickListener());
        mSearchGoButton.setOnKeyListener(new ButtonsKeyListener());

        mVoiceSearchButton.setOnClickListener(new VoiceSearchButtonClickListener());
        mVoiceSearchButton.setOnKeyListener(new ButtonsKeyListener());

        mUpdateSuggestions = true;

        // First get setup from intent
        Intent intent = getIntent();
        setupFromIntent(intent);
        // Then restore any saved instance state
        restoreInstanceState(savedInstanceState);

        // Do this at the end, to avoid updating the list view when setSource()
        // is called.
        mSuggestionsView.setAdapter(mSuggestionsAdapter);
    }

    @Override
    protected void onNewIntent(Intent intent) {
        setIntent(intent);
        setupFromIntent(intent);
    }

    protected void restoreInstanceState(Bundle savedInstanceState) {
        if (savedInstanceState == null) return;
        ComponentName sourceName = savedInstanceState.getParcelable(INSTANCE_KEY_SOURCE);
        String query = savedInstanceState.getString(INSTANCE_KEY_USER_QUERY);
        setSource(getSourceByComponentName(sourceName));
        setUserQuery(query);
    }

    @Override
    protected void onSaveInstanceState(Bundle outState) {
        super.onSaveInstanceState(outState);
        // We don't save appSearchData, since we always get the value
        // from the intent and the user can't change it.
        outState.putParcelable(INSTANCE_KEY_SOURCE, getSourceName());
        outState.putString(INSTANCE_KEY_USER_QUERY, mUserQuery);
    }

    private void setupFromIntent(Intent intent) {
        if (DBG) Log.d(TAG, "setupFromIntent(" + intent.toUri(0) + ")");
        if (intent.hasExtra(EXTRA_KEY_SEARCH_SOURCE)) {
            Source source = getSourceByName(intent.getStringExtra(EXTRA_KEY_SEARCH_SOURCE));
            setSource(source);
            // The source was selected by the user, save it.
            setLastSelectedSource(source);
        } else {
            Source source = getSources().getLastSelectedSource();
            if (DBG) Log.d(TAG, "Setting source from preferences: " + source);
            setSource(source);
        }
        // TODO: Should this be SearchManager.INITIAL_QUERY?
        setUserQuery(intent.getStringExtra(SearchManager.QUERY));
        // TODO: Expose SearchManager.SELECT_INITIAL_QUERY
        mSelectAll = false;
        setAppSearchData(intent.getBundleExtra(SearchManager.APP_DATA));
    }

    private void setLastSelectedSource(Source source) {
        getSources().setLastSelectedSource(source);
        // Update search widgets to show the new source.
        SearchWidgetProvider.updateSearchWidgets(this);
    }

    private Source getSourceByName(String sourceNameStr) {
        if (sourceNameStr == null) return null;
        ComponentName sourceName = ComponentName.unflattenFromString(sourceNameStr);
        if (sourceName == null) {
            Log.w(TAG, "Malformed source name: " + sourceName);
            return null;
        }
        return getSourceByComponentName(sourceName);
    }

    private Source getSourceByComponentName(ComponentName sourceName) {
        Source source = getSources().getSourceByComponentName(sourceName);
        if (source == null) {
            Log.w(TAG, "Unknown source " + sourceName);
            return null;
        }
        return source;
    }

    private void setSource(Source source) {
        if (DBG) Log.d(TAG, "setSource(" + source + ")");
        mSource = source;
        Drawable sourceIcon;
        if (source == null) {
            sourceIcon = getSuggestionViewFactory().getGlobalSearchIcon();
        } else {
            sourceIcon = source.getSourceIcon();
        }
        ComponentName sourceName = getSourceName();
        mSuggestionsAdapter.setSource(sourceName);
        mSourceSelector.setSource(sourceName);
        mSourceSelector.setSourceIcon(sourceIcon);
    }

    private ComponentName getSourceName() {
        return mSource == null ? null : mSource.getComponentName();
    }

    private QsbApplication getQsbApplication() {
        return (QsbApplication) getApplication();
    }

    private Config getConfig() {
        return getQsbApplication().getConfig();
    }

    private SourceLookup getSources() {
        return getQsbApplication().getSources();
    }

    private ShortcutRepository getShortcutRepository() {
        return getQsbApplication().getShortcutRepository();
    }

    private SuggestionsProvider getSuggestionsProvider(Source source) {
        return getQsbApplication().getSuggestionsProvider(source);
    }

    private SuggestionViewFactory getSuggestionViewFactory() {
        return getQsbApplication().getSuggestionViewFactory();
    }

    @Override
    protected void onDestroy() {
        if (DBG) Log.d(TAG, "onDestroy()");
        super.onDestroy();
        mSuggestionsView.setAdapter(null);  // closes mSuggestionsAdapter
    }

    @Override
    protected void onStop() {
        if (DBG) Log.d(TAG, "onStop()");
        // Close all open suggestion cursors. The query will be redone in onResume()
        // if we come back to this activity.
        mSuggestionsAdapter.setSuggestions(null);
        super.onStop();
    }

    @Override
    protected void onResume() {
        if (DBG) Log.d(TAG, "onResume()");
        super.onResume();
        setQuery(mUserQuery, mSelectAll);
        // Only select everything the first time after creating the activity.
        mSelectAll = false;
        updateSuggestions(mUserQuery);
        mQueryTextView.requestFocus();
    }

    @Override
    public boolean onCreateOptionsMenu(Menu menu) {
        super.onCreateOptionsMenu(menu);

        Intent settings = new Intent(INTENT_ACTION_SEARCH_SETTINGS);
        settings.addFlags(Intent.FLAG_ACTIVITY_CLEAR_WHEN_TASK_RESET);
        // Don't show activity chooser if there are multiple search settings activities,
        // e.g. from different QSB implementations.
        settings.setPackage(this.getPackageName());
        menu.add(Menu.NONE, Menu.NONE, 0, R.string.menu_settings)
                .setIcon(android.R.drawable.ic_menu_preferences).setAlphabeticShortcut('P')
                .setIntent(settings);

        return true;
    }

    /**
     * Sets the query as typed by the user. Does not update the suggestions
     * or the text in the query box.
     */
    protected void setUserQuery(String userQuery) {
        if (userQuery == null) userQuery = "";
        mUserQuery = userQuery;
        mSourceSelector.setQuery(mUserQuery);
    }

    protected void setAppSearchData(Bundle appSearchData) {
        mLauncher.setAppSearchData(appSearchData);
        mSourceSelector.setAppSearchData(appSearchData);
    }

    protected String getQuery() {
        CharSequence q = mQueryTextView.getText();
        return q == null ? "" : q.toString();
    }

    /**
     * Restores the query entered by the user.
     */
    private void restoreUserQuery() {
        if (DBG) Log.d(TAG, "Restoring query to '" + mUserQuery + "'");
        setQuery(mUserQuery, false);
    }

    /**
     * Sets the text in the query box. Does not update the suggestions,
     * and does not change the saved user-entered query.
     * {@link #restoreUserQuery()} will restore the query to the last
     * user-entered query.
     */
    private void setQuery(String query, boolean selectAll) {
        mUpdateSuggestions = false;
        mQueryTextView.setText(query);
        setTextSelection(selectAll);
        mUpdateSuggestions = true;
    }

    /**
     * Sets the text selection in the query text view.
     *
     * @param selectAll If {@code true}, selects the entire query.
     *        If {@false}, no characters are selected, and the cursor is placed
     *        at the end of the query.
     */
    private void setTextSelection(boolean selectAll) {
        if (selectAll) {
            mQueryTextView.setSelection(0, mQueryTextView.length());
        } else {
            mQueryTextView.setSelection(mQueryTextView.length());
        }
    }

    protected void onSearchClicked() {
        String query = getQuery();
        if (DBG) Log.d(TAG, "Search clicked, query=" + query);
        mLauncher.startSearch(mSource, query);
    }

    protected void onVoiceSearchClicked() {
        if (DBG) Log.d(TAG, "Voice Search clicked");
        // TODO: should this start voice search in the current source?
        mLauncher.startVoiceSearch();
    }

    protected boolean launchSuggestion(SuggestionPosition suggestion) {
        return launchSuggestion(suggestion, KeyEvent.KEYCODE_UNKNOWN, null);
    }

    protected boolean launchSuggestion(SuggestionPosition suggestion,
            int actionKey, String actionMsg) {
        if (DBG) Log.d(TAG, "Launching suggestion " + suggestion);
        mLauncher.launchSuggestion(suggestion, actionKey, actionMsg);
        getShortcutRepository().reportClick(suggestion);
        // Update search widgets, since the top shortcuts can have changed.
        SearchWidgetProvider.updateSearchWidgets(this);
        return true;
    }

    protected boolean launchSuggestionSecondary(SuggestionPosition suggestion, Rect target) {
      if (DBG) Log.d(TAG, "Clicked on suggestion icon " + suggestion);
      mLauncher.launchSuggestionSecondary(suggestion, target);
      getShortcutRepository().reportClick(suggestion);
      return true;
    }

    protected boolean onSuggestionLongClicked(SuggestionPosition suggestion) {
        SuggestionCursor sourceResult = suggestion.getSuggestion();
        if (DBG) Log.d(TAG, "Long clicked on suggestion " + sourceResult.getSuggestionText1());
        return false;
    }

    protected void onSuggestionSelected(SuggestionPosition suggestion) {
        SuggestionCursor sourceResult = suggestion.getSuggestion();
        String displayQuery = sourceResult.getSuggestionDisplayQuery();
        if (DBG) {
            Log.d(TAG, "Selected suggestion " + sourceResult.getSuggestionText1()
                    + ",displayQuery="+ displayQuery);
        }
        if (TextUtils.isEmpty(displayQuery)) {
            restoreUserQuery();
        } else {
            setQuery(displayQuery, false);
        }
    }

    protected boolean onSuggestionKeyDown(SuggestionPosition suggestion,
            int keyCode, KeyEvent event) {
        // Treat enter or search as a click
        if (keyCode == KeyEvent.KEYCODE_ENTER || keyCode == KeyEvent.KEYCODE_SEARCH) {
            return launchSuggestion(suggestion);
        }

        // Handle source-specified action keys
        String actionMsg = suggestion.getSuggestion().getActionKeyMsg(keyCode);
        if (actionMsg != null) {
            return launchSuggestion(suggestion, keyCode, actionMsg);
        }

        return false;
    }

    protected void onSourceSelected() {
        if (DBG) Log.d(TAG, "No suggestion selected");
        restoreUserQuery();
    }

    protected int getSelectedPosition() {
        return mSuggestionsView.getSelectedPosition();
    }

    protected SuggestionPosition getSelectedSuggestion() {
        return mSuggestionsView.getSelectedSuggestion();
    }

    /**
     * Hides the input method.
     */
    protected void hideInputMethod() {
        InputMethodManager imm = (InputMethodManager) getSystemService(INPUT_METHOD_SERVICE);
        if (imm != null) {
            imm.hideSoftInputFromWindow(mQueryTextView.getWindowToken(), 0);
        }
    }

    /**
     * Hides the input method when the suggestions get focus.
     */
    private class SuggestListFocusListener implements OnFocusChangeListener {
        public void onFocusChange(View v, boolean focused) {
            if (v == mQueryTextView) {
                if (!focused) {
                    hideInputMethod();
                } else {
                    // TODO: clear list selection?
                }
            }
        }
    }

    private void startSearchProgress() {
        // TODO: Cache animation between calls?
        mSearchGoButton.setImageResource(R.drawable.searching);
        Animatable animation = (Animatable) mSearchGoButton.getDrawable();
        animation.start();
    }

    private void stopSearchProgress() {
        Drawable animation = mSearchGoButton.getDrawable();
        if (animation instanceof Animatable) {
            // TODO: Is this needed, or is it done automatically when the
            // animation is removed?
            ((Animatable) animation).stop();
        }
        mSearchGoButton.setImageResource(R.drawable.ic_btn_search);
    }

    private void updateSuggestions(String query) {
        LatencyTracker latency = new LatencyTracker(TAG);
        Suggestions suggestions = getSuggestionsProvider(mSource).getSuggestions(query);
        latency.addEvent("getSuggestions_done");
        if (!suggestions.isDone()) {
            suggestions.registerDataSetObserver(new ProgressUpdater(suggestions));
            startSearchProgress();
        } else {
            stopSearchProgress();
        }
        mSuggestionsAdapter.setSuggestions(suggestions);
        latency.addEvent("shortcuts_shown");
        long userVisibleLatency = latency.getUserVisibleLatency();
        if (DBG) {
            Log.d(TAG, "User visible latency (shortcuts): " + userVisibleLatency + " ms.");
        }
    }

    /**
     * Filters the suggestions list when the search text changes.
     */
    private class SearchTextWatcher implements TextWatcher {
        public void afterTextChanged(Editable s) {
            if (mUpdateSuggestions) {
                String query = s == null ? "" : s.toString();
                setUserQuery(query);
                updateSuggestions(query);
            }
        }

        public void beforeTextChanged(CharSequence s, int start, int count, int after) {
        }

        public void onTextChanged(CharSequence s, int start, int before, int count) {
        }
    }

    /**
     * Handles non-text keys in the query text view.
     */
    private class QueryTextViewKeyListener implements View.OnKeyListener {
        public boolean onKey(View view, int keyCode, KeyEvent event) {
            // Handle IME search action key
            if (keyCode == KeyEvent.KEYCODE_ENTER && event.getAction() == KeyEvent.ACTION_UP) {
                onSearchClicked();
            }
            return false;
        }
    }

    /**
     * Handles key events on the search and voice search buttons,
     * by refocusing to EditText.
     */
    private class ButtonsKeyListener implements View.OnKeyListener {
        public boolean onKey(View v, int keyCode, KeyEvent event) {
            if (!event.isSystem() &&
                    (keyCode != KeyEvent.KEYCODE_DPAD_UP) &&
                    (keyCode != KeyEvent.KEYCODE_DPAD_LEFT) &&
                    (keyCode != KeyEvent.KEYCODE_DPAD_RIGHT) &&
                    (keyCode != KeyEvent.KEYCODE_DPAD_CENTER)) {
                if (mQueryTextView.requestFocus()) {
                    return mQueryTextView.dispatchKeyEvent(event);
                }
            }
            return false;
        }
    }

    /**
     * Handles key events on the suggestions list view.
     */
    private class SuggestionsViewKeyListener implements View.OnKeyListener {
        public boolean onKey(View v, int keyCode, KeyEvent event) {
            if (event.getAction() == KeyEvent.ACTION_DOWN) {
                SuggestionPosition suggestion = getSelectedSuggestion();
                if (suggestion != null) {
                    return onSuggestionKeyDown(suggestion, keyCode, event);
                }
            }
            return false;
        }
    }

    private class InputMethodCloser implements SuggestionsView.InteractionListener {
        public void onInteraction() {
            hideInputMethod();
        }
    }

    private class ClickHandler implements SuggestionClickListener {
       public void onSuggestionClicked(SuggestionPosition suggestion) {
           launchSuggestion(suggestion);
       }

       public boolean onSuggestionLongClicked(SuggestionPosition suggestion) {
           return SearchActivity.this.onSuggestionLongClicked(suggestion);
       }

       public void onSuggestionSelected(SuggestionPosition suggestion) {
           SearchActivity.this.onSuggestionSelected(suggestion);
       }

       public void onSuggestionIconClicked(SuggestionPosition suggestion, Rect rect) {
           launchSuggestionSecondary(suggestion, rect);
       }
    }

    /**
     * Listens for clicks on the search button.
     */
    private class SearchGoButtonClickListener implements View.OnClickListener {
        public void onClick(View view) {
            onSearchClicked();
        }
    }

    /**
     * Listens for clicks on the voice search button.
     */
    private class VoiceSearchButtonClickListener implements View.OnClickListener {
        public void onClick(View view) {
            onVoiceSearchClicked();
        }
    }

    /**
     * Updates the progress bar when the suggestions adapter changes its progress.
     */
    private class ProgressUpdater extends DataSetObserver {
        private final Suggestions mSuggestions;

        public ProgressUpdater(Suggestions suggestions) {
            mSuggestions = suggestions;
        }

        @Override
        public void onChanged() {
            if (mSuggestions.isDone()) {
                stopSearchProgress();
            }
        }
    }

}