BMLMod.h

#pragma once

#include "IMod.h"
#include "Version.h"
#include "Gui.h"
#include <queue>
#include "ICommand.h"
#include <map>

#define MSG_MAXSIZE 35

class Config;
class Property;

class GuiList : public BGui::Gui {
public:
	GuiList();
	void Init(int size, int maxsize);

	virtual BGui::Button* CreateButton(int index) = 0;
	virtual std::string GetButtonText(int index) = 0;
	virtual BGui::Gui* CreateSubGui(int index) = 0;
	virtual BGui::Gui* GetParentGui() = 0;

	virtual void SetPage(int page);
	void PreviousPage() { if (m_curpage > 0) SetPage(m_curpage - 1); }
	void NextPage() { if (m_curpage < m_maxpage - 1) SetPage(m_curpage + 1); }
	virtual void Exit();
	virtual void SetVisible(bool visible) override;

protected:
	std::vector<BGui::Button*> m_buttons;
	int m_curpage, m_maxpage, m_size, m_maxsize;
	std::vector<BGui::Gui*> m_guiList;
	BGui::Button* m_left, * m_right;
};

class GuiModOption : public GuiList {
public:
	GuiModOption();

	virtual BGui::Button* CreateButton(int index) override;
	virtual std::string GetButtonText(int index) override;
	virtual BGui::Gui* CreateSubGui(int index) override;
	virtual BGui::Gui* GetParentGui() override;
	virtual void Exit() override;
};

class GuiModMenu : public GuiList {
public:
	GuiModMenu(IMod* mod);

	virtual void Process() override;
	virtual void SetVisible(bool visible) override;
	virtual BGui::Button* CreateButton(int index) override;
	virtual std::string GetButtonText(int index) override;
	virtual BGui::Gui* CreateSubGui(int index) override;
	virtual BGui::Gui* GetParentGui() override;

private:
	Config* m_config;
	std::vector<std::string> m_categories;

	BGui::Panel* m_comment_bg;
	BGui::Label* m_comment;
	int m_curcmt = -1;
};

class GuiCustomMap : public GuiList {
public:
	GuiCustomMap(class BMLMod* mod);

	virtual BGui::Button* CreateButton(int index) override;
	virtual std::string GetButtonText(int index) override;
	virtual BGui::Gui* CreateSubGui(int index) override;
	virtual BGui::Gui* GetParentGui() override;
	virtual void SetPage(int page);
	virtual void Exit() override;

private:
	struct MapInfo {
		std::string displayname, searchname, filepath;
		bool operator <(const MapInfo& o) {
			return displayname < o.displayname;
		}
	};

	std::vector<MapInfo> m_maps;
	std::vector<MapInfo*> m_searchRes;
	std::vector<BGui::Text*> m_texts;
	BGui::Input* m_searchBar = nullptr;
	BGui::Button* m_exit = nullptr;
	class BMLMod* m_mod;
};

class GuiModCategory : public BGui::Gui {
public:
	GuiModCategory(GuiModMenu* parent, Config* config, std::string category);

	virtual void Process() override;
	virtual void SetVisible(bool visible) override;
	void SetPage(int page);
	void PreviousPage() { if (m_curpage > 0) SetPage(m_curpage - 1); }
	void NextPage() { if (m_curpage < m_maxpage - 1) SetPage(m_curpage + 1); }

	void SaveAndExit();
	void Exit();

private:
	std::vector<Property*> m_data;
	std::vector<std::vector<BGui::Element*>> m_elements;
	std::vector<std::pair<BGui::Element*, BGui::Element*>> m_inputs;
	int m_curpage, m_maxpage, m_size;
	BGui::Button* m_left, * m_right;

	GuiModMenu* m_parent;
	Config* m_config;
	std::string m_category;
	BGui::Button* m_exit;

	std::vector<std::vector<std::pair<Property*, BGui::Element*>>> m_comments;
	BGui::Panel* m_comment_bg;
	BGui::Label* m_comment;
	Property* m_curcmt = nullptr;
};

class ScreenModeHook : public BGui::Gui {
public:
	virtual void OnScreenModeChanged() override;
};

class CommandTravel : public ICommand {
	virtual std::string GetName() override { return "travel"; };
	virtual std::string GetAlias() override { return ""; };
	virtual std::string GetDescription() override { return "Switch to First-Person Camera."; };
	virtual bool IsCheat() override { return false; };

	virtual void Execute(IBML* bml, const std::vector<std::string>& args) override;
	virtual const std::vector<std::string> GetTabCompletion(IBML* bml, const std::vector<std::string>& args) override { return std::vector<std::string>(); };
public:
	CommandTravel(class BMLMod* mod) : m_mod(mod) {};

private:
	class BMLMod* m_mod;
};

class BMLMod : public IMod {
	friend class CommandClear;
	friend class CommandSector;
	friend class CommandSpeed;
	friend class GuiModMenu;
	friend class GuiCustomMap;
public:
	BMLMod(IBML* bml) : IMod(bml) {}

	virtual CKSTRING GetID() override { return "BML"; }
	virtual CKSTRING GetVersion() override { return BML_VERSION; }
	virtual CKSTRING GetName() override { return "Ballance Mod Loader"; }
	virtual CKSTRING GetAuthor() override { return "Gamepiaynmo & YingChe"; }
	virtual CKSTRING GetDescription() override { return "Implementation of functions provided by Ballance Mod Loader."
		"\n\n https://github.com/Gamepiaynmo/BallanceModLoader"; }
	DECLARE_BML_VERSION;

