ImprovedTalentLoadouts/ImprovedTalentLoadoutsImportDialog.lua at main · Lardeck/ImprovedTalentLoadouts · 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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
local addonName, TalentLoadouts = ...

ImprovedTalentLoadoutsDialogMixin = {};

function ImprovedTalentLoadoutsDialogMixin:OnLoad()
	self.Title:SetText(self.titleText);
end

ImprovedTalentLoadoutsDialogInputControlMixin = {};

function ImprovedTalentLoadoutsDialogInputControlMixin:OnLoad()
	local editBox = self:GetEditBox();
	editBox:SetScript("OnTextChanged", GenerateClosure(self.OnTextChanged, self));
	editBox:SetScript("OnEnterPressed", GenerateClosure(self.OnEnterPressed, self));
	editBox:SetScript("OnEscapePressed", GenerateClosure(self.OnEscapePressed, self));
	self.Label:SetText(self.labelText);
end

function ImprovedTalentLoadoutsDialogInputControlMixin:OnShow()
	self:GetEditBox():SetText("");
end

function ImprovedTalentLoadoutsDialogInputControlMixin:GetText()
	return self:GetEditBox():GetText();
end

function ImprovedTalentLoadoutsDialogInputControlMixin:SetText(text)
	return self:GetEditBox():SetText(text);
end

function ImprovedTalentLoadoutsDialogInputControlMixin:HasText()
	return UserEditBoxNonEmpty(self:GetEditBox());
end

function ImprovedTalentLoadoutsDialogInputControlMixin:OnEnterPressed()
end

function ImprovedTalentLoadoutsDialogInputControlMixin:OnEscapePressed()
end

function ImprovedTalentLoadoutsDialogInputControlMixin:OnTextChanged()
end

function ImprovedTalentLoadoutsDialogInputControlMixin:GetEditBox()
end

ImprovedTalentLoadoutsDialogNameControlMixin = CreateFromMixins(ImprovedTalentLoadoutsDialogInputControlMixin);

function ImprovedTalentLoadoutsDialogNameControlMixin:GetEditBox()
	return self.EditBox;
end

ImprovedTalentLoadoutsImportDialogMixin = {}

function ImprovedTalentLoadoutsImportDialogMixin:OnLoad()
	self.exclusive = true;
	self.AcceptButton:SetOnClickHandler(GenerateClosure(self.OnAccept, self));
	self.CancelButton:SetOnClickHandler(GenerateClosure(self.OnCancel, self));
	ImprovedTalentLoadoutsDialogMixin.OnLoad(self);

	self.NameControl:GetEditBox():SetAutoFocus(false);
	self.ImportControl:GetEditBox():SetAutoFocus(true);
end

function ImprovedTalentLoadoutsImportDialogMixin:OnHide()

end

function ImprovedTalentLoadoutsImportDialogMixin:OnCancel()
	StaticPopupSpecial_Hide(self);
end

function ImprovedTalentLoadoutsImportDialogMixin:OnAccept()
	if self.AcceptButton:IsEnabled() then
		local importString = self.ImportControl:GetText();
		local loadoutName = self.NameControl:GetText();

        local fakeConfigID
        local treeType = self.treeType
        local categoryInfo = self.categoryInfo

        if not treeType or treeType == 1 then
            fakeConfigID = TalentLoadouts:ImportLoadout(importString, loadoutName, categoryInfo and categoryInfo.key)
        elseif treeType == 2 then
            loadoutName = string.format("[C] %s", loadoutName)
            fakeConfigID = TalentLoadouts:ImportClassLoadout(importString, loadoutName, categoryInfo and categoryInfo.key)
        elseif treeType == 3 then
            loadoutName = string.format("[S] %s", loadoutName)
            fakeConfigID = TalentLoadouts:ImportSpecLoadout(importString, loadoutName, categoryInfo and categoryInfo.key)
        end

        if fakeConfigID then
			StaticPopupSpecial_Hide(self);

            if IsShiftKeyDown() then
                TalentLoadouts:LoadLoadoutByConfigID(fakeConfigID, self.categoryInfo)
            end
        end
	end
end

function ImprovedTalentLoadoutsImportDialogMixin:UpdateAcceptButtonEnabledState()
	local importTextFilled = self.ImportControl:HasText();
	local nameTextFilled = self.NameControl:HasText();
	self.AcceptButton:SetEnabled(importTextFilled and nameTextFilled);
end

function ImprovedTalentLoadoutsImportDialogMixin:OnTextChanged()
	self:UpdateAcceptButtonEnabledState();
end

function ImprovedTalentLoadoutsImportDialogMixin:ShowDialog(treeType, categoryInfo)
    self.treeType = treeType
    self.categoryInfo = categoryInfo

	StaticPopupSpecial_Show(self);
end


ImprovedTalentLoadoutsImportDialogImportControlMixin = CreateFromMixins(ImprovedTalentLoadoutsDialogInputControlMixin);

function ImprovedTalentLoadoutsImportDialogImportControlMixin:OnShow()
	ImprovedTalentLoadoutsDialogInputControlMixin.OnShow(self);
end

function ImprovedTalentLoadoutsImportDialogImportControlMixin:OnEnterPressed()
	self:GetParent():OnAccept();
end

function ImprovedTalentLoadoutsImportDialogImportControlMixin:OnEscapePressed()
	self:GetParent():OnCancel();
end

function ImprovedTalentLoadoutsImportDialogImportControlMixin:OnTextChanged()
	self:GetParent():OnTextChanged();
	InputScrollFrame_OnTextChanged(self.InputContainer.EditBox);
end

function ImprovedTalentLoadoutsImportDialogImportControlMixin:GetEditBox()
	return self.InputContainer.EditBox;
end

ImprovedTalentLoadoutsImportDialogNameControlMixin = {}

function ImprovedTalentLoadoutsImportDialogNameControlMixin:OnEnterPressed()
	self:GetParent():OnAccept();
end

function ImprovedTalentLoadoutsImportDialogNameControlMixin:OnEscapePressed()
	self:GetParent():OnCancel();
end

function ImprovedTalentLoadoutsImportDialogNameControlMixin:OnTextChanged()
	self:GetParent():OnTextChanged();
end