Files
ox/src/nostalgia/tools/pack/pack.cpp
T
2021-04-17 17:55:47 -05:00

117 lines
3.6 KiB
C++

/*
* Copyright 2016 - 2021 gary@drinkingtea.net
*
* 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 <ox/claw/read.hpp>
#include "pack.hpp"
namespace nostalgia {
/**
* Convert path references to inodes to save space
* @param buff buffer holding file
* @return error
* stub for now
*/
static ox::Error pathToInode(ox::Vector<uint8_t>*) noexcept {
return OxError(0);
}
// just strip header for now...
static ox::Error toMetalClaw(ox::Vector<uint8_t> *buff) noexcept {
oxRequire(mc, ox::stripClawHeader(ox::bit_cast<char*>(buff->data()), buff->size()));
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
static ox::Error transformClaw(ox::FileSystem *dest, const ox::String &path) noexcept {
// copy
oxTracef("pack::transformClaw", "path: {}", path);
oxRequire(fileList, dest->ls(path));
for (auto i = 0u; i < fileList.size(); ++i) {
auto &name = fileList[i];
auto filePath = path + name;
oxRequire(stat, dest->stat(filePath.c_str()));
if (stat.fileType == ox::FileType_Directory) {
const auto dir = path + name + '/';
oxReturnError(transformClaw(dest, dir));
} else {
// do transforms
if (name.endsWith(".ng") || name.endsWith(".npal")) {
// load file
ox::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);
}
static ox::Error verifyFile(ox::FileSystem *fs, const ox::String &path, const ox::Vector<uint8_t> &expected) noexcept {
ox::Vector<uint8_t> buff(expected.size());
oxReturnError(fs->read(path.c_str(), buff.data(), buff.size()));
return OxError(buff == expected ? 0 : 1);
}
struct VerificationPair {
ox::String path;
ox::Vector<uint8_t> buff;
};
static ox::Error copy(ox::FileSystem *src, ox::FileSystem *dest, const ox::String &path) noexcept {
oxOutf("copying directory: {}\n", path);
ox::Vector<VerificationPair> verificationPairs;
// copy
oxRequire(fileList, src->ls(path));
for (auto i = 0u; i < fileList.size(); ++i) {
auto &name = fileList[i];
auto currentFile = path + name;
if (currentFile == "/.nostalgia") {
continue;
}
oxOutf("reading {}\n", name);
oxRequire(stat, src->stat(currentFile.c_str()));
if (stat.fileType == ox::FileType_Directory) {
oxReturnError(dest->mkdir(currentFile.c_str(), true));
oxReturnError(copy(src, dest, currentFile + '/'));
} else {
ox::Vector<uint8_t> buff;
// load file
buff.resize(stat.size);
oxReturnError(src->read(currentFile.c_str(), buff.data(), buff.size()));
// write file to dest
oxOutf("writing {}\n", currentFile);
oxReturnError(dest->write(currentFile.c_str(), buff.data(), buff.size()));
oxReturnError(verifyFile(dest, currentFile, buff));
verificationPairs.push_back({currentFile, buff});
}
}
// verify all at once in addition to right after the files are written
for (auto i = 0u; i < verificationPairs.size(); ++i) {
const auto &v = verificationPairs[i];
oxReturnError(verifyFile(dest, v.path, v.buff));
}
return OxError(0);
}
ox::Error pack(ox::FileSystem *src, ox::FileSystem *dest) noexcept {
oxReturnError(copy(src, dest, "/"));
oxReturnError(transformClaw(dest, "/"));
return OxError(0);
}
}