Version bump

[someplayer] / src / dbstorage.cpp

diff --git a/src/dbstorage.cpp b/src/dbstorage.cpp
index a9b811d..171f2ae 100644 (file)
--- a/src/dbstorage.cpp
+++ b/src/dbstorage.cpp
@@ -72,7 +72,7 @@ void DbStorage::_prepare_queries() {
                                                                "(SELECT tracks.id AS track_id, artist.name AS artist, title, count, source, tracks.album_id, length FROM "
                                                                "tracks JOIN artist ON tracks.artist_id = artist.id) "
                                                                "JOIN album ON album_id = album.id "
-                                                               "WHERE count = 0");
+                                                               "WHERE count = 0 LIMIT 0, :max");
 
        _get_recently_added_query = new QSqlQuery(db);
        _get_recently_added_query->prepare("SELECT track_id as id, title, artist, album.name as album, source, count, length FROM "
@@ -118,6 +118,9 @@ void DbStorage::_prepare_queries() {
        _check_directory_query = new QSqlQuery(db);
        _check_directory_query->prepare("SELECT id from directories WHERE path = :path");
 
+       _check_favorite_query = new QSqlQuery(db);
+       _check_favorite_query->prepare("SELECT 'yes' from favorites WHERE track_id in (SELECT id from tracks WHERE source = :source)");
+
        _insert_artist_query = new QSqlQuery(db);
        _insert_artist_query->prepare("INSERT INTO artist (name, uname) values (:name, :uname)");
 
@@ -161,6 +164,9 @@ void DbStorage::_prepare_queries() {
 
        _remove_directory_query = new QSqlQuery(db);
        _remove_directory_query->prepare("DELETE FROM directories WHERE path = :path");
+
+       _remove_track_from_favorites_query = new QSqlQuery(db);
+       _remove_track_from_favorites_query->prepare("DELETE FROM favorites WHERE track_id in (SELECT id FROM tracks WHERE source = :source)");
 }
 
 void DbStorage::_create_database_structure() {
@@ -243,6 +249,7 @@ DbStorage::~DbStorage() {
        delete _remove_track_query;
        delete _remove_directory_query;
        delete _remove_tracks_from_query;
+       delete _remove_track_from_favorites_query;
        db.close();
 }
 
@@ -454,12 +461,15 @@ void DbStorage::addToFavorites(Track track) {
        QSqlQuery *query = _get_track_id_by_source_query;
        query->bindValue(":source", track.source());
        query->exec();
-       if (query->next()) {
-               int id = query->value(0).toInt();
-               query = _insert_favorites_query;
-               query->bindValue(":track_id", id);
+       if (!query->next()) {
+               addTrack(track);
                query->exec();
+               query->next();
        }
+       int id = query->value(0).toInt();
+       query = _insert_favorites_query;
+       query->bindValue(":track_id", id);
+       query->exec();
 }
 
 void DbStorage::updateTrackCount(Track track) {
@@ -653,3 +663,20 @@ void DbStorage::checkTracksFrom(QString path) {
                _cleanup();
        }
 }
+
+void DbStorage::removeFromFavorites(Track track) {
+       QSqlQuery *query = _remove_track_from_favorites_query;
+       query->bindValue(":source", track.source());
+       query->exec();
+}
+
+bool DbStorage::isFavorite(Track track) {
+       QSqlQuery *query = _check_favorite_query;
+       query->bindValue(":source", track.source());
+       query->exec();
+       if (query->next()) {
+               QString ans = query->value(0).toString();
+               return ans == "yes";
+       }
+       return false;
+}