02c6985201
- Introduced `RomLoadOptions` struct to manage various loading configurations for ROM files, including options for stripping headers, populating metadata, and loading Zelda 3 content. - Updated `Rom::LoadFromFile` and `Rom::LoadFromData` methods to accept `RomLoadOptions`, allowing for more flexible ROM loading behavior. - Implemented `MaybeStripSmcHeader` function to conditionally remove SMC headers from ROM data. - Added new command handler `RomInfo` to display basic ROM information, including title and size. - Created `ProposalRegistry` class to manage agent-generated proposals, including creation, logging, and status updates. - Enhanced CLI commands to support proposal listing and detailed diff viewing, improving user interaction with agent-generated modifications. - Updated resource catalog to include new actions for ROM info and agent proposal management.
135 lines
3.9 KiB
C++
135 lines
3.9 KiB
C++
#include "cli/z3ed.h"
|
|
#include "absl/flags/flag.h"
|
|
#include "absl/flags/declare.h"
|
|
#include "absl/strings/str_format.h"
|
|
|
|
ABSL_DECLARE_FLAG(std::string, rom);
|
|
|
|
namespace yaze {
|
|
namespace cli {
|
|
|
|
absl::Status RomInfo::Run(const std::vector<std::string>& arg_vec) {
|
|
std::string rom_file = absl::GetFlag(FLAGS_rom);
|
|
if (rom_file.empty()) {
|
|
return absl::InvalidArgumentError("ROM file must be provided via --rom flag.");
|
|
}
|
|
|
|
RETURN_IF_ERROR(rom_.LoadFromFile(rom_file, RomLoadOptions::CliDefaults()));
|
|
if (!rom_.is_loaded()) {
|
|
return absl::AbortedError("Failed to load ROM.");
|
|
}
|
|
|
|
std::cout << "ROM Information:" << std::endl;
|
|
std::cout << " Title: " << rom_.title() << std::endl;
|
|
std::cout << " Size: 0x" << std::hex << rom_.size() << " bytes" << std::endl;
|
|
std::cout << " Filename: " << rom_file << std::endl;
|
|
|
|
return absl::OkStatus();
|
|
}
|
|
|
|
absl::Status RomValidate::Run(const std::vector<std::string>& arg_vec) {
|
|
std::string rom_file = absl::GetFlag(FLAGS_rom);
|
|
if (rom_file.empty()) {
|
|
return absl::InvalidArgumentError("ROM file must be provided via --rom flag.");
|
|
}
|
|
|
|
RETURN_IF_ERROR(rom_.LoadFromFile(rom_file, RomLoadOptions::CliDefaults()));
|
|
if (!rom_.is_loaded()) {
|
|
return absl::AbortedError("Failed to load ROM.");
|
|
}
|
|
|
|
bool all_ok = true;
|
|
std::cout << "Validating ROM: " << rom_file << std::endl;
|
|
|
|
// Checksum validation
|
|
std::cout << " - Verifying checksum... " << std::flush;
|
|
// Basic ROM validation - check if ROM is loaded and has reasonable size
|
|
if (rom_.is_loaded() && rom_.size() > 0) {
|
|
std::cout << "✅ PASSED" << std::endl;
|
|
} else {
|
|
std::cout << "❌ FAILED" << std::endl;
|
|
all_ok = false;
|
|
}
|
|
|
|
// Header validation
|
|
std::cout << " - Verifying header... " << std::flush;
|
|
if (rom_.title() == "THE LEGEND OF ZELDA") {
|
|
std::cout << "✅ PASSED" << std::endl;
|
|
} else {
|
|
std::cout << "❌ FAILED (Invalid title: " << rom_.title() << ")" << std::endl;
|
|
all_ok = false;
|
|
}
|
|
|
|
std::cout << std::endl;
|
|
if (all_ok) {
|
|
std::cout << "✅ ROM validation successful." << std::endl;
|
|
} else {
|
|
std::cout << "❌ ROM validation failed." << std::endl;
|
|
}
|
|
|
|
return absl::OkStatus();
|
|
}
|
|
|
|
absl::Status RomDiff::Run(const std::vector<std::string>& arg_vec) {
|
|
if (arg_vec.size() < 2) {
|
|
return absl::InvalidArgumentError("Usage: rom diff <rom_a> <rom_b>");
|
|
}
|
|
|
|
Rom rom_a;
|
|
auto status_a = rom_a.LoadFromFile(arg_vec[0], RomLoadOptions::CliDefaults());
|
|
if (!status_a.ok()) {
|
|
return status_a;
|
|
}
|
|
|
|
Rom rom_b;
|
|
auto status_b = rom_b.LoadFromFile(arg_vec[1], RomLoadOptions::CliDefaults());
|
|
if (!status_b.ok()) {
|
|
return status_b;
|
|
}
|
|
|
|
if (rom_a.size() != rom_b.size()) {
|
|
std::cout << "ROMs have different sizes: " << rom_a.size() << " vs " << rom_b.size() << std::endl;
|
|
}
|
|
|
|
int differences = 0;
|
|
for (size_t i = 0; i < rom_a.size(); ++i) {
|
|
if (rom_a.vector()[i] != rom_b.vector()[i]) {
|
|
differences++;
|
|
std::cout << absl::StrFormat("Difference at 0x%08X: 0x%02X vs 0x%02X\n", i, rom_a.vector()[i], rom_b.vector()[i]);
|
|
}
|
|
}
|
|
|
|
if (differences == 0) {
|
|
std::cout << "ROMs are identical." << std::endl;
|
|
} else {
|
|
std::cout << "Found " << differences << " differences." << std::endl;
|
|
}
|
|
|
|
return absl::OkStatus();
|
|
}
|
|
|
|
absl::Status RomGenerateGolden::Run(const std::vector<std::string>& arg_vec) {
|
|
if (arg_vec.size() < 2) {
|
|
return absl::InvalidArgumentError("Usage: rom generate-golden <rom_file> <golden_file>");
|
|
}
|
|
|
|
Rom rom;
|
|
auto status = rom.LoadFromFile(arg_vec[0], RomLoadOptions::CliDefaults());
|
|
if (!status.ok()) {
|
|
return status;
|
|
}
|
|
|
|
std::ofstream file(arg_vec[1], std::ios::binary);
|
|
if (!file.is_open()) {
|
|
return absl::NotFoundError("Could not open file for writing.");
|
|
}
|
|
|
|
file.write(reinterpret_cast<const char*>(rom.vector().data()), rom.size());
|
|
|
|
std::cout << "Successfully generated golden file: " << arg_vec[1] << std::endl;
|
|
|
|
return absl::OkStatus();
|
|
}
|
|
|
|
} // namespace cli
|
|
} // namespace yaze
|