1/*
2 * Copyright (C) 2013 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17package com.android.documentsui;
18
19import static com.android.documentsui.DocumentsActivity.TAG;
20import static com.android.documentsui.model.DocumentInfo.getCursorLong;
21import static com.android.documentsui.model.DocumentInfo.getCursorString;
22
23import android.database.AbstractCursor;
24import android.database.Cursor;
25import android.os.Bundle;
26import android.provider.DocumentsContract.Document;
27import android.util.Log;
28
29/**
30 * Cursor wrapper that filters MIME types not matching given list.
31 */
32public class FilteringCursorWrapper extends AbstractCursor {
33    private final Cursor mCursor;
34
35    private final int[] mPosition;
36    private int mCount;
37
38    public FilteringCursorWrapper(Cursor cursor, String[] acceptMimes) {
39        this(cursor, acceptMimes, null, Long.MIN_VALUE);
40    }
41
42    public FilteringCursorWrapper(Cursor cursor, String[] acceptMimes, String[] rejectMimes) {
43        this(cursor, acceptMimes, rejectMimes, Long.MIN_VALUE);
44    }
45
46    public FilteringCursorWrapper(
47            Cursor cursor, String[] acceptMimes, String[] rejectMimes, long rejectBefore) {
48        mCursor = cursor;
49
50        final int count = cursor.getCount();
51        mPosition = new int[count];
52
53        cursor.moveToPosition(-1);
54        while (cursor.moveToNext() && mCount < count) {
55            final String mimeType = getCursorString(cursor, Document.COLUMN_MIME_TYPE);
56            final long lastModified = getCursorLong(cursor, Document.COLUMN_LAST_MODIFIED);
57            if (rejectMimes != null && MimePredicate.mimeMatches(rejectMimes, mimeType)) {
58                continue;
59            }
60            if (lastModified < rejectBefore) {
61                continue;
62            }
63            if (MimePredicate.mimeMatches(acceptMimes, mimeType)) {
64                mPosition[mCount++] = cursor.getPosition();
65            }
66        }
67
68        Log.d(TAG, "Before filtering " + cursor.getCount() + ", after " + mCount);
69    }
70
71    @Override
72    public Bundle getExtras() {
73        return mCursor.getExtras();
74    }
75
76    @Override
77    public void close() {
78        super.close();
79        mCursor.close();
80    }
81
82    @Override
83    public boolean onMove(int oldPosition, int newPosition) {
84        return mCursor.moveToPosition(mPosition[newPosition]);
85    }
86
87    @Override
88    public String[] getColumnNames() {
89        return mCursor.getColumnNames();
90    }
91
92    @Override
93    public int getCount() {
94        return mCount;
95    }
96
97    @Override
98    public double getDouble(int column) {
99        return mCursor.getDouble(column);
100    }
101
102    @Override
103    public float getFloat(int column) {
104        return mCursor.getFloat(column);
105    }
106
107    @Override
108    public int getInt(int column) {
109        return mCursor.getInt(column);
110    }
111
112    @Override
113    public long getLong(int column) {
114        return mCursor.getLong(column);
115    }
116
117    @Override
118    public short getShort(int column) {
119        return mCursor.getShort(column);
120    }
121
122    @Override
123    public String getString(int column) {
124        return mCursor.getString(column);
125    }
126
127    @Override
128    public int getType(int column) {
129        return mCursor.getType(column);
130    }
131
132    @Override
133    public boolean isNull(int column) {
134        return mCursor.isNull(column);
135    }
136}
137