#ifndef TPROTOCOL_H
#define TPROTOCOL_H

#include <global_types.h>
#include <string.h>
#include <pso/macros.h>
#include <pso/forward.h>
#include <pso/packet_classes.h>
#include <pso/protocol.h>
#include <pso/PSOV3EncryptionTCP.h>
#include <pso/THeap.h>
#include <pso/TMainTask.h>
#include <pso/TMenuList.h>
#include <pso/TObject.h>
#include <pso/TPlyCharData.h>
#include <pso/TPlyClientConfig.h>
#include <pso/TPlyGuildCardTag.h>
#include <pso/TPlySmth.h>
#include <pso/TSocket.h>
#include <pso/TTcpSocket.h>

EXTERN_OBJECT_NAME(TProtocol);
extern int enable_v2_features;
extern u32 new_ip_addr;
extern u16 new_port;
extern TPlyMeetUserExtension meet_user_settings;
extern u32 game_variations[16][2];
extern void copy_packet(struct packet *pkt);

// NOTE: Having this outside of `flex_packet` is required in some cases.
//       This is because some of the senders allocate a different sized
//       packet from what `TProtocol` uses, but also call `_packet::bswap()`.
struct _packet {
	void bswap() {
		header.bswap();
	};

	packet_header header;
	u8 data[];
};

template<size_t n = 0>
union flex_packet {
	struct _packet pkt;
	u8 bytes[(n) ? n : sizeof(packet_header)];
};

struct packet : public flex_packet<0x7c00> {};

#define RECV_PACKET_HANDLERS \
	o(handle_unused_login, void) \
	o(handle_03_recv_regist, u8 state) \
	o(handle_04_recv_login, u8 error_code) \
	o(handle_05_recv_logout, void) \
	o(handle_07_A0_A1_recv_dir_list, int entry_count, GameListEntry *entries, GameListEntry &entry_0) \
	o(handle_08_recv_game_list, int entry_count, GameListEntry *entries, GameListEntry &entry_0) \
	o(handle_01_recv_error, char *mesg) \
	o(handle_06_recv_chat, TPlyGuildCardTag &tag, char *mesg) \
	o(handle_11_recv_message, char *mesg, u8 unused) \
	o(handle_1A_D5_recv_text, char *text) \
	o(handle_0E_recv_start_game, TPlyJoinLobbyData *entries, u8 entry_count, int lobby_entry_idx) \
	o(unused7, void) \
	o(handle_64_recv_start_game3, TPlyJoinGame &join_data) \
	o(unused8, void) \
	o(handle_67_recv_start_lobby2, TPlyJoinLobbyEntry *entries, u8 entry_count, int client_id, int leader_id, int lobby_number, int block_number, int smth, int event) \
	o(handle_80_recv_generate_id, TRecvGenerateID gen_id) \
	o(unused9, void) \
	o(handle_65_add_player_to_game_packet, void) \
	o(handle_66_player_left_game_packet, void) \
	o(handle_68_add_player_to_lobby_packet, void) \
	o(handle_69_player_left_lobby_packet, void) \
	o(handle_18_90_9A_login_response_packet, void) \
	o(handle_92_9C_register_response_packet, void) \
	o(unused10, void) \
	o(handle_95_request_character_data_packet, void) \
	o(handle_81_recv_chat_message, TChatMessage &chat_message) \
	o(handle_41_recv_user_ans, TUserAns &user_ans) \
	o(send_96_unused, void) \
	o(handle_97_checksum_reply_packet, void) \
	o(handle_B1_current_time_packet, void) \
	o(handle_C0_choice_search_option_packet, void) \
	o(handle_C4_choice_search_reply_packet, void) \
	o(handle_D8_infoboard_packet, void) \
	o(handle_A2_recv_quest_menu_list, TMenuList<QuestListEntry, 30, 0> &quest_list) \
	o(handle_A3_recv_quest_menu_info, TPlyText<288> &info) \
	o(handle_44_recv_download_head, TRecvDownloadHead &download_head) \
	o(handle_13_recv_download, TRecvDownload &download) \
	o(handle_A4_recv_quest_list, u8 entry_count, TMenuList<QuestListEntry, 30, 0> &quest_list) \
	o(handle_A5_recv_quest_info, TPlyText<288> &info) \
	o(handle_A6_recv_vm_download_head, TRecvDownloadHead &download_head) \
	o(handle_A7_recv_vm_download, TRecvDownload &download) \
	o(unused11, void) \
	o(handle_1F_recv_text_list, int entry_count, GameListEntry *entries, GameListEntry &entry_0) \
	o(handle_B0_recv_emergency_call, char *mesg) \
	o(handle_88_player_arrow_color_list_packet, void) \
	o(handle_8A_lobby_name_packet, void) \
	o(handle_C5_player_challenge_data_packet, void) \
	o(handle_DA_lobby_event_packet, void) \
	o(handle_AB_quest_stats_response_packet, void) \
	o(handle_D3_execute_trade_packet, void) \
	o(handle_D4_trade_result_packet, void) \
	o(handle_D1_advance_trade_state_packet, void)

#define recv_packet_handler(name) recv_packet_handler_##name

