/*==LICENSE==*

CyanWorlds.com Engine - MMOG client, server and tools
Copyright (C) 2011  Cyan Worlds, Inc.

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 3 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, see <http://www.gnu.org/licenses/>.

You can contact Cyan Worlds, Inc. by email legal@cyan.com
 or by snail mail at:
      Cyan Worlds, Inc.
      14617 N Newport Hwy
      Mead, WA   99021

*==LICENSE==*/
#include "hsFiles.h"

#if HS_BUILD_FOR_UNIX

#include <errno.h>
#include <dirent.h>
#include <limits.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
#include <glob.h>
#include "hsTemplates.h"
#include "plFileUtils.h"
#include "hsStlUtils.h"

struct hsFolderIterator_Data {
    glob_t fGlobBuf;
    bool fInited;
    int fCnt;
    hsFolderIterator_Data() : fInited(false), fCnt(0) {}
    // ~hsFolderIterator_Data() { fInited=false; globfree(&fData->fGlobBuf); }
};

hsFolderIterator::hsFolderIterator(const char path[], bool)
{
    fData = TRACKED_NEW hsFolderIterator_Data;

    this->SetPath(path);
}

hsFolderIterator::~hsFolderIterator()
{
    this->Reset();
    delete fData;
}

void hsFolderIterator::SetPath(const char path[])
{
    fPath[0] = 0;
    if (path)
    {
        ::strcpy(fPath, path);
    }
    this->Reset();
}

void hsFolderIterator::Reset()
{
    if (fData->fInited)
    {
        globfree(&fData->fGlobBuf);
        fData->fCnt = 0;
        fData->fInited=false;
    }
}
hsBool hsFolderIterator::NextFile()
{
    if (fData->fInited == false)
    {   
        std::string path=fPath;
        if(!(strchr(fPath,'*')  || strchr(fPath,'?')  || strchr(fPath,'[')))
        {
            if (fPath[strlen(fPath)-1] != PATH_SEPARATOR)
                path = path + PATH_SEPARATOR_STR + "*";
            else
                path = path + "*";
        }

        if(glob(path.c_str(), 0, NULL, &fData->fGlobBuf) != 0 ) {
            return false;
        }
        fData->fInited=true;
        fData->fCnt = 0;
    }

    return fData->fCnt++ < fData->fGlobBuf.gl_pathc;
}

const char* hsFolderIterator::GetFileName() const
{
    if (!fData->fInited || fData->fCnt > fData->fGlobBuf.gl_pathc)
        throw "end of folder";

    const char* fn=fData->fGlobBuf.gl_pathv[fData->fCnt-1];
    return plFileUtils::GetFileName(fn);
}

hsBool  hsFolderIterator::IsDirectory( void ) const
{
    // rob, please forgive me, this is my best attempt...
    if(fData->fCnt > fData->fGlobBuf.gl_pathc )
        return false;

    struct stat info;
    const char* fn=fData->fGlobBuf.gl_pathv[fData->fCnt-1];
    if( stat( fn, &info ) )
    {
        printf("Error calling stat(): %s errno=%d\n", strerror(errno), errno);
        return false;
    }
    return ( info.st_mode & S_IFDIR ) ? true : false;
}

#endif