#pragma once

#include "framework/headless_editor_test.h"
#include "app/editor/overworld/overworld_editor.h"
#include "rom/rom.h"
#include "rom/snes.h"
#include "gtest/gtest.h"
#include "zelda3/game_data.h"

namespace yaze {
namespace test {

class OverworldEditorTest : public HeadlessEditorTest {
 protected:
  void SetUp() override {
    HeadlessEditorTest::SetUp();

    // Load ROM
    const char* paths[] = {"assets/zelda3.sfc", "build/bin/zelda3.sfc", "zelda3.sfc"};
    bool loaded = false;
    for (const char* path : paths) {
        rom_ = std::make_unique<Rom>();
        if (rom_->LoadFromFile(path).ok()) {
            loaded = true;
            break;
        }
    }
    ASSERT_TRUE(loaded) << "Could not load zelda3.sfc from any location";

    // Load GameData
    game_data_ = std::make_unique<zelda3::GameData>(rom_.get());
    ASSERT_TRUE(zelda3::LoadGameData(*rom_, *game_data_).ok());

    // Create Dependencies
    editor::EditorDependencies deps;
    deps.rom = rom_.get();
    deps.game_data = game_data_.get();
    deps.panel_manager = panel_manager_.get();
    deps.renderer = renderer_.get();

    // Create Editor
    overworld_editor_ = std::make_unique<editor::OverworldEditor>(rom_.get(), deps);
    overworld_editor_->SetGameData(game_data_.get());
    
    // Initialize and Load
    overworld_editor_->Initialize();
    ASSERT_TRUE(overworld_editor_->Load().ok());
  }

  void TearDown() override {
    overworld_editor_.reset();
    game_data_.reset();
    HeadlessEditorTest::TearDown();
  }

  std::unique_ptr<editor::OverworldEditor> overworld_editor_;
  std::unique_ptr<zelda3::GameData> game_data_;
};

}  // namespace test
}  // namespace yaze