tile16_proposal_generator.h

Go to the documentation of this file.

#ifndef YAZE_CLI_SERVICE_TILE16_PROPOSAL_GENERATOR_H_
#define YAZE_CLI_SERVICE_TILE16_PROPOSAL_GENERATOR_H_
 
#include <chrono>
#include <string>
#include <vector>
 
#include "absl/status/status.h"
#include "absl/status/statusor.h"
#include "app/gfx/bitmap.h"
#include "app/rom.h"
 
namespace yaze {
namespace cli {
 
struct Tile16Change {
  int map_id;             // Overworld map ID
  int x;                  // Tile16 X coordinate
  int y;                  // Tile16 Y coordinate
  uint16_t old_tile;      // Original tile16 ID (for rollback)
  uint16_t new_tile;      // New tile16 ID to apply
  
  std::string ToString() const;
};
 
struct Tile16Proposal {
  std::string id;                                    // Unique proposal ID (UUID-like)
  std::string prompt;                                 // Original user prompt
  std::vector<Tile16Change> changes;                  // List of tile changes
  std::string reasoning;                              // AI's explanation
  std::string ai_service;                             // "gemini", "ollama", "mock"
  std::chrono::system_clock::time_point created_at;   // Timestamp
  
  // Proposal state
  enum class Status {
    PENDING,    // Generated but not reviewed
    ACCEPTED,   // User accepted, changes applied
    REJECTED,   // User rejected, changes discarded
    APPLIED     // Successfully applied to ROM
  };
  Status status = Status::PENDING;
  
  std::string ToJson() const;
  static absl::StatusOr<Tile16Proposal> FromJson(const std::string& json);
};
 
class Tile16ProposalGenerator {
 public:
  Tile16ProposalGenerator() = default;
  
  // Allow testing of private methods
  friend class Tile16ProposalGeneratorTest;
  
  absl::StatusOr<Tile16Proposal> GenerateFromCommands(
      const std::string& prompt,
      const std::vector<std::string>& commands,
      const std::string& ai_service,
      Rom* rom);
  
  absl::Status ApplyProposal(const Tile16Proposal& proposal, Rom* rom);
  
  absl::StatusOr<gfx::Bitmap> GenerateDiff(
      const Tile16Proposal& proposal,
      Rom* before_rom,
      Rom* after_rom);
  
  absl::Status SaveProposal(const Tile16Proposal& proposal,
                            const std::string& path);
  
  absl::StatusOr<Tile16Proposal> LoadProposal(const std::string& path);
  
 private:
  absl::StatusOr<Tile16Change> ParseSetTileCommand(
      const std::string& command,
      Rom* rom);
  
  absl::StatusOr<std::vector<Tile16Change>> ParseSetAreaCommand(
      const std::string& command,
      Rom* rom);
  
  absl::StatusOr<std::vector<Tile16Change>> ParseReplaceTileCommand(
      const std::string& command,
      Rom* rom);
  
  std::string GenerateProposalId() const;
};
 
}  // namespace cli
}  // namespace yaze
 
#endif  // YAZE_CLI_SERVICE_TILE16_PROPOSAL_GENERATOR_H_