database.lua

----------------------------------------------------------------
-- KiwiHUD: Manage SavedVariables Database
----------------------------------------------------------------

local addon = KiwiHUD

local rootDB = { global = {}, profiles = {}, profileChars = {} }

local GetNumSpecializations = not addon.isClassic and GetNumSpecializations or function() return 1 end
local GetSpecialization     = not addon.isClassic and GetSpecialization     or function() return 1 end
local GetSpecializationInfo = not addon.isClassic and GetSpecializationInfo or function() return 1, "Default" end

----------------------------------------------------------------
-- Initialize Database
----------------------------------------------------------------

function addon:InitializeDB()
	local key = addon.addonName ..'DB'
	local db = _G[key]
	if not db then
		db = rootDB
		_G[key] = db
	end
	local charKey = UnitName("player") .. " - " .. GetRealmName()
	if not db.profiles['default']  then
		db.profiles['default'] = CopyTable(addon.defaults)
	end
	local profileChar = db.profileChars[charKey]
	if not profileChar then
		profileChar = {}
		for i=1,GetNumSpecializations() or 1 do
			profileChar[i] = 'default'
		end
		db.profileChars[charKey] = profileChar
		if not db.profiles[charKey] then
			db.profiles[charKey] = CopyTable(addon.defaults)
		end
	end
	addon.__profileChar = profileChar
	addon.__db = db
	addon:PLAYER_TALENT_UPDATE()
	if not addon.isClassic then
		addon:RegisterEvent("PLAYER_TALENT_UPDATE")
	end
end

----------------------------------------------------------------
-- Talent change management
----------------------------------------------------------------

function addon:PLAYER_TALENT_UPDATE()
	local profileKey = self.__profileChar[ GetSpecialization() or 1 ] or 'default'
	if profileKey ~= self.__profileKey then
		local old_db = self.db
		local new_db = self.__db.profiles[profileKey] or self.__db.profiles.default
		self.__profileKey = profileKey
		self.db = new_db
		if old_db then
			self:OnProfileChanged()
		end
	end
end

----------------------------------------------------------------
-- Profile management options
----------------------------------------------------------------

local values, undeletable = {}, {}

local function GetProfiles()
	wipe(values)
	for key in pairs(addon.__db.profiles) do
		values[key] = key
	end
	return values
end

local function GetSpecName(i)
	if i<=GetNumSpecializations() then
		local name = select(2,GetSpecializationInfo(i))
		if i==GetSpecialization() then
			name = name .. ' (active)'
		end
		return name
	else
		return ''
	end
end

local function GetSpecProfile(specIndex)
	return addon.__profileChar[specIndex] or 'default'
end

local function SetSpecProfile(specIndex, key)
	addon.__profileChar[specIndex] = key
	if GetSpecialization() or specIndex == 1 then
		addon:PLAYER_TALENT_UPDATE()
	end
end

-- profile selection per specialization

local options = addon:SetupOptions( 'Profiles', 'Profiles', {} )
for i=1,6 do
	options['spec'..i] = {
		type   = 'select',
		order  = i,
		name   = function() return GetSpecName(i) end,
		get    = function() return GetSpecProfile(i) end,
		set    = function(_, key) SetSpecProfile(i,key) end,
		hidden = function() return i>GetNumSpecializations() end,
		values = GetProfiles,
	}
end

-- profile database maintenance operations

addon:SetupOptions( 'Profiles', 'Operations', {
	newDesc = {
		type = 'description',
		order = 0.5,
		name = "\nYou can create a new profile by entering a name in the editbox.",
	},
	newProfile = {
		type = 'input',
		name = 'New Profile Name',
		order = 1,
		get = function() end,
		set = function(info,name)
			name = strtrim(name)
			addon.__db.profiles[name] = CopyTable( addon.defaults )
			SetSpecProfile( GetSpecialization() or 1, name)
		end,
		validate = function(info,name)
			name = strtrim(name)
			return strlen(name)>2 and not addon.__db.profiles[name]
		end,
	},
	copyDesc = {
		type = 'description',
		order = 1.5,
		name = "\nCopy the settings from one existing profile into the currently active profile.",
	},
	copyProfile = {
		type   = 'select',
		order  = 2,
		name   = 'Copy From',
		desc   = "Copy the settings from one existing profile into the currently active profile.",
		get    = function() end,
		set    = function(_, key)
			local profiles = addon.__db.profiles
			profiles[addon.__profileKey] = CopyTable( profiles[key] )
			addon.__profileKey = nil
			addon:PLAYER_TALENT_UPDATE()
		end,
		confirm = function() return "Selected profile will be copied into the current profile and current profile settings will be lost. Are you sure ?" end,
		values = function()
			wipe(values)
			for key in pairs(addon.__db.profiles) do
				if key ~= addon.__profileKey then
					values[key] = key
				end
			end
			return values
		end,
	},
	deleteDesc = {
		type = 'description',
		order = 2.5,
		name = "\nYou can delete unused profiles from the database to save space.",
	},
	deleteProfile = {
		type   = 'select',
		order  = 3,
		name   = 'Delete Profile',
		get    = function() end,
		set    = function(_, key)
			addon.__db.profiles[key] = nil
			addon:Update()
		end,
		confirm = function() return "Are you sure you want to delete the selected profile?" end,
		values = function()
			wipe(values)
			wipe(undeletable)
			for _,keys in pairs(addon.__db.profileChars) do
				for _,key in pairs(keys) do	undeletable[key] = true	end
			end
			for key in pairs(addon.__db.profiles) do
				if not undeletable[key] then values[key] = key end
			end
			return values
		end,
	},

	footer = { type = "description", order = 100, name = " " },

} )