/* * Copyright (C) 2014 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.support.v17.leanback.app; import android.app.Fragment; import android.content.Intent; import android.graphics.drawable.Drawable; import android.os.Bundle; import android.os.Handler; import android.speech.SpeechRecognizer; import android.speech.RecognizerIntent; import android.support.v17.leanback.widget.ObjectAdapter; import android.support.v17.leanback.widget.ObjectAdapter.DataObserver; import android.support.v17.leanback.widget.OnItemClickedListener; import android.support.v17.leanback.widget.OnItemSelectedListener; import android.support.v17.leanback.widget.OnItemViewClickedListener; import android.support.v17.leanback.widget.OnItemViewSelectedListener; import android.support.v17.leanback.widget.Row; import android.support.v17.leanback.widget.RowPresenter; import android.support.v17.leanback.widget.SearchBar; import android.support.v17.leanback.widget.VerticalGridView; import android.support.v17.leanback.widget.Presenter.ViewHolder; import android.support.v17.leanback.widget.SpeechRecognitionCallback; import android.util.Log; import android.view.LayoutInflater; import android.view.View; import android.view.ViewGroup; import android.widget.FrameLayout; import android.support.v17.leanback.R; import java.util.ArrayList; import java.util.List; /** * A fragment to handle searches. An application will supply an implementation * of the {@link SearchResultProvider} interface to handle the search and return * an {@link ObjectAdapter} containing the results. The results are rendered * into a {@link RowsFragment}, in the same way that they are in a {@link * BrowseFragment}. * *

If you do not supply a callback via * {@link #setSpeechRecognitionCallback(SpeechRecognitionCallback)}, an internal speech * recognizer will be used for which your application will need to request * android.permission.RECORD_AUDIO. *

*

* Speech recognition is automatically started when fragment is created, but * not when fragment is restored from an instance state. Activity may manually * call {@link #startRecognition()}, typically in onNewIntent(). *

*/ public class SearchFragment extends Fragment { private static final String TAG = SearchFragment.class.getSimpleName(); private static final boolean DEBUG = false; private static final String EXTRA_LEANBACK_BADGE_PRESENT = "LEANBACK_BADGE_PRESENT"; private static final String ARG_PREFIX = SearchFragment.class.getCanonicalName(); private static final String ARG_QUERY = ARG_PREFIX + ".query"; private static final String ARG_TITLE = ARG_PREFIX + ".title"; private static final long SPEECH_RECOGNITION_DELAY_MS = 300; private static final int RESULTS_CHANGED = 0x1; private static final int QUERY_COMPLETE = 0x2; /** * Search API to be provided by the application. */ public static interface SearchResultProvider { /** *

Method invoked some time prior to the first call to onQueryTextChange to retrieve * an ObjectAdapter that will contain the results to future updates of the search query.

* *

As results are retrieved, the application should use the data set notification methods * on the ObjectAdapter to instruct the SearchFragment to update the results.

* * @return ObjectAdapter The result object adapter. */ public ObjectAdapter getResultsAdapter(); /** *

Method invoked when the search query is updated.

* *

This is called as soon as the query changes; it is up to the application to add a * delay before actually executing the queries if needed. * *

This method might not always be called before onQueryTextSubmit gets called, in * particular for voice input. * * @param newQuery The current search query. * @return whether the results changed as a result of the new query. */ public boolean onQueryTextChange(String newQuery); /** * Method invoked when the search query is submitted, either by dismissing the keyboard, * pressing search or next on the keyboard or when voice has detected the end of the query. * * @param query The query entered. * @return whether the results changed as a result of the query. */ public boolean onQueryTextSubmit(String query); } private final DataObserver mAdapterObserver = new DataObserver() { @Override public void onChanged() { // onChanged() may be called multiple times e.g. the provider add // rows to ArrayObjectAdapter one by one. mHandler.removeCallbacks(mResultsChangedCallback); mHandler.post(mResultsChangedCallback); } }; private final Handler mHandler = new Handler(); private final Runnable mResultsChangedCallback = new Runnable() { @Override public void run() { if (DEBUG) Log.v(TAG, "results changed, new size " + mResultAdapter.size()); if (mRowsFragment != null && mRowsFragment.getAdapter() != mResultAdapter) { if (!(mRowsFragment.getAdapter() == null && mResultAdapter.size() == 0)) { mRowsFragment.setAdapter(mResultAdapter); mRowsFragment.setSelectedPosition(0); } } mStatus |= RESULTS_CHANGED; if ((mStatus & QUERY_COMPLETE) != 0) { updateFocus(); } updateSearchBarNextFocusId(); } }; /** * Runs when a new provider is set AND when the fragment view is created. */ private final Runnable mSetSearchResultProvider = new Runnable() { @Override public void run() { if (mRowsFragment == null) { // We'll retry once we have a rows fragment return; } // Retrieve the result adapter ObjectAdapter adapter = mProvider.getResultsAdapter(); if (DEBUG) Log.v(TAG, "Got results adapter " + adapter); if (adapter != mResultAdapter) { boolean firstTime = mResultAdapter == null; releaseAdapter(); mResultAdapter = adapter; if (mResultAdapter != null) { mResultAdapter.registerObserver(mAdapterObserver); } if (DEBUG) Log.v(TAG, "mResultAdapter " + mResultAdapter + " size " + (mResultAdapter == null ? 0 : mResultAdapter.size())); // delay the first time to avoid setting a empty result adapter // until we got first onChange() from the provider if (!(firstTime && (mResultAdapter == null || mResultAdapter.size() == 0))) { mRowsFragment.setAdapter(mResultAdapter); } executePendingQuery(); } updateSearchBarNextFocusId(); if (DEBUG) Log.v(TAG, "mAutoStartRecognition " + mAutoStartRecognition + " mResultAdapter " + mResultAdapter + " adapter " + mRowsFragment.getAdapter()); if (mAutoStartRecognition) { mHandler.removeCallbacks(mStartRecognitionRunnable); mHandler.postDelayed(mStartRecognitionRunnable, SPEECH_RECOGNITION_DELAY_MS); } else { updateFocus(); } } }; private final Runnable mStartRecognitionRunnable = new Runnable() { @Override public void run() { mAutoStartRecognition = false; mSearchBar.startRecognition(); } }; private RowsFragment mRowsFragment; private SearchBar mSearchBar; private SearchResultProvider mProvider; private String mPendingQuery = null; private OnItemSelectedListener mOnItemSelectedListener; private OnItemClickedListener mOnItemClickedListener; private OnItemViewSelectedListener mOnItemViewSelectedListener; private OnItemViewClickedListener mOnItemViewClickedListener; private ObjectAdapter mResultAdapter; private SpeechRecognitionCallback mSpeechRecognitionCallback; private String mTitle; private Drawable mBadgeDrawable; private ExternalQuery mExternalQuery; private SpeechRecognizer mSpeechRecognizer; private int mStatus; private boolean mAutoStartRecognition = true; /** * @param args Bundle to use for the arguments, if null a new Bundle will be created. */ public static Bundle createArgs(Bundle args, String query) { return createArgs(args, query, null); } public static Bundle createArgs(Bundle args, String query, String title) { if (args == null) { args = new Bundle(); } args.putString(ARG_QUERY, query); args.putString(ARG_TITLE, title); return args; } /** * Create a search fragment with a given search query. * *

You should only use this if you need to start the search fragment with a * pre-filled query. * * @param query The search query to begin with. * @return A new SearchFragment. */ public static SearchFragment newInstance(String query) { SearchFragment fragment = new SearchFragment(); Bundle args = createArgs(null, query); fragment.setArguments(args); return fragment; } @Override public void onCreate(Bundle savedInstanceState) { if (mAutoStartRecognition) { mAutoStartRecognition = savedInstanceState == null; } super.onCreate(savedInstanceState); } @Override public View onCreateView(LayoutInflater inflater, ViewGroup container, Bundle savedInstanceState) { View root = inflater.inflate(R.layout.lb_search_fragment, container, false); FrameLayout searchFrame = (FrameLayout) root.findViewById(R.id.lb_search_frame); mSearchBar = (SearchBar) searchFrame.findViewById(R.id.lb_search_bar); mSearchBar.setSearchBarListener(new SearchBar.SearchBarListener() { @Override public void onSearchQueryChange(String query) { if (DEBUG) Log.v(TAG, String.format("onSearchQueryChange %s %s", query, null == mProvider ? "(null)" : mProvider)); if (null != mProvider) { retrieveResults(query); } else { mPendingQuery = query; } } @Override public void onSearchQuerySubmit(String query) { if (DEBUG) Log.v(TAG, String.format("onSearchQuerySubmit %s", query)); submitQuery(query); } @Override public void onKeyboardDismiss(String query) { if (DEBUG) Log.v(TAG, String.format("onKeyboardDismiss %s", query)); queryComplete(); } }); mSearchBar.setSpeechRecognitionCallback(mSpeechRecognitionCallback); applyExternalQuery(); readArguments(getArguments()); if (null != mBadgeDrawable) { setBadgeDrawable(mBadgeDrawable); } if (null != mTitle) { setTitle(mTitle); } // Inject the RowsFragment in the results container if (getChildFragmentManager().findFragmentById(R.id.lb_results_frame) == null) { mRowsFragment = new RowsFragment(); getChildFragmentManager().beginTransaction() .replace(R.id.lb_results_frame, mRowsFragment).commit(); } else { mRowsFragment = (RowsFragment) getChildFragmentManager() .findFragmentById(R.id.lb_results_frame); } mRowsFragment.setOnItemViewSelectedListener(new OnItemViewSelectedListener() { @Override public void onItemSelected(ViewHolder itemViewHolder, Object item, RowPresenter.ViewHolder rowViewHolder, Row row) { int position = mRowsFragment.getVerticalGridView().getSelectedPosition(); if (DEBUG) Log.v(TAG, String.format("onItemSelected %d", position)); mSearchBar.setVisibility(0 >= position ? View.VISIBLE : View.GONE); if (null != mOnItemSelectedListener) { mOnItemSelectedListener.onItemSelected(item, row); } if (null != mOnItemViewSelectedListener) { mOnItemViewSelectedListener.onItemSelected(itemViewHolder, item, rowViewHolder, row); } } }); mRowsFragment.setOnItemViewClickedListener(new OnItemViewClickedListener() { @Override public void onItemClicked(ViewHolder itemViewHolder, Object item, RowPresenter.ViewHolder rowViewHolder, Row row) { int position = mRowsFragment.getVerticalGridView().getSelectedPosition(); if (DEBUG) Log.v(TAG, String.format("onItemClicked %d", position)); if (null != mOnItemClickedListener) { mOnItemClickedListener.onItemClicked(item, row); } if (null != mOnItemViewClickedListener) { mOnItemViewClickedListener.onItemClicked(itemViewHolder, item, rowViewHolder, row); } } }); mRowsFragment.setExpand(true); if (null != mProvider) { onSetSearchResultProvider(); } return root; } private void resultsAvailable() { if ((mStatus & QUERY_COMPLETE) != 0) { focusOnResults(); } updateSearchBarNextFocusId(); } @Override public void onStart() { super.onStart(); VerticalGridView list = mRowsFragment.getVerticalGridView(); int mContainerListAlignTop = getResources().getDimensionPixelSize(R.dimen.lb_search_browse_rows_align_top); list.setItemAlignmentOffset(0); list.setItemAlignmentOffsetPercent(VerticalGridView.ITEM_ALIGN_OFFSET_PERCENT_DISABLED); list.setWindowAlignmentOffset(mContainerListAlignTop); list.setWindowAlignmentOffsetPercent(VerticalGridView.WINDOW_ALIGN_OFFSET_PERCENT_DISABLED); list.setWindowAlignment(VerticalGridView.WINDOW_ALIGN_NO_EDGE); } @Override public void onResume() { super.onResume(); if (mSpeechRecognitionCallback == null && null == mSpeechRecognizer) { mSpeechRecognizer = SpeechRecognizer.createSpeechRecognizer(getActivity()); mSearchBar.setSpeechRecognizer(mSpeechRecognizer); } // Ensure search bar state consistency when using external recognizer mSearchBar.stopRecognition(); } @Override public void onPause() { releaseRecognizer(); super.onPause(); } @Override public void onDestroy() { releaseAdapter(); super.onDestroy(); } private void releaseRecognizer() { if (null != mSpeechRecognizer) { mSearchBar.setSpeechRecognizer(null); mSpeechRecognizer.destroy(); mSpeechRecognizer = null; } } /** * Starts speech recognition. Typical use case is that * activity receives onNewIntent() call when user clicks a MIC button. * Note that SearchFragment automatically starts speech recognition * at first time created, there is no need to call startRecognition() * when fragment is created. */ public void startRecognition() { mSearchBar.startRecognition(); } /** * Set the search provider that is responsible for returning results for the * search query. */ public void setSearchResultProvider(SearchResultProvider searchResultProvider) { if (mProvider != searchResultProvider) { mProvider = searchResultProvider; onSetSearchResultProvider(); } } /** * Sets an item selection listener for the results. * * @param listener The item selection listener to be invoked when an item in * the search results is selected. * @deprecated Use {@link #setOnItemViewSelectedListener(OnItemViewSelectedListener)} */ public void setOnItemSelectedListener(OnItemSelectedListener listener) { mOnItemSelectedListener = listener; } /** * Sets an item clicked listener for the results. * * @param listener The item clicked listener to be invoked when an item in * the search results is clicked. * @deprecated Use {@link #setOnItemViewClickedListener(OnItemViewClickedListener)} */ public void setOnItemClickedListener(OnItemClickedListener listener) { mOnItemClickedListener = listener; } /** * Sets an item selection listener for the results. * * @param listener The item selection listener to be invoked when an item in * the search results is selected. */ public void setOnItemViewSelectedListener(OnItemViewSelectedListener listener) { mOnItemViewSelectedListener = listener; } /** * Sets an item clicked listener for the results. * * @param listener The item clicked listener to be invoked when an item in * the search results is clicked. */ public void setOnItemViewClickedListener(OnItemViewClickedListener listener) { mOnItemViewClickedListener = listener; } /** * Sets the title string to be be shown in an empty search bar. The title * may be placed in a call-to-action, such as "Search title" or * "Speak to search title". */ public void setTitle(String title) { mTitle = title; if (null != mSearchBar) { mSearchBar.setTitle(title); } } /** * Returns the title set in the search bar. */ public String getTitle() { if (null != mSearchBar) { return mSearchBar.getTitle(); } return null; } /** * Sets the badge drawable that will be shown inside the search bar next to * the title. */ public void setBadgeDrawable(Drawable drawable) { mBadgeDrawable = drawable; if (null != mSearchBar) { mSearchBar.setBadgeDrawable(drawable); } } /** * Returns the badge drawable in the search bar. */ public Drawable getBadgeDrawable() { if (null != mSearchBar) { return mSearchBar.getBadgeDrawable(); } return null; } /** * Display the completions shown by the IME. An application may provide * a list of query completions that the system will show in the IME. * * @param completions A list of completions to show in the IME. Setting to * null or empty will clear the list. */ public void displayCompletions(List completions) { mSearchBar.displayCompletions(completions); } /** * Set this callback to have the fragment pass speech recognition requests * to the activity rather than using an internal recognizer. */ public void setSpeechRecognitionCallback(SpeechRecognitionCallback callback) { mSpeechRecognitionCallback = callback; if (mSearchBar != null) { mSearchBar.setSpeechRecognitionCallback(mSpeechRecognitionCallback); } if (callback != null) { releaseRecognizer(); } } /** * Sets the text of the search query and optionally submits the query. Either * {@link SearchResultProvider#onQueryTextChange onQueryTextChange} or * {@link SearchResultProvider#onQueryTextSubmit onQueryTextSubmit} will be * called on the provider if it is set. * * @param query The search query to set. * @param submit Whether to submit the query. */ public void setSearchQuery(String query, boolean submit) { if (DEBUG) Log.v(TAG, "setSearchQuery " + query + " submit " + submit); if (query == null) { return; } mExternalQuery = new ExternalQuery(query, submit); applyExternalQuery(); if (mAutoStartRecognition) { mAutoStartRecognition = false; mHandler.removeCallbacks(mStartRecognitionRunnable); } } /** * Sets the text of the search query based on the {@link RecognizerIntent#EXTRA_RESULTS} in * the given intent, and optionally submit the query. If more than one result is present * in the results list, the first will be used. * * @param intent Intent received from a speech recognition service. * @param submit Whether to submit the query. */ public void setSearchQuery(Intent intent, boolean submit) { ArrayList matches = intent.getStringArrayListExtra(RecognizerIntent.EXTRA_RESULTS); if (matches != null && matches.size() > 0) { setSearchQuery(matches.get(0), submit); } } /** * Returns an intent that can be used to request speech recognition. * Built from the base {@link RecognizerIntent#ACTION_RECOGNIZE_SPEECH} plus * extras: * *

* * For handling the intent returned from the service, see * {@link #setSearchQuery(Intent, boolean)}. */ public Intent getRecognizerIntent() { Intent recognizerIntent = new Intent(RecognizerIntent.ACTION_RECOGNIZE_SPEECH); recognizerIntent.putExtra(RecognizerIntent.EXTRA_LANGUAGE_MODEL, RecognizerIntent.LANGUAGE_MODEL_FREE_FORM); recognizerIntent.putExtra(RecognizerIntent.EXTRA_PARTIAL_RESULTS, true); if (mSearchBar != null && mSearchBar.getHint() != null) { recognizerIntent.putExtra(RecognizerIntent.EXTRA_PROMPT, mSearchBar.getHint()); } recognizerIntent.putExtra(EXTRA_LEANBACK_BADGE_PRESENT, mBadgeDrawable != null); return recognizerIntent; } private void retrieveResults(String searchQuery) { if (DEBUG) Log.v(TAG, "retrieveResults " + searchQuery); if (mProvider.onQueryTextChange(searchQuery)) { mStatus &= ~QUERY_COMPLETE; } } private void submitQuery(String query) { queryComplete(); if (null != mProvider) { mProvider.onQueryTextSubmit(query); } } private void queryComplete() { if (DEBUG) Log.v(TAG, "queryComplete"); mStatus |= QUERY_COMPLETE; focusOnResults(); } private void updateSearchBarNextFocusId() { if (mSearchBar == null || mResultAdapter == null) { return; } final int viewId = (mResultAdapter.size() == 0 || mRowsFragment == null || mRowsFragment.getVerticalGridView() == null) ? 0 : mRowsFragment.getVerticalGridView().getId(); mSearchBar.setNextFocusDownId(viewId); } private void updateFocus() { if (mResultAdapter != null && mResultAdapter.size() > 0 && mRowsFragment != null && mRowsFragment.getAdapter() == mResultAdapter) { focusOnResults(); } else { mSearchBar.requestFocus(); } } private void focusOnResults() { if (mRowsFragment == null || mRowsFragment.getVerticalGridView() == null || mResultAdapter.size() == 0) { return; } if (mRowsFragment.getVerticalGridView().requestFocus()) { mStatus &= ~RESULTS_CHANGED; } } private void onSetSearchResultProvider() { mHandler.removeCallbacks(mSetSearchResultProvider); mHandler.post(mSetSearchResultProvider); } private void releaseAdapter() { if (mResultAdapter != null) { mResultAdapter.unregisterObserver(mAdapterObserver); mResultAdapter = null; } } private void executePendingQuery() { if (null != mPendingQuery && null != mResultAdapter) { String query = mPendingQuery; mPendingQuery = null; retrieveResults(query); } } private void applyExternalQuery() { if (mExternalQuery == null || mSearchBar == null) { return; } mSearchBar.setSearchQuery(mExternalQuery.mQuery); if (mExternalQuery.mSubmit) { submitQuery(mExternalQuery.mQuery); } mExternalQuery = null; } private void readArguments(Bundle args) { if (null == args) { return; } if (args.containsKey(ARG_QUERY)) { setSearchQuery(args.getString(ARG_QUERY)); } if (args.containsKey(ARG_TITLE)) { setTitle(args.getString(ARG_TITLE)); } } private void setSearchQuery(String query) { mSearchBar.setSearchQuery(query); } static class ExternalQuery { String mQuery; boolean mSubmit; ExternalQuery(String query, boolean submit) { mQuery = query; mSubmit = submit; } } }