#pragma once

#define DBKEY_ID "id"

class CMTProto;
typedef void (CMTProto::*TG_QUERY_HANDLER)(td::ClientManager::Response &response);

struct TG_REQUEST
{
	TG_REQUEST(td::ClientManager::RequestId _1, TG_QUERY_HANDLER _2) :
		requestId(_1),
		pHandler(_2)
	{}

	td::ClientManager::RequestId requestId;
	TG_QUERY_HANDLER pHandler;
};

struct TG_USER
{
	TG_USER(uint64_t _1, MCONTACT _2, bool _3 = false) :
		id(_1),
		hContact(_2),
		isGroupChat(_3)
	{}

	uint64_t id;
	MCONTACT hContact;
	bool isGroupChat;
};

class CMTProto : public PROTO<CMTProto>
{
	std::unique_ptr<td::ClientManager> m_pClientMmanager;
	TD::object_ptr<TD::AuthorizationState> pAuthState;

	bool m_bAuthorized, m_bRunning = false, m_bTerminated;
	int32_t m_iClientId;
	uint64_t m_iQueryId;

	OBJLIST<TG_REQUEST> m_arRequests;

	static INT_PTR CALLBACK EnterPhoneCode(void *param);
	static INT_PTR CALLBACK EnterPassword(void *param);

	CMStringW GetProtoFolder() const
	{	return CMStringW(VARSW(L"%miranda_userdata%")) + L"\\" + _A2T(m_szModuleName);
	}

	void OnUpdateAuth(td::ClientManager::Response &response);

	void LogOut(void);
	void OnLoggedIn(void);
	void ProcessResponse(td::ClientManager::Response);
	void SendQuery(TD::Function *pFunc, TG_QUERY_HANDLER pHandler = nullptr);

	void ProcessAuth(TD::updateAuthorizationState *pObj);
	void ProcessChat(TD::updateNewChat *pObj);
	void ProcessGroups(TD::updateChatFilters *pObj);
	void ProcessMessage(TD::updateNewMessage *pObj);
	void ProcessUser(TD::updateUser *pObj);

	void UpdateString(MCONTACT hContact, const char *pszSetting, const std::string &str);

	// Users
	int64_t m_iOwnId;
	OBJLIST<TG_USER> m_arUsers;

	TG_USER* FindUser(uint64_t id);
	TG_USER* AddUser(uint64_t id, bool bIsChat);

	// Popups
	HANDLE m_hPopupClass;

	void InitPopups(void);
	void Popup(MCONTACT hContact, const wchar_t *szMsg, const wchar_t *szTitle);

public:
	//////////////////////////////////////////////////////////////////////////////////////
	// Ctors

	CMTProto(const char *protoName, const wchar_t *userName);
	~CMTProto();

	//////////////////////////////////////////////////////////////////////////////////////
	// Virtual functions

	INT_PTR GetCaps(int type, MCONTACT hContact = NULL) override;
	
	int SetStatus(int iNewStatus) override;

	void OnModulesLoaded() override;
	void OnShutdown() override;
	void OnErase() override;

	// Services //////////////////////////////////////////////////////////////////////////

	INT_PTR __cdecl SvcCreateAccMgrUI(WPARAM, LPARAM);
	
	// Events ////////////////////////////////////////////////////////////////////////////
	
	int __cdecl OnOptionsInit(WPARAM, LPARAM);

	// Options ///////////////////////////////////////////////////////////////////////////
	
	CMOption<wchar_t*> m_szOwnPhone;       // our own phone number
	CMOption<wchar_t*> m_wszDefaultGroup;  // clist group to store contacts
	CMOption<bool>     m_bHideGroupchats;  // do not open chat windows on creation
	CMOption<bool>     m_bUsePopups;

	// Processing Threads ////////////////////////////////////////////////////////////////

	void __cdecl ServerThread(void *);
};