package org.codeaurora.gallery3d.video;

import android.content.ContentResolver;
import android.content.ContentUris;
import android.content.ContentValues;
import android.content.Context;
import android.database.Cursor;
import android.net.Uri;
import android.util.Log;

import com.android.gallery3d.R;

public class BookmarkEnhance {
    private static final String TAG = "BookmarkEnhance";
    private static final boolean LOG = true;

    private static final Uri BOOKMARK_URI = Uri.parse("content://media/internal/bookmark");

    public static final String COLUMN_ID = "_id";
    public static final String COLUMN_DATA = "_data";
    public static final String COLUMN_TITLE = "_display_name";
    public static final String COLUMN_ADD_DATE = "date_added";
    public static final String COLUMN_MEDIA_TYPE = "mime_type";
    private static final String COLUMN_POSITION = "position";
    private static final String COLUMN_MIME_TYPE = "media_type";

    private static final String NULL_HOCK = COLUMN_POSITION;
    public static final String ORDER_COLUMN = COLUMN_ADD_DATE + " ASC ";
    private static final String VIDEO_STREAMING_MEDIA_TYPE = "streaming";

    public static final int INDEX_ID = 0;
    public static final int INDEX_DATA = 1;
    public static final int INDEX_TITLE = 2;
    public static final int INDEX_ADD_DATE = 3;
    public static final int INDEX_MIME_TYPE = 4;
    private static final int INDEX_POSITION = 5;
    private static final int INDEX_MEDIA_TYPE = 6;

    public static final String[] PROJECTION = new String[] {
            COLUMN_ID,
            COLUMN_DATA,
            COLUMN_TITLE,
            COLUMN_ADD_DATE,
            COLUMN_MIME_TYPE,
    };

    private final Context mContext;
    private final ContentResolver mCr;

    public BookmarkEnhance(final Context context) {
        mContext = context;
        mCr = context.getContentResolver();
    }

    public Uri insert(final String title, final String uri, final String mimeType,
            final long position) {
        final ContentValues values = new ContentValues();
        final String mytitle = (title == null ? mContext.getString(R.string.default_title) : title);
        values.put(COLUMN_TITLE, mytitle);
        values.put(COLUMN_DATA, uri);
        values.put(COLUMN_POSITION, position);
        values.put(COLUMN_ADD_DATE, System.currentTimeMillis());
        values.put(COLUMN_MEDIA_TYPE, VIDEO_STREAMING_MEDIA_TYPE);
        values.put(COLUMN_MIME_TYPE, mimeType);
        final Uri insertUri = mCr.insert(BOOKMARK_URI, values);
        if (LOG) {
            Log.v(TAG, "insert(" + title + "," + uri + ", " + position + ") return "
                    + insertUri);
        }
        return insertUri;
    }

    public int delete(final long id) {
        final Uri uri = ContentUris.withAppendedId(BOOKMARK_URI, id);
        final int count = mCr.delete(uri, null, null);
        if (LOG) {
            Log.v(TAG, "delete(" + id + ") return " + count);
        }
        return count;
    }

    public int deleteAll() {
        final int count = mCr.delete(BOOKMARK_URI, COLUMN_MEDIA_TYPE + "=? ", new String[] {
            VIDEO_STREAMING_MEDIA_TYPE
        });
        if (LOG) {
            Log.v(TAG, "deleteAll() return " + count);
        }
        return count;
    }

    public boolean exists(final String uri) {
        final Cursor cursor = mCr.query(BOOKMARK_URI,
                PROJECTION,
                COLUMN_DATA + "=? and " + COLUMN_MEDIA_TYPE + "=? ",
                new String[] {
                        uri, VIDEO_STREAMING_MEDIA_TYPE
                },
                null
                );
        boolean exist = false;
        if (cursor != null) {
            exist = cursor.moveToFirst();
            cursor.close();
        }
        if (LOG) {
            Log.v(TAG, "exists(" + uri + ") return " + exist);
        }
        return exist;
    }

    public Cursor query() {
        final Cursor cursor = mCr.query(BOOKMARK_URI,
                PROJECTION,
                COLUMN_MEDIA_TYPE + "='" + VIDEO_STREAMING_MEDIA_TYPE + "' ",
                null,
                ORDER_COLUMN
                );
        if (LOG) {
            Log.v(TAG, "query() return cursor=" + (cursor == null ? -1 : cursor.getCount()));
        }
        return cursor;
    }

    public int update(final long id, final String title, final String uri, final int position) {
        final ContentValues values = new ContentValues();
        values.put(COLUMN_TITLE, title);
        values.put(COLUMN_DATA, uri);
        values.put(COLUMN_POSITION, position);
        final Uri updateUri = ContentUris.withAppendedId(BOOKMARK_URI, id);
        final int count = mCr.update(updateUri, values, null, null);
        if (LOG) {
            Log.v(TAG, "update(" + id + ", " + title + ", " + uri + ", " + position + ")" +
                    " return " + count);
        }
        return count;
    }
}