/*
 * 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.gallery3d.data;

import android.content.ContentProvider;
import android.content.ContentResolver;
import android.database.Cursor;
import android.database.sqlite.SQLiteDatabase;
import android.database.sqlite.SQLiteQueryBuilder;
import android.net.Uri;
import android.os.Looper;
import android.test.AndroidTestCase;
import android.test.mock.MockContentProvider;
import android.test.mock.MockContentResolver;
import android.test.suitebuilder.annotation.MediumTest;
import android.util.Log;

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

public class LocalDataTest extends AndroidTestCase {
    @SuppressWarnings("unused")
    private static final String TAG = "LocalDataTest";
    private static final long DEFAULT_TIMEOUT = 1000; // one second

    @MediumTest
    public void testLocalAlbum() throws Exception {
        new TestZeroImage().run();
        new TestOneImage().run();
        new TestMoreImages().run();
        new TestZeroVideo().run();
        new TestOneVideo().run();
        new TestMoreVideos().run();
        new TestDeleteOneImage().run();
        new TestDeleteOneAlbum().run();
    }

    abstract class TestLocalAlbumBase {
        private boolean mIsImage;
        protected GalleryAppStub mApp;
        protected LocalAlbumSet mAlbumSet;

        TestLocalAlbumBase(boolean isImage) {
            mIsImage = isImage;
        }

        public void run() throws Exception {
            SQLiteDatabase db = SQLiteDatabase.create(null);
            prepareData(db);
            mApp = newGalleryContext(db, Looper.getMainLooper());
            Path.clearAll();
            Path path = Path.fromString(
                    mIsImage ? "/local/image" : "/local/video");
            mAlbumSet = new LocalAlbumSet(path, mApp);
            mAlbumSet.reload();
            verifyResult();
        }

        abstract void prepareData(SQLiteDatabase db);
        abstract void verifyResult() throws Exception;
    }

    abstract class TestLocalImageAlbum extends TestLocalAlbumBase {
        TestLocalImageAlbum() {
            super(true);
        }
    }

    abstract class TestLocalVideoAlbum extends TestLocalAlbumBase {
        TestLocalVideoAlbum() {
            super(false);
        }
    }

    class TestZeroImage extends TestLocalImageAlbum {
        @Override
        public void prepareData(SQLiteDatabase db) {
            createImageTable(db);
        }

        @Override
        public void verifyResult() {
            assertEquals(0, mAlbumSet.getMediaItemCount());
            assertEquals(0, mAlbumSet.getSubMediaSetCount());
            assertEquals(0, mAlbumSet.getTotalMediaItemCount());
         }
    }

    class TestOneImage extends TestLocalImageAlbum {
        @Override
        public void prepareData(SQLiteDatabase db) {
            createImageTable(db);
            insertImageData(db);
        }

        @Override
        public void verifyResult() {
            assertEquals(0, mAlbumSet.getMediaItemCount());
            assertEquals(1, mAlbumSet.getSubMediaSetCount());
            assertEquals(1, mAlbumSet.getTotalMediaItemCount());
            MediaSet sub = mAlbumSet.getSubMediaSet(0);
            assertEquals(1, sub.getMediaItemCount());
            assertEquals(0, sub.getSubMediaSetCount());
            LocalMediaItem item = (LocalMediaItem) sub.getMediaItem(0, 1).get(0);
            assertEquals(1, item.id);
            assertEquals("IMG_0072", item.caption);
            assertEquals("image/jpeg", item.mimeType);
            assertEquals(12.0, item.latitude);
            assertEquals(34.0, item.longitude);
            assertEquals(0xD000, item.dateTakenInMs);
            assertEquals(1280395646L, item.dateAddedInSec);
            assertEquals(1275934796L, item.dateModifiedInSec);
            assertEquals("/mnt/sdcard/DCIM/100CANON/IMG_0072.JPG", item.filePath);
        }
    }

    class TestMoreImages extends TestLocalImageAlbum {
        @Override
        public void prepareData(SQLiteDatabase db) {
            // Albums are sorted by names, and items are sorted by
            // dateTimeTaken (descending)
            createImageTable(db);
            // bucket 0xB000
            insertImageData(db, 1000, 0xB000, "second");  // id 1
            insertImageData(db, 2000, 0xB000, "second");  // id 2
            // bucket 0xB001
            insertImageData(db, 3000, 0xB001, "first");   // id 3
        }

        @Override
        public void verifyResult() {
            assertEquals(0, mAlbumSet.getMediaItemCount());
            assertEquals(2, mAlbumSet.getSubMediaSetCount());
            assertEquals(3, mAlbumSet.getTotalMediaItemCount());

            MediaSet first = mAlbumSet.getSubMediaSet(0);
            assertEquals(1, first.getMediaItemCount());
            LocalMediaItem item = (LocalMediaItem) first.getMediaItem(0, 1).get(0);
            assertEquals(3, item.id);
            assertEquals(3000L, item.dateTakenInMs);

            MediaSet second = mAlbumSet.getSubMediaSet(1);
            assertEquals(2, second.getMediaItemCount());
            item = (LocalMediaItem) second.getMediaItem(0, 1).get(0);
            assertEquals(2, item.id);
            assertEquals(2000L, item.dateTakenInMs);
            item = (LocalMediaItem) second.getMediaItem(1, 1).get(0);
            assertEquals(1, item.id);
            assertEquals(1000L, item.dateTakenInMs);
        }
    }

    class OnContentDirtyLatch implements ContentListener {
        private CountDownLatch mLatch = new CountDownLatch(1);

        public void onContentDirty() {
            mLatch.countDown();
        }

        public boolean isOnContentDirtyBeCalled(long timeout)
                throws InterruptedException {
            return mLatch.await(timeout, TimeUnit.MILLISECONDS);
        }
    }

    class TestDeleteOneAlbum extends TestLocalImageAlbum {
        @Override
        public void prepareData(SQLiteDatabase db) {
            // Albums are sorted by names, and items are sorted by
            // dateTimeTaken (descending)
            createImageTable(db);
            // bucket 0xB000
            insertImageData(db, 1000, 0xB000, "second");  // id 1
            insertImageData(db, 2000, 0xB000, "second");  // id 2
            // bucket 0xB001
            insertImageData(db, 3000, 0xB001, "first");   // id 3
        }

        @Override
        public void verifyResult() throws Exception {
            MediaSet sub = mAlbumSet.getSubMediaSet(1);  // "second"
            assertEquals(2, mAlbumSet.getSubMediaSetCount());
            OnContentDirtyLatch latch = new OnContentDirtyLatch();
            sub.addContentListener(latch);
            assertTrue((sub.getSupportedOperations() & MediaSet.SUPPORT_DELETE) != 0);
            sub.delete();
            mAlbumSet.fakeChange();
            latch.isOnContentDirtyBeCalled(DEFAULT_TIMEOUT);
            mAlbumSet.reload();
            assertEquals(1, mAlbumSet.getSubMediaSetCount());
        }
    }

    class TestDeleteOneImage extends TestLocalImageAlbum {

        @Override
        public void prepareData(SQLiteDatabase db) {
            createImageTable(db);
            insertImageData(db);
        }

        @Override
        public void verifyResult() {
            MediaSet sub = mAlbumSet.getSubMediaSet(0);
            LocalMediaItem item = (LocalMediaItem) sub.getMediaItem(0, 1).get(0);
            assertEquals(1, sub.getMediaItemCount());
            assertTrue((sub.getSupportedOperations() & MediaSet.SUPPORT_DELETE) != 0);
            sub.delete();
            sub.reload();
            assertEquals(0, sub.getMediaItemCount());
        }
    }

    static void createImageTable(SQLiteDatabase db) {
        // This is copied from MediaProvider
        db.execSQL("CREATE TABLE IF NOT EXISTS images (" +
                "_id INTEGER PRIMARY KEY," +
                "_data TEXT," +
                "_size INTEGER," +
                "_display_name TEXT," +
                "mime_type TEXT," +
                "title TEXT," +
                "date_added INTEGER," +
                "date_modified INTEGER," +
                "description TEXT," +
                "picasa_id TEXT," +
                "isprivate INTEGER," +
                "latitude DOUBLE," +
                "longitude DOUBLE," +
                "datetaken INTEGER," +
                "orientation INTEGER," +
                "mini_thumb_magic INTEGER," +
                "bucket_id TEXT," +
                "bucket_display_name TEXT" +
               ");");
    }

    static void insertImageData(SQLiteDatabase db) {
        insertImageData(db, 0xD000, 0xB000, "name");
    }

    static void insertImageData(SQLiteDatabase db, long dateTaken,
            int bucketId, String bucketName) {
        db.execSQL("INSERT INTO images (title, mime_type, latitude, longitude, "
                + "datetaken, date_added, date_modified, bucket_id, "
                + "bucket_display_name, _data, orientation) "
                + "VALUES ('IMG_0072', 'image/jpeg', 12, 34, "
                + dateTaken + ", 1280395646, 1275934796, '" + bucketId + "', "
                + "'" + bucketName + "', "
                + "'/mnt/sdcard/DCIM/100CANON/IMG_0072.JPG', 0)");
    }

    class TestZeroVideo extends TestLocalVideoAlbum {
        @Override
        public void prepareData(SQLiteDatabase db) {
            createVideoTable(db);
        }

        @Override
        public void verifyResult() {
            assertEquals(0, mAlbumSet.getMediaItemCount());
            assertEquals(0, mAlbumSet.getSubMediaSetCount());
            assertEquals(0, mAlbumSet.getTotalMediaItemCount());
        }
    }

    class TestOneVideo extends TestLocalVideoAlbum {
        @Override
        public void prepareData(SQLiteDatabase db) {
            createVideoTable(db);
            insertVideoData(db);
        }

        @Override
        public void verifyResult() {
            assertEquals(0, mAlbumSet.getMediaItemCount());
            assertEquals(1, mAlbumSet.getSubMediaSetCount());
            assertEquals(1, mAlbumSet.getTotalMediaItemCount());
            MediaSet sub = mAlbumSet.getSubMediaSet(0);
            assertEquals(1, sub.getMediaItemCount());
            assertEquals(0, sub.getSubMediaSetCount());
            LocalMediaItem item = (LocalMediaItem) sub.getMediaItem(0, 1).get(0);
            assertEquals(1, item.id);
            assertEquals("VID_20100811_051413", item.caption);
            assertEquals("video/mp4", item.mimeType);
            assertEquals(11.0, item.latitude);
            assertEquals(22.0, item.longitude);
            assertEquals(0xD000, item.dateTakenInMs);
            assertEquals(1281503663L, item.dateAddedInSec);
            assertEquals(1281503662L, item.dateModifiedInSec);
            assertEquals("/mnt/sdcard/DCIM/Camera/VID_20100811_051413.3gp",
                    item.filePath);
        }
    }

    class TestMoreVideos extends TestLocalVideoAlbum {
        @Override
        public void prepareData(SQLiteDatabase db) {
            // Albums are sorted by names, and items are sorted by
            // dateTimeTaken (descending)
            createVideoTable(db);
            // bucket 0xB002
            insertVideoData(db, 1000, 0xB000, "second");  // id 1
            insertVideoData(db, 2000, 0xB000, "second");  // id 2
            // bucket 0xB001
            insertVideoData(db, 3000, 0xB001, "first");   // id 3
        }

        @Override
        public void verifyResult() {
            assertEquals(0, mAlbumSet.getMediaItemCount());
            assertEquals(2, mAlbumSet.getSubMediaSetCount());
            assertEquals(3, mAlbumSet.getTotalMediaItemCount());

            MediaSet first = mAlbumSet.getSubMediaSet(0);
            assertEquals(1, first.getMediaItemCount());
            LocalMediaItem item = (LocalMediaItem) first.getMediaItem(0, 1).get(0);
            assertEquals(3, item.id);
            assertEquals(3000L, item.dateTakenInMs);

            MediaSet second = mAlbumSet.getSubMediaSet(1);
            assertEquals(2, second.getMediaItemCount());
            item = (LocalMediaItem) second.getMediaItem(0, 1).get(0);
            assertEquals(2, item.id);
            assertEquals(2000L, item.dateTakenInMs);
            item = (LocalMediaItem) second.getMediaItem(1, 1).get(0);
            assertEquals(1, item.id);
            assertEquals(1000L, item.dateTakenInMs);
        }
    }

    static void createVideoTable(SQLiteDatabase db) {
        db.execSQL("CREATE TABLE IF NOT EXISTS video (" +
                   "_id INTEGER PRIMARY KEY," +
                   "_data TEXT NOT NULL," +
                   "_display_name TEXT," +
                   "_size INTEGER," +
                   "mime_type TEXT," +
                   "date_added INTEGER," +
                   "date_modified INTEGER," +
                   "title TEXT," +
                   "duration INTEGER," +
                   "artist TEXT," +
                   "album TEXT," +
                   "resolution TEXT," +
                   "description TEXT," +
                   "isprivate INTEGER," +   // for YouTube videos
                   "tags TEXT," +           // for YouTube videos
                   "category TEXT," +       // for YouTube videos
                   "language TEXT," +       // for YouTube videos
                   "mini_thumb_data TEXT," +
                   "latitude DOUBLE," +
                   "longitude DOUBLE," +
                   "datetaken INTEGER," +
                   "mini_thumb_magic INTEGER" +
                   ");");
        db.execSQL("ALTER TABLE video ADD COLUMN bucket_id TEXT;");
        db.execSQL("ALTER TABLE video ADD COLUMN bucket_display_name TEXT");
    }

    static void insertVideoData(SQLiteDatabase db) {
        insertVideoData(db, 0xD000, 0xB000, "name");
    }

    static void insertVideoData(SQLiteDatabase db, long dateTaken,
            int bucketId, String bucketName) {
        db.execSQL("INSERT INTO video (title, mime_type, latitude, longitude, "
                + "datetaken, date_added, date_modified, bucket_id, "
                + "bucket_display_name, _data, duration) "
                + "VALUES ('VID_20100811_051413', 'video/mp4', 11, 22, "
                + dateTaken + ", 1281503663, 1281503662, '" + bucketId + "', "
                + "'" + bucketName + "', "
                + "'/mnt/sdcard/DCIM/Camera/VID_20100811_051413.3gp', 2964)");
    }

    static GalleryAppStub newGalleryContext(SQLiteDatabase db, Looper mainLooper) {
        MockContentResolver cr = new MockContentResolver();
        ContentProvider cp = new DbContentProvider(db, cr);
        cr.addProvider("media", cp);
        return new GalleryAppMock(null, cr, mainLooper);
    }
}

class DbContentProvider extends MockContentProvider {
    private static final String TAG = "DbContentProvider";
    private SQLiteDatabase mDatabase;
    private ContentResolver mContentResolver;

    DbContentProvider(SQLiteDatabase db, ContentResolver cr) {
        mDatabase = db;
        mContentResolver = cr;
    }

    @Override
    public Cursor query(Uri uri, String[] projection,
            String selection, String[] selectionArgs, String sortOrder) {
        // This is a simplified version extracted from MediaProvider.

        String tableName = getTableName(uri);
        if (tableName == null) return null;

        SQLiteQueryBuilder qb = new SQLiteQueryBuilder();
        qb.setTables(tableName);

        String groupBy = null;
        String limit = uri.getQueryParameter("limit");

        if (uri.getQueryParameter("distinct") != null) {
            qb.setDistinct(true);
        }

        Log.v(TAG, "query = " + qb.buildQuery(projection, selection,
                selectionArgs, groupBy, null, sortOrder, limit));

        if (selectionArgs != null) {
            for (String s : selectionArgs) {
                Log.v(TAG, "  selectionArgs = " + s);
            }
        }

        Cursor c = qb.query(mDatabase, projection, selection,
                selectionArgs, groupBy, null, sortOrder, limit);

        return c;
    }

    @Override
    public int delete(Uri uri, String whereClause, String[] whereArgs) {
        Log.v(TAG, "delete " + uri + "," + whereClause + "," + whereArgs[0]);
        String tableName = getTableName(uri);
        if (tableName == null) return 0;
        int count = mDatabase.delete(tableName, whereClause, whereArgs);
        mContentResolver.notifyChange(uri, null);
        return count;
    }

    private String getTableName(Uri uri) {
        String uriString = uri.toString();
        if (uriString.startsWith("content://media/external/images/media")) {
            return "images";
        } else if (uriString.startsWith("content://media/external/video/media")) {
            return "video";
        } else {
            return null;
        }
    }
}