7#include "absl/strings/str_cat.h"
8#include "absl/strings/str_format.h"
9#include "absl/strings/str_join.h"
18 escaped.reserve(input.size());
19 for (
char c : input) {
22 escaped.append(
"\\\\");
25 escaped.append(
"\\\"");
28 escaped.append(
"\\n");
31 escaped.append(
"\\r");
34 escaped.append(
"\\t");
44 const std::vector<std::string>& values) {
46 for (
size_t i = 0; i < values.size(); ++i) {
57 static bool initialized =
false;
66 std::unique_ptr<resources::CommandHandler> handler,
68 std::string name = handler->GetName();
74 for (
const auto& alias : metadata.
aliases) {
83 std::vector<std::unique_ptr<resources::CommandHandler>> handlers) {
84 for (
auto& handler : handlers) {
85 std::string name = handler->GetName();
92 auto descriptor = handler->Describe();
95 metadata.
usage = handler->GetUsage();
101 if (name.find(
"resource-") == 0) {
103 metadata.
description =
"Resource inspection and search";
104 if (name ==
"resource-list") {
105 metadata.
examples = {
"z3ed resource-list --type=dungeon --format=json",
106 "z3ed resource-list --type=sprite --format=table"};
108 }
else if (name.find(
"dungeon-") == 0) {
110 metadata.
description =
"Dungeon inspection and editing";
111 if (name ==
"dungeon-describe-room") {
113 "z3ed dungeon-describe-room --room=5 --format=json"};
114 }
else if (name ==
"dungeon-stream-plan") {
116 "Read-only stream inventory, alias/overlap, and free-space "
117 "diagnostics; replacement-aware immutable move output is pending";
119 "z3ed dungeon-stream-plan --kind=objects "
120 "--manifest=hack_manifest.json --rom=game.sfc --format=json",
121 "z3ed dungeon-stream-plan --kind=pot_items "
122 "--manifest=hack_manifest.json --rom=game.sfc --format=text"};
123 }
else if (name ==
"dungeon-place-sprite") {
125 "Place a dungeon sprite (dry-run by default, --write to apply)";
127 "z3ed dungeon-place-sprite --room=0x77 --id=0xA3 --x=16 --y=21 "
128 "--subtype=4 --rom=Roms/oos168.sfc --format=json",
129 "z3ed dungeon-place-sprite --room=0x77 --id=0xA3 --x=16 --y=21 "
130 "--subtype=4 --write --rom=/tmp/oos-work.sfc --format=json"};
131 }
else if (name ==
"dungeon-remove-sprite") {
133 "Remove a dungeon sprite by index or exact coordinates";
135 "z3ed dungeon-remove-sprite --room=0x77 --index=2 "
136 "--rom=/tmp/oos-work.sfc --format=json",
137 "z3ed dungeon-remove-sprite --room=0x77 --x=16 --y=21 --write "
138 "--rom=/tmp/oos-work.sfc --format=json"};
139 }
else if (name ==
"dungeon-place-object") {
141 "Place a dungeon object with immutable capacity/COW preflight "
142 "(dry-run by default)";
144 "z3ed dungeon-place-object --room=0x98 --id=0x0031 --x=20 --y=20 "
145 "--size=4 --rom=/tmp/oos-work.sfc --format=json",
146 "z3ed dungeon-place-object --room=0x98 --id=0x0031 --x=20 --y=20 "
147 "--size=4 --manifest=hack_manifest.json --write "
148 "--rom=/tmp/oos-work.sfc --format=json"};
149 }
else if (name ==
"dungeon-get-palette") {
151 "Resolve a US/OOS room's full 8-bit palette-set ID to its shared "
152 "global dungeon palette and report every affected room; other "
153 "regional layouts fail closed";
155 "z3ed dungeon-get-palette --room=0xA8 "
156 "--rom=Roms/oos168.sfc --format=json",
157 "z3ed dungeon-get-palette --room=0xA8 --index=0x07 "
158 "--rom=Roms/oos168.sfc --format=json"};
159 }
else if (name ==
"dungeon-set-palette-color") {
161 "Dry-run or persist one US/OOS shared dungeon palette color with "
162 "palette-set/index/color CAS, manifest ownership, exact fence, "
163 "clean-file baseline, required backup, atomic save, and external "
166 "z3ed dungeon-set-palette-color --room=0xA8 --index=7 "
167 "--expect-palette-set=0x07 --expect-palette-index=7 "
168 "--expect-color=0x7FFF --color=0x7FFE "
169 "--manifest=Roms/hack_manifest.json "
170 "--rom=Roms/oos168.sfc --format=json",
171 "z3ed dungeon-set-palette-color --room=0xA8 --index=7 "
172 "--expect-palette-set=0x07 --expect-palette-index=7 "
173 "--expect-color=0x7FFF --color=0x7FFE "
174 "--manifest=Roms/hack_manifest.json --write "
175 "--rom=/tmp/oos-work.sfc --format=json"};
176 }
else if (name ==
"dungeon-set-door-type") {
178 "Change one existing door type with coordinate/CAS and manifest "
179 "guards (dry-run by default)";
181 "z3ed dungeon-set-door-type --room=0xA8 --x=46 --y=58 "
182 "--type=0x00 --expect-type=0x18 "
183 "--manifest=hack_manifest.json --rom=Roms/oos168.sfc "
185 "z3ed dungeon-set-door-type --room=0xA8 --x=46 --y=58 "
186 "--type=0x00 --expect-type=0x18 "
187 "--manifest=hack_manifest.json --write "
188 "--rom=/tmp/oos-work.sfc --format=json"};
189 }
else if (name ==
"dungeon-list-pot-items") {
191 "List existing pot items with raw positions and ROM addresses";
193 "z3ed dungeon-list-pot-items --room=0xA8 "
194 "--rom=Roms/oos168.sfc --format=json"};
195 }
else if (name ==
"dungeon-set-pot-item") {
197 "Change one existing pot item with index/position/item CAS and "
198 "manifest guards (dry-run by default)";
200 "z3ed dungeon-set-pot-item --room=0xA8 --index=0 "
201 "--expect-position=0x198C --expect-item=0x05 --item=0x06 "
202 "--manifest=Roms/hack_manifest.json --rom=Roms/oos168.sfc "
204 "z3ed dungeon-set-pot-item --room=0xA8 --index=0 "
205 "--expect-position=0x198C --expect-item=0x05 --item=0x06 "
206 "--manifest=Roms/hack_manifest.json --write "
207 "--rom=/tmp/oos-work.sfc --format=json"};
208 }
else if (name ==
"dungeon-oracle-preflight") {
210 "Oracle ROM safety preflight: water-fill region/table, collision "
211 "maps, and optional required-room checks";
213 "z3ed dungeon-oracle-preflight --rom oos168x.sfc --format=json",
214 "z3ed dungeon-oracle-preflight --rom oos168x.sfc "
215 "--required-water-fill-rooms=0x25,0x27 --format=json",
216 "z3ed dungeon-oracle-preflight --rom oos168x.sfc "
217 "--report=/tmp/preflight.json"};
218 }
else if (name ==
"dungeon-set-collision-tile") {
220 "Set one or more custom collision tiles in a dungeon room";
222 "z3ed dungeon-set-collision-tile --room=0xB8 "
223 "--tiles=\"10,5,0xB7;50,45,0xBA\" --rom=/tmp/oos-work.sfc "
225 "z3ed dungeon-set-collision-tile --room=0xB8 "
226 "--tiles=\"10,5,0xB7;50,45,0xBA\" --write "
227 "--rom=/tmp/oos-work.sfc --format=json"};
229 }
else if (name.find(
"overworld-") == 0) {
231 metadata.
description =
"Overworld inspection and editing";
232 if (name ==
"overworld-find-tile") {
234 "z3ed overworld-find-tile --tile=0x42 --format=json"};
236 }
else if (name.find(
"project-bundle-") == 0) {
239 if (name ==
"project-bundle-verify") {
240 metadata.
description =
"Project bundle verification";
242 "z3ed project-bundle-verify --project MyProject.yazeproj "
244 "z3ed project-bundle-verify --project project.yaze "
245 "--report report.json",
246 "z3ed project-bundle-verify --project MyProject.yazeproj "
247 "--check-rom-hash --format=json"};
248 }
else if (name ==
"project-bundle-pack") {
249 metadata.
description =
"Pack bundle into zip archive";
251 "z3ed project-bundle-pack --project MyProject.yazeproj "
252 "--out MyProject.zip --format=json",
253 "z3ed project-bundle-pack --project MyProject.yazeproj "
254 "--out MyProject.zip --overwrite"};
255 }
else if (name ==
"project-bundle-unpack") {
256 metadata.
description =
"Unpack zip archive into bundle";
258 "z3ed project-bundle-unpack --archive MyProject.zip "
259 "--out ./projects --format=json",
260 "z3ed project-bundle-unpack --archive MyProject.zip "
261 "--out ./projects --overwrite",
262 "z3ed project-bundle-unpack --archive MyProject.zip "
263 "--out ./projects --keep-partial-output",
264 "z3ed project-bundle-unpack --archive MyProject.zip "
265 "--out ./projects --dry-run"};
267 }
else if (name.find(
"rom-") == 0) {
269 metadata.
description =
"ROM inspection and validation";
270 if (name ==
"rom-info") {
271 metadata.
examples = {
"z3ed rom-info --rom=zelda3.sfc"};
272 }
else if (name ==
"rom-validate") {
273 metadata.
examples = {
"z3ed rom-validate --rom=zelda3.sfc"};
274 }
else if (name ==
"rom-doctor") {
275 metadata.
examples = {
"z3ed rom-doctor --rom=zelda3.sfc --format=json"};
276 }
else if (name ==
"rom-diff") {
278 "z3ed rom-diff --rom_a=base.sfc --rom_b=target.sfc"};
279 }
else if (name ==
"rom-compare") {
281 "z3ed rom-compare --rom=target.sfc --baseline=vanilla.sfc"};
283 }
else if (name.find(
"emulator-") == 0) {
285 metadata.
description =
"Emulator control and debugging";
287 if (name ==
"emulator-set-breakpoint") {
289 "z3ed emulator-set-breakpoint --address=0x83D7 --description='NMI "
292 }
else if (name.find(
"mesen-") == 0) {
294 metadata.
description =
"Mesen2 socket automation and introspection";
296 if (name ==
"mesen-state-capture") {
298 "z3ed mesen-state-capture --state=foo.state "
299 "--rom-file=Roms/oos168x.sfc --scenario=d6_room_88",
300 "z3ed mesen-state-capture --state=Roms/SaveStates/d6_room_88.state "
301 "--rom-file=Roms/oos168x.sfc --slot=1 "
302 "--states-dir=~/Library/Application\\ Support/Mesen2/SaveStates"};
303 }
else if (name ==
"mesen-state-hook") {
305 "z3ed mesen-state-hook --state=foo.state "
306 "--rom-file=Roms/oos168x.sfc --scenario=d6_room_88",
307 "z3ed mesen-state-hook --state=foo.state "
308 "--rom-file=Roms/oos168x.sfc --format=json"};
310 }
else if (name.find(
"gui-") == 0) {
314 }
else if (name.find(
"hex-") == 0) {
317 }
else if (name.find(
"palette-") == 0) {
320 }
else if (name.find(
"sprite-") == 0) {
323 }
else if (name.find(
"message-") == 0 || name.find(
"dialogue-") == 0) {
325 metadata.
description = name.find(
"message-") == 0 ?
"Message inspection"
326 :
"Dialogue inspection";
327 if (name ==
"message-source-sync") {
329 "Dry-run or CAS-publish a complete canonical expanded-message "
330 "bundle and deterministic no-org Asar include";
332 "z3ed message-source-sync --project=Oracle-of-Secrets.yaze "
333 "--file=/tmp/message-edits.json --format=json",
334 "z3ed message-source-sync --project=Oracle-of-Secrets.yaze "
335 "--file=/tmp/message-edits.json "
336 "--expected-source-sha256=<sha256> --write --format=json"};
338 }
else if (name.find(
"music-") == 0) {
341 }
else if (name.find(
"oracle-") == 0) {
343 metadata.
description =
"Oracle-of-Secrets project tooling";
344 if (name ==
"oracle-menu-index") {
346 "z3ed oracle-menu-index --project=/path/to/oracle-of-secrets "
348 "z3ed oracle-menu-index --table=Menu_ItemCursorPositions "
349 "--missing-bins --format=json"};
350 }
else if (name ==
"oracle-menu-set-offset") {
352 "z3ed oracle-menu-set-offset "
353 "--asm=Menu/menu_select_item.asm "
354 "--table=Menu_ItemCursorPositions --index=0 --row=7 --col=2",
355 "z3ed oracle-menu-set-offset "
356 "--asm=Menu/menu_select_item.asm "
357 "--table=Menu_ItemCursorPositions --index=0 --row=7 --col=2 "
359 }
else if (name ==
"oracle-menu-validate") {
361 "z3ed oracle-menu-validate --project=/path/to/oracle-of-secrets",
362 "z3ed oracle-menu-validate --project=/path/to/oracle-of-secrets "
363 "--strict --max-row=31 --max-col=31"};
364 }
else if (name ==
"oracle-smoke-check") {
366 "z3ed oracle-smoke-check --rom oos168.sfc --format=json",
367 "z3ed oracle-smoke-check --rom oos168.sfc --strict-readiness "
369 "z3ed oracle-smoke-check --rom oos168.sfc "
370 "--min-d6-track-rooms=4 --format=json",
371 "z3ed oracle-smoke-check --rom oos168.sfc "
372 "--report=/tmp/smoke.json"};
374 }
else if (name ==
"simple-chat" || name ==
"chat") {
380 "z3ed simple-chat --rom=zelda3.sfc",
381 "z3ed simple-chat \"What dungeons exist?\" --rom=zelda3.sfc"};
382 }
else if (name.find(
"tools-") == 0) {
384 if (name ==
"tools-list") {
385 metadata.
description =
"List available test helper tools";
388 }
else if (name ==
"tools-harness-state") {
389 metadata.
description =
"Generate WRAM state for test harnesses";
391 "z3ed tools-harness-state --rom=zelda3.sfc --output=state.h"};
392 }
else if (name ==
"tools-extract-values") {
393 metadata.
description =
"Extract vanilla ROM values";
394 metadata.
examples = {
"z3ed tools-extract-values --rom=zelda3.sfc"};
395 }
else if (name ==
"tools-extract-golden") {
396 metadata.
description =
"Extract comprehensive golden data for testing";
398 "z3ed tools-extract-golden --rom=zelda3.sfc --output=golden.h"};
399 }
else if (name ==
"tools-patch-v3") {
400 metadata.
description =
"Create v3 ZSCustomOverworld patched ROM";
402 "z3ed tools-patch-v3 --rom=zelda3.sfc --output=patched.sfc"};
406 }
else if (name.find(
"test-") == 0) {
408 metadata.
description =
"Test discovery and execution";
409 if (name ==
"test-list") {
411 metadata.
examples = {
"z3ed test-list",
"z3ed test-list --format json"};
412 }
else if (name ==
"test-run") {
413 metadata.
examples = {
"z3ed test-run --label stable",
414 "z3ed test-run --label gui"};
415 }
else if (name ==
"test-status") {
417 metadata.
examples = {
"z3ed test-status --format json"};
425 if (!descriptor.summary.empty() &&
426 descriptor.summary !=
"Command summary not provided.") {
431 if (!descriptor.todo_reference.empty() &&
432 descriptor.todo_reference !=
"todo#unassigned") {
436 Register(std::move(handler), metadata);
444 return it->second.get();
448 auto alias_it =
aliases_.find(name);
450 auto handler_it =
handlers_.find(alias_it->second);
452 return handler_it->second.get();
460 const std::string& name)
const {
462 std::string canonical_name = name;
463 auto alias_it =
aliases_.find(name);
465 canonical_name = alias_it->second;
468 auto it =
metadata_.find(canonical_name);
469 return (it !=
metadata_.end()) ? &it->second :
nullptr;
473 const std::string& category)
const {
474 std::vector<std::string> result;
475 for (
const auto& [name, metadata] :
metadata_) {
476 if (metadata.category == category) {
477 result.push_back(name);
484 std::vector<std::string> categories;
485 for (
const auto& [_, metadata] :
metadata_) {
486 if (std::find(categories.begin(), categories.end(), metadata.category) ==
488 categories.push_back(metadata.category);
495 std::vector<std::string> result;
496 for (
const auto& [name, metadata] :
metadata_) {
497 if (metadata.available_to_agent) {
498 result.push_back(name);
505 std::ostringstream out;
506 out <<
"{\n \"commands\": [\n";
509 for (
const auto& [_, metadata] :
metadata_) {
515 out <<
" \"name\": \"" << EscapeJson(metadata.name) <<
"\",\n";
516 out <<
" \"category\": \"" << EscapeJson(metadata.category) <<
"\",\n";
517 out <<
" \"description\": \"" << EscapeJson(metadata.description)
519 out <<
" \"usage\": \"" << EscapeJson(metadata.usage) <<
"\",\n";
520 out <<
" \"available_to_agent\": "
521 << (metadata.available_to_agent ?
"true" :
"false") <<
",\n";
522 out <<
" \"requires_rom\": "
523 << (metadata.requires_rom ?
"true" :
"false") <<
",\n";
524 out <<
" \"requires_grpc\": "
525 << (metadata.requires_grpc ?
"true" :
"false") <<
",\n";
526 if (!metadata.todo_reference.empty()) {
527 out <<
" \"todo_reference\": \""
528 << EscapeJson(metadata.todo_reference) <<
"\",\n";
530 out <<
" \"todo_reference\": \"\",\n";
532 out <<
" \"aliases\": ";
533 AppendStringArray(out, metadata.aliases);
535 out <<
" \"examples\": ";
536 AppendStringArray(out, metadata.examples);
548 return absl::StrFormat(
"Command '%s' not found", name);
551 std::ostringstream help;
552 help <<
"\n\033[1;36m" << metadata->name <<
"\033[0m - "
553 << metadata->description <<
"\n\n";
554 help <<
"\033[1;33mUsage:\033[0m\n";
555 help <<
" " << metadata->usage <<
"\n\n";
557 if (!metadata->examples.empty()) {
558 help <<
"\033[1;33mExamples:\033[0m\n";
559 for (
const auto& example : metadata->examples) {
560 help <<
" " << example <<
"\n";
565 if (metadata->requires_rom) {
566 help <<
"\033[1;33mRequires:\033[0m ROM file (--rom=<path>)\n";
568 if (metadata->requires_grpc) {
569 help <<
"\033[1;33mRequires:\033[0m YAZE running with gRPC enabled\n";
572 if (!metadata->aliases.empty()) {
573 help <<
"\n\033[1;33mAliases:\033[0m "
574 << absl::StrJoin(metadata->aliases,
", ") <<
"\n";
581 const std::string& category)
const {
583 if (commands.empty()) {
584 return absl::StrFormat(
"No commands in category '%s'", category);
587 std::ostringstream help;
588 help <<
"\n\033[1;36m" << category <<
" commands:\033[0m\n\n";
590 for (
const auto& cmd : commands) {
593 help <<
" \033[1;33m" << cmd <<
"\033[0m\n";
594 help <<
" " << metadata->description <<
"\n";
595 if (!metadata->usage.empty()) {
596 help <<
" Usage: " << metadata->usage <<
"\n";
606 std::ostringstream help;
607 help <<
"\n\033[1;36mAll z3ed Commands:\033[0m\n\n";
610 for (
const auto& category : categories) {
618 const std::vector<std::string>& args,
620 std::string* captured_output) {
621 auto* handler =
Get(name);
623 return absl::NotFoundError(absl::StrFormat(
"Command '%s' not found", name));
626 const bool command_help_requested =
627 std::find(args.begin(), args.end(),
"--help") != args.end() ||
628 std::find(args.begin(), args.end(),
"-h") != args.end();
629 if (command_help_requested) {
631 if (captured_output !=
nullptr) {
632 *captured_output = help;
634 std::cout << help <<
"\n";
636 return absl::OkStatus();
639 absl::Status status = handler->Run(args, rom_context, captured_output);
643 if (!status.ok() && status.code() == absl::StatusCode::kInvalidArgument &&
646 if (captured_output !=
nullptr) {
647 if (!captured_output->empty()) {
648 captured_output->append(
"\n\n");
650 captured_output->append(help);
652 std::cout << help <<
"\n";
660 return Get(name) !=
nullptr;
The Rom class is used to load, save, and modify Rom data. This is a generic SNES ROM container and do...
Single source of truth for all z3ed commands.
void RegisterCliCommands()
std::map< std::string, std::string > aliases_
std::vector< std::string > GetCommandsInCategory(const std::string &category) const
Get all commands in a category.
static CommandRegistry & Instance()
std::string GenerateCategoryHelp(const std::string &category) const
Generate category help text.
std::string ExportFunctionSchemas() const
Export function schemas for AI tool calling (JSON)
const CommandMetadata * GetMetadata(const std::string &name) const
Get command metadata.
std::string GenerateCompleteHelp() const
Generate complete help text (all commands)
std::vector< std::string > GetAgentCommands() const
Get all commands available to AI agents.
std::map< std::string, CommandMetadata > metadata_
absl::Status Execute(const std::string &name, const std::vector< std::string > &args, Rom *rom_context=nullptr, std::string *captured_output=nullptr)
Execute a command by name.
std::map< std::string, std::unique_ptr< resources::CommandHandler > > handlers_
resources::CommandHandler * Get(const std::string &name) const
Get a command handler by name or alias.
bool HasCommand(const std::string &name) const
Check if command exists.
std::vector< std::string > GetCategories() const
Get all categories.
void Register(std::unique_ptr< resources::CommandHandler > handler, const CommandMetadata &metadata)
Register a command handler.
std::string GenerateHelp(const std::string &name) const
Generate help text for a command.
void RegisterHandlers(std::vector< std::unique_ptr< resources::CommandHandler > > handlers)
Register a set of command handlers (idempotent)
Base class for CLI command handlers.
std::string EscapeJson(const std::string &input)
void AppendStringArray(std::ostringstream &out, const std::vector< std::string > &values)
std::vector< std::unique_ptr< resources::CommandHandler > > CreateCliCommandHandlers()
Factory function to create all CLI-level command handlers.