databroker.lua

local ADDON_NAME, Addon = ...;
local L = LibStub("AceLocale-3.0"):GetLocale(ADDON_NAME);
local LibDataBroker = LibStub("LibDataBroker-1.1");

local bagTotal = "";
local bankTotal = "";
local grandTotal = "";
local gbankTotal = "";
local accountBankTotal = "";
local bankLastUpdated = "never";
local gbankLastUpdated = "never";
local accountBankLastUpdated = "never";
local ldbLabelText = "";
local topContributors = {};

local private = {};

local settings = {
  type = "data source",
  label = ADDON_NAME,
  text = ADDON_NAME .. " " .. L["text_display"],
  icon = "Interface\\Icons\\Inv_Ingot_03",
  OnTooltipShow = function(tooltip)
    Addon.DisplayToolTip(tooltip)
  end,
  OnClick = function(self, button, down)
    if button == "LeftButton" then
      Settings.OpenToCategory(Addon.CONST.METADATA.NAME)
      Settings.OpenToCategory(Addon.CONST.METADATA.NAME)
    end
  end,
};

function Addon.DisplayToolTip(tooltip)
  if (tooltip and tooltip.AddLine) then
    tooltip:AddDoubleLine(ADDON_NAME, ldbLabelText, 1, 1, 1)
    tooltip:AddLine(" ");
    tooltip:AddDoubleLine(L["bags"] .. ":", bagTotal, 1, 1, 1)
    private.addTopContributorSection(tooltip, "Bag")
    tooltip:AddDoubleLine(L["bank"] .. ":", bankTotal, 1, 1, 1)
    private.addTopContributorSection(tooltip, "Bank")
    if Addon.GetFromDb("accountBank", "enabled") and Addon.GetFromDb("bagFilter", "selectedBags") == "all" then
      tooltip:AddDoubleLine(L["account_bank"] .. ":", accountBankTotal, 1, 1, 1)
      private.addTopContributorSection(tooltip, "AccountBank")
    end
    if Addon.GetFromDb("guildBank", "enabled") and IsInGuild() then
      tooltip:AddDoubleLine(L["guild_bank"] .. ":", gbankTotal, 1, 1, 1)
      private.addTopContributorSection(tooltip, "GuildBank")
    end
    tooltip:AddDoubleLine(L["total"] .. ":", grandTotal, 1, 1, 1)
    tooltip:AddLine(" ");
    tooltip:AddDoubleLine(L["based_on"] .. ":", Addon.CONST.PRICE_SOURCE[Addon.GetFromDb("pricesource", "source")])
    tooltip:AddDoubleLine(L["bank_last_updated"] .. ":", bankLastUpdated)
    if Addon.GetFromDb("accountBank", "enabled") and Addon.GetFromDb("bagFilter", "selectedBags") == "all" then
      tooltip:AddDoubleLine(L["account_bank_last_updated"] .. ":", accountBankLastUpdated)
    end
    if Addon.GetFromDb("guildBank", "enabled") and IsInGuild() then
      tooltip:AddDoubleLine(L["guild_bank_last_updated"] .. ":", gbankLastUpdated)
    end
    tooltip:AddLine(" ");
    tooltip:AddLine(L["open_options"]);
  end
end

function private.addTopContributorSection(tooltip, source)
  -- Addon.Debug.Log("attempting to add top contributor section for " .. source .. " to tooltip")
  -- bSettingEnabled = Addon.GetFromDb("topContributors", "enabled")
  -- Addon.Debug.Log("topContributors setting is " .. tostring(bSettingEnabled))
  -- Addon.Debug.Log("topContributors table is " .. tostring(topContributors))
  -- Addon.Debug.Log("topContributors table size is " .. tostring(Addon:tableSize(topContributors)))
  -- Addon.Debug.Log("topContributors table source size is " .. tostring(Addon:tableSize(topContributors[source])))
  if Addon.GetFromDb("topContributors", "enabled") and topContributors and topContributors[source] and Addon:tableSize(topContributors[source]) > 0 then
    -- Addon.Debug.Log("topContributors section is enabled and has data")
    local precision = Addon.GetFromDb("moneyPrecision", "tooltip")
    tooltip:AddLine(format(L["top_contributors"], Addon.GetFromDb("topContributors", "limit")))
    for k, v in ipairs(topContributors[source]) do
      tooltip:AddDoubleLine(format("%s x%d", v.itemLink, v.count),
        GetMoneyString(Addon:round(v.totalValue, precision), true))
    end
    tooltip:AddLine(" ");
  end
  -- Addon.Debug.Log("end top contributor section for " .. source)
end

function Addon:InitializeDataBroker()
  Addon.BrokerModule = LibDataBroker:NewDataObject(ADDON_NAME, settings);
  Addon.icon = LibStub("LibDBIcon-1.0");
  minimapIconSettings = Addon.GetFromDb("minimapIcon");
  Addon.icon:Register(Addon.CONST.METADATA.NAME, Addon.BrokerModule, minimapIconSettings)
  if minimapIconSettings.hide == true then
    Addon.icon:Hide(Addon.CONST.METADATA.NAME)
  else
    Addon.icon:Show(Addon.CONST.METADATA.NAME)
  end
end

function Addon:UpdateDataBrokerText(text)
  if (Addon.BrokerModule ~= nil) then
    Addon.BrokerModule.text = text;
  end
  ldbLabelText = text;
end

function Addon:UpdateBagTotalText(text)
  bagTotal = text;
end

function Addon:UpdateBankTotalText(text)
  bankTotal = text;
end

function Addon:UpdateGrandTotalText(text)
  grandTotal = text;
end

function Addon:UpdateBankLastUpdatedText(text)
  bankLastUpdated = text;
end

function Addon:UpdateGBankLastUpdatedText(text)
  gbankLastUpdated = text;
end

function Addon:UpdateAccountBankLastUpdatedText(text)
  accountBankLastUpdated = text;
end

function Addon:UpdateGBankTotalText(text)
  gbankTotal = text;
end

function Addon:UpdateAccountBankTotalText(text)
  accountBankTotal = text;
end

function Addon:SetTopContributors(tbl)
  topContributors = tbl;
end

function Addon:tableSize(tbl)
  local size = 0;
  for k, v in pairs(tbl) do
    size = size + 1;
  end
  return size;
end