Fixed bug with no_repeate mode, improved prev button behaviour
[someplayer] / src / dbstorage.cpp
index 485a59d..03755dd 100644 (file)
@@ -82,15 +82,13 @@ void DbStorage::_prepare_queries() {
                                                                "WHERE track_id IN "
                                                                "(SELECT track_id FROM adding_date ORDER BY date DESC LIMIT 0, :max)");
 
-       _get_track_count = new QSqlQuery(db);
-       _get_track_count->prepare("SELECT count from tracks WHERE id = :id");
+       _get_track_count_query = new QSqlQuery(db);
+       _get_track_count_query->prepare("SELECT count from tracks WHERE id = :id");
 
        _get_tracks_by_pattern_query = new QSqlQuery(db);
        _get_tracks_by_pattern_query->prepare("SELECT id, title, artist, album, source, count, length, year FROM "
                                              "entire WHERE "
-                                             "utitle LIKE (SELECT '%' || :ptitle || '%') OR "
-                                             "album_uname LIKE (SELECT '%' || :palbum || '%') OR "
-                                             "artist_uname LIKE (SELECT '%' || :partist || '%') "
+                                             "utitle LIKE (SELECT '%' || :ptitle || '%') "
                                              "ORDER BY artist_uname, year");
 
        _get_track_id_by_source_query = new QSqlQuery(db);
@@ -99,6 +97,18 @@ void DbStorage::_prepare_queries() {
        _get_directories_query = new QSqlQuery(db);
        _get_directories_query->prepare("SELECT id, path FROM directories");
 
+       _get_artists_count_query = new QSqlQuery(db);
+       _get_artists_count_query->prepare("SELECT COUNT(id) AS cnt FROM artist");
+
+       _get_albums_count_query = new QSqlQuery(db);
+       _get_albums_count_query->prepare("SELECT COUNT(id) AS cnt FROM album");
+
+       _get_tracks_count_query = new QSqlQuery(db);
+       _get_tracks_count_query->prepare("SELECT COUNT(id) AS cnt FROM tracks");
+
+       _get_tracks_source_from_query = new QSqlQuery(db);
+       _get_tracks_source_from_query->prepare("SELECT source FROM tracks WHERE directory = :directory_id");
+
        _check_artist_query = new QSqlQuery(db);
        _check_artist_query->prepare("SELECT id FROM artist WHERE uname = :uname");
 
@@ -108,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)");
 
@@ -145,6 +158,15 @@ void DbStorage::_prepare_queries() {
                                             "(SELECT COUNT(tracks.id) AS cnt, artist.id FROM "
                                             "artist LEFT OUTER JOIN tracks ON artist.id = tracks.artist_id "
                                             "GROUP BY artist.id) WHERE cnt = 0)");
+
+       _remove_tracks_from_query = new QSqlQuery(db);
+       _remove_tracks_from_query->prepare("DELETE FROM tracks WHERE directory = :directory_id");
+
+       _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() {
@@ -209,6 +231,10 @@ DbStorage::~DbStorage() {
        delete _get_tracks_for_album_query;
        delete _get_tracks_by_pattern_query;
        delete _get_directories_query;
+       delete _get_artists_count_query;
+       delete _get_albums_count_query;
+       delete _get_tracks_count_query;
+       delete _get_tracks_source_from_query;
        delete _check_album_query;
        delete _check_artist_query;
        delete _check_directory_query;
@@ -221,6 +247,9 @@ DbStorage::~DbStorage() {
        delete _insert_directory_query;
        delete _update_track_count_query;
        delete _remove_track_query;
+       delete _remove_directory_query;
+       delete _remove_tracks_from_query;
+       delete _remove_track_from_favorites_query;
        db.close();
 }
 
@@ -432,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) {
@@ -446,7 +478,7 @@ void DbStorage::updateTrackCount(Track track) {
        query->exec();
        if (query->next()) {
                int id = query->value(0).toInt();
-               query = _get_track_count;
+               query = _get_track_count_query;
                query->bindValue(":id", id);
                query->exec();
                if (query->next()) {
@@ -517,12 +549,10 @@ int DbStorage::_check_add_album(QString album, int artist_id, int year) {
        }
 }
 
-QList<Track> DbStorage::search(QString pattern) {
+QList<Track> DbStorage::searchTracks(QString pattern) {
        QList <Track> found;
        QSqlQuery *query = _get_tracks_by_pattern_query;
        query->bindValue(":ptitle", pattern.toUpper()); // with :pattern only doesn't work
-       query->bindValue(":palbum", pattern.toUpper());
-       query->bindValue(":partist", pattern.toUpper());
        query->exec();
        // id, title, artist, album, source, count, length, year
        while (query->next()) {
@@ -570,3 +600,83 @@ QList<QString> DbStorage::getDirectories() {
        }
        return directories;
 }
+
+int DbStorage::getArtistsCount() {
+       QSqlQuery *query = _get_artists_count_query;
+       query->exec();
+       if (query->next()) {
+               return query->value(0).toInt();
+       }
+       return 0;
+}
+
+int DbStorage::getAlbumsCount() {
+       QSqlQuery *query = _get_albums_count_query;
+       query->exec();
+       if (query->next()) {
+               return query->value(0).toInt();
+       }
+       return 0;
+}
+
+int DbStorage::getTracksCount() {
+       QSqlQuery *query = _get_tracks_count_query;
+       query->exec();
+       if (query->next()) {
+               return query->value(0).toInt();
+       }
+       return 0;
+}
+
+void DbStorage::deleteTracksFrom(QString path) {
+       QSqlQuery *query = _check_directory_query;
+       query->bindValue(":path", path);
+       query->exec();
+       if (query->next()) {
+               int id = query->value(0).toInt();
+               query = _remove_tracks_from_query;
+               query->bindValue(":directory_id", id);
+               query->exec();
+               query = _remove_directory_query;
+               query->bindValue(":path", path);
+               query->exec();
+               _cleanup();
+       }
+}
+
+void DbStorage::checkTracksFrom(QString path) {
+       QSqlQuery *query = _check_directory_query;
+       query->bindValue(":path", path);
+       query->exec();
+       if (query->next()) {
+               int id = query->value(0).toInt();
+               query = _get_tracks_source_from_query;
+               query->bindValue(":directory_id", id);
+               query->exec();
+               while (query->next()) {
+                       QString source = query->value(0).toString();
+                       if (!QFile::exists(source)) {
+                               Track track(TrackMetadata(), source);   // removeTrack uses only source field, so
+                               removeTrack(track);                     // we can use this method
+                       }
+               }
+               _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;
+}