DbAdapterMovies.java

/*
 * Copyright (C) 2014 Michell Bak
 *
 * 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.miz.db;

import android.content.ContentValues;
import android.content.Context;
import android.database.Cursor;
import android.database.SQLException;
import android.database.sqlite.SQLiteException;
import android.text.TextUtils;

import com.miz.mizuu.MizuuApplication;

import java.util.ArrayList;

public class DbAdapterMovies extends AbstractDbAdapter {

    // Database fields
    public static final String KEY_TMDB_ID = "tmdbid"; // Unidentified movies and .nfo files without TMDb ID use the filepath
    public static final String KEY_TITLE = "title";
    public static final String KEY_PLOT = "plot";
    public static final String KEY_IMDB_ID = "imdbid";
    public static final String KEY_RATING = "rating";
    public static final String KEY_TAGLINE = "tagline";
    public static final String KEY_RELEASEDATE = "release";
    public static final String KEY_CERTIFICATION = "certification";
    public static final String KEY_RUNTIME = "runtime";
    public static final String KEY_TRAILER = "trailer";
    public static final String KEY_GENRES = "genres";
    public static final String KEY_FAVOURITE = "favourite";
    public static final String KEY_ACTORS = "actors";
    public static final String KEY_COLLECTION_ID = "collection_id";
    public static final String KEY_TO_WATCH = "to_watch";
    public static final String KEY_HAS_WATCHED = "has_watched";
    public static final String KEY_DATE_ADDED = "date_added";

    public static final String DATABASE_TABLE = "movie";

    public static final String UNIDENTIFIED_ID = "invalid";

    public static final String[] SELECT_ALL = new String[] {KEY_TMDB_ID, KEY_TITLE, KEY_PLOT, KEY_IMDB_ID, KEY_RATING, KEY_TAGLINE, KEY_RELEASEDATE, KEY_CERTIFICATION,
            KEY_RUNTIME, KEY_TRAILER, KEY_GENRES, KEY_FAVOURITE, KEY_ACTORS, KEY_COLLECTION_ID, KEY_TO_WATCH, KEY_HAS_WATCHED, KEY_DATE_ADDED};

    public DbAdapterMovies(Context context) {
        super(context);
    }

    /**
     * Create a new movie
     */
    public void createMovie(String tmdbid, String title, String plot, String imdbid, String rating, String tagline, String release,
                            String certification, String runtime, String trailer, String genres, String favourite, String actors, String collection,
                            String collectionId, String toWatch, String hasWatched, String date) {

        if (tmdbid.equals(UNIDENTIFIED_ID))
            return; // We're not interesting in adding this to the movie database

        if (movieExists(tmdbid))
            return; // It already exists, let's not create it again

        ContentValues initialValues = createContentValues(tmdbid, title, plot, imdbid, rating, tagline, release, certification, runtime, trailer, genres, favourite, actors, collectionId, toWatch, hasWatched, date, true);
        mDatabase.insert(DATABASE_TABLE, null, initialValues);

        MizuuApplication.getCollectionsAdapter().createCollection(tmdbid, collectionId, collection);
    }

    public void createOrUpdateMovie(String tmdbid, String title, String plot, String imdbid, String rating, String tagline, String release,
                                    String certification, String runtime, String trailer, String genres, String favourite, String actors, String collection,
                                    String collectionId, String toWatch, String hasWatched, String date) {
        if (movieExists(tmdbid)) {
            // It already exists - let's update it while keeping
            // favourite, watchlist and watched status data intact
            updateMovie(tmdbid, title, plot, imdbid, rating, tagline, release, certification, runtime, trailer, genres,
                    actors, collection, collectionId, date);
        } else {
            // It doesn't exist! Let's create it
            createMovie(tmdbid, title, plot, imdbid, rating, tagline, release, certification, runtime, trailer, genres,
                    favourite, actors, collection, collectionId, toWatch, hasWatched, date);
        }
    }

    /**
     * Update the movie
     */
    private boolean updateMovie(String tmdbid, String title, String plot, String imdbid, String rating, String tagline, String release,
                                String certification, String runtime, String trailer, String genres, String actors, String collection, String collectionId, String date) {

        ContentValues updateValues = createUpdateContentValues(title, plot, imdbid, rating, tagline, release, certification,
                runtime, trailer, genres, actors, collectionId, date);

        return mDatabase.update(DATABASE_TABLE, updateValues, KEY_TMDB_ID + " = ?", new String[]{tmdbid}) > 0 &&
                MizuuApplication.getCollectionsAdapter().createCollection(tmdbid, collectionId, collection) > 0;
    }

    public boolean updateMovieSingleItem(String tmdbId, String column, String value) {
        ContentValues values = new ContentValues();
        values.put(column, value);

        return mDatabase.update(DATABASE_TABLE, values, KEY_TMDB_ID + " = ?", new String[]{tmdbId}) > 0;
    }

    public String getSingleItem(String tmdbId, String column) {
        if (TextUtils.isEmpty(tmdbId))
            return "";

        String singleItem = "";
        Cursor c = mDatabase.query(DATABASE_TABLE, new String[]{column}, KEY_TMDB_ID + " = ?", new String[]{tmdbId}, null, null, null);
        if (c != null) {
            try {
                if (c.moveToFirst())
                    singleItem = c.getString(c.getColumnIndex(column));
            } catch (SQLException e) {} finally {
                c.close();
            }
        }

        return singleItem;
    }

    public boolean editMovie(String movieId, String title, String tagline, String description,
                             String genres, String runtime, String rating, String releaseDate, String certification) {
        ContentValues cv = new ContentValues();
        cv.put(KEY_TITLE, title);
        cv.put(KEY_PLOT, description);
        cv.put(KEY_RATING, rating);
        cv.put(KEY_TAGLINE, tagline);
        cv.put(KEY_RELEASEDATE, releaseDate);
        cv.put(KEY_CERTIFICATION, certification);
        cv.put(KEY_RUNTIME, runtime);
        cv.put(KEY_GENRES, genres);
        return mDatabase.update(DATABASE_TABLE, cv, KEY_TMDB_ID + " = ?", new String[]{movieId}) > 0;
    }

    /**
     * Deletes movie
     */
    public boolean deleteMovie(String tmdbId) {
        return mDatabase.delete(DATABASE_TABLE, KEY_TMDB_ID + " = ?", new String[]{tmdbId}) > 0;
    }

    public boolean deleteAllMovies() {
        return mDatabase.delete(DATABASE_TABLE, null, null) > 0;
    }

    public String getCollectionId(String tmdbId) {
        if (TextUtils.isEmpty(tmdbId))
            return "";
        return getSingleItem(tmdbId, KEY_COLLECTION_ID);
    }

    /**
     * Return a Cursor over the list of all movies in the database
     */
    public Cursor fetchAllMovies(String sort) {
        return mDatabase.query(DATABASE_TABLE, SELECT_ALL, null, null, null, null, sort);
    }

    public Cursor getAllMovies() {
        return mDatabase.query(DATABASE_TABLE, SELECT_ALL, null, null, null, null, KEY_TITLE + " ASC");
    }

    public Cursor getCollections() {
        return mDatabase.query(DATABASE_TABLE, SELECT_ALL, "NOT(" + KEY_COLLECTION_ID + " = '')", null, KEY_COLLECTION_ID, null, KEY_TITLE + " ASC");
    }

    public Cursor getFavorites() {
        return mDatabase.query(DATABASE_TABLE, SELECT_ALL, KEY_FAVOURITE + " = 1", null, null, null, KEY_TITLE + " ASC");
    }

    public Cursor getNewReleases() {
        return mDatabase.query(DATABASE_TABLE, SELECT_ALL, KEY_RELEASEDATE +
                " >= date('now', '-9 months')", null, null, null, KEY_RELEASEDATE + " DESC");
    }

    public Cursor getWatched() {
        return mDatabase.query(DATABASE_TABLE, SELECT_ALL, KEY_HAS_WATCHED + " = 1", null, null, null, KEY_TITLE + " ASC");
    }

    public Cursor getUnwatched() {
        return mDatabase.query(DATABASE_TABLE, SELECT_ALL, KEY_HAS_WATCHED + " = 0", null, null, null, KEY_TITLE + " ASC");
    }

    public Cursor getWatchlist() {
        return mDatabase.query(DATABASE_TABLE, SELECT_ALL, KEY_TO_WATCH + " = 1", null, null, null, KEY_TITLE + " ASC");
    }

    public Cursor fetchMovie(String movieId) throws SQLException {
        Cursor cursor = mDatabase.query(DATABASE_TABLE, SELECT_ALL, KEY_TMDB_ID + " = ?", new String[]{movieId}, null, null, null, null);
        if (cursor != null) {
            cursor.moveToFirst();
        }
        return cursor;
    }

    public boolean movieExists(String movieId) {
        Cursor cursor = mDatabase.query(DATABASE_TABLE, SELECT_ALL, KEY_TMDB_ID + " = ?", new String[]{movieId}, null, null, null, null);

        if (cursor != null)
            try {
                return cursor.getCount() > 0;
            } catch (SQLiteException e) {} finally {
                cursor.close();
            }

        return false;
    }

    private ContentValues createContentValues(String tmdbid, String title,
                                              String plot, String imdbid, String rating, String tagline, String release,
                                              String certification, String runtime, String trailer, String genres, String favourite, String actors,
                                              String collectionId, String toWatch, String hasWatched, String date, boolean includeTmdbId) {
        ContentValues values = new ContentValues();
        values.put(KEY_TITLE, title);
        values.put(KEY_PLOT, plot);
        if (includeTmdbId)
            values.put(KEY_TMDB_ID, tmdbid);
        values.put(KEY_IMDB_ID, imdbid);
        values.put(KEY_RATING, rating);
        values.put(KEY_TAGLINE, tagline);
        values.put(KEY_RELEASEDATE, release);
        values.put(KEY_CERTIFICATION, certification);
        values.put(KEY_RUNTIME, runtime);
        values.put(KEY_TRAILER, trailer);
        values.put(KEY_GENRES, genres);
        values.put(KEY_FAVOURITE, favourite);
        values.put(KEY_ACTORS, actors);
        values.put(KEY_COLLECTION_ID, collectionId);
        values.put(KEY_TO_WATCH, toWatch);
        values.put(KEY_HAS_WATCHED, hasWatched);
        values.put(KEY_DATE_ADDED, date);
        return values;
    }

    private ContentValues createUpdateContentValues(String title, String plot, String imdbid,
                                                    String rating, String tagline, String release, String certification, String runtime,
                                                    String trailer, String genres, String actors, String collectionId, String date) {
        ContentValues values = new ContentValues();
        values.put(KEY_TITLE, title);
        values.put(KEY_PLOT, plot);
        values.put(KEY_IMDB_ID, imdbid);
        values.put(KEY_RATING, rating);
        values.put(KEY_TAGLINE, tagline);
        values.put(KEY_RELEASEDATE, release);
        values.put(KEY_CERTIFICATION, certification);
        values.put(KEY_RUNTIME, runtime);
        values.put(KEY_TRAILER, trailer);
        values.put(KEY_GENRES, genres);
        values.put(KEY_ACTORS, actors);
        values.put(KEY_COLLECTION_ID, collectionId);
        values.put(KEY_DATE_ADDED, date);
        return values;
    }

    public int count() {
        Cursor c = mDatabase.query(DATABASE_TABLE, new String[]{KEY_TITLE, KEY_TMDB_ID}, "NOT(" + KEY_TMDB_ID + " = '" + UNIDENTIFIED_ID + "')", null, null, null, null);
        int count = c.getCount();
        c.close();
        return count;
    }

    public int countWatchlist() {
        Cursor c = mDatabase.query(DATABASE_TABLE, new String[] {KEY_TO_WATCH}, KEY_TO_WATCH + " = '1'", null, null, null, null);
        int count = c.getCount();
        c.close();
        return count;
    }

    public ArrayList<String> getCertifications() {
        ArrayList<String> certifications = new ArrayList<String>();
        Cursor cursor = mDatabase.query(DATABASE_TABLE, new String[]{KEY_CERTIFICATION}, null, null, KEY_CERTIFICATION, null, null);

        if (cursor != null) {
            try {
                while (cursor.moveToNext()) {
                    String certification = cursor.getString(cursor.getColumnIndex(KEY_CERTIFICATION));
                    if (!TextUtils.isEmpty(certification))
                        certifications.add(certification);
                }
            } catch (Exception e) {
            } finally {
                cursor.close();
            }
        }

        return certifications;
    }
}