vcmi/lib/Filesystem/CFilesystemLoader.cpp

#include "StdInc.h"
#include "CFilesystemLoader.h"

#include "CFileInfo.h"
#include "CFileInputStream.h"

CFilesystemLoader::CFilesystemLoader()
{

}

CFilesystemLoader::CFilesystemLoader(const std::string & baseDirectory, size_t depth)
{
	open(baseDirectory, depth);
}

CFilesystemLoader::CFilesystemLoader(const CFileInfo & baseDirectory, size_t depth)
{
	open(baseDirectory.getName(), depth);
}

void CFilesystemLoader::open(const std::string & baseDirectory, size_t depth)
{
	// Indexes all files in the directory and store them
	this->baseDirectory = baseDirectory;
	CFileInfo directory(baseDirectory);
	std::unique_ptr<std::list<CFileInfo> > fileList = directory.listFiles(depth);
	if(fileList)
	{
		this->fileList = std::move(*fileList);
	}
}

std::unique_ptr<CInputStream> CFilesystemLoader::load(const std::string & resourceName) const
{
	std::unique_ptr<CInputStream> stream(new CFileInputStream(getOrigin() + '/' + resourceName));
	return stream;
}

bool CFilesystemLoader::existsEntry(const std::string & resourceName) const
{
	for(auto it = fileList.begin(); it != fileList.end(); ++it)
	{
		if(it->getName() == resourceName)
		{
			return true;
		}
	}

	return false;
}

std::list<std::string> CFilesystemLoader::getEntries() const
{
	std::list<std::string> retList;

	for(auto it = fileList.begin(); it != fileList.end(); ++it)
	{
		retList.push_back(it->getName());
	}

	return std::move(retList);
}

std::string CFilesystemLoader::getOrigin() const
{
	return baseDirectory;
}