136 lines
4.1 KiB
C++
136 lines
4.1 KiB
C++
/*
|
|
* Copyright 2016 - 2019 gtalent2@gmail.com
|
|
*
|
|
* 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 http://mozilla.org/MPL/2.0/.
|
|
*/
|
|
|
|
#include <iostream>
|
|
#include <string_view>
|
|
#include <vector>
|
|
|
|
#include <ox/claw/read.hpp>
|
|
|
|
#include "pack.hpp"
|
|
|
|
namespace nostalgia {
|
|
|
|
namespace {
|
|
|
|
[[nodiscard]] static constexpr bool endsWith(std::string_view str, std::string_view ending) {
|
|
return str.size() >= ending.size() && str.substr(str.size() - ending.size()) == ending;
|
|
}
|
|
|
|
static_assert(endsWith("asdf", "df"));
|
|
static_assert(!endsWith("asdf", "awefawe"));
|
|
static_assert(!endsWith("asdf", "eu"));
|
|
|
|
/**
|
|
* Convert path references to inodes to save space
|
|
* @param buff buffer holding file
|
|
* @return error
|
|
* stub for now
|
|
*/
|
|
ox::Error pathToInode(std::vector<uint8_t>*) {
|
|
return OxError(0);
|
|
}
|
|
|
|
// stub for now
|
|
ox::Error toMetalClaw(std::vector<uint8_t> *buff) {
|
|
auto [mc, err] = ox::stripClawHeader(ox::bit_cast<char*>(buff->data()), buff->size());
|
|
oxReturnError(err);
|
|
buff->resize(mc.size());
|
|
ox_memcpy(buff->data(), mc.data(), mc.size());
|
|
return OxError(0);
|
|
}
|
|
|
|
// claw file transformations are broken out because path to inode
|
|
// transformations need to be done after the copy to the new FS is complete
|
|
ox::Error transformClaw(ox::FileSystem32 *dest, std::string path) {
|
|
// copy
|
|
oxTrace("pack::transformClaw") << "path:" << path.c_str();
|
|
return dest->ls(path.c_str(), [dest, path](const char *name, ox::InodeId_t) {
|
|
auto filePath = path + name;
|
|
auto [stat, err] = dest->stat(filePath.c_str());
|
|
oxReturnError(err);
|
|
if (stat.fileType == ox::FileType_Directory) {
|
|
const auto dir = path + name + '/';
|
|
oxReturnError(transformClaw(dest, dir));
|
|
} else {
|
|
// do transforms
|
|
if (endsWith(name, ".ng") || endsWith(name, ".npal")) {
|
|
// load file
|
|
std::vector<uint8_t> buff(stat.size);
|
|
oxReturnError(dest->read(filePath.c_str(), buff.data(), buff.size()));
|
|
// do transformations
|
|
oxReturnError(pathToInode(&buff));
|
|
oxReturnError(toMetalClaw(&buff));
|
|
// write file to dest
|
|
oxReturnError(dest->write(filePath.c_str(), buff.data(), buff.size()));
|
|
}
|
|
}
|
|
return OxError(0);
|
|
});
|
|
}
|
|
|
|
ox::Error verifyFile(ox::FileSystem32 *fs, const std::string &path, const std::vector<uint8_t> &expected) noexcept {
|
|
std::vector<uint8_t> buff(expected.size());
|
|
oxReturnError(fs->read(path.c_str(), buff.data(), buff.size()));
|
|
return OxError(buff == expected ? 0 : 1);
|
|
}
|
|
|
|
struct VerificationPair {
|
|
std::string path;
|
|
std::vector<uint8_t> buff;
|
|
};
|
|
|
|
ox::Error copy(ox::PassThroughFS *src, ox::FileSystem32 *dest, std::string path) {
|
|
std::cout << "copying directory: " << path << '\n';
|
|
std::vector<VerificationPair> verficationPairs;
|
|
// copy
|
|
oxReturnError(src->ls(path.c_str(), [&verficationPairs, src, dest, path](std::string name, ox::InodeId_t) {
|
|
auto currentFile = path + name;
|
|
if (currentFile == "/.nostalgia") {
|
|
return OxError(0);
|
|
}
|
|
std::cout << "reading " << name << '\n';
|
|
auto [stat, err] = src->stat((currentFile).c_str());
|
|
oxReturnError(err);
|
|
if (stat.fileType == ox::FileType_Directory) {
|
|
oxReturnError(dest->mkdir(currentFile.c_str(), true));
|
|
oxReturnError(copy(src, dest, currentFile + '/'));
|
|
} else {
|
|
std::vector<uint8_t> buff;
|
|
// do transforms
|
|
//const std::string OldExt = path.substr(path.find_last_of('.'));
|
|
// load file
|
|
buff.resize(stat.size);
|
|
oxReturnError(src->read(currentFile.c_str(), buff.data(), buff.size()));
|
|
// write file to dest
|
|
std::cout << "writing " << currentFile << '\n';
|
|
oxReturnError(dest->write(currentFile.c_str(), buff.data(), buff.size()));
|
|
oxReturnError(verifyFile(dest, currentFile, buff));
|
|
verficationPairs.push_back({currentFile, buff});
|
|
}
|
|
return OxError(0);
|
|
}));
|
|
|
|
// verify all at once in addition to right after the files are written
|
|
for (auto v : verficationPairs) {
|
|
oxReturnError(verifyFile(dest, v.path, v.buff));
|
|
}
|
|
|
|
return OxError(0);
|
|
}
|
|
|
|
}
|
|
|
|
ox::Error pack(ox::PassThroughFS *src, ox::FileSystem32 *dest) {
|
|
oxReturnError(copy(src, dest, "/"));
|
|
oxReturnError(transformClaw(dest, "/"));
|
|
return OxError(0);
|
|
}
|
|
|
|
}
|