Phobos/src/Commands/AdvancedCommandBarButtons.cpp at 30e31ee1b955bda4f08f353fef96a70100ecaba2 · Phobos-developers/Phobos · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

#include "AdvancedCommandBarButtons.h"


#include "DistributionMode.h"

#include <ShapeButtonClass.h>
#include "Commands.h"

#include <Utilities/Macro.h>

void AdvancedCommandBarButton::RegisterButtons()
{
	// Vanilla buttons
	AdvancedCommandBarButton::TotalButtonCount = AdvancedCommandBarButton::OldButtonCount;

	// New button register here
	AdvancedCommandBarButton::Array.push_back(std::make_unique<DistributionModeHoldDownButtonClass>());
}

ShapeButtonClass* AdvancedCommandBarButton::GetShapeButton(const char* name)
{
	for (const auto& ptr : AdvancedCommandBarButton::Array)
	{
		AdvancedCommandBarButton* pNewButton = ptr.get();
		if (_strcmpi(pNewButton->GetName(), name) == 0)
		{
			return ShapeButtonClass::GetButton(pNewButton->ID);
		}
	}

	return nullptr;
}

#pragma region Hooks

DEFINE_HOOK(0x6CFD08, AdvancedCommandBarClass_GetButtonIdxByName_NewButton, 0x5)
{
	enum { SetButtonIndex = 0x6CFD0D };

	GET(const char*, name, ECX);

	for (const auto& ptr : AdvancedCommandBarButton::Array)
	{
		AdvancedCommandBarButton* pButton = ptr.get();
		if (_strcmpi(name, pButton->GetName()) == 0)
		{
			R->EAX(pButton->ID);
			return SetButtonIndex;
		}
	}

	return 0;
}

DEFINE_HOOK(0x6D0233, AdvancedCommandBarClass_Init_InitNewButtonIndex, 0x6)
{
	AdvancedCommandBarButton::RegisterButtons();
	return 0;
}

DEFINE_HOOK(0x6D0827, AdvancedCommandBarClass_Update_UpdateNewButton, 0x6)
{
	GET(const int, index, EAX);

	for (const auto& ptr : AdvancedCommandBarButton::Array)
	{
		AdvancedCommandBarButton* pNewButton = ptr.get();

		if (pNewButton->ID == index)
		{
			pNewButton->Execute(ShapeButtonClass::GetButton(index)->IsOn);
			break;
		}
	}

	return 0;
}

DEFINE_HOOK(0x6D10DF, AdvancedCommandBarClass_InitButtonIO_InitNewHoldDownButton, 0x6)
{
	for (const auto& ptr : AdvancedCommandBarButton::Array)
	{
		AdvancedCommandBarButton* pNewButton = ptr.get();

		if (!pNewButton->CanHoldDown())
			continue;

		if (auto pButton = ShapeButtonClass::GetButton(pNewButton->ID))
		{
			pButton->ToggleType = 1;
			pButton->UseFlash = true;
		}
	}

	return 0;
}

DEFINE_HOOK(0x6D14DD, AdvancedCommandBarClass_InitToolTip_InitNewButtonToolTip, 0x5)
{
	for (const auto& ptr : AdvancedCommandBarButton::Array)
	{
		AdvancedCommandBarButton* pNewButton = ptr.get();
		ShapeButtonClass::SetToolTip(ShapeButtonClass::GetButton(pNewButton->ID), pNewButton->GetTipName());
	}

	return 0;
}

#pragma endregion