#include "ofxDirList.h"

//----------------------------------------------------------
ofxDirList::ofxDirList()
{

	allowedFileExt = new char*[OF_DL_MAX_EXTS];
	nameArray = new char*[OF_DL_MAX_RESULTS];
	pathArray = new char*[OF_DL_MAX_RESULTS];

	for(int k = 0; k < OF_DL_MAX_EXTS; k++)
		allowedFileExt[k] = new char[OF_DL_EXT_SIZE];

	for(int k = 0; k < OF_DL_MAX_RESULTS; k++){
		nameArray[k] = new char[OF_DL_FILELEN];
		pathArray[k] = new char[OF_DL_FILELEN];
	}
	//an allowCount of 0 means no extensions are to be checked,
	allowCount  = 0;
	verbose		= true;

}

//----------------------------------------------------------
void ofxDirList::reset(){
	allowCount  = 0;
}

//----------------------------------------------------------
void ofxDirList::setVerbose(bool _verbose)
{
	verbose = _verbose;
}

//----------------------------------------------------------
bool ofxDirList::allowExt(string ext)
{

	if(allowCount >= OF_DL_MAX_EXTS -1){if(verbose)printf("ofxDirList - no of extensions > OF_DL_MAX_EXTS\n");   return false; }
	if(ext.length() >= OF_DL_EXT_SIZE){if(verbose)printf("ofxDirList - extension is > OF_DL_EXT_SIZE\n");   return false; }

	sprintf(allowedFileExt[allowCount], "%s", (char *)ext.c_str());

	allowCount++;

	return true;
}

//----------------------------------------------------------
string ofxDirList::getName(int pos)
{
	if(pos >= count) return 0;
	return string(nameArray[pos]);
}

//----------------------------------------------------------
string ofxDirList::getPath(int pos)
{
	if(pos >= count) return 0;
	return string(pathArray[pos]);
}

//----------------------------------------------------------
int ofxDirList::listDir(string directory)
{
	directory = ofToDataPath(directory);

	DIR *dir;
	char *extension;
	struct dirent *entry;
	bool skip = false;

	//if no dir is specified look in same folder as executable
	if(directory.length() == 0) 	dir = opendir(".");
	else dir = opendir(directory.c_str());

	if (dir == NULL){
		if(verbose)printf("ofxDirList - error opening directory\n");
		return 0;
	}

	count = 0;
	while ((entry = readdir(dir)) != NULL && count < OF_DL_MAX_RESULTS)
	{

		//make sure we are not looking at ./  or ../ on unix based systems
		char * pch;
		pch = (char*) memchr ((char *)entry->d_name, '.', 1);
		if(pch != NULL) continue;

		//by default we don't skip files unless we are checking extensions
		skip = false;

		if(allowCount > 0)
		{
			//we will skip this files unless it has an allowed extension
			skip = true;
			for(int i = 0; i < allowCount; i++)
			{
				//if the wildecard * (42) has been entered for an ext type then don't check any extensions
				if( allowedFileExt[i][0] == 42){ skip = false; break; }

				//get the lengths of both the filename and the extention
				int extLen  = strlen(allowedFileExt[i]);
				int fileLen = strlen(entry->d_name);

				//the extension has to be shorter than the filename
				if(extLen >= fileLen) continue;

				//lets check if there is a fullstop/peroid where we expect it to be
				if(entry->d_name[fileLen-extLen-1] != '.') continue;

				//so now lets see if the extensions match
				bool checkExt = true;
				for(int j = 0; j < extLen; j++)
				{
					if( tolower(entry->d_name[fileLen-j]) != tolower(allowedFileExt[i][extLen-j]) ){checkExt = false; break; }
				}

				//they do match so we don't skip this entry
				if(checkExt == true){ skip = false; break;}
			}
		}

		if(skip) continue;

		//if the file is too long for our string
		if(strlen(entry->d_name) >= OF_DL_FILELEN)
		{
			if(verbose)printf("ofxDirList - file name > OF_DL_FILELEN\n");
			return 0;
		}

		//Otherwise lets copy it to our results list


		//MAKE OUR PATH ARRAY
		//lets check that the full path can fit in the char * we have for it
		if(OF_DL_FILELEN > strlen(entry->d_name) + directory.length() )
		{
			//if some muppet forgot the trailing slash lets add it for them
			if(directory.substr(directory.length()-1) != "/") sprintf(pathArray[count],"%s/%s",directory.c_str(),entry->d_name);
			else  sprintf(pathArray[count],"%s%s",directory.c_str(),entry->d_name);
		}
		else
		{
			if(verbose)printf("ofxDirList - path %s too long for string",entry->d_name);
			continue;
		}

		//MAKE OUR NAME ARRAY
		//lets check that the filename can fit in the char * we have for it
		if(OF_DL_FILELEN > strlen(entry->d_name) )sprintf(nameArray[count],entry->d_name);
		else
		{
			if(verbose)printf("ofxDirList - path %s too long for string",entry->d_name);
			continue;
		}


		if(verbose)printf("ofxDirList - listing %s \n", nameArray[count]);
		count++;
	}

	if(verbose)printf("ofxDirList - listed %i files in %s\n", count, directory.c_str());
	return count;
}

//Clean up
//----------------------------------------------------------
ofxDirList::~ofxDirList()
{
	delete pathArray;
	delete nameArray;
	delete allowedFileExt;
}