vcmi/lib/Filesystem/CFilesystemLoader.cpp

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

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

CFilesystemLoader::CFilesystemLoader(const std::string & baseDirectory, size_t depth, bool initial):
    baseDirectory(baseDirectory),
    fileList(listFiles(depth, initial))
{
}

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->second == resourceName)
		{
			return true;
		}
	}

	return false;
}

boost::unordered_map<ResourceID, std::string> CFilesystemLoader::getEntries() const
{
	return fileList;
}

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

bool CFilesystemLoader::createEntry(std::string filename)
{
	ResourceID res(filename);
	if (fileList.find(res) != fileList.end())
		return false;

	fileList[res] = filename;
	std::ofstream newfile (baseDirectory + "/" + filename);
	return true;
}


boost::unordered_map<ResourceID, std::string> CFilesystemLoader::listFiles(size_t depth, bool initial) const
{
	std::set<EResType::Type> initialTypes;
	initialTypes.insert(EResType::DIRECTORY);
	initialTypes.insert(EResType::TEXT);
	initialTypes.insert(EResType::ARCHIVE_LOD);
	initialTypes.insert(EResType::ARCHIVE_VID);
	initialTypes.insert(EResType::ARCHIVE_SND);

	assert(boost::filesystem::is_directory(baseDirectory));
	boost::unordered_map<ResourceID, std::string> fileList;

	std::vector<std::string> path;//vector holding relative path to our file

	boost::filesystem::recursive_directory_iterator enddir;
	boost::filesystem::recursive_directory_iterator it(baseDirectory, boost::filesystem::symlink_option::recurse);

	for(; it != enddir; ++it)
	{
		EResType::Type type;

		if (boost::filesystem::is_directory(it->status()))
		{
			path.resize(it.level()+1);
			path.back() = it->path().leaf().string();
			it.no_push(depth <= it.level()); // don't iterate into directory if depth limit reached

			type = EResType::DIRECTORY;
		}
		else
			type = EResTypeHelper::getTypeFromExtension(boost::filesystem::extension(*it));

		if (!initial || vstd::contains(initialTypes, type))
		{
			//reconstruct relative filename (not possible via boost AFAIK)
			std::string filename;
			for (size_t i=0; i<it.level() && i<path.size(); i++)
				filename += path[i] + '/';
			filename += it->path().leaf().string();

			fileList[ResourceID(filename, type)] = filename;
		}
	}

	return fileList;
}