Fix evil native form button

Ghostopheles · Jan 4, 2024 · 3d46f48 · 3d46f48
1 parent 7953784
commit 3d46f48
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 15 deletions.
diff --git a/Datamine/Modules/Unified/Templates/ModelScene/ModelScene.lua b/Datamine/Modules/Unified/Templates/ModelScene/ModelScene.lua
@@ -173,24 +173,11 @@ function DatamineModelSceneMixin:OnLoad_Custom()
     self:ReleaseAllActors();
     EventUtil.RegisterOnceFrameEventAndCallback("PLAYER_ENTERING_WORLD", function() self:SetFromModelSceneID(596) end);
 
-    self.NativeFormToggleButton:SetScript("OnClick", function()
-        local useNativeForm = self:GetUseNativeForm();
-        self:SetUseNativeForm(not useNativeForm);
-    end);
-
     self.actorTemplate = "DatamineModelSceneActorTemplate";
 
     Registry:RegisterCallback(Events.MODEL_LOADED_INTERNAL, self.OnModelLoaded_Internal, self);
 
-    self.NativeFormToggleButton:SetScript("OnEnter", function()
-        GameTooltip:SetOwner(self.NativeFormToggleButton, "ANCHOR_TOP");
-        GameTooltip:SetText("Toggle Alternate Form", 1, 1, 1);
-        GameTooltip:Show();
-    end);
-
-    self.NativeFormToggleButton:SetScript("OnLeave", function()
-        GameTooltip:Hide();
-    end);
+    self:UpdateNativeFormButton();
 end
 
 function DatamineModelSceneMixin:OnFirstShow()
@@ -263,6 +250,32 @@ function DatamineModelSceneMixin:OnModelLoaded_Internal(actor)
     Registry:TriggerEvent(Events.MODEL_OUTFIT_UPDATED);
 end
 
+function DatamineModelSceneMixin:UpdateNativeFormButton()
+    local _, raceFileName = UnitRace("player");
+    if raceFileName == "Dracthyr" or raceFileName == "Worgen" then
+        self.NativeFormToggleButton:SetScript("OnClick", function()
+            local useNativeForm = self:GetUseNativeForm();
+            self:SetUseNativeForm(not useNativeForm);
+        end);
+
+        self.NativeFormToggleButton:SetScript("OnEnter", function()
+            GameTooltip:SetOwner(self.NativeFormToggleButton, "ANCHOR_TOP");
+            GameTooltip:SetText("Toggle Alternate Form", 1, 1, 1);
+            GameTooltip:Show();
+        end);
+
+        self.NativeFormToggleButton:SetScript("OnLeave", function()
+            GameTooltip:Hide();
+        end);
+
+        self.NativeFormToggleButton:Show();
+        self.NativeFormToggleButton:Enable();
+    else
+        self.NativeFormToggleButton:Hide();
+        self.NativeFormToggleButton:Disable();
+    end
+end
+
 function DatamineModelSceneMixin:StartPanning()
     self.IsPanning = true;
     local actor = self:GetActiveActor();

diff --git a/Datamine/Modules/Unified/Templates/ModelScene/ModelScene.xml b/Datamine/Modules/Unified/Templates/ModelScene/ModelScene.xml
@@ -187,7 +187,8 @@
                     <Anchor point="TOP" y="-10" />
                 </Anchors>
             </Frame>
-            <Button parentKey="NativeFormToggleButton" inherits="DatamineGenericButtonTemplate">
+            <Button parentKey="NativeFormToggleButton" inherits="DatamineGenericButtonTemplate"
+                hidden="true" motionScriptsWhileDisabled="false">
                 <Size x="50" y="50" />
                 <Anchors>
                     <Anchor point="TOPRIGHT" x="-15" y="-15" />