#pragma once

#include "Emu/Cell/ErrorCodes.h"
#include "Emu/Memory/vm_ptr.h"
#include "util/BitField.h"
#include "util/atomic.hpp"
#include "util/types.hpp"

#include <functional>
#include <string>

enum
{
	CELL_MSGDIALOG_PROGRESSBAR_STRING_SIZE = 64,
	CELL_MSGDIALOG_STRING_SIZE = 512,
};

enum CellMsgDialogError : u32
{
	CELL_MSGDIALOG_ERROR_PARAM = 0x8002b301,
	CELL_MSGDIALOG_ERROR_DIALOG_NOT_OPENED = 0x8002b302,
};

enum : u32
{
	CELL_MSGDIALOG_TYPE_SE_TYPE_ERROR = 0 << 0,
	CELL_MSGDIALOG_TYPE_SE_TYPE_NORMAL = 1 << 0,

	CELL_MSGDIALOG_TYPE_SE_MUTE_OFF = 0 << 1,
	CELL_MSGDIALOG_TYPE_SE_MUTE_ON = 1 << 1,

	CELL_MSGDIALOG_TYPE_BG_VISIBLE = 0 << 2,
	CELL_MSGDIALOG_TYPE_BG_INVISIBLE = 1 << 2,

	CELL_MSGDIALOG_TYPE_BUTTON_TYPE_NONE = 0 << 4,
	CELL_MSGDIALOG_TYPE_BUTTON_TYPE_YESNO = 1 << 4,
	CELL_MSGDIALOG_TYPE_BUTTON_TYPE_OK = 2 << 4,

	CELL_MSGDIALOG_TYPE_DISABLE_CANCEL_OFF = 0 << 7,
	CELL_MSGDIALOG_TYPE_DISABLE_CANCEL_ON = 1 << 7,

	CELL_MSGDIALOG_TYPE_DEFAULT_CURSOR_NONE = 0 << 8,
	CELL_MSGDIALOG_TYPE_DEFAULT_CURSOR_YES = 0 << 8,
	CELL_MSGDIALOG_TYPE_DEFAULT_CURSOR_NO = 1 << 8,
	CELL_MSGDIALOG_TYPE_DEFAULT_CURSOR_OK = 0 << 8,

	CELL_MSGDIALOG_TYPE_PROGRESSBAR_NONE = 0 << 12,
	CELL_MSGDIALOG_TYPE_PROGRESSBAR_SINGLE = 1 << 12,
	CELL_MSGDIALOG_TYPE_PROGRESSBAR_DOUBLE = 2 << 12,
};

// MsgDialog Button Type
enum : s32
{
	CELL_MSGDIALOG_BUTTON_NONE = -1,
	CELL_MSGDIALOG_BUTTON_INVALID = 0,
	CELL_MSGDIALOG_BUTTON_OK = 1,
	CELL_MSGDIALOG_BUTTON_YES = 1,
	CELL_MSGDIALOG_BUTTON_NO = 2,
	CELL_MSGDIALOG_BUTTON_ESCAPE = 3,
};

enum CellMsgDialogProgressBarIndex
{
	CELL_MSGDIALOG_PROGRESSBAR_INDEX_SINGLE =
		0, // the only bar in a single bar dialog
	CELL_MSGDIALOG_PROGRESSBAR_INDEX_DOUBLE_UPPER =
		0, // the upper bar in a double bar dialog
	CELL_MSGDIALOG_PROGRESSBAR_INDEX_DOUBLE_LOWER =
		1, // the lower bar in a double bar dialog
};

using CellMsgDialogCallback = void(s32 buttonType, vm::ptr<void> userData);

union MsgDialogType
{
	u32 value;

	bf_t<u32, 0, 1> se_normal;
	bf_t<u32, 1, 1> se_mute_on;
	bf_t<u32, 2, 1> bg_invisible;
	bf_t<u32, 4, 3> button_type;
	bf_t<u32, 7, 1> disable_cancel;
	bf_t<u32, 8, 2> default_cursor;
	bf_t<u32, 12, 2> progress_bar_count;
};

enum class MsgDialogState
{
	Open,
	Abort,
	Close,
};

enum class msg_dialog_source
{
	_cellMsgDialog,
	_cellSaveData,
	_cellGame,
	_cellCrossController,
	_sceNp,
	_sceNpTrophy,
	sys_progress,
	shader_loading,
};

void close_msg_dialog();
error_code open_msg_dialog(bool is_blocking, u32 type, vm::cptr<char> msgString,
	msg_dialog_source source,
	vm::ptr<CellMsgDialogCallback> callback = vm::null,
	vm::ptr<void> userData = vm::null,
	vm::ptr<void> extParam = vm::null,
	s32* return_code = nullptr);
error_code open_exit_dialog(const std::string& message, bool is_exit_requested,
	msg_dialog_source source);

class MsgDialogBase
{
protected:
	// the progressbar that will be represented in the taskbar. Use -1 to combine
	// the progress.
	s32 taskbar_index = 0;

public:
	atomic_t<MsgDialogState> state{MsgDialogState::Close};

	MsgDialogType type{};
	msg_dialog_source source = msg_dialog_source::_cellMsgDialog;

	std::function<void(s32 status)> on_close = nullptr;

	virtual ~MsgDialogBase();
	virtual void Create(const std::string& msg,
		const std::string& title = "") = 0;
	virtual void Close(bool success) = 0;
	virtual void SetMsg(const std::string& msg) = 0;
	virtual void ProgressBarSetMsg(u32 progressBarIndex,
		const std::string& msg) = 0;
	virtual void ProgressBarReset(u32 progressBarIndex) = 0;
	virtual void ProgressBarInc(u32 progressBarIndex, u32 delta) = 0;
	virtual void ProgressBarSetValue(u32 progressBarIndex, u32 value) = 0;
	virtual void ProgressBarSetLimit(u32 index, u32 limit) = 0;

	void ProgressBarSetTaskbarIndex(s32 index)
	{
		taskbar_index = index;
	}
};
