resource_commands.cc

Go to the documentation of this file.

#include "cli/handlers/tools/resource_commands.h"
 
#include "absl/strings/ascii.h"
#include "absl/strings/match.h"
#include "absl/strings/str_format.h"
#include "absl/strings/str_split.h"
#include "cli/service/resources/resource_context_builder.h"
#include "util/macro.h"
 
namespace yaze {
namespace cli {
namespace handlers {
 
absl::Status ResourceListCommandHandler::Execute(Rom* rom, const resources::ArgumentParser& parser,
                      resources::OutputFormatter& formatter)  {
  auto type = parser.GetString("type").value();
  
  ResourceContextBuilder builder(rom);
  ASSIGN_OR_RETURN(auto labels, builder.GetLabels(type));
  
  formatter.BeginObject(absl::StrFormat("%s Labels", absl::AsciiStrToUpper(type)));
  for (const auto& [key, value] : labels) {
    formatter.AddField(key, value);
  }
  formatter.EndObject();
  
  return absl::OkStatus();
}
 
absl::Status ResourceSearchCommandHandler::Execute(Rom* rom, const resources::ArgumentParser& parser,
                      resources::OutputFormatter& formatter)  {
  auto query = parser.GetString("query").value();
  auto type = parser.GetString("type").value_or("all");
  
  ResourceContextBuilder builder(rom);
  
  std::vector<std::string> categories = {"overworld", "dungeon", "entrance", 
                                       "room", "sprite", "palette", "item"};
  if (type != "all") {
    categories = {type};
  }
  
  formatter.BeginObject("Resource Search Results");
  formatter.AddField("query", query);
  formatter.AddField("search_type", type);
  
  int total_matches = 0;
  formatter.BeginArray("matches");
  
  for (const auto& category : categories) {
    auto labels_or = builder.GetLabels(category);
    if (!labels_or.ok()) continue;
    
    auto labels = labels_or.value();
    for (const auto& [key, value] : labels) {
      if (absl::StrContains(absl::AsciiStrToLower(value), 
                           absl::AsciiStrToLower(query))) {
        formatter.AddArrayItem(absl::StrFormat("%s:%s = %s", 
                                              category, key, value));
        total_matches++;
      }
    }
  }
  
  formatter.EndArray();
  formatter.AddField("total_matches", total_matches);
  formatter.EndObject();
  
  return absl::OkStatus();
}
 
}  // namespace handlers
}  // namespace cli
}  // namespace yaze