Lua: Ensure Custom badges inheritance

Author: manups4e

ScaleformUI_Lua/src/Menus/UIMenu/Items/UIMenuCheckboxItem.lua

@@ -186,9 +186,21 @@ function UIMenuCheckboxItem:LeftBadge(Badge)
     end
 end
 
+function UIMenuCheckboxItem:CustomLeftBadge(txd,txn)
+    if txd ~= nil and txd ~= "" and txn ~= nil and txn ~= "" then
+        self.Base:CustomLeftBadge(txd,txn, self)
+    else
+        return self.Base:LeftBadge()
+    end
+end
+
 ---RightBadge
 function UIMenuCheckboxItem:RightBadge()
-    error("This item does not support badges")
+    error("This item does not support right badges")
+end
+
+function UIMenuCheckboxItem:CustomRightBadge()
+    error("This item does not support right badges")
 end
 
 ---RightLabel

ScaleformUI_Lua/src/Menus/UIMenu/Items/UIMenuDynamicListItem.lua

@@ -225,9 +225,21 @@ function UIMenuDynamicListItem:LeftBadge(Badge)
     end
 end
 
+function UIMenuDynamicListItem:CustomLeftBadge(txd,txn)
+    if txd ~= nil and txd ~= "" and txn ~= nil and txn ~= "" then
+        self.Base:CustomLeftBadge(txd,txn, self)
+    else
+        return self.Base:LeftBadge()
+    end
+end
+
 ---RightBadge
 function UIMenuDynamicListItem:RightBadge()
-    error("This item does not support badges")
+    error("This item does not support right badges")
+end
+
+function UIMenuDynamicListItem:CustomRightBadge()
+    error("This item does not support right badges")
 end
 
 ---RightLabel

ScaleformUI_Lua/src/Menus/UIMenu/Items/UIMenuListItem.lua

@@ -277,9 +277,23 @@ function UIMenuListItem:LeftBadge(Badge)
     end
 end
 
+--- CustomLeftBadge
+function UIMenuListItem:CustomLeftBadge(txd,txn)
+    if txd ~= nil and txd ~= "" and txn ~= nil and txn ~= "" then
+        self.Base:CustomLeftBadge(txd,txn, self)
+    else
+        return self.Base:LeftBadge()
+    end
+end
+
+
 ---RightBadge
 function UIMenuListItem:RightBadge()
-    error("This item does not support badges")
+    error("This item does not support right badges")
+end
+
+function UIMenuListItem:CustomRightBadge()
+    error("This item does not support right badges")
 end
 
 ---RightLabel

ScaleformUI_Lua/src/Menus/UIMenu/Items/UIMenuProgressItem.lua

@@ -241,9 +241,21 @@ function UIMenuProgressItem:LeftBadge(Badge)
     end
 end
 
+function UIMenuProgressItem:CustomLeftBadge(txd,txn)
+    if txd ~= nil and txd ~= "" and txn ~= nil and txn ~= "" then
+        self.Base:CustomLeftBadge(txd,txn, self)
+    else
+        return self.Base:LeftBadge()
+    end
+end
+
 ---RightBadge
 function UIMenuProgressItem:RightBadge()
-    error("This item does not support badges")
+    error("This item does not support right badges")
+end
+
+function UIMenuProgressItem:CustomRightBadge()
+    error("This item does not support right badges")
 end
 
 ---RightLabel

ScaleformUI_Lua/src/Menus/UIMenu/Items/UIMenuSeperatorItem.lua

@@ -166,6 +166,15 @@ function UIMenuSeparatorItem:RightBadge()
     error("This item does not support badges")
 end
 
+function UIMenuSeparatorItem:CustomLeftBadge()
+    error("This item does not support badges")
+end
+
+---RightBadge
+function UIMenuSeparatorItem:CustomRightBadge()
+    error("This item does not support badges")
+end
+
 ---RightLabel
 function UIMenuSeparatorItem:RightLabel()
     error("This item does not support a right label")

ScaleformUI_Lua/src/Menus/UIMenu/Items/UIMenuSliderItem.lua

@@ -231,8 +231,20 @@ function UIMenuSliderItem:LeftBadge(Badge)
     end
 end
 
+function UIMenuSliderItem:CustomLeftBadge(txd,txn)
+    if txd ~= nil and txd ~= "" and txn ~= nil and txn ~= "" then
+        self.Base:CustomLeftBadge(txd,txn, self)
+    else
+        return self.Base:LeftBadge()
+    end
+end
+
 function UIMenuSliderItem:RightBadge()
-    error("This item does not support badges")
+    error("This item does not support right badges")
+end
+
+function UIMenuSliderItem:CustomRightBadge()
+    error("This item does not support right badges")
 end
 
 function UIMenuSliderItem:RightLabel()

ScaleformUI_Lua/src/Menus/UIMenu/Items/UIMenuStatsItem.lua

@@ -221,6 +221,14 @@ function UIMenuStatsItem:RightBadge()
     error("This item does not support badges")
 end
 
+function UIMenuStatsItem:CustomLeftBadge()
+    error("This item does not support badges")
+end
+
+function UIMenuStatsItem:CustomRightBadge()
+    error("This item does not support badges")
+end
+
 ---RightLabel
 function UIMenuStatsItem:RightLabel()
     error("This item does not support a right label")