#define o(name, ...) typedef void (*recv_packet_handler(name))(__VA_ARGS__);
RECV_PACKET_HANDLERS;
#undef o

struct recv_packet_handlers {
	#define o(name, ...) recv_packet_handler(name) name;
	RECV_PACKET_HANDLERS;
	#undef o
};

typedef void (TProtocol::*command_handler)(packet &pkt);

struct command_handler_entry {
	int command;
	command_handler handler;
};


class TProtocol : public TTcpSocket {
public:
	TPlySmth m_smth;
	TPlyGuildCardTag m_guildcard_tag;
	u16 m_sub_version;
	int m_login_response_state;
	int m_connected;
	int m_joined_game;
	int m_has_meet_user_settings;
	int m_handle_pings_only;
	int m_entry_count;
	int m_unused;
	int m_lobby_list_count;
	int m_unused2;
	int m_udp_disabled;
	int m_language;
	u8 m_unused3[80];
	TPlyCharData m_character_data;
	u32 m_unused4;
	GameListEntry m_game_entries[64];
	GameListEntry m_game_entries2[64];
	LobbyListEntry m_lobby_entries[16];
	QuestListEntry m_quest_entries[30];
	u8 m_unused5[128];
	char m_serial_number3[17];
	char m_password2[17];
	char m_serial_number[48];
	char m_access_key[48];
	char m_password[64];
	char m_player_name[16];
	char m_serial_number2[17];
	char m_access_key2[17];
	TPlyClientConfig m_client_config;
	int m_packet_offset;
	int m_packet_size;
	struct packet m_packet;
	struct recv_packet_handlers m_recv_handlers;
private:
	static void operator delete(void *ptr) { obj_heap->heap_free(ptr); };
public:
	TProtocol(TObject *parent, u16 sub_version, int language, char *serial_number, char *access_key, char *password);
	virtual ~TProtocol();

	virtual void run_task();
	virtual short send(u8 *data, size_t size);

	void some_stub();
	int handle_command(struct packet *pkt);
	void parse_packet();
	void seq_jump(u8 *some_struct, TPlyMeetUserExtension &extension);

	// Command handlers.
	// 0x01
	void recv_error(packet &pkt);
	// 0x1A/0xD5
	void recv_text(packet &pkt);
	// 0x03
	void recv_regist(packet &pkt);
	// 0x04
	void recv_login(packet &pkt);
	// 0x05
	void recv_logout(packet &pkt);
	// 0x06
	void recv_chat(packet &pkt);
	// 0x07/0xA0/0xA1
	void recv_dir_list(packet &pkt);
	// 0x08
	void recv_game_list(packet &pkt);
	// 0x1F
	void recv_text_list(packet &pkt);
	// 0x0E
	void recv_start_game(packet &pkt);
	// 0x64
	void recv_start_game3(packet &pkt);
	// 0x67
	void recv_start_lobby2(packet &pkt);
	// 0x12
	void recv_banner(packet &pkt);
	// 0x16
	void recv_banner_head(packet &pkt);
	// 0x44
	void recv_download_head(packet &pkt);
	// 0xA6
	void recv_vm_download_head(packet &pkt);
	// 0xA7
	void recv_vm_download(packet &pkt);
	// 0x13
	void recv_download(packet &pkt);
	// 0x14
	void recv_upload(packet &pkt);
	// 0x11
	void recv_message(packet &pkt);
	// 0x19
	void recv_port(packet &pkt);
	// 0x1B
	void recv_battle_data(packet &pkt);
	// 0x1C
	void recv_system_file(packet &pkt);
	// 0x60/0x62
	void recv_pso_data(packet &pkt);
	// 0x6C/0x6D
	void recv_pso_data_long(packet &pkt);
	// 0x80
	void recv_generate_id(packet &pkt);
	// 0x83
	void recv_room_info(packet &pkt);
	// 0x41
	void recv_user_ans(packet &pkt);
	// 0x1D
	void recv_ping(packet &pkt);
	// 0x81
	void recv_chat_message(packet &pkt);
	// 0xA2
	void recv_quest_menu_list(packet &pkt);
	// 0xA3
	void recv_quest_menu_info(packet &pkt);
	// 0xA4
	void recv_quest_list(packet &pkt);
	// 0xA5
	void recv_quest_info(packet &pkt);
	// 0xB0
	void recv_emergency_call(packet &pkt);

	// Send command handlers.
	// 0x03
	void send_regist();
	// 0x04
	void send_login2(int disable_udp);
	// Doesn't send a command.
	int send_login3();
	// 0x05
	void send_logout();
	// 0x06
	void send_chat(TPlyGuildCardTag &tag, char *mesg);
	// 0x10
	void send_action(GameListEntry &selection, char *action_name, char *password);
	void send_action(GameListEntry &selection, u8 flags, char *action_name, char *password);
	// 0x09
	void send_info(TPlyGuildCardTag &tag);
	// 0x08
	void send_game_list();
	// 0x1F
	void send_text_list();
	// 0x61
	void send_chara_data_v2(TPlyCharData &char_data, TPlyChallenge &challenge, TPlyChoiceSearchConfig &choice_search_config, TBlockedSenders &blocked_senders, TPlyText<512> &auto_reply, char *info_board);
};

#endif