Big commit:
[someplayer] / src / filestorage.cpp
1 #include "filestorage.h"
2 #include <QDir>
3 #include <QDirIterator>
4 #include <QFileInfo>
5 #include <QTextStream>
6 #include <QRegExp>
7 #include <QDebug>
8
9 using namespace SomePlayer::Storage;
10 using namespace SomePlayer::DataObjects;
11
12 FileStorage::FileStorage(QString path) {
13         _path_prefix = path;
14         _meta_regexp.setPattern("#META\\ +\\[(\\d+)\\]\\[(\\d+)\\].*::(.+)::,::(.+)::,::(.+)::");
15         _path_regexp.setPattern("#PATH (.+)");
16
17         Playlist current = getCurrentPlaylist();
18         if (current.name() == PLAYLIST_BAD_NAME) {
19                 current.setName(_CURRENT_PLAYLIST_NAME_);
20                 saveCurrentPlaylist(current);
21         }
22 }
23
24 QList<Playlist> FileStorage::getPlaylists() {
25         QList<Playlist> stub;
26         return stub;
27 }
28
29 Playlist FileStorage::getPlaylist(QString name) {
30         QFile playlistFile (_path_prefix+"/"+name+"."+_PLAYLIST_FILE_EXTENSION_);
31         Playlist playlist;
32         playlist.setName(PLAYLIST_BAD_NAME);
33         if (playlistFile.exists()) {
34                 playlist.setName(name);
35                 playlistFile.open(QFile::ReadOnly);
36                 QTextStream stream(&playlistFile);
37                 QString buffer = stream.readLine();
38                 if (buffer.startsWith(_PLAYLIST_SIGNATURE_)) {
39                         while (!stream.atEnd()) {
40                                 buffer = stream.readLine();
41                                 if (_meta_regexp.indexIn(buffer) != -1) {
42                                         int id = _meta_regexp.cap(1).toInt();
43                                         int seconds = _meta_regexp.cap(2).toInt();
44                                         QString artist = _meta_regexp.cap(3);
45                                         QString album = _meta_regexp.cap(4);
46                                         QString title = _meta_regexp.cap(5);
47                                         buffer = stream.readLine();
48                                         if (_path_regexp.indexIn(buffer) != -1) {
49                                                 QString source = _path_regexp.cap(1);
50                                                 TrackMetadata meta(title, artist, album, seconds);
51                                                 Track track(id, meta, source);
52                                                 playlist.addTrack(track);
53                                         }
54                                 }
55                         }
56                 }
57         }
58         return playlist;
59 }
60
61 QStringList FileStorage::getPlaylistsNames() {
62         QDir directory(_path_prefix);
63         QDirIterator iterator(directory, QDirIterator::FollowSymlinks);
64         QStringList playlistNames;
65         while (iterator.hasNext()) {
66                 QString entry = iterator.next();
67                 QFileInfo info(entry);
68                 QString suffix = info.suffix().toLower();
69                 if (suffix == _PLAYLIST_FILE_EXTENSION_) {
70                         QString name = info.fileName().replace(QString(".%1").arg(_PLAYLIST_FILE_EXTENSION_), "", Qt::CaseInsensitive);
71                         playlistNames.append(name);
72                 }
73         }
74         return playlistNames;
75 }
76
77 void FileStorage::savePlaylist(Playlist playlist) {
78         QString filename = _path_prefix + "/" +playlist.name()+"."_PLAYLIST_FILE_EXTENSION_;
79         QFile playlistFile(filename);
80         if (playlistFile.exists()) {
81                 playlistFile.remove();
82         }
83         playlistFile.open(QFile::WriteOnly);
84         QTextStream stream(&playlistFile);
85         stream << _PLAYLIST_SIGNATURE_ << endl;
86         const QList<Track> &tracks = playlist.tracks();
87         foreach (Track track, tracks) {
88                 stream << _PLAYLIST_META_KEYWORD_ << " [" << track.id() << "]" << "[" << track.metadata().length() << "],::"
89                                 << track.metadata().artist() << "::,::" << track.metadata().album() << "::,::"
90                                 << track.metadata().title() << "::" << endl;
91                 stream << _PLAYLIST_PATH_KEYWORD_ << " " << track.source() << endl;
92         }
93 }
94
95 void FileStorage::removePlaylist(Playlist playlist) {
96         QString filename = _path_prefix + "/" + playlist.name() + "." + _PLAYLIST_FILE_EXTENSION_;
97         QFile file(filename);
98         file.remove();
99 }
100
101 void FileStorage::removePlaylist(QString name) {
102         QString filename = _path_prefix + "/" + name + "." + _PLAYLIST_FILE_EXTENSION_;
103         QFile file(filename);
104         file.remove();
105 }
106
107 Playlist FileStorage::getCurrentPlaylist() {
108         return getPlaylist(_CURRENT_PLAYLIST_NAME_);
109 }
110
111 void FileStorage::saveCurrentPlaylist(Playlist playlist) {
112         playlist.setName(_CURRENT_PLAYLIST_NAME_);
113         savePlaylist(playlist);
114 }