#pragma once

#include "constants.hpp"
#include "game/registry.hpp"
#include "algos/simplefsm.hpp"
#include "gui/debug_ui.hpp"
#include "gui/main_ui.hpp"
#include "gui/status_ui.hpp"
#include "gui/loot_ui.hpp"
#include "events.hpp"
#include "gui/event_router.hpp"
#include "gui/dnd_loot.hpp"
#include "events.hpp"
#include "gui/scene_ui.hpp"
#include "gui/storyboard.hpp"

namespace gui {
  enum class State {
    START=__LINE__,
    INTRO=__LINE__,
    START_SCENE=__LINE__,
    DEATH_SCENE=__LINE__,
    WIN_SCENE=__LINE__,
    NEXT_LEVEL_SCENE=__LINE__,
    MOVING=__LINE__,
    ATTACKING=__LINE__,
    ROTATING=__LINE__,
    LOOTING=__LINE__,
    IDLE=__LINE__,
    END_QUIT=__LINE__,
    END_PLAYER_DIED=__LINE__,
    END_PLAY_AGAIN=__LINE__,
  };

  class FSM : public DeadSimpleFSM<State, game::Event> {
    public:
      sf::RenderWindow& $window;
      bool $draw_stats = false;
      DebugUI $debug_ui;
      MainUI $main_ui;
      StatusUI $status_ui{STATUS_UI_X, STATUS_UI_Y, STATUS_UI_WIDTH, STATUS_UI_HEIGHT};
      LootUI $loot_ui;
      gui::routing::Router $router;
      DNDLoot $dnd_loot;
      System::Registry $systems;
      std::unordered_map<std::string, std::shared_ptr<gui::SceneUI>> $scenes{
        {"STARTING", std::make_shared<gui::SceneUI>("STARTING")},
        {"DEATH", std::make_shared<gui::SceneUI>("DEATH")},
        {"WIN", std::make_shared<gui::SceneUI>("WIN")},
        {"NEXT_LEVEL", std::make_shared<gui::SceneUI>("NEXT_LEVEL")},
      };

      std::shared_ptr<gui::SceneUI> $cur_scene = nullptr;
      std::shared_ptr<storyboard::UI> $story = nullptr;
      bool $display_help = false;

      FSM(sf::RenderWindow& window);

      void event(game::Event ev, std::any data={});

      void START(game::Event ev);
      void INTRO(game::Event ev);
      void START_SCENE(game::Event ev);
      void WIN_SCENE(game::Event ev);
      void DEATH_SCENE(game::Event ev);
      void NEXT_LEVEL_SCENE(game::Event ev);
      void IDLE(game::Event ev, std::any data);
      void MOVING(game::Event ev);
      void ATTACKING(game::Event ev, std::any data);
      void MAPPING(game::Event ev);
      void ROTATING(game::Event ev);
      void LOOTING(game::Event ev, std::any data);
      void END_QUIT(game::Event ev);
      void END_PLAYER_DIED(game::Event ev);
      void END_PLAY_AGAIN(game::Event ev);

      void try_move(int dir, bool strafe);
      sf::Vector2f mouse_position();
      void mouse_action(guecs::Modifiers mods);
      void handle_keyboard_mouse();
      void draw_gui();
      void update();
      void render();
      bool active();
      void run_systems();
      void handle_world_events();
      void next_level();
      void debug_render();
      void take_screenshot();
      void show_scene(const std::string& name);
      void close_scene();
      void player_died();
      void check_player_wins();
      void toggle_help();
  };
}