Refactored PlayList handling to recursive function.
[vlc-remote] / browsemainwindow.cpp
1 /*   VLC-REMOTE for MAEMO 5
2 *   Copyright (C) 2010 Schutz Sacha <istdasklar@gmail.com>, Dru Moore <usr@dru-id.co.uk>, Yann Nave <yannux@onbebop.net>
3 *   This program is free software; you can redistribute it and/or modify
4 *   it under the terms of the GNU General Public License version 2,
5 *   or (at your option) any later version, as published by the Free
6 *   Software Foundation
7 *
8 *   This program is distributed in the hope that it will be useful,
9 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
10 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11 *   GNU General Public License for more details
12 *
13 *   You should have received a copy of the GNU General Public
14 *   License along with this program; if not, write to the
15 *   Free Software Foundation, Inc.,
16 *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
17 */
18 #include "browsemainwindow.h"
19 #include "ui_browsemainwindow.h"
20 #include <QSettings>
21 #if defined(Q_WS_S60) || defined(Q_WS_MAEMO_5)
22 #include <QMaemo5InformationBox>
23 #endif
24 #include "configdialog.h"
25 #include "aboutdialog.h"
26 #include "vlcbrowseelement.h"
27 #include "appsettings.h"
28
29 BrowseMainWindow::BrowseMainWindow(QWidget *parent) :
30         QMainWindow(parent),
31         ui(new Ui::BrowseMainWindow)
32 {
33
34     ui->setupUi(this);
35     mCurrentDir = "~/"; //AppSettings::getHomeDirectory().path; // This works on win as well as linux, would guess mac too.
36     setWindowTitle("Vlc remote");
37
38
39     mNetManager = new QNetworkAccessManager(this);
40
41     mContents = new QList<VlcBrowseElement>();
42
43     ui->listWidget->setTextElideMode(Qt::ElideMiddle);
44     ui->listWidget->setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
45
46     ui->playButton->setIcon(QIcon::fromTheme("camera_playback"));
47     ui->addButton->setIcon(QIcon::fromTheme("general_add"));
48     ui->browseButton->setIcon(QIcon::fromTheme("filemanager_media_folder"));
49     ui->browseButton->setDisabled(true);
50     ui->playButton->setDisabled(true);
51     ui->addButton->setDisabled(true);
52
53     connect(ui->browseButton,SIGNAL(clicked()),this,SLOT(onBrowse()));
54     connect(ui->addButton,SIGNAL(clicked()),this,SLOT(onAddToPlaylist()));
55     connect(ui->playButton,SIGNAL(clicked()),this,SLOT(onPlay()));
56     connect(ui->listWidget, SIGNAL(itemSelectionChanged()), this, SLOT(onListSelectionChanged()));
57
58     connect(ui->actionGoHome, SIGNAL(triggered()), this, SLOT(showHomeFolder()));
59     connect(ui->actionSetHome, SIGNAL(triggered()), this, SLOT(setHomeFolder()));
60     connect(ui->actionViewFavourites, SIGNAL(triggered()), this, SLOT(showFavourites()));
61     connect(ui->actionSetFavourite, SIGNAL(triggered()), this, SLOT(setFavourite()));
62
63     init();
64
65
66 }
67 void BrowseMainWindow::init()  // THIS METHOD IS CALLED WHEN CONFIG CHANGED...
68 {
69     mIp = AppSettings::getCurrentIp(); // AccountDialog::currentIp();
70     setHomeDirectory();
71 }
72 void BrowseMainWindow::setHomeDirectory()
73 {
74     mCurrentDir = AppSettings::getHomeDirectory().path;
75 }
76 void BrowseMainWindow::showCurrentDirectory()  // THIS METHOD IS CALLED WHEN WINDOW IS OPENED...
77 {
78     browseDirectory(mCurrentDir);
79 }
80
81 BrowseMainWindow::~BrowseMainWindow()
82 {
83     delete ui;
84 }
85
86 void BrowseMainWindow::changeEvent(QEvent *e)
87 {
88     QMainWindow::changeEvent(e);
89     switch (e->type()) {
90     case QEvent::LanguageChange:
91         ui->retranslateUi(this);
92         break;
93     default:
94         break;
95     }
96 }
97
98 void BrowseMainWindow::showHomeFolder() {
99     browseDirectory(AppSettings::getHomeDirectory().path);
100 }
101 void BrowseMainWindow::setHomeFolder() {
102     if (0 < mCurrentDir.length()) {
103         VlcDirectory dir;
104         dir.name = "Home";
105         dir.path = mCurrentDir;
106         AppSettings::setHomeDirectory(dir);
107     }
108 }
109 void BrowseMainWindow::showFavourites() {}
110 void BrowseMainWindow::setFavourite() {}
111
112 void BrowseMainWindow::onListSelectionChanged() {
113     QList<QListWidgetItem *> items = ui->listWidget->selectedItems();
114     if (0 < items.count()) {
115         mCurrentElement = getElementFromText(items.at(0)->text());
116         // are we up dir?
117         if (0 == QString::compare("..", mCurrentElement.name)) {
118             ui->browseButton->setDisabled(true);
119             ui->playButton->setDisabled(true);
120             ui->addButton->setDisabled(true);
121             mCurrentDir = mCurrentElement.path;
122             browseDirectory(mCurrentDir);
123         }
124         else {
125             // can we browse?
126             if (0 == QString::compare("directory", mCurrentElement.type)) {
127                 ui->browseButton->setDisabled(false);
128             }
129             else {
130                 ui->browseButton->setDisabled(true);
131             }
132             // can we play?
133             ui->playButton->setDisabled(false);
134             // can we playlist?
135             ui->addButton->setDisabled(false);
136         }
137     }
138 }
139
140 VlcBrowseElement BrowseMainWindow::getElementFromText(QString text) {
141     for (int idx = 0; idx < mContents->count(); ++idx) {
142         if (0 == QString::compare(text, mContents->at(idx).name)) {
143             return mContents->at(idx);
144         }
145     }
146     return *(new VlcBrowseElement());
147 }
148
149 void BrowseMainWindow::onBrowse() {
150     // check for directory
151     if (0 == QString::compare("directory", mCurrentElement.type)) {
152         // call browseDirectory
153         mCurrentDir = mCurrentElement.path;
154         browseDirectory(mCurrentDir);
155     }
156     else {
157         ui->browseButton->setDisabled(true);
158     }
159 }
160
161 void BrowseMainWindow::onAddToPlaylist() {
162     QUrl url = QUrl("http://"+mIp+"/requests/status.xml?command=in_enqueue");
163     url.addEncodedQueryItem(QByteArray("input"), QUrl::toPercentEncoding(mCurrentElement.path.replace("\\", "\\\\")));
164     mNetManager->get(QNetworkRequest(url));
165     //mNetManager->get(QNetworkRequest(QUrl("http://"+mIp+"/requests/status.xml?command=in_enqueue&input=" + mCurrentElement.path.replace("\\", "\\\\"))));
166 }
167
168 void BrowseMainWindow::onPlay() {
169     QUrl url = QUrl("http://"+mIp+"/requests/status.xml?command=in_play");
170     url.addEncodedQueryItem(QByteArray("input"), QUrl::toPercentEncoding(mCurrentElement.path.replace("\\", "\\\\")));
171     mNetManager->get(QNetworkRequest(url));
172     //mNetManager->get(QNetworkRequest(QUrl("http://"+mIp+"/requests/status.xml?command=in_play&input=" + mCurrentElement.path.replace("\\", "\\\\"))));
173 }
174
175 void BrowseMainWindow::browseDirectory(QString dir) {
176     mResponse.clear();
177     QUrl url = QUrl("http://"+mIp+"/requests/browse.xml");
178     url.addEncodedQueryItem(QByteArray("dir"), QUrl::toPercentEncoding(dir));
179     QNetworkReply * reply = mNetManager->get(QNetworkRequest(url));
180     //QNetworkReply * reply =  mNetManager->get(QNetworkRequest(QUrl("http://"+mIp+"/requests/browse.xml?dir=" + dir.replace("&", "%26").replace("\\", "\\\\"))));
181     connect(reply,SIGNAL(readyRead()),this,SLOT(readReady()));
182     connect(reply,SIGNAL(error(QNetworkReply::NetworkError)),this,SLOT(error(QNetworkReply::NetworkError)));
183     connect(mNetManager,SIGNAL(finished(QNetworkReply *)),this,SLOT(finished(QNetworkReply *)));
184 }
185 void BrowseMainWindow::error(QNetworkReply::NetworkError code) {
186     qDebug() << code;
187 }
188 void BrowseMainWindow::readReady() {
189     QNetworkReply * reply = qobject_cast<QNetworkReply*>(sender());
190     // append to buffer
191     mResponse += reply->readAll();
192 }
193 void BrowseMainWindow::finished(QNetworkReply * reply) {
194     // now we can call parseXmlDirectory to process the full buffers
195     this->parseXmlDirectory();
196     // only interested in finished signals
197     disconnect(mNetManager,SIGNAL(finished(QNetworkReply *)),this,SLOT(finished(QNetworkReply *)));
198 }
199 void BrowseMainWindow::parseXmlDirectory() {
200     QDomDocument doc;
201     doc.setContent(this->mResponse);
202     QDomElement docElem = doc.documentElement();
203     QDomNodeList elements = docElem.elementsByTagName("element");
204     // we can sort by folders then files alphabetically by running to lists and appending them at the end
205     // vlc alpha sorts everything in the incoming stream, we just need to seperate files from folders.
206     if (0 < elements.count()) {
207         QList<VlcBrowseElement>* files = new QList<VlcBrowseElement>();
208         int idx = 0;
209         mContents->clear();
210         do {
211             QDomNode node = elements.at(idx);
212             VlcBrowseElement* dir = new VlcBrowseElement();
213             dir->type = node.attributes().namedItem("type").nodeValue();
214             dir->size = node.attributes().namedItem("size").nodeValue().toInt();
215             dir->date = QDate::fromString(node.attributes().namedItem("date").nodeValue());
216             dir->path = node.attributes().namedItem("path").nodeValue();
217             dir->name = node.attributes().namedItem("name").nodeValue();
218             dir->extension = getExtension(dir->path, node.attributes().namedItem("extension").nodeValue());
219             ++idx;
220             if (0 != QString::compare("directory", dir->type)) {
221                 files->append(*dir);
222             }
223             else if (0 == QString::compare("..", dir->name)) {
224                 this->mContents->prepend(*dir);
225             }
226             else {
227                 this->mContents->append(*dir);
228             }
229             delete dir;
230         } while (idx < elements.count());
231         if (0 < files->count()) {
232             mContents->append(*files);
233         }
234         delete files;
235         // Update UI
236         this->updateList();
237     }
238     else {
239         // alert user of error / empty directory
240         qDebug() << "can't browse dir: " << mCurrentDir;
241 #if defined(Q_WS_S60) || defined(Q_WS_MAEMO_5)
242         QMaemo5InformationBox::information(this, tr("Directory could not be browsed!"), QMaemo5InformationBox::DefaultTimeout);
243 #endif
244     }
245     mResponse.clear();
246
247 }
248
249 QString BrowseMainWindow::getExtension(QString path, QString extension) {
250     // return extension if exists
251     if (!extension.isNull() && !extension.isEmpty()) return extension;
252     // return blank if no path
253     if (path.isNull() || path.isEmpty()) return "";
254     // otherwise extract the extension
255     int dot_pos = path.lastIndexOf('.');
256     if (0 < dot_pos) {
257         return path.right(path.length() - (dot_pos + 1));
258     }
259     else { // no dot
260         return "";
261     }
262 }
263
264 void BrowseMainWindow::writeFile(QString path, QByteArray text) {
265     QFile file(path);
266     if (!file.open(QIODevice::WriteOnly | QIODevice::Text))
267         return;
268
269     QTextStream out(&file);
270     out << text;
271 }
272
273 void BrowseMainWindow::updateList() {
274     ui->listWidget->clear();
275     int ct = this->mContents->count();
276     if (0 < ct) {
277         QIcon icon_up     = QIcon::fromTheme("filemanager_folder_up");
278         QIcon icon_folder = QIcon::fromTheme("general_folder");
279         QIcon icon_audio  = QIcon::fromTheme("general_audio_file");
280         QIcon icon_video  = QIcon::fromTheme("general_video_file");
281         QIcon icon_image  = QIcon::fromTheme("general_image");
282         QIcon icon_flash  = QIcon::fromTheme("filemanager_flash_file");
283         QIcon icon_real   = QIcon::fromTheme("filemanager_real_music");
284         QIcon icon_unknown= QIcon::fromTheme("filemanager_unknown_file");
285         for (int idx = 0; idx < ct; ++idx) {
286             VlcBrowseElement dir = mContents->at(idx);
287             QListWidgetItem* item;
288             bool item_good = false;
289             if (0 == QString::compare("directory", dir.type)) {
290                 if (0 == QString::compare("..", dir.name)) {
291                     item = new QListWidgetItem(icon_up, dir.name, ui->listWidget, 0);
292                     item_good = true;
293                 }
294                 else {
295                     item = new QListWidgetItem(icon_folder, dir.name, ui->listWidget, 0);
296                     item_good = true;
297                 }
298             }
299             else if (0 == QString::compare("file", dir.type)) {
300                 if ( 0 == QString::compare(dir.extension, "jpg")  ||
301                      0 == QString::compare(dir.extension, "jpeg") ||
302                      0 == QString::compare(dir.extension, "gif")  ||
303                      0 == QString::compare(dir.extension, "png")  ||
304                      0 == QString::compare(dir.extension, "bmp")  ) {
305                     item_good = true;
306                     item = new QListWidgetItem(icon_image, dir.name, ui->listWidget, 0); // .jpg, .jpeg, .gif, .png, .bmp
307                 }
308                 else if ( 0 == QString::compare(dir.extension, "mp3")  ||
309                           0 == QString::compare(dir.extension, "m4a")  ||
310                           0 == QString::compare(dir.extension, "ogg")  ||
311                           0 == QString::compare(dir.extension, "oga")  ||
312                           0 == QString::compare(dir.extension, "wav")  ||
313                           0 == QString::compare(dir.extension, "flac")  ) {
314                     item_good = true;
315                     item = new QListWidgetItem(icon_audio, dir.name, ui->listWidget, 0); // .mp3, .m4a, .ogg, .oga, .wav, .flac
316                 }
317                 else if ( 0 == QString::compare(dir.extension, "avi")  ||
318                           0 == QString::compare(dir.extension, "mpeg") ||
319                           0 == QString::compare(dir.extension, "mpg")  ||
320                           0 == QString::compare(dir.extension, "mov")  ||
321                           0 == QString::compare(dir.extension, "mp4")  ||
322                           0 == QString::compare(dir.extension, "m4v")  ||
323                           0 == QString::compare(dir.extension, "wmv")  ||
324                           0 == QString::compare(dir.extension, "mkv")  ||
325                           0 == QString::compare(dir.extension, "ogv")  ) {
326                     item_good = true;
327                     item = new QListWidgetItem(icon_video, dir.name, ui->listWidget, 0); // .avi, .mpg, .mpeg, .mov, .mp4, .m4v, .wmv, .mkv, .ogv
328                 }
329                 else if ( 0 == QString::compare(dir.extension, "rm")  ||
330                           0 == QString::compare(dir.extension, "ra")  ||
331                           0 == QString::compare(dir.extension, "ram")  ) {
332                     item = new QListWidgetItem(icon_real, dir.name, ui->listWidget, 0); // .ram, 'rm, 'ra
333                 }
334                 else if ( 0 == QString::compare(dir.extension, "flv")  ) {
335                     item_good = true;
336                     item = new QListWidgetItem(icon_flash, dir.name, ui->listWidget, 0); // .flv
337                 }
338                 else {
339                     if (dir.name.startsWith("Flash")) {
340                         item_good = true;
341                         item = new QListWidgetItem(icon_flash, dir.name, ui->listWidget, 0);
342                     }
343                     else {
344                         item_good = false;
345                         //item = new QListWidgetItem(icon_unknown, dir.name, ui->listWidget, 0);
346                     }
347                 }
348             }
349             if (item_good) {
350                 ui->listWidget->addItem(item);
351             }
352             // other types ignored
353         }
354     }
355 }
356
357