TvShowContentProvider.java

/*
 * Copyright (C) 2014 Michell Bak
 *
 * 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.miz.contentprovider;

import android.app.SearchManager;
import android.content.ContentValues;
import android.content.SearchRecentSuggestionsProvider;
import android.database.Cursor;
import android.database.MatrixCursor;
import android.net.Uri;
import android.provider.BaseColumns;
import android.text.TextUtils;
import android.util.Log;

import com.miz.db.DbAdapterTvShows;
import com.miz.functions.ColumnIndexCache;
import com.miz.functions.MizLib;
import com.miz.mizuu.MizuuApplication;
import com.miz.mizuu.TvShow;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Locale;
import java.util.regex.Pattern;

public class TvShowContentProvider extends SearchRecentSuggestionsProvider {

	static final String TAG = TvShowContentProvider.class.getSimpleName();
	public static final String AUTHORITY = TvShowContentProvider.class.getName();
	public static final int MODE = DATABASE_MODE_QUERIES | DATABASE_MODE_2LINES;
	private static final String[] COLUMNS = {
		BaseColumns._ID, // must include this column
		SearchManager.SUGGEST_COLUMN_TEXT_1, // First line (title)
		SearchManager.SUGGEST_COLUMN_TEXT_2, // Second line (smaller text)
		SearchManager.SUGGEST_COLUMN_INTENT_DATA, // Icon
		SearchManager.SUGGEST_COLUMN_ICON_1, // Icon
		SearchManager.SUGGEST_COLUMN_INTENT_EXTRA_DATA, // TV show ID
		SearchManager.SUGGEST_COLUMN_INTENT_ACTION,
		SearchManager.SUGGEST_COLUMN_SHORTCUT_ID };

	public TvShowContentProvider() {
		setupSuggestions(AUTHORITY, MODE);
	}

	@Override
	public Cursor query(Uri uri, String[] projection, String selection,
			String[] selectionArgs, String sortOrder) {

		String query = selectionArgs[0];
		if (query == null || query.length() == 0) {
			return null;
		}

		MatrixCursor cursor = new MatrixCursor(COLUMNS);

		try {
			List<TvShow> list = getSearchResults(query);
			for (int i = 0; i < list.size(); i++) {
				cursor.addRow(createRow(i, list.get(i).getTitle(), list.get(i).getFirstAirdateYear(), "'" + Uri.fromFile(list.get(i).getThumbnail()) + "' AS " + SearchManager.SUGGEST_COLUMN_ICON_1, list.get(i).getId()));
			}
		} catch (Exception e) {
			Log.e(TAG, "Failed to lookup " + query, e);
		}
		return cursor;
	}

	@Override
	public Uri insert(Uri uri, ContentValues values) {
		throw new UnsupportedOperationException();
	}

	@Override
	public int delete(Uri uri, String selection, String[] selectionArgs) {
		throw new UnsupportedOperationException();
	}

	@Override
	public int update(Uri uri, ContentValues values, String selection,
			String[] selectionArgs) {
		throw new UnsupportedOperationException();
	}

	private Object[] createRow(Integer id, String text1, String text2, String icon, String rowId) {
		return new Object[] {
				id, // _id
				text1, // text1
				text2, // text2
				icon,
				icon,
				rowId,
				"android.intent.action.SEARCH", // action
				SearchManager.SUGGEST_NEVER_MAKE_SHORTCUT };
	}

	private List<TvShow> getSearchResults(String query) {
		List<TvShow> shows = new ArrayList<TvShow>();
		if (!TextUtils.isEmpty(query)) {

			DbAdapterTvShows db = MizuuApplication.getTvDbAdapter();

			query = query.toLowerCase(Locale.ENGLISH);

			Cursor c = db.getAllShows();
			Pattern p = Pattern.compile(MizLib.CHARACTER_REGEX); // Use a pre-compiled pattern as it's a lot faster (approx. 3x for ~700 movies)
			ColumnIndexCache cache = new ColumnIndexCache();

			try {
				while (c.moveToNext()) {
					String title = c.getString(cache.getColumnIndex(c, DbAdapterTvShows.KEY_SHOW_TITLE)).toLowerCase(Locale.ENGLISH);

					if (title.indexOf(query) != -1 ||  p.matcher(title).replaceAll("").indexOf(query) != -1) {
						shows.add(new TvShow(
								getContext(),
								c.getString(cache.getColumnIndex(c, DbAdapterTvShows.KEY_SHOW_ID)),
								c.getString(cache.getColumnIndex(c, DbAdapterTvShows.KEY_SHOW_TITLE)),
								c.getString(cache.getColumnIndex(c, DbAdapterTvShows.KEY_SHOW_PLOT)),
								c.getString(cache.getColumnIndex(c, DbAdapterTvShows.KEY_SHOW_RATING)),
								c.getString(cache.getColumnIndex(c, DbAdapterTvShows.KEY_SHOW_GENRES)),
								c.getString(cache.getColumnIndex(c, DbAdapterTvShows.KEY_SHOW_ACTORS)),
								c.getString(cache.getColumnIndex(c, DbAdapterTvShows.KEY_SHOW_CERTIFICATION)),
								c.getString(cache.getColumnIndex(c, DbAdapterTvShows.KEY_SHOW_FIRST_AIRDATE)),
								c.getString(cache.getColumnIndex(c, DbAdapterTvShows.KEY_SHOW_RUNTIME)),
								c.getString(cache.getColumnIndex(c, DbAdapterTvShows.KEY_SHOW_FAVOURITE)),
								MizuuApplication.getTvEpisodeDbAdapter().getLatestEpisodeAirdate(c.getString(cache.getColumnIndex(c, DbAdapterTvShows.KEY_SHOW_ID)))
								));
					}
				}
			} catch (Exception e) {
			} finally {
				c.close();
				cache.clear();
			}
			Collections.sort(shows);
		}
		return shows;
	}
}