Version bump
[someplayer] / src / mediascanner.cpp
index 1f9adf0..3d1017f 100644 (file)
@@ -1,24 +1,69 @@
+/*
+ * SomePlayer - An alternate music player for Maemo 5
+ * Copyright (C) 2010 Nikolay (somebody) Tischenko <niktischenko@gmail.com>
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License
+ * as published by the Free Software Foundation; either version 2
+ * of the License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+ */
+
 #include "mediascanner.h"
 
 using namespace SomePlayer::Storage;
 
 #include <QMap>
+#include <QDir>
 
 MediaScanner::MediaScanner(QObject *parent) :
                QThread(parent), _stopped(false), _initialized(false)
 {
-       REGISTERED_FILE_EXTENSIONS << "mp3" << "flac" << "wma" << "acc";
-       _iterator = NULL;
+       REGISTERED_FILE_EXTENSIONS << "mp3" << "flac" << "wma" << "aac" << "ogg" << "asf" << "ape" << "wav" << "m4a";
 }
 
 void MediaScanner::run() {
        if (!_initialized)
                return;
        _foundMedia.clear();
-       while(!_stopped && _iterator->hasNext()) {
-               QString entry(_iterator->next());
-               QFileInfo info(entry);
-               if (info.isReadable()) {
+       _scan_directory(_dir);
+       emit scanFinish(_foundMedia);
+       _stopped = true;
+}
+
+QStringList MediaScanner::singleScan(QString path) {
+       _dir = path;
+       _foundMedia.clear();
+       return _scan_directory(_dir);
+}
+
+QStringList MediaScanner::scanForPlaylists(QString path) {
+       QStringList extensions_backup(REGISTERED_FILE_EXTENSIONS);
+       REGISTERED_FILE_EXTENSIONS.clear();
+       REGISTERED_FILE_EXTENSIONS << "m3u";
+       _dir = path;
+       _foundMedia.clear();
+       QStringList result = _scan_directory(_dir);
+       REGISTERED_FILE_EXTENSIONS.clear();
+       REGISTERED_FILE_EXTENSIONS.append(extensions_backup);
+       return result;
+}
+
+QStringList MediaScanner::_scan_directory(QDir dir) {
+       QFileInfoList items = dir.entryInfoList(QDir::Files | QDir::Dirs | QDir::NoDotAndDotDot | QDir::Hidden);
+       foreach (QFileInfo info, items) {
+               if (info.isDir()) {
+                       QDir ndir(info.absoluteFilePath());
+                       _scan_directory(ndir);
+               } else {
                        QString suffix = info.suffix().toLower();
                        if (REGISTERED_FILE_EXTENSIONS.contains(suffix)) {
                                if (!_foundMedia.contains(info.absoluteFilePath()))
@@ -26,8 +71,7 @@ void MediaScanner::run() {
                        }
                }
        }
-       emit scanFinish(_foundMedia);
-       _stopped = true;
+       return _foundMedia;
 }
 
 void MediaScanner::stop() {
@@ -38,7 +82,5 @@ void MediaScanner::stop() {
 void MediaScanner::init(QString dir) {
        _stopped = false;
        _initialized = true;
-       if (!_iterator)
-               delete _iterator;
-       _iterator = new QDirIterator(QDir(dir), QDirIterator::Subdirectories | QDirIterator::FollowSymlinks);
+       _dir = dir;
 }