Fixed broken image issue in taglib
[someplayer] / src / dbstorage.cpp
index dac2868..a9b811d 100644 (file)
 #include "dbstorage.h"
 #include <QSqlQuery>
 #include <QSqlResult>
+#include <QDebug>
+#include <QFileInfo>
 
 using namespace SomePlayer::Storage;
 using namespace SomePlayer::DataObjects;
 
 DbStorage::DbStorage(QString path) {
-       QString dbname = path+_DATABASE_NAME_;
+       QString dbname = path + _DATABASE_NAME_;
        db = QSqlDatabase::addDatabase("QSQLITE");
        db.setDatabaseName(dbname);
        if (!db.open()) {
@@ -38,65 +40,92 @@ DbStorage::DbStorage(QString path) {
 
 void DbStorage::_prepare_queries() {
        _get_artists_query = new QSqlQuery(db);
-       _get_artists_query->prepare("SELECT name FROM artist");
+       _get_artists_query->prepare("SELECT name FROM artist ORDER BY uname");
 
-       _get_albums_for_artist_query = new QSqlQuery(db);
-       _get_albums_for_artist_query->prepare("SELECT name FROM album WHERE artist_id in (SELECT id from artist WHERE name = :name);");
+       _get_albums_for_artist_sort_name_query = new QSqlQuery(db);
+       _get_albums_for_artist_sort_name_query->prepare("SELECT name, year FROM album WHERE artist_id in (SELECT id from artist WHERE uname = :uname) ORDER BY uname;");
+
+       _get_albums_for_artist_sort_year_query = new QSqlQuery(db);
+       _get_albums_for_artist_sort_year_query->prepare("SELECT name, year FROM album WHERE artist_id in (SELECT id from artist WHERE uname = :uname) ORDER BY year;");
 
        _get_tracks_for_album_query = new QSqlQuery(db);
        _get_tracks_for_album_query->prepare("SELECT id, title, source, count, length FROM tracks WHERE artist_id IN "
-                                                                                "(SELECT id FROM artist WHERE name = :artist_name) AND album_id IN "
-                                                                                "(SELECT id FROM album WHERE name = :album_name);");
+                                                               "(SELECT id FROM artist WHERE uname = :artist_uname) AND album_id IN "
+                                                               "(SELECT id FROM album WHERE uname = :album_uname);");
 
        _get_favorites_query = new QSqlQuery(db);
        _get_favorites_query->prepare("SELECT track_id as id, title, artist, album.name as album, source, count, length FROM "
-                                                                 "(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 track_id IN "
-                                                                 "(SELECT track_id FROM favorites);");
+                                                               "(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 track_id IN "
+                                                               "(SELECT track_id FROM favorites);");
 
        _get_most_played_query = new QSqlQuery(db);
        _get_most_played_query->prepare("SELECT track_id as id, title, artist, album.name as album, source, count, length FROM "
-                                                                       "(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 ORDER BY count DESC "
-                                                                       "LIMIT 0, :max");
+                                                               "(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 ORDER BY count DESC "
+                                                               "LIMIT 0, :max");
 
        _get_never_played_query = new QSqlQuery(db);
        _get_never_played_query->prepare("SELECT track_id as id, title, artist, album.name as album, source, count, length FROM "
-                                                                        "(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");
+                                                               "(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");
 
        _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 "
-                                                                          "(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 track_id IN "
-                                                                          "(SELECT track_id FROM adding_date ORDER BY date DESC LIMIT 0, :max)");
+                                                               "(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 track_id IN "
+                                                               "(SELECT track_id FROM adding_date ORDER BY date DESC LIMIT 0, :max)");
+
+       _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 || '%') "
+                                             "ORDER BY artist_uname, year");
+
+       _get_track_id_by_source_query = new QSqlQuery(db);
+       _get_track_id_by_source_query->prepare("SELECT id FROM tracks WHERE source = :source");
+
+       _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_track_count = new QSqlQuery(db);
-       _get_track_count->prepare("SELECT count from tracks WHERE id = :id");
+       _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 name = :name");
+       _check_artist_query->prepare("SELECT id FROM artist WHERE uname = :uname");
 
        _check_album_query = new QSqlQuery(db);
-       _check_album_query->prepare("SELECT id FROM album WHERE name = :name AND artist_id = :artist_id");
+       _check_album_query->prepare("SELECT id FROM album WHERE uname = :uname AND artist_id = :artist_id");
 
