jasper/src/olympic/keel/src/pack-applib.cpp

/*
 * Copyright 2016 - 2024 Gary Talent (gary@drinkingtea.net). All rights reserved.
 */

#include <fstream>

#include <ox/clargs/clargs.hpp>
#include <ox/fs/fs.hpp>
#include <ox/logconn/def.hpp>
#include <ox/logconn/logconn.hpp>
#include <ox/oc/write.hpp>

#include <keel/keel.hpp>

static ox::Error writeFileBuff(ox::StringView path, ox::BufferView const buff) noexcept {
	try {
		std::ofstream f(std::string(toStdStringView(path)), std::ios::binary);
		f.write(buff.data(), static_cast<intptr_t>(buff.size()));
	} catch (std::fstream::failure const&) {
		return ox::Error(2, "failed to write file");
	}
	return {};
}

static ox::Result<ox::Buffer> readFileBuff(ox::StringView path) noexcept {
	std::ifstream file(std::string(toStdStringView(path)), std::ios::binary | std::ios::ate);
	if (!file.good()) {
		oxErrorf("Could not find OxFS file: {}", path);
		return ox::Error(1, "Could not find OxFS file");
	}
	try {
		auto const size = static_cast<std::size_t>(file.tellg());
		ox::Buffer buff(size);
		file.seekg(0, std::ios::beg);
		file.read(buff.data(), static_cast<std::streamsize>(buff.size()));
		return buff;
	} catch (std::ios_base::failure const&e) {
		oxErrorf("Could not read OxFS file: {}", e.what());
		return ox::Error(2, "Could not read OxFS file");
	}
}

static ox::Error generateTypes(ox::TypeStore &ts) noexcept {
	for (auto const mod : keel::modules()) {
		for (auto gen : mod->types()) {
			OX_RETURN_ERROR(gen(ts));
		}
	}
	return {};
}

static ox::Error pack(
		ox::StringView argSrc,
		ox::StringView argRomBin,
		ox::StringView argManifest,
		ox::StringView projectDataDir) noexcept {
	ox::Buffer dstBuff(32 * ox::units::MB);
	OX_RETURN_ERROR(ox::FileSystem32::format(dstBuff.data(), dstBuff.size()));
	ox::FileSystem32 dst(dstBuff);
	OX_REQUIRE(ctx, keel::init(ox::make_unique<ox::PassThroughFS>(argSrc), "keel-pack"));
	keel::TypeStore ts(*ctx->rom, ox::sfmt("{}/type_descriptors", projectDataDir));
	OX_RETURN_ERROR(generateTypes(ts));
	keel::Manifest manifest;
	OX_RETURN_ERROR(keel::pack(manifest, *ctx, ts, dst));
	OX_REQUIRE_M(pl, keel::GbaPreloader::make());
	OX_RETURN_ERROR(preload(manifest, ts, dst, *pl));
	OX_RETURN_ERROR(dst.resize());
	// resize buffer
	OX_REQUIRE(dstSize, dst.size());
	dstBuff.resize(dstSize);
	// concatenate ROM segments
	OX_REQUIRE_M(romBuff, readFileBuff(argRomBin));
	oxOutf("Input exe size: {} bytes\n", romBuff.size());
	oxOutf("Dest FS size: {} bytes\n", dstSize);
	oxOutf("Preload buff size: {} bytes\n", pl->buff().size());
	OX_RETURN_ERROR(appendBinary(romBuff, dstBuff, *pl));
	oxOutf("Final ROM buff size: {} bytes\n", romBuff.size());
	OX_RETURN_ERROR(writeFileBuff(argRomBin, romBuff));
	OX_REQUIRE(manifestJson, ox::writeOCString(manifest));
	OX_RETURN_ERROR(writeFileBuff(argManifest, {manifestJson.data(), manifestJson.len()}));
	return {};
}

ox::Error run(
		[[maybe_unused]] ox::StringView project,
		[[maybe_unused]] ox::StringView appName,
		ox::StringView projectDataDir,
		ox::SpanView<ox::CString> argv) noexcept {
	ox::ClArgs const args(argv);
	auto const argSrc = args.getString("src", "");
	auto const argRomBin = args.getString("rom-bin", "");
	auto const argManifest = args.getString("manifest", "");
	if (argSrc == "") {
		oxErr("\033[31;1;1merror:\033[0m must specify a source directory\n");
		return ox::Error(1, "must specify a source directory");
	}
	if (argRomBin == "") {
		oxErr("\033[31;1;1merror:\033[0m must specify a path for ROM file\n");
		return ox::Error(1, "must specify a path for preload file");
	}
	return pack(argSrc, argRomBin, argManifest, projectDataDir);
}