	virtual void OnLoad() override;
	virtual void OnLoadObject(CKSTRING filename, BOOL isMap, CKSTRING masterName,
		CK_CLASSID filterClass, BOOL addtoscene, BOOL reuseMeshes, BOOL reuseMaterials,
		BOOL dynamic, XObjectArray* objArray, CKObject* masterObj) override;
	virtual void OnLoadScript(CKSTRING filename, CKBehavior* script) override;
	virtual void OnProcess() override;
	virtual void OnCheatEnabled(bool enable) override;
	virtual void OnModifyConfig(CKSTRING category, CKSTRING key, IProperty* prop) override;

	virtual void OnPostResetLevel() override;
	virtual void OnStartLevel() override;
	virtual void OnPostExitLevel() override;
	virtual void OnPauseLevel() override;
	virtual void OnUnpauseLevel() override;
	virtual void OnCounterActive() override;
	virtual void OnCounterInactive() override;

	void AddIngameMessage(CKSTRING msg);
	void ShowCheatBanner(bool show);
	void ShowWatermark(bool show);
	void ShowModOptions();
	void ShowGui(BGui::Gui* gui);
	void CloseCurrentGui();
	bool IsWatermarkEnabled();

	float GetSRScore() { return m_srtimer; }
	int GetHSScore();

	void EnterTravelCam();
	void ExitTravelCam();
	bool IsInTravelCam();

private:
	void OnEditScript_Base_DefaultLevel(CKBehavior* script);
	void OnEditScript_Base_EventHandler(CKBehavior* script);
	void OnEditScript_Menu_MenuInit(CKBehavior* script);
	void OnEditScript_Menu_OptionsMenu(CKBehavior* script);
	void OnEditScript_Gameplay_Ingame(CKBehavior* script);
	void OnEditScript_Gameplay_Energy(CKBehavior* script);
	void OnEditScript_Gameplay_Events(CKBehavior* script);
	void OnEditScript_Levelinit_build(CKBehavior* script);
	void OnEditScript_ExtraLife_Fix(CKBehavior* script);

	void OnCmdEdit(CKDWORD key);

	BGui::Gui* m_cmdBar = nullptr;
	bool m_cmdTyping = false;
	BGui::Input* m_cmdInput;
	std::vector<std::string> m_cmdHistory;
	unsigned int m_historyPos;

	BGui::Gui* m_msgLog = nullptr;
	int m_msgCnt = 0;
	struct {
		BGui::Panel* m_bg;
		BGui::Label* m_text;
		float timer;
	} m_msg[MSG_MAXSIZE];
	float m_msgMaxTimer = 6000; // in milliseconds

	BGui::Gui* m_ingameBanner = nullptr;
	BGui::Label* m_watermark, * m_cheat, * m_fps, * m_srScore, * m_srTitle = nullptr;
	int m_fpscnt = 0, m_fpstimer = 0;
	float m_srtimer = 0.0f;
	bool m_sractive = false;

	BGui::Gui* m_currentGui = nullptr;
	GuiModOption* m_modOption = nullptr;

	IProperty* m_skipAnim;
	IProperty* m_fullscreenKey;
	IProperty* m_unlockRes;
	IProperty* m_showFPS;
	IProperty* m_showSR;
	IProperty* m_fixLifeBall;
	IProperty* m_customMapNumber;
	IProperty* m_msgDisplayDuration; // in seconds

	IProperty* m_ballCheat[2];
	IProperty* m_suicide, * m_suicideOn;
	CKParameterLocal* m_ballForce[2] = { 0 };
	bool m_suicideCd = false;

	IProperty* m_camRot[2], * m_camY[2], * m_camZ[2], * m_cam45, * m_camReset, * m_camOn;
	CK3dEntity* m_camPos, * m_camOrient, * m_camOrientRef, * m_camTarget;

	IProperty* m_overclock;
	CKBehaviorLink* m_oclinks[3];
	CKBehaviorIO* m_oclinkio[3][2];

	IProperty* m_changeBall[3];
	CKBehavior* m_setNewBall;
	CKParameter* m_curTrafo;
	CKDataArray* m_curLevel, * m_ingameParam;
	int m_changeBallCd = 0;
	IProperty* m_speedupBall, * m_speedNotification;
	bool m_speedup;

	IProperty* m_resetBall;
	CKParameter* m_curSector;
	CKBehavior* m_phyNewBall, * m_dynamicPos;

	IProperty* m_addLife;
	bool m_addLifeCd = false;

	IProperty* m_addBall[4];
	int m_curSel = -1;
	CK3dEntity* m_curObj = nullptr;
	CK3dEntity* m_pBalls[4];
	std::vector<std::pair<int, CK3dEntity*>> m_tempBalls;
	IProperty* m_moveKeys[6];

	IProperty* m_fpsKeys[6];
	CKCamera* m_travelCam;

	GuiCustomMap* m_mapsGui = nullptr;
	CK2dEntity* m_level01;
	CKBehavior* m_exitStart;
	BGui::Button* m_customMaps;
	CKParameter* m_loadCustom, * m_mapFile, * m_levelRow;

	ScreenModeHook* m_smHook = nullptr;
};