-       _check_track_query = new QSqlQuery(db);
-       _check_track_query->prepare("SELECT id FROM tracks WHERE source = :source");
+       _check_directory_query = new QSqlQuery(db);
+       _check_directory_query->prepare("SELECT id from directories WHERE path = :path");
 
        _insert_artist_query = new QSqlQuery(db);
-       _insert_artist_query->prepare("INSERT INTO artist (name) values (:name)");
+       _insert_artist_query->prepare("INSERT INTO artist (name, uname) values (:name, :uname)");
 
        _insert_album_query = new QSqlQuery(db);
-       _insert_album_query->prepare("INSERT INTO album (name, artist_id) values (:name, :artist_id)");
+       _insert_album_query->prepare("INSERT INTO album (name, uname, artist_id, year) values (:name, :uname, :artist_id, :year)");
 
        _insert_track_query = new QSqlQuery(db);
-       _insert_track_query->prepare("INSERT INTO tracks (title, artist_id, album_id, source, length) values (:title, :artist_id, :album_id, :source, :length)");
+       _insert_track_query->prepare("INSERT INTO tracks (title, utitle, artist_id, album_id, source, directory, length) values (:title, :utitle, :artist_id, :album_id, :source, :directory_id, :length)");
 
        _insert_date_query = new QSqlQuery(db);
        _insert_date_query->prepare("INSERT INTO adding_date (track_id, date) values (:track_id, strftime('%s', 'now'))");
@@ -104,56 +133,116 @@ void DbStorage::_prepare_queries() {
        _insert_favorites_query = new QSqlQuery(db);
        _insert_favorites_query->prepare("INSERT INTO favorites (track_id) values (:track_id)");
 
+       _insert_directory_query = new QSqlQuery(db);
+       _insert_directory_query->prepare("INSERT INTO directories (path) values (:path)");
+
        _update_track_count_query = new QSqlQuery(db);
        _update_track_count_query->prepare("UPDATE tracks SET count = :count where id = :id");
+
+       _remove_track_query = new QSqlQuery(db);
+       _remove_track_query->prepare("DELETE FROM tracks WHERE id = :id");
+
+       _remove_empty_albums_query = new QSqlQuery(db);
+       _remove_empty_albums_query->prepare("DELETE FROM album WHERE album.id IN "
+                                           "(SELECT id FROM "
+                                           "(SELECT COUNT(tracks.id) AS cnt, album.id FROM "
+                                           "album LEFT OUTER JOIN tracks ON album.id = tracks.album_id "
+                                           "GROUP BY album.id) WHERE cnt = 0)");
+
+       _remove_empty_artists_query = new QSqlQuery(db);
+       _remove_empty_artists_query->prepare("DELETE FROM artist WHERE artist.id IN "
+                                            "(SELECT id FROM "
+                                            "(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");
 }
 
 void DbStorage::_create_database_structure() {
        QSqlQuery *query = new QSqlQuery(db);
        query->exec("create table artist (id integer primary key, "
-                                                                               "name text "
-                                                                               ");");
+                                                               "name text, "
+                                                               "uname text "
+                                                               ");");
        query->exec("create table album (id integer primary key, "
-                                                                               "artist_id integer, "
-                                                                               "name text, "
-                                                                               "foreign key(artist_id) references arist(id) "
-                                                                               ");");
+                                                               "artist_id integer, "
+                                                               "name text, "
+                                                               "uname text, "
+                                                               "year int, "
+                                                               "foreign key(artist_id) references arist(id) "
+                                                               ");");
        query->exec("create table tracks (id integer primary key, "
-                                                                               "artist_id integer, "
-                                                                               "album_id integer, "
-                                                                               "title text, "
-                                                                               "source text, "
-                                                                               "count integer default 0, "
-                                                                               "length integer default 0, "
-                                                                               "foreign key(artist_id) references artist(id), "
-                                                                               "foreign key(album_id) references album(id) "
-                                                                               ");");
+                                                               "artist_id integer, "
+                                                               "album_id integer, "
+                                                               "title text, "
+                                                               "utitle text, "
+                                                               "source text, "
+                                                               "directory integer, "
+                                                               "count integer default 0, "
+                                                               "length integer default 0, "
+                                                               "foreign key(artist_id) references artist(id), "
+                                                               "foreign key(album_id) references album(id) "
+                                                               ");");
        query->exec("create table favorites (track_id integer, "
-                                                                               "foreign key(track_id) references tracks(id) "
-                                                                               ");");
+                                                               "foreign key(track_id) references tracks(id) "
+                                                               ");");
        query->exec("create table adding_date (track_id integer, "
-                                                                               "date integer, "
-                                                                               "foreign key(track_id) references tracks(id) "
-                                                                               ");");
+                                                               "date integer, "
+                                                               "foreign key(track_id) references tracks(id) "
+                                                               ");");
+       query->exec("create view entire as "
+                   "select id, title, artist_name as artist, album_name as album, source, count, length, album_year as year, utitle, artist_uname, album_uname from "
+                       "tracks left join "
+                               "(select artist.id as aartist_id, "
+                               "artist.uname as artist_uname, "
+                               "artist.name as artist_name, "
+                               "album.uname as album_uname, "
+                               "album.year as album_year, "
+                               "album.id as aalbum_id, "
+                               "album.name as album_name from "
+                                       "artist "
+                                       "join album "
+                                       "on album.artist_id = artist.id) "
+                               "on aartist_id = tracks.artist_id "
+                               "and aalbum_id = tracks.album_id");
+
+       query->exec("create table directories (id integer primary key, path text)");
 }
 
 DbStorage::~DbStorage() {
-       delete _get_albums_for_artist_query;
+       delete _get_albums_for_artist_sort_name_query;
+       delete _get_albums_for_artist_sort_year_query;
        delete _get_artists_query;
        delete _get_favorites_query;
        delete _get_most_played_query;
        delete _get_never_played_query;
        delete _get_recently_added_query;
        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_track_query;
+       delete _check_directory_query;
+       delete _get_track_id_by_source_query;
        delete _insert_album_query;
        delete _insert_artist_query;
        delete _insert_date_query;
        delete _insert_track_query;
        delete _insert_favorites_query;
+       delete _insert_directory_query;
        delete _update_track_count_query;
+       delete _remove_track_query;
+       delete _remove_directory_query;
+       delete _remove_tracks_from_query;
        db.close();
 }
 
