86 lines
2.4 KiB
C++
86 lines
2.4 KiB
C++
/*
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at https://mozilla.org/MPL/2.0/.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "IO/FileDescription.hpp"
|
|
#include <vector>
|
|
|
|
namespace OpenVulkano
|
|
{
|
|
class IArchiveWriter
|
|
{
|
|
protected:
|
|
IArchiveWriter() = default;
|
|
|
|
public:
|
|
virtual ~IArchiveWriter() = default;
|
|
|
|
virtual bool AddFile(const FileDescription& description, const void* buffer) = 0;
|
|
virtual bool AddFile(const FileDescription& description, const std::vector<std::pair<const void*, size_t>>& buffers) = 0;
|
|
virtual bool AddFile(const char* fileName, const char* inArchiveName) = 0;
|
|
|
|
bool AddFile(const char* filePath)
|
|
{
|
|
std::filesystem::path path(filePath);
|
|
std::string fileName = path.filename().string();
|
|
if (std::filesystem::is_regular_file(path))
|
|
{
|
|
return AddFile(filePath, fileName.c_str());
|
|
}
|
|
else if (std::filesystem::is_directory(path))
|
|
{
|
|
AddFiles(filePath, fileName.c_str());
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool AddFile(const char* inArchiveName, const void* buffer, size_t length)
|
|
{
|
|
FileDescription description = FileDescription::MakeDescriptionForFile(inArchiveName, length);
|
|
return AddFile(description, buffer);
|
|
}
|
|
|
|
bool AddFile(const char* fileName, const std::vector<std::pair<const void*, size_t>>& buffers)
|
|
{
|
|
size_t size = 0;
|
|
for(const auto& buffer : buffers)
|
|
{
|
|
if (buffer.first == nullptr) continue;
|
|
size += buffer.second;
|
|
}
|
|
FileDescription description = FileDescription::MakeDescriptionForFile(fileName, size);
|
|
return AddFile(description, buffers);
|
|
}
|
|
|
|
bool AddFiles(const char* dirName)
|
|
{
|
|
return AddFiles(dirName, "");
|
|
}
|
|
|
|
bool AddFiles(const char* dirName, const char* inArchiveDirName)
|
|
{
|
|
return AddFiles(std::filesystem::path(dirName), std::string(inArchiveDirName));
|
|
}
|
|
|
|
bool AddFiles(const std::filesystem::path& dirName, const std::string& inArchiveDirName)
|
|
{
|
|
std::string sDirName = dirName.string();
|
|
AddFile(sDirName.c_str(), inArchiveDirName.c_str());
|
|
for(const auto& entry : std::filesystem::directory_iterator(dirName))
|
|
{
|
|
std::string fPath = inArchiveDirName + "/" + entry.path().filename().string();
|
|
if (entry.is_directory()) AddFiles(entry, fPath);
|
|
else
|
|
{
|
|
std::string entryName = entry.path().string();
|
|
AddFile(entryName.c_str(), fPath.c_str());
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
};
|
|
} |