From 6ac4b9af0649b1c51d4a016ef24febd94c1c33e5 Mon Sep 17 00:00:00 2001
From: Hendrik Leppkes <h.leppkes@gmail.com>
Date: Wed, 25 Feb 2009 16:48:05 +0100
Subject: [PATCH] Split StateBar configuration out of ActionBar code

---
 Options/ActionBar.lua                         |  9 -----
 Options/Options.xml                           |  2 +-
 Options/{ActionBarStates.lua => StateBar.lua} | 37 ++++++++++++-------
 3 files changed, 25 insertions(+), 23 deletions(-)
 rename Options/{ActionBarStates.lua => StateBar.lua} (88%)

diff --git a/Options/ActionBar.lua b/Options/ActionBar.lua
index 42571ac..8fbdc4f 100644
--- a/Options/ActionBar.lua
+++ b/Options/ActionBar.lua
@@ -87,15 +87,6 @@ function module:GetOptionsObject()
 			},
 		}
 		obj:AddElementGroup("general", cat_general)
-
-		local states = {
-			type = "group",
-			name = L["State Configuration"],
-			order = 5,
-			args = self:GetStateOptionsTable(),
-		}
-		obj:NewCategory("state", states)
-
 		self.baroptions = obj
 	end
 
diff --git a/Options/Options.xml b/Options/Options.xml
index 726aecf..c437bb0 100644
--- a/Options/Options.xml
+++ b/Options/Options.xml
@@ -4,9 +4,9 @@
 	
 	<Script file="Bar.lua"/>
 	<Script file="ButtonBar.lua"/>
+	<Script file="StateBar.lua"/>
 	
 	<Script file="ActionBar.lua"/>
-	<Script file="ActionBarStates.lua"/>
 	<Script file="BagBar.lua"/>
 	<Script file="MicroMenu.lua"/>
 	<Script file="PetBar.lua"/>
diff --git a/Options/ActionBarStates.lua b/Options/StateBar.lua
similarity index 88%
rename from Options/ActionBarStates.lua
rename to Options/StateBar.lua
index e6a713d..7239d13 100644
--- a/Options/ActionBarStates.lua
+++ b/Options/StateBar.lua
@@ -1,11 +1,13 @@
 local L = LibStub("AceLocale-3.0"):GetLocale("Bartender4")
-local ActionBar = Bartender4.ActionBar
 
-local module = Bartender4:GetModule("ActionBars")
+local Bar = Bartender4.Bar.prototype
+local ButtonBar = Bartender4.ButtonBar.prototype
+local StateBar = Bartender4.StateBar.prototype
 
-local optGetter, optSetter
+local optGetter, optSetter, getBar
 do
-	local getBar, optionMap, callFunc
+	local optionMap, callFunc
+	local barregistry = Bartender4.Bar.barregistry
 
 	optionMap = {
 		stance = "StanceStateOption",
@@ -19,17 +21,17 @@ do
 		custom = "StateOption",
 		customCopy = "CopyCustomConditionals",
 	}
-	-- retrieves a valid bar object from the modules actionbars table
+	-- retrieves a valid bar object from the barregistry table
 	function getBar(id)
-		local bar = module.actionbars[tonumber(id)]
-		assert(bar, "Invalid bar id in options table.")
+		local bar = barregistry[tostring(id)]
+		assert(bar, ("Invalid bar id in options table. (%s)"):format(id))
 		return bar
 	end
 
 	-- calls a function on the bar
 	function callFunc(bar, type, option, ...)
 		local func = type .. (optionMap[option] or option)
-		assert(bar[func], "Invalid get/set function."..func)
+		assert(bar[func], ("Invalid get/set function %s in bar %s."):format(func, bar.id))
 		return bar[func](bar, ...)
 	end
 
@@ -80,21 +82,22 @@ local function createOptionGroup(k, id)
 end
 
 local disabledFunc = function(info)
-	local bar = module.actionbars[tonumber(info[2])]
+	local bar = getBar(info[2])
 	return not bar:GetStateOption("enabled")
 end
 
 local stateOffOrCustomOn = function(info)
-	local bar = module.actionbars[tonumber(info[2])]
+	local bar = getBar(info[2])
 	return (not bar:GetStateOption("enabled")) or (bar:GetStateOption("customEnabled"))
 end
 
 local stateOffOrCustomOff = function(info)
-	local bar = module.actionbars[tonumber(info[2])]
+	local bar = getBar(info[2])
 	return (not bar:GetStateOption("enabled")) or (not bar:GetStateOption("customEnabled"))
 end
 
-function module:GetStateOptionsTable()
+function StateBar:GetOptionObject()
+	local obj = ButtonBar.GetOptionObject()
 	local options = {
 		enabled = {
 			order = 1,
@@ -267,5 +270,13 @@ function module:GetStateOptionsTable()
 		end
 	end
 
-	return options
+	local states = {
+		type = "group",
+		name = L["State Configuration"],
+		order = 5,
+		args = options,
+	}
+	obj:NewCategory("state", states)
+
+	return obj
 end