@@ -168,14 +257,21 @@ QList<QString> DbStorage::getArtists() {
        return artists;
 }
 
-QList<QString> DbStorage::getAlbumsForArtist(QString artist) {
-       QList<QString> albums;
-       QSqlQuery *query = _get_albums_for_artist_query;
-       query->bindValue(":name", artist);
+QMap<QString, int> DbStorage::getAlbumsForArtist(QString artist) {
+       QMap<QString, int> albums;
+       Config config;
+       QString sort = config.getValue("ui/albumsorting").toString();
+       QSqlQuery *query = NULL;
+       if (sort == "date") {
+               query = _get_albums_for_artist_sort_year_query;
+       } else {
+               query = _get_albums_for_artist_sort_name_query;
+       }
+       query->bindValue(":uname", artist.toUpper());
        query->exec();
        while (query->next()) {
                QString name = query->value(0).toString();
-               albums.append(name);
+               albums[name] = query->value(1).toInt();
        }
        return albums;
 }
@@ -183,18 +279,17 @@ QList<QString> DbStorage::getAlbumsForArtist(QString artist) {
 QList<Track> DbStorage::getTracksForAlbum(QString album, QString artist) {
        QList<Track> tracks;
        QSqlQuery *query = _get_tracks_for_album_query;
-       query->bindValue(":artist_name", artist);
-       query->bindValue(":album_name", album);
+       query->bindValue(":artist_uname", artist.toUpper());
+       query->bindValue(":album_uname", album.toUpper());
        query->exec();
 
        while (query->next()) {
-               int id = query->value(0).toInt();
                QString title = query->value(1).toString();
                QString source = query->value(2).toString();
                int count = query->value(3).toInt();
                int length = query->value(4).toInt();
                TrackMetadata meta (title, artist, album, length);
-               Track track(id, meta, source);
+               Track track(meta, source);
                track.setCount(count);
                tracks.append(track);
        }
@@ -207,7 +302,6 @@ Playlist DbStorage::getFavorites() {
        QSqlQuery *query = _get_favorites_query;
        query->exec();
        while(query->next()) {
-               int id = query->value(0).toInt();
                QString title = query->value(1).toString();
                QString artist = query->value(2).toString();
                QString album = query->value(3).toString();
@@ -215,7 +309,7 @@ Playlist DbStorage::getFavorites() {
                int count = query->value(5).toInt();
                int length = query->value(6).toInt();
                TrackMetadata meta(title, artist, album, length);
-               Track track(id, meta, source);
+               Track track(meta, source);
                track.setCount(count);
                playlist.addTrack(track);
                playlist.setName("Favorites");
@@ -229,7 +323,6 @@ Playlist DbStorage::getMostPlayed() {
        query->bindValue(":max", _DYNAMIC_PLAYLIST_MAX_COUNT_);
        query->exec();
        while (query->next()) {
-               int id = query->value(0).toInt();
                QString title = query->value(1).toString();
                QString artist = query->value(2).toString();
                QString album = query->value(3).toString();
@@ -237,7 +330,7 @@ Playlist DbStorage::getMostPlayed() {
                int count = query->value(5).toInt();
                int length = query->value(6).toInt();
                TrackMetadata meta(title, artist, album, length);
-               Track track(id, meta, source);
+               Track track(meta, source);
                track.setCount(count);
                playlist.addTrack(track);
                playlist.setName("Most popular");
@@ -251,7 +344,6 @@ Playlist DbStorage::getNeverPlayed() {
        query->bindValue(":max", _DYNAMIC_PLAYLIST_MAX_COUNT_);
        query->exec();
        while (query->next()) {
-               int id = query->value(0).toInt();
                QString title = query->value(1).toString();
                QString artist = query->value(2).toString();
                QString album = query->value(3).toString();
@@ -259,7 +351,7 @@ Playlist DbStorage::getNeverPlayed() {
                int count = query->value(5).toInt();
                int length = query->value(6).toInt();
                TrackMetadata meta(title, artist, album, length);
-               Track track(id, meta, source);
+               Track track(meta, source);
                track.setCount(count);
                playlist.addTrack(track);
                playlist.setName("Never played");
@@ -273,7 +365,6 @@ Playlist DbStorage::getRecentlyAdded() {
        query->bindValue(":max", _DYNAMIC_PLAYLIST_MAX_COUNT_);
        query->exec();
        while (query->next()) {
-               int id = query->value(0).toInt();
                QString title = query->value(1).toString();
                QString artist = query->value(2).toString();
                QString album = query->value(3).toString();
@@ -281,7 +372,7 @@ Playlist DbStorage::getRecentlyAdded() {
                int count = query->value(5).toInt();
                int length = query->value(6).toInt();
                TrackMetadata meta(title, artist, album, length);
-               Track track(id, meta, source);
+               Track track(meta, source);
                track.setCount(count);
                playlist.addTrack(track);
                playlist.setName("Recently added");
@@ -290,17 +381,22 @@ Playlist DbStorage::getRecentlyAdded() {
 }
 
 void DbStorage::removeTrack(Track track) {
-       int id = track.id();
-       QSqlQuery *query = new QSqlQuery(db);
-       query->prepare("DELETE FROM tracks WHERE id = :id;");
-       query->bindValue(":id", id);
-       query->exec();
-       query->prepare("DELETE FROM favorites WHERE track_id = :id;");
-       query->bindValue(":id", id);
-       query->exec();
-       query->prepare("DELETE FROM adding_date WHERE id = :id;");
-       query->bindValue(":id", id);
+       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 = new QSqlQuery(db);
+               query->prepare("DELETE FROM tracks WHERE id = :id;");
+               query->bindValue(":id", id);
+               query->exec();
+               query->prepare("DELETE FROM favorites WHERE track_id = :id;");
+               query->bindValue(":id", id);
+               query->exec();
+               query->prepare("DELETE FROM adding_date WHERE id = :id;");
+               query->bindValue(":id", id);
+               query->exec();
+       }
 }
 
 void DbStorage::addTrack(Track track) {
@@ -308,13 +404,16 @@ void DbStorage::addTrack(Track track) {
        QString artist = track.metadata().artist();
        QString album = track.metadata().album();
        QString source = track.source();
+       QFileInfo info(source);
+       QString path = info.canonicalPath();
+       int year = track.metadata().year();
        int artist_id = _check_add_artist(artist);
-       int album_id = _check_add_album(album, artist_id);
+       int album_id = _check_add_album(album, artist_id, year);
        if (artist_id == -1 || album_id == -1) {
                //big bang
                return;
        }
-       QSqlQuery *query = _check_track_query;
+       QSqlQuery *query = _get_track_id_by_source_query;
        query->bindValue(":source", source);
        query->exec();
        if (query->next()) {
@@ -323,13 +422,15 @@ void DbStorage::addTrack(Track track) {
        }
        query = _insert_track_query;
        query->bindValue(":title", title);
+       query->bindValue(":utitle", title.toUpper());
        query->bindValue(":artist_id", artist_id);
        query->bindValue(":album_id", album_id);
        query->bindValue(":source", source);
+       query->bindValue(":directory", _check_add_directory(path));
        query->bindValue(":length", track.metadata().length());
        if (query->exec()) {
                //ok
-               query = _check_track_query;
+               query = _get_track_id_by_source_query;
                query->bindValue(":source", source);
                query->exec();
                if (query->next()) {
@@ -350,27 +451,54 @@ void DbStorage::addTrack(Track track) {
 }
 
 void DbStorage::addToFavorites(Track track) {
-       QSqlQuery *query = _insert_favorites_query;
-       query->bindValue(":track_id", track.id());
+       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);
+               query->exec();
+       }
 }
 
 void DbStorage::updateTrackCount(Track track) {
-       QSqlQuery *query = _get_track_count;
-       query->bindValue(":id", track.id());
+       QSqlQuery *query = _get_track_id_by_source_query;
+       query->bindValue(":source", track.source());
        query->exec();
        if (query->next()) {
-               int count = query->value(0).toInt();
-               query = _update_track_count_query;
-               query->bindValue(":count", count+1);
-               query->bindValue(":id", track.id());
+               int id = query->value(0).toInt();
+               query = _get_track_count_query;
+               query->bindValue(":id", id);
                query->exec();
+               if (query->next()) {
+                       int count = query->value(0).toInt();
+                       query = _update_track_count_query;
+                       query->bindValue(":count", count+1);
+                       query->bindValue(":id", id);
+                       query->exec();
+               }
        }
 }
 
+Track DbStorage::updateTrack(Track track) {
+       QSqlQuery *query = _get_track_id_by_source_query;
+       query->bindValue(":source", track.source());
+       query->exec();
+       if (query->next()) { // found track in library
+               int id = query->value(0).toInt();
+               query = _remove_track_query;
+               query->bindValue(":id", id);
+               query->exec();
+               addTrack(track);
+               _cleanup();
+       }
+       return track;
+}
+
 int DbStorage::_check_add_artist(QString artist) {
        QSqlQuery *query = _check_artist_query;
-       query->bindValue(":name", artist);
+       query->bindValue(":uname", artist.toUpper());
        query->exec();
        if (query->next()) {
                int id = query->value(0).toInt();
@@ -378,6 +506,7 @@ int DbStorage::_check_add_artist(QString artist) {
        } else {
                query = _insert_artist_query;
                query->bindValue(":name", artist);
+               query->bindValue(":uname", artist.toUpper());
                if (query->exec()) {
                        return _check_add_artist(artist);
                } else {
@@ -387,9 +516,9 @@ int DbStorage::_check_add_artist(QString artist) {
        }
 }
 
-int DbStorage::_check_add_album(QString album, int artist_id) {
+int DbStorage::_check_add_album(QString album, int artist_id, int year) {
        QSqlQuery *query = _check_album_query;
-       query->bindValue(":name", album);
+       query->bindValue(":uname", album.toUpper());
        query->bindValue(":artist_id", artist_id);
        query->exec();
        if (query->next()) {
@@ -398,12 +527,129 @@ int DbStorage::_check_add_album(QString album, int artist_id) {
        } else {
                query = _insert_album_query;
                query->bindValue(":name", album);
+               query->bindValue(":uname", album.toUpper());
                query->bindValue(":artist_id", artist_id);
+               query->bindValue(":year", year);
                if (query->exec()) {
-                       return _check_add_album(album, artist_id);
+                       return _check_add_album(album, artist_id, year);
                } else {
                        // big bang
                        return -1;
                }
        }
 }
+
+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->exec();
+       // id, title, artist, album, source, count, length, year
+       while (query->next()) {
+               QString title = query->value(1).toString();
+               QString artist = query->value(2).toString();
+               QString album = query->value(3).toString();
+               QString source = query->value(4).toString();
+               int count = query->value(5).toInt();
+               int length = query->value(6).toInt();
+               int year = query->value(7).toInt();
+               TrackMetadata meta(title, artist, album, length);
+               meta.setYear(year);
+               Track track(meta, source);
+               track.setCount(count);
+               found.append(track);
+       }
+       return found;
+}
+
+void DbStorage::_cleanup() {
+       _remove_empty_albums_query->exec();
+       _remove_empty_artists_query->exec();
+}
+
+int DbStorage::_check_add_directory(QString path) {
+       QSqlQuery *query = _check_directory_query;
+       query->bindValue(":path", path);
+       query->exec();
+       if (query->next()) {
+               return query->value(0).toInt();
+       } else {
+               query = _insert_directory_query;
+               query->bindValue(":path", path);
+               query->exec();
+               return _check_add_directory(path);
+       }
+}
+
+QList<QString> DbStorage::getDirectories() {
+       QSqlQuery *query = _get_directories_query;
+       query->exec();
+       QList<QString> directories;
+       while (query->next()) {
+               directories.append(query->value(1).toString());
+       }
+       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();
+       }
+}