project_graph_tool.h

Go to the documentation of this file.

#ifndef YAZE_CLI_SERVICE_AGENT_TOOLS_PROJECT_GRAPH_TOOL_H_
#define YAZE_CLI_SERVICE_AGENT_TOOLS_PROJECT_GRAPH_TOOL_H_
 
#include <string>
#include <vector>
 
#include "absl/status/status.h"
#include "cli/service/resources/command_context.h"
#include "cli/service/resources/command_handler.h"
#include "core/project.h" // For YazeProject
#include "core/asar_wrapper.h" // For AsarWrapper
 
namespace yaze {
namespace cli {
namespace agent {
namespace tools {
 
class ProjectGraphTool : public resources::CommandHandler {
 public:
  ProjectGraphTool() = default;
  ~ProjectGraphTool() override = default;
 
  std::string GetName() const override { return "project-graph"; }
  std::string GetUsage() const override {
    return "project-graph --query=<info|files|symbols> [--path=<folder>]";
  }
  bool RequiresRom() const override { return true; } // Requires ROM to get symbols
 
  // Override to receive project and asar contexts
  void SetProjectContext(project::YazeProject* project) override { project_ = project; }
  void SetAsarWrapper(core::AsarWrapper* asar_wrapper) override { asar_wrapper_ = asar_wrapper; }
 
 protected:
  absl::Status ValidateArgs(const resources::ArgumentParser& parser) override;
  absl::Status Execute(Rom* rom, const resources::ArgumentParser& parser,
                       resources::OutputFormatter& formatter) override;
 
 private:
  absl::Status GetProjectInfo(resources::OutputFormatter& formatter) const;
  absl::Status GetFileStructure(const std::string& path, resources::OutputFormatter& formatter) const;
  absl::Status GetSymbolTable(resources::OutputFormatter& formatter) const;
};
 
}  // namespace tools
}  // namespace agent
}  // namespace cli
}  // namespace yaze
 
#endif  // YAZE_CLI_SERVICE_AGENT_TOOLS_PROJECT_GRAPH_TOOL_H_