xref: /packages/apps/QuickSearchBox/tests/partial/src/com/android/quicksearchbox/tests/partial/PartialSuggestionProvider.java

/*
 * Copyright (C) 2010 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.tests.partial;

import android.app.SearchManager;
import android.content.ContentProvider;
import android.content.ContentValues;
import android.content.Intent;
import android.database.Cursor;
import android.database.MatrixCursor;
import android.net.Uri;
import android.os.Bundle;
import android.os.Handler;
import android.os.Message;
import android.util.Log;

public class PartialSuggestionProvider extends ContentProvider {

    private static final String TAG = PartialSuggestionProvider.class.getSimpleName();

    private static final int MSG_COMPLETE = 1;

    private MutableMatrixCursor mCursor;
    private int mType = -1;

    private static final String[] COLUMNS = {
        "_id",
        SearchManager.SUGGEST_COLUMN_TEXT_1,
        SearchManager.SUGGEST_COLUMN_TEXT_2,
        SearchManager.SUGGEST_COLUMN_INTENT_ACTION,
        SearchManager.SUGGEST_COLUMN_INTENT_DATA,
    };

    Handler mHandler = new Handler() {
        @Override
        public void handleMessage(Message msg) {
            switch (msg.what) {
            case MSG_COMPLETE:
                mCursor.getExtras().putBoolean(SearchManager.CURSOR_EXTRA_KEY_IN_PROGRESS, false);
                addRows(mCursor, false);
                mCursor.notifyChange();
            }

        }
    };

    private static class MutableMatrixCursor extends MatrixCursor {

        Bundle mBundle;

        MutableMatrixCursor(String[] columns) {
            super(columns);
        }

        @Override
        public Bundle getExtras() {
            if (mBundle == null) mBundle = new Bundle();
            return mBundle;
        }

        public void notifyChange() {
            onChange(false);
        }
    }

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

    private void addRows(MatrixCursor cursor, boolean partial) {
        for (int i = 0; i < 3; i++) {
            cursor.addRow(new Object[]{
                i,
                (partial? "Partial" : "Final ") + " suggestion " + i,
                "This is a suggestion",
                Intent.ACTION_VIEW,
                "content://com.android.quicksearchbox.partial/partial/" + i
            });
        }
    }

    @Override
    public Cursor query(Uri uri, String[] projectionIn, String selection,
            String[] selectionArgs, String sortOrder) {
        Log.d(TAG, "query(" + uri + ")");
        mType = (mType + 1) % 3;

        if (mType == 0) {
            Log.d(TAG, "returning null cursor");
            return null;
        }

        MutableMatrixCursor cursor = new MutableMatrixCursor(COLUMNS);
        if (mType == 1) {
            addRows(cursor, true);
        } else {
            Log.d(TAG, "returning empty cursor");
        }
        cursor.getExtras().putBoolean(SearchManager.CURSOR_EXTRA_KEY_IN_PROGRESS, true);
        mCursor = cursor;
        // Update the cursor in 2 seconds
        mHandler.removeMessages(MSG_COMPLETE);
        mHandler.sendMessageDelayed(mHandler.obtainMessage(MSG_COMPLETE), 2000);
        return cursor;
    }

    @Override
    public String getType(Uri uri) {
        return SearchManager.SUGGEST_MIME_TYPE;
    }

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

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

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

}