Fixes #65 - Added check for mounted or flight form

Bestride · Oct 28, 2018 · 069f0ea · 069f0ea
1 parent e3b4f2c
commit 069f0ea
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/BeStride_Logic.lua b/BeStride_Logic.lua
@@ -301,7 +301,7 @@ function BeStride_Logic:IsDruidAndSpecial()
 	if self:IsDruid() then
 		if not self:DruidFlying() then
 			return false
-		elseif IsMounted() and IsFlying() and self:IsFlyable() and self:DruidFlying() and self:DruidFlyingMTFF() then
+		elseif (IsMounted() or self:IsDruidTraveling()) and IsFlying() and self:IsFlyable() and self:DruidFlying() and self:DruidFlyingMTFF() then
 			return true
 		elseif IsFlying() and self:NoDismountWhileFlying() then
 			return false
@@ -1122,6 +1122,15 @@ end
 -- Druid --
 -- ----- --
 
+function BeStride_Logic:IsDruidTraveling()
+  if self:IsDruid() then
+    local index = GetShapeshiftForm()
+    if index == 3 then
+      return true
+    end
+  end
+end
+
 function BeStride_Logic:DruidFlying()
 	if self:IsDruid() then
 		if self:DruidCanFly() then