/*
* Created by Angel Leon (@gubatron), Alden Torres (aldenml)
* Copyright (c) 2011, 2012, FrostWire(TM). All rights reserved.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package com.bt.download.android.core.providers;
import java.util.HashMap;
import java.util.logging.Logger;
import android.content.ContentProvider;
import android.content.ContentUris;
import android.content.ContentValues;
import android.content.Context;
import android.content.UriMatcher;
import android.content.res.Resources;
import android.database.Cursor;
import android.database.SQLException;
import android.database.sqlite.SQLiteDatabase;
import android.database.sqlite.SQLiteOpenHelper;
import android.database.sqlite.SQLiteQueryBuilder;
import android.net.Uri;
import android.os.Environment;
import android.text.TextUtils;
import com.bt.download.android.core.providers.UniversalStore.Documents;
import com.bt.download.android.core.providers.UniversalStore.Documents.DocumentsColumns;
import com.bt.download.android.core.providers.UniversalStore.Documents.Media;
/**
* @author gubatron
* @author aldenml
*
*/
public class DocumentsProvider extends ContentProvider {
private static final Logger LOG = Logger.getLogger(DocumentsProvider.class.getName());
private static final String DATABASE_NAME = "documents.db";
private static final int DATABASE_VERSION = 1;
private static final String DOCUMENTS_TABLE_NAME = "documents";
private static final int DOCUMENTS_ALL = 1;
private static final int DOCUMENTS_ID = 2;
private static final UriMatcher uriMatcher;
private static HashMap<String, String> projectionMap;
private DatabaseHelper databaseHelper;
static {
uriMatcher = new UriMatcher(UriMatcher.NO_MATCH);
uriMatcher.addURI(UniversalStore.UNIVERSAL_DOCUMENTS_AUTHORITY, "documents", DOCUMENTS_ALL);
uriMatcher.addURI(UniversalStore.UNIVERSAL_DOCUMENTS_AUTHORITY, "documents/#", DOCUMENTS_ID);
projectionMap = new HashMap<String, String>();
projectionMap.put(DocumentsColumns._ID, DocumentsColumns._ID);
projectionMap.put(DocumentsColumns.DATA, DocumentsColumns.DATA);
projectionMap.put(DocumentsColumns.SIZE, DocumentsColumns.SIZE);
projectionMap.put(DocumentsColumns.DISPLAY_NAME, DocumentsColumns.DISPLAY_NAME);
projectionMap.put(DocumentsColumns.TITLE, DocumentsColumns.TITLE);
projectionMap.put(DocumentsColumns.DATE_ADDED, DocumentsColumns.DATE_ADDED);
projectionMap.put(DocumentsColumns.DATE_MODIFIED, DocumentsColumns.DATE_MODIFIED);
projectionMap.put(DocumentsColumns.MIME_TYPE, DocumentsColumns.MIME_TYPE);
}
@Override
public boolean onCreate() {
databaseHelper = new DatabaseHelper(getContext());
return true;
}
@Override
public Cursor query(Uri uri, String[] projection, String selection, String[] selectionArgs, String sortOrder) {
if (!accept()) {
return null;
}
SQLiteQueryBuilder qb = new SQLiteQueryBuilder();
qb.setTables(DOCUMENTS_TABLE_NAME);
switch (uriMatcher.match(uri)) {
case DOCUMENTS_ALL:
qb.setProjectionMap(projectionMap);
break;
case DOCUMENTS_ID:
qb.setProjectionMap(projectionMap);
qb.appendWhere(DocumentsColumns._ID + "=" + uri.getPathSegments().get(1));
break;
default:
throw new IllegalArgumentException("Unknown URI " + uri);
}
// If no sort order is specified use the default
String orderBy;
if (TextUtils.isEmpty(sortOrder)) {
orderBy = Documents.DEFAULT_SORT_ORDER;
} else {
orderBy = sortOrder;
}
// Get the database and run the query
SQLiteDatabase db = databaseHelper.getReadableDatabase();
Cursor c = qb.query(db, projection, selection, selectionArgs, null, null, orderBy);
// Tell the cursor what uri to watch, so it knows when its source data
// changes
c.setNotificationUri(getContext().getContentResolver(), uri);
return c;
}
@Override
public String getType(Uri uri) {
switch (uriMatcher.match(uri)) {
case DOCUMENTS_ALL:
return Media.CONTENT_TYPE;
case DOCUMENTS_ID:
return Media.CONTENT_TYPE_ITEM;
default:
throw new IllegalArgumentException("Unknown URI " + uri);
}
}
@Override
public Uri insert(Uri uri, ContentValues initialValues) {
if (!accept()) {
return null;
}
// Validate the requested uri
if (uriMatcher.match(uri) != DOCUMENTS_ALL) {
throw new IllegalArgumentException("Unknown URI " + uri);
}
ContentValues values;
if (initialValues != null) {
values = new ContentValues(initialValues);
} else {
values = new ContentValues();
}
Long now = Long.valueOf(System.currentTimeMillis()/1000);
// Make sure that the fields are all set
if (values.containsKey(DocumentsColumns.DATA) == false) {
values.put(DocumentsColumns.DATA, "");
}
if (values.containsKey(DocumentsColumns.SIZE) == false) {
values.put(DocumentsColumns.SIZE, 0);
}
if (values.containsKey(DocumentsColumns.DISPLAY_NAME) == false) {
values.put(DocumentsColumns.DISPLAY_NAME, "");
}
if (values.containsKey(DocumentsColumns.TITLE) == false) {
Resources r = Resources.getSystem();
values.put(DocumentsColumns.TITLE, r.getString(android.R.string.untitled));
}
if (values.containsKey(DocumentsColumns.DATE_ADDED) == false) {
values.put(DocumentsColumns.DATE_ADDED, now);
}
if (values.containsKey(DocumentsColumns.DATE_MODIFIED) == false) {
values.put(DocumentsColumns.DATE_MODIFIED, now);
}
if (values.containsKey(DocumentsColumns.MIME_TYPE) == false) {
values.put(DocumentsColumns.MIME_TYPE, "");
}
SQLiteDatabase db = databaseHelper.getWritableDatabase();
long rowId = db.insert(DOCUMENTS_TABLE_NAME, "", values);
if (rowId > 0) {
Uri documentUri = ContentUris.withAppendedId(Documents.Media.CONTENT_URI, rowId);
getContext().getContentResolver().notifyChange(documentUri, null);
return documentUri;
}
throw new SQLException("Failed to insert row into " + uri);
}
@Override
public int delete(Uri uri, String where, String[] whereArgs) {
if (!accept()) {
return 0;
}
SQLiteDatabase db = databaseHelper.getWritableDatabase();
int count;
switch (uriMatcher.match(uri)) {
case DOCUMENTS_ALL:
count = db.delete(DOCUMENTS_TABLE_NAME, where, whereArgs);
break;
case DOCUMENTS_ID:
String documentId = uri.getPathSegments().get(1);
count = db.delete(DOCUMENTS_TABLE_NAME, DocumentsColumns._ID + "=" + documentId + (!TextUtils.isEmpty(where) ? " AND (" + where + ")" : ""), whereArgs);
break;
default:
throw new IllegalArgumentException("Unknown URI " + uri);
}
getContext().getContentResolver().notifyChange(uri, null);
return count;
}
@Override
public int update(Uri uri, ContentValues values, String where, String[] whereArgs) {
if (!accept()) {
return 0;
}
SQLiteDatabase db = databaseHelper.getWritableDatabase();
int count;
switch (uriMatcher.match(uri)) {
case DOCUMENTS_ALL:
count = db.update(DOCUMENTS_TABLE_NAME, values, where, whereArgs);
break;
case DOCUMENTS_ID:
String documentId = uri.getPathSegments().get(1);
count = db.update(DOCUMENTS_TABLE_NAME, values, DocumentsColumns._ID + "=" + documentId + (!TextUtils.isEmpty(where) ? " AND (" + where + ")" : ""), whereArgs);
break;
default:
throw new IllegalArgumentException("Unknown URI " + uri);
}
getContext().getContentResolver().notifyChange(uri, null);
return count;
}
private boolean accept() {
return Environment.MEDIA_MOUNTED.equals(Environment.getExternalStorageState());
}
/**
* This class helps open, create, and upgrade the database file.
*/
private static class DatabaseHelper extends SQLiteOpenHelper {
DatabaseHelper(Context context) {
super(context, DATABASE_NAME, null, DATABASE_VERSION);
}
@Override
public void onCreate(SQLiteDatabase db) {
db.execSQL("CREATE TABLE " + DOCUMENTS_TABLE_NAME + " (" + DocumentsColumns._ID + " INTEGER PRIMARY KEY AUTOINCREMENT," + DocumentsColumns.DATA + " TEXT," + DocumentsColumns.SIZE + " INTEGER," + DocumentsColumns.DISPLAY_NAME + " TEXT," + DocumentsColumns.TITLE + " TEXT,"
+ DocumentsColumns.DATE_ADDED + " INTEGER," + DocumentsColumns.DATE_MODIFIED + " INTEGER," + DocumentsColumns.MIME_TYPE + " TEXT" + ");");
}
@Override
public void onUpgrade(SQLiteDatabase db, int oldVersion, int newVersion) {
LOG.warning("Upgrading documents database from version " + oldVersion + " to " + newVersion + ", which will destroy all old data");
db.execSQL("DROP TABLE IF EXISTS " + DOCUMENTS_TABLE_NAME);
onCreate(db);
}
}
}