gmenu2x/src/filelister.cpp

/***************************************************************************
 *   Copyright (C) 2006 by Massimiliano Torromeo   *
 *   massimiliano.torromeo@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.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/

//for browsing the filesystem
#include <sys/stat.h>
#include <sys/types.h>
#include <dirent.h>
#include <errno.h>
#include <iostream>

#include "filelister.h"
#include "utilities.h"

using namespace std;

FileLister::FileLister(string startPath, bool showDirectories, bool showFiles) {
	this->showDirectories = showDirectories;
	this->showFiles = showFiles;
	setPath(startPath,false);
}

string FileLister::getPath() {
	return path;
}
void FileLister::setPath(string path, bool doBrowse) {
	if (path[path.length()-1]!='/') path += "/";
	this->path = path;
	if (doBrowse)
		browse();
}

string FileLister::getFilter() {
	return filter;
}
void FileLister::setFilter(string filter) {
	this->filter = filter;
}

void FileLister::browse() {
	directories.clear();
	files.clear();

	if (showDirectories || showFiles) {
		DIR *dirp;
		if ((dirp = opendir(path.c_str())) == NULL) {
			cout << "Error: opendir(" << path << ")" << endl;
			return;
		}

		vector<string> vfilter;
		split(vfilter,getFilter(),",");

		string filepath, file;
		struct stat st;
		struct dirent *dptr;

		while ((dptr = readdir(dirp))) {
			file = dptr->d_name;
			if (file[0]=='.' && file!="..") continue;
			filepath = path+file;
			int statRet = stat(filepath.c_str(), &st);
			if (statRet == -1) {
				cout << "\033[0;34mGMENU2X:\033[0;31m stat failed on '" << filepath << "' with error '" << strerror(errno) << "'\033[0m" << endl;
				continue;
			}

			if (find(exclude.begin(), exclude.end(), file) != exclude.end())
				continue;

			if (S_ISDIR(st.st_mode)) {
				if (!showDirectories) continue;
#ifdef TARGET_GP2X
//				if (!(path=="/card/" && (file!="sd" && file!="ext" && file!="nand")))
#endif
					directories.push_back(file);
			} else {
				if (!showFiles) continue;
				bool filterOk = false;
				for (uint i = 0; i<vfilter.size() && !filterOk; i++)
					if (vfilter[i].length()<=file.length())
						filterOk = file.substr(file.length()-vfilter[i].length(),vfilter[i].length())==vfilter[i];
				if (filterOk) files.push_back(file);
			}
		}

		closedir(dirp);
		sort(files.begin(),files.end(),case_less());
		sort(directories.begin(),directories.end(),case_less());
	}
}

uint FileLister::size() {
	return files.size()+directories.size();
}
uint FileLister::dirCount() {
	return directories.size();
}
uint FileLister::fileCount() {
	return files.size();
}

string FileLister::operator[](uint x) {
	return at(x);
}

string FileLister::at(uint x) {
	if (x<directories.size())
		return directories[x];
	else
		return files[x-directories.size()];
}

bool FileLister::isFile(uint x) {
	return x>=directories.size() && x<size();
}

bool FileLister::isDirectory(uint x) {
	return x<directories.size();
}