rename command.h to z3ed.h, update structure

src/cli/z3ed.h

@@ -0,0 +1,256 @@
+#ifndef YAZE_CLI_Z3ED_H
+#define YAZE_CLI_Z3ED_H
+
+#include <cstdint>
+#include <iostream>
+#include <memory>
+#include <sstream>
+#include <string>
+#include <unordered_map>
+#include <vector>
+
+#include "absl/status/status.h"
+#include "app/core/common.h"
+#include "app/core/constants.h"
+#include "app/rom.h"
+
+namespace yaze {
+namespace cli {
+
+enum ColorCode {
+  FG_RED = 31,
+  FG_GREEN = 32,
+  FG_YELLOW = 33,
+  FG_BLUE = 36,
+  FG_MAGENTA = 35,
+  FG_DEFAULT = 39,
+  FG_RESET = 0,
+  FG_UNDERLINE = 4,
+  BG_RED = 41,
+  BG_GREEN = 42,
+  BG_BLUE = 44,
+  BG_DEFAULT = 49
+};
+class ColorModifier {
+  ColorCode code;
+
+ public:
+  explicit ColorModifier(ColorCode pCode) : code(pCode) {}
+  friend std::ostream& operator<<(std::ostream& os, const ColorModifier& mod) {
+    return os << "\033[" << mod.code << "m";
+  }
+};
+
+class CommandHandler {
+ public:
+  CommandHandler() = default;
+  virtual ~CommandHandler() = default;
+  virtual absl::Status handle(const std::vector<std::string>& arg_vec) = 0;
+
+  Rom rom_;
+};
+
+class ApplyPatch : public CommandHandler {
+ public:
+  absl::Status handle(const std::vector<std::string>& arg_vec) override;
+};
+
+class AsarPatch : public CommandHandler {
+ public:
+  absl::Status handle(const std::vector<std::string>& arg_vec) override;
+};
+
+class CreatePatch : public CommandHandler {
+ public:
+  absl::Status handle(const std::vector<std::string>& arg_vec) override;
+};
+
+/**
+ * @brief Open a Rom file and display information about it.
+ */
+class Open : public CommandHandler {
+ public:
+  absl::Status handle(const std::vector<std::string>& arg_vec) override {
+    ColorModifier green(ColorCode::FG_GREEN);
+    ColorModifier blue(ColorCode::FG_BLUE);
+    ColorModifier reset(ColorCode::FG_RESET);
+    auto const& arg = arg_vec[0];
+    RETURN_IF_ERROR(rom_.LoadFromFile(arg))
+    std::cout << "Title: " << green << rom_.title() << std::endl;
+    std::cout << reset << "Size: " << blue << "0x" << std::hex << rom_.size()
+              << reset << std::endl;
+    return absl::OkStatus();
+  }
+};
+
+/**
+ * @brief Backup a Rom file.
+ */
+class Backup : public CommandHandler {
+ public:
+  absl::Status handle(const std::vector<std::string>& arg_vec) override {
+    RETURN_IF_ERROR(rom_.LoadFromFile(arg_vec[0]))
+    if (arg_vec.size() == 2) {
+      // Optional filename added
+      RETURN_IF_ERROR(rom_.SaveToFile(/*backup=*/true, false, arg_vec[1]))
+    } else {
+      RETURN_IF_ERROR(rom_.SaveToFile(/*backup=*/true))
+    }
+    return absl::OkStatus();
+  }
+};
+
+// Compress Graphics
+class Compress : public CommandHandler {
+ public:
+  absl::Status handle(const std::vector<std::string>& arg_vec) override;
+};
+
+// Decompress (Export) Graphics
+//
+// -e <rom_file> <bin_file> --mode=<optional:mode>
+//
+// mode:
+class Decompress : public CommandHandler {
+ public:
+  absl::Status handle(const std::vector<std::string>& arg_vec) override;
+};
+
+/**
+ * @brief Convert a SNES address to a PC address.
+
+  * @param arg_vec `-s <address>`
+  * @return absl::Status
+*/
+class SnesToPc : public CommandHandler {
+ public:
+  absl::Status handle(const std::vector<std::string>& arg_vec) override {
+    auto arg = arg_vec[0];
+    std::stringstream ss(arg.data());
+    uint32_t snes_address;
+    ss >> std::hex >> snes_address;
+    uint32_t pc_address = core::SnesToPc(snes_address);
+    std::cout << std::hex << pc_address << std::endl;
+    return absl::OkStatus();
+  }
+};
+
+/**
+ * @brief Convert a PC address to a SNES address.
+
+  * @param arg_vec `-p <address>`
+  * @return absl::Status
+*/
+class PcToSnes : public CommandHandler {
+ public:
+  absl::Status handle(const std::vector<std::string>& arg_vec) override {
+    auto arg = arg_vec[0];
+    std::stringstream ss(arg.data());
+    uint32_t pc_address;
+    ss >> std::hex >> pc_address;
+    uint32_t snes_address = core::PcToSnes(pc_address);
+    ColorModifier blue(ColorCode::FG_BLUE);
+    std::cout << "SNES LoROM Address: ";
+    std::cout << blue << "$" << std::uppercase << std::hex << snes_address
+              << "\n";
+    return absl::OkStatus();
+  }
+};
+
+/**
+ * @brief Read from a Rom file.
+
+  * @param arg_vec `-r <rom_file> <address> <optional:length, default: 0x01>`
+  * @return absl::Status
+*/
+class ReadFromRom : public CommandHandler {
+ public:
+  absl::Status handle(const std::vector<std::string>& arg_vec) override {
+    RETURN_IF_ERROR(rom_.LoadFromFile(arg_vec[0]))
+
+    std::stringstream ss(arg_vec[1].data());
+    uint32_t offset;
+    ss >> std::hex >> offset;
+    uint32_t length = 0x01;
+    if (!arg_vec[2].empty()) {
+      length = std::stoi(arg_vec[2]);
+    }
+
+    if (length > 1) {
+      auto returned_bytes_status = rom_.ReadByteVector(offset, length);
+      if (!returned_bytes_status.ok()) {
+        return returned_bytes_status.status();
+      }
+      auto returned_bytes = returned_bytes_status.value();
+      for (const auto& each : returned_bytes) {
+        std::cout << each;
+      }
+      std::cout << std::endl;
+    } else {
+      auto byte = rom_.ReadByte(offset);
+      std::cout << std::hex << byte.value() << std::endl;
+    }
+
+    return absl::OkStatus();
+  }
+};
+
+/**
+ * @brief Transfer tile 16 data from one Rom to another.
+
+  * @param arg_vec `-t <src_rom> <dest_rom> "<tile32_id_list:csv>"`
+  * @return absl::Status
+*/
+class Tile16Transfer : public CommandHandler {
+ public:
+  absl::Status handle(const std::vector<std::string>& arg_vec) override;
+};
+
+/**
+ * @brief Expand a Rom file.
+
+  * @param arg_vec `-x <rom_file> <file_size>`
+  * @return absl::Status
+*/
+class Expand : public CommandHandler {
+ public:
+  absl::Status handle(const std::vector<std::string>& arg_vec) override {
+    RETURN_IF_ERROR(rom_.LoadFromFile(arg_vec[0]))
+
+    std::stringstream ss(arg_vec[1].data());
+    uint32_t size;
+    ss >> std::hex >> size;
+
+    rom_.Expand(size);
+
+    std::cout << "Successfully expanded ROM to " << std::hex << size
+              << std::endl;
+
+    return absl::OkStatus();
+  }
+};
+
+/**
+ * @brief Command handler for the CLI.
+ */
+struct Commands {
+  std::unordered_map<std::string, std::shared_ptr<CommandHandler>> handlers = {
+      {"-a", std::make_shared<ApplyPatch>()},
+      {"-asar", std::make_shared<AsarPatch>()},
+      {"-c", std::make_shared<CreatePatch>()},
+      {"-o", std::make_shared<Open>()},
+      {"-b", std::make_shared<Backup>()},
+      {"-x", std::make_shared<Expand>()},
+      {"-i", std::make_shared<Compress>()},    // Import
+      {"-e", std::make_shared<Decompress>()},  // Export
+      {"-s", std::make_shared<SnesToPc>()},
+      {"-p", std::make_shared<PcToSnes>()},
+      {"-t", std::make_shared<Tile16Transfer>()},
+      {"-r", std::make_shared<ReadFromRom>()}  // Read from Rom
+  };
+};
+
+}  // namespace cli
+}  // namespace yaze
+
+#endif