#pragma once

#include "game/character.hpp"
#include "game/world_packets.hpp"
#include <imgui.h>
#include <string>
#include <functional>
#include <vector>
#include <memory>
#include <cstdint>

namespace wowee {
namespace game { class GameHandler; }
namespace pipeline { class AssetManager; }
namespace rendering { class CharacterPreview; }

namespace ui {

class CharacterCreateScreen {
public:
    CharacterCreateScreen();
    ~CharacterCreateScreen();

    void render(game::GameHandler& gameHandler);
    void update(float deltaTime);
    void setOnCreate(std::function<void(const game::CharCreateData&)> cb) { onCreate = std::move(cb); }
    void setOnCancel(std::function<void()> cb) { onCancel = std::move(cb); }
    void setStatus(const std::string& msg, bool isError = false);
    void reset();
    void initializePreview(pipeline::AssetManager* am);

    /** Set allowed races/classes from expansion profile. Empty = allow all (WotLK default). */
    void setExpansionConstraints(const std::vector<uint32_t>& races, const std::vector<uint32_t>& classes);

private:
    char nameBuffer[13] = {};  // WoW max name = 12 chars + null
    int raceIndex = 0;
    int classIndex = 0;
    int genderIndex = 0;
    int bodyTypeIndex = 0;  // For nonbinary: 0=masculine, 1=feminine
    int skin = 0, face = 0, hairStyle = 0, hairColor = 0, facialHair = 0;
    int maxSkin = 9, maxFace = 9, maxHairStyle = 11, maxHairColor = 9, maxFacialHair = 8;
    std::string statusMessage;
    bool statusIsError = false;

    // For many races/styles, CharSections hair color IDs are not guaranteed to be contiguous.
    // We expose an index (hairColor) in the UI and map it to the actual DBC hairColorId here.
    std::vector<uint8_t> hairColorIds_;

    std::vector<game::Class> availableClasses;
    void updateAvailableClasses();

    // Expansion-filtered race/class lists
    std::vector<game::Race> availableRaces_;    // Alliance-first, then horde order
    int allianceRaceCount_ = 0;                 // How many of availableRaces_ are alliance
    std::vector<game::Class> expansionClasses_; // Allowed classes (empty = all)

    std::function<void(const game::CharCreateData&)> onCreate;
    std::function<void()> onCancel;

    // 3D model preview
    std::unique_ptr<rendering::CharacterPreview> preview_;
    pipeline::AssetManager* assetManager_ = nullptr;
    int prevRaceIndex_ = -1;
    int prevGenderIndex_ = -1;
    int prevBodyTypeIndex_ = -1;
    int prevSkin_ = -1;
    int prevFace_ = -1;
    int prevHairStyle_ = -1;
    int prevHairColor_ = -1;
    int prevFacialHair_ = -1;
    int prevRangeRace_ = -1;
    int prevRangeGender_ = -1;
    int prevRangeSkin_ = -1;
    int prevRangeHairStyle_ = -1;
    float createTimer_ = -1.0f;  // >=0 while waiting for SMSG_CHAR_CREATE response
    bool draggingPreview_ = false;
    float dragStartX_ = 0.0f;

    void updatePreviewIfNeeded();
    void updateAppearanceRanges();
};

} // namespace ui
} // namespace wowee