MasterMerchant_Guild.lua

-- MasterMerchant Utility Functions File
-- Last Updated September 15, 2014
-- Written February 2015 by Chris Lasswell (@Philgo68) - Philgo68@gmail.com
-- Released under terms in license accompanying this file.
-- Distribution without license is prohibited!
local internal = _G["LibGuildStore_Internal"]

function MasterMerchant:CalculateCut(amount)
  local listingFee, tradingHouseCut, expectedProfit = GetTradingHousePostPriceInfo(amount)
  local cutAmount = zo_floor(tradingHouseCut / 2)
  return cutAmount
end

local MMSeller = {
  guild = {},
  sellerName = '',
  sales = {},
  tax = {},
  count = {},
  stack = {},
  rank = {},
  outsideBuyer = false,
  searchText = nil
}

function MMSeller:new(_guild, _name, _outsideBuyer, _searchText)
  local o = {}   -- create object if user does not provide one
  setmetatable(o, self)
  self.__index = self

  o.guild = _guild
  o.sellerName = _name
  o.sales = {}
  o.tax = {}
  o.count = {}
  o.stack = {}
  o.rank = {}
  o.outsideBuyer = _outsideBuyer
  o.searchText = zo_strlower(_searchText or '')
  return o
end

function MMSeller:addSale(rankIndex, amount, stack, sort, tax)
  if sort == nil then sort = not internal.isDatabaseBusy end

  if not self.sales[rankIndex] then
    self.sales[rankIndex] = 0
    self.tax[rankIndex] = 0
    self.count[rankIndex] = 0
    self.stack[rankIndex] = 0
  end

  self.sales[rankIndex] = self.sales[rankIndex] + amount
  self.tax[rankIndex] = self.tax[rankIndex] + MasterMerchant:CalculateCut(amount)
  self.count[rankIndex] = self.count[rankIndex] + 1
  self.stack[rankIndex] = self.stack[rankIndex] + stack

  local guildRanks = self.guild.ranks[rankIndex]

  if (not self.rank[rankIndex]) then
    -- add myself to the guild ranking list, then note my current place
    table.insert(guildRanks, self)
    self.rank[rankIndex] = #guildRanks
  end

  if sort then
    self:SortByIndexAndRanks(rankIndex, guildRanks)
  end
end

function MMSeller:SortByIndexAndRanks(rankIndex, guildRanks)
  while ((self.rank[rankIndex] or 0) > 1) and (guildRanks[self.rank[rankIndex] - 1]) and ((self.sales[rankIndex] or 0) > (guildRanks[self.rank[rankIndex] - 1].sales[rankIndex] or 0)) do

    local swapSeller = guildRanks[self.rank[rankIndex] - 1]

    local tempRank = swapSeller.rank[rankIndex]
    swapSeller.rank[rankIndex] = self.rank[rankIndex]
    self.rank[rankIndex] = tempRank

    -- Make sure guild lists point to the right sellers
    guildRanks[self.rank[rankIndex]] = self
    guildRanks[swapSeller.rank[rankIndex]] = swapSeller
  end
end

function MMSeller:removeSale(rankIndex, amount, stack)
  self.sales[rankIndex] = (self.sales[rankIndex] or 0) - amount
  self.tax[rankIndex] = (self.tax[rankIndex] or 0) - MasterMerchant:CalculateCut(amount)
  self.count[rankIndex] = (self.count[rankIndex] or 0) - 1
  self.stack[rankIndex] = (self.stack[rankIndex] or 0) - stack
end

function MMSeller:removeRankIndex(rankIndex)
  if (self.sales[rankIndex]) then self.sales[rankIndex] = nil end
  if (self.tax[rankIndex]) then self.tax[rankIndex] = nil end
  if (self.count[rankIndex]) then self.count[rankIndex] = nil end
  if (self.rank[rankIndex]) then self.rank[rankIndex] = nil end
end

MMGuild = {
  guildName = '',
  sellers = {},
  ranks = {},
  sales = {},
  tax = {},
  count = {},
  stack = {},
  rankIsDirty = {},
  initDateTime = GetTimeStamp(),
}

local function guild_system_offline()
  local weekCutoff = 1595962800 -- Tuesday, 28-Jul-20 19:00:00 UTC

  if GetWorldName() == 'EU Megaserver' then
    weekCutoff = 1595941200  -- Tuesday, 28-Jul-20 13:00:00 UTC
  end

  while weekCutoff + (7 * ZO_ONE_DAY_IN_SECONDS) < GetTimeStamp() do
    weekCutoff = weekCutoff + (7 * ZO_ONE_DAY_IN_SECONDS)
  end
  return weekCutoff
end

function MasterMerchant:BuildDateRangeTable()
  local _, weekCutoff = GetGuildKioskCycleTimes()
  local cycleTimes = {}
  if weekCutoff == 0 then
    -- guild system is down, do something about it
    weekCutoff = guild_system_offline() -- do not subtract time because of while loop
    cycleTimes.week_start = weekCutoff -- this is 7 day back already
    cycleTimes.kiosk_cycle = weekCutoff + (7 * ZO_ONE_DAY_IN_SECONDS) -- add 7 days for when week would end
  end

  -- Calc Day Cutoff in Local Time
  local dayCutoff = GetTimeStamp() - GetSecondsSinceMidnight()

  cycleTimes.oneStart = dayCutoff -- Today

  cycleTimes.twoStart = dayCutoff - ZO_ONE_DAY_IN_SECONDS -- Yesterday
  cycleTimes.twoEnd = dayCutoff

  -- This Week
  cycleTimes.threeStart = weekCutoff - (7 * ZO_ONE_DAY_IN_SECONDS) -- GetGuildKioskCycleTimes() minus 7 days
  cycleTimes.threeEnd = weekCutoff -- GetGuildKioskCycleTimes()

  -- Last Week
  cycleTimes.fourStart = cycleTimes.threeStart - (7 * ZO_ONE_DAY_IN_SECONDS) -- last week Tuesday flip
  cycleTimes.fourEnd = cycleTimes.threeStart -- last week end

  -- Previous Week
  cycleTimes.fiveStart = cycleTimes.fourStart - (7 * ZO_ONE_DAY_IN_SECONDS)
  cycleTimes.fiveEnd = cycleTimes.fourStart -- prior week end

  cycleTimes.sixStart = dayCutoff - (7 * ZO_ONE_DAY_IN_SECONDS) -- last 7 days
  cycleTimes.sevenStart = dayCutoff - (10 * ZO_ONE_DAY_IN_SECONDS) -- last 10 days
  cycleTimes.eightStart = dayCutoff - (30 * ZO_ONE_DAY_IN_SECONDS) -- last 30 days

  if MasterMerchant.systemSavedVariables.customTimeframeType == GetString(MM_CUSTOM_TIMEFRAME_HOURS) then
    cycleTimes.nineStart = dayCutoff - (MasterMerchant.systemSavedVariables.customTimeframe * ZO_ONE_HOUR_IN_SECONDS) -- last x hours
    cycleTimes.nineEnd = dayCutoff + (7 * ZO_ONE_DAY_IN_SECONDS)
  end
  if MasterMerchant.systemSavedVariables.customTimeframeType == GetString(MM_CUSTOM_TIMEFRAME_DAYS) then
    cycleTimes.nineStart = dayCutoff - (MasterMerchant.systemSavedVariables.customTimeframe * ZO_ONE_DAY_IN_SECONDS) -- last x days
    cycleTimes.nineEnd = dayCutoff + (7 * ZO_ONE_DAY_IN_SECONDS)
  end
  if MasterMerchant.systemSavedVariables.customTimeframeType == GetString(MM_CUSTOM_TIMEFRAME_WEEKS) then
    cycleTimes.nineStart = dayCutoff - ((MasterMerchant.systemSavedVariables.customTimeframe * ZO_ONE_DAY_IN_SECONDS) * 7) -- last x weeks
    cycleTimes.nineEnd = dayCutoff + (7 * ZO_ONE_DAY_IN_SECONDS)
  end
  if MasterMerchant.systemSavedVariables.customTimeframeType == GetString(MM_CUSTOM_TIMEFRAME_GUILD_WEEKS) then
    cycleTimes.nineStart = weekCutoff - ((MasterMerchant.systemSavedVariables.customTimeframe * ZO_ONE_DAY_IN_SECONDS) * 7) -- last x full guild weeks
    cycleTimes.nineEnd = weekCutoff
  end
  if cycleTimes.nineStart == nil then
    -- Default custom timeframe to Last 3 days if it's undefined
    MasterMerchant.systemSavedVariables.customTimeframeType = GetString(MM_CUSTOM_TIMEFRAME_DAYS)
    MasterMerchant.systemSavedVariables.customTimeframe = 3
    cycleTimes.nineStart = dayCutoff - (MasterMerchant.systemSavedVariables.customTimeframe * ZO_ONE_DAY_IN_SECONDS) -- last x days
    cycleTimes.nineEnd = dayCutoff + (7 * ZO_ONE_DAY_IN_SECONDS)
  end

  MasterMerchant.dateRanges = { }
  MasterMerchant.dateRanges[MM_DATERANGE_TODAY] = { }
  MasterMerchant.dateRanges[MM_DATERANGE_TODAY].startTimestamp = cycleTimes.oneStart
  MasterMerchant.dateRanges[MM_DATERANGE_YESTERDAY] = { }
  MasterMerchant.dateRanges[MM_DATERANGE_YESTERDAY].startTimestamp = cycleTimes.twoStart
  MasterMerchant.dateRanges[MM_DATERANGE_YESTERDAY].endTimestamp = cycleTimes.twoEnd
  MasterMerchant.dateRanges[MM_DATERANGE_THISWEEK] = { }
  MasterMerchant.dateRanges[MM_DATERANGE_THISWEEK].startTimestamp = cycleTimes.threeStart
  MasterMerchant.dateRanges[MM_DATERANGE_THISWEEK].endTimestamp = cycleTimes.threeEnd
  MasterMerchant.dateRanges[MM_DATERANGE_LASTWEEK] = { }
  MasterMerchant.dateRanges[MM_DATERANGE_LASTWEEK].startTimestamp = cycleTimes.fourStart
  MasterMerchant.dateRanges[MM_DATERANGE_LASTWEEK].endTimestamp = cycleTimes.fourEnd
  MasterMerchant.dateRanges[MM_DATERANGE_PRIORWEEK] = { }
  MasterMerchant.dateRanges[MM_DATERANGE_PRIORWEEK].startTimestamp = cycleTimes.fiveStart
  MasterMerchant.dateRanges[MM_DATERANGE_PRIORWEEK].endTimestamp = cycleTimes.fiveEnd
  MasterMerchant.dateRanges[MM_DATERANGE_7DAY] = { }
  MasterMerchant.dateRanges[MM_DATERANGE_7DAY].startTimestamp = cycleTimes.sixStart
  MasterMerchant.dateRanges[MM_DATERANGE_10DAY] = { }
  MasterMerchant.dateRanges[MM_DATERANGE_10DAY].startTimestamp = cycleTimes.sevenStart
  MasterMerchant.dateRanges[MM_DATERANGE_30DAY] = { }
  MasterMerchant.dateRanges[MM_DATERANGE_30DAY].startTimestamp = cycleTimes.eightStart
  MasterMerchant.dateRanges[MM_DATERANGE_CUSTOM] = { }
  MasterMerchant.dateRanges[MM_DATERANGE_CUSTOM].startTimestamp = cycleTimes.nineStart
  MasterMerchant.dateRanges[MM_DATERANGE_CUSTOM].endTimestamp = cycleTimes.nineEnd
end

function MMGuild:new(_name)
  local o = {}   -- create object if user does not provide one
  setmetatable(o, self)
  self.__index = self

  o.guildName = _name
  o.sellers = {}
  o.ranks = {}
  o.count = {}
  o.stack = {}
  o.sales = {}
  o.tax = {}
  o.kiosk_cycle = 0
  o.week_start = 0

  -- /script MasterMerchant:dm("Info", { GetGuildKioskCycleTimes() } )
  -- Calc Guild Week Cutoff
  local _, weekCutoff = GetGuildKioskCycleTimes()
  if weekCutoff == 0 then
    -- guild system is down, do something about it
    weekCutoff = guild_system_offline() -- do not subtract time because of while loop
    o.week_start = weekCutoff -- this is 7 day back already
    o.kiosk_cycle = weekCutoff + (7 * ZO_ONE_DAY_IN_SECONDS) -- add 7 days for when week would end
  end

  -- Calc Day Cutoff in Local Time
  local dayCutoff = GetTimeStamp() - GetSecondsSinceMidnight()

  o.oneStart = dayCutoff -- Today

  o.twoStart = dayCutoff - ZO_ONE_DAY_IN_SECONDS -- Yesterday
  o.twoEnd = dayCutoff

  -- This Week
  o.threeStart = weekCutoff - (7 * ZO_ONE_DAY_IN_SECONDS) -- GetGuildKioskCycleTimes() minus 7 days
  o.threeEnd = weekCutoff -- GetGuildKioskCycleTimes()

  -- Last Week
  o.fourStart = o.threeStart - (7 * ZO_ONE_DAY_IN_SECONDS) -- last week Tuesday flip
  o.fourEnd = o.threeStart -- last week end

  -- Previous Week
  o.fiveStart = o.fourStart - (7 * ZO_ONE_DAY_IN_SECONDS)
  o.fiveEnd = o.fourStart -- prior week end

  o.sixStart = dayCutoff - (7 * ZO_ONE_DAY_IN_SECONDS) -- last 7 days
  o.sevenStart = dayCutoff - (10 * ZO_ONE_DAY_IN_SECONDS) -- last 10 days
  o.eightStart = dayCutoff - (30 * ZO_ONE_DAY_IN_SECONDS) -- last 30 days

  if MasterMerchant.systemSavedVariables.customTimeframeType == GetString(MM_CUSTOM_TIMEFRAME_HOURS) then
    o.nineStart = dayCutoff - (MasterMerchant.systemSavedVariables.customTimeframe * ZO_ONE_HOUR_IN_SECONDS) -- last x hours
    o.nineEnd = dayCutoff + (7 * ZO_ONE_DAY_IN_SECONDS)
  end
  if MasterMerchant.systemSavedVariables.customTimeframeType == GetString(MM_CUSTOM_TIMEFRAME_DAYS) then
    o.nineStart = dayCutoff - (MasterMerchant.systemSavedVariables.customTimeframe * ZO_ONE_DAY_IN_SECONDS) -- last x days
    o.nineEnd = dayCutoff + (7 * ZO_ONE_DAY_IN_SECONDS)
  end
  if MasterMerchant.systemSavedVariables.customTimeframeType == GetString(MM_CUSTOM_TIMEFRAME_WEEKS) then
    o.nineStart = dayCutoff - ((MasterMerchant.systemSavedVariables.customTimeframe * ZO_ONE_DAY_IN_SECONDS) * 7) -- last x weeks
    o.nineEnd = dayCutoff + (7 * ZO_ONE_DAY_IN_SECONDS)
  end
  if MasterMerchant.systemSavedVariables.customTimeframeType == GetString(MM_CUSTOM_TIMEFRAME_GUILD_WEEKS) then
    o.nineStart = weekCutoff - ((MasterMerchant.systemSavedVariables.customTimeframe * ZO_ONE_DAY_IN_SECONDS) * 7) -- last x full guild weeks
    o.nineEnd = weekCutoff
  end
  if o.nineStart == nil then
    -- Default custom timeframe to Last 3 days if it's undefined
    MasterMerchant.systemSavedVariables.customTimeframeType = GetString(MM_CUSTOM_TIMEFRAME_DAYS)
    MasterMerchant.systemSavedVariables.customTimeframe = 3
    o.nineStart = dayCutoff - (MasterMerchant.systemSavedVariables.customTimeframe * ZO_ONE_DAY_IN_SECONDS) -- last x days
    o.nineEnd = dayCutoff + (7 * ZO_ONE_DAY_IN_SECONDS)
  end

  return o
end

function MMGuild:CreateGuildDataMap(map, key)
  if map[key] == nil then
    map[key] = MMGuild:new(key)
  end
  return map[key]
end

function MMGuild:addSale(sellerName, rankIndex, amount, stack, wasKiosk, sort, searchText)
  if sort == nil then sort = not internal.isDatabaseBusy end
  amount = tonumber(amount)
  if type(stack) ~= 'number' then stack = 1 end

  if not self.ranks[rankIndex] then
    self.ranks[rankIndex] = {}
    self.sales[rankIndex] = 0
    self.tax[rankIndex] = 0
    self.count[rankIndex] = 0
    self.stack[rankIndex] = 0
  end

  self.sales[rankIndex] = self.sales[rankIndex] + amount
  self.tax[rankIndex] = self.tax[rankIndex] + MasterMerchant:CalculateCut(amount)
  self.count[rankIndex] = self.count[rankIndex] + 1
  self.stack[rankIndex] = self.stack[rankIndex] + stack

  self.sellers[sellerName] = self.sellers[sellerName] or MMSeller:new(self, sellerName, wasKiosk, searchText)
  self.sellers[sellerName]:addSale(rankIndex, amount, stack, sort, tax)
end

function MMGuild:removeSale(sellerName, rankIndex, amount, stack)
  if (self.sellers[sellerName]) then self.sellers[sellerName]:removeSale(rankIndex, amount, stack) end

  self.sales[rankIndex] = (self.sales[rankIndex] or 0) - amount
  self.tax[rankIndex] = (self.tax[rankIndex] or 0) - MasterMerchant:CalculateCut(amount)
  self.count[rankIndex] = (self.count[rankIndex] or 0) - 1
  self.stack[rankIndex] = (self.stack[rankIndex] or 0) - stack
end

function MMGuild:removeRankIndex(rankIndex)
  --if (self.sellers[sellersName]) then self.sellers[sellersName]:removeRankIndex(rankIndex) end

  if (self.sales[rankIndex]) then self.sales[rankIndex] = nil end
  if (self.tax[rankIndex]) then self.tax[rankIndex] = nil end
  if (self.count[rankIndex]) then self.count[rankIndex] = nil end
  if (self.rank[rankIndex]) then self.rank[rankIndex] = nil end
end

function MMGuild:MarkDirty(rankIndex)
  self.rankIsDirty[rankIndex] = true
end

function MMGuild:MarkClean(rankIndex)
  self.rankIsDirty[rankIndex] = false
end

function MMGuild:IsDirty(rankIndex)
  return self.rankIsDirty[rankIndex]
end

function MMGuild:addSaleByDate(sellerName, timestamp, amount, stack, wasKiosk, sort, searchText, salesId)
  if sellerName == nil then return end
  if timestamp == nil then return end
  if type(timestamp) ~= 'number' then return end
  if salesId ~= nil and not MasterMerchant:ShouldUseSale(salesId) then return end

  if (timestamp >= self.oneStart) then
    self:MarkDirty(MM_DATERANGE_TODAY)
    self:addSale(sellerName, MM_DATERANGE_TODAY, amount, stack, wasKiosk, sort, searchText)
  end
  if (timestamp >= self.twoStart and timestamp < self.twoEnd) then
    self:MarkDirty(MM_DATERANGE_YESTERDAY)
    self:addSale(sellerName, MM_DATERANGE_YESTERDAY, amount, stack, wasKiosk, sort, searchText)
  end
  if (timestamp >= self.threeStart and timestamp < self.threeEnd) then
    self:MarkDirty(MM_DATERANGE_THISWEEK)
    self:addSale(sellerName, MM_DATERANGE_THISWEEK, amount, stack, wasKiosk, sort, searchText)
  end
  if (timestamp >= self.fourStart and timestamp < self.fourEnd) then
    self:MarkDirty(MM_DATERANGE_LASTWEEK)
    self:addSale(sellerName, MM_DATERANGE_LASTWEEK, amount, stack, wasKiosk, sort, searchText)
  end
  if (timestamp >= self.fiveStart and timestamp < self.fiveEnd) then
    self:MarkDirty(MM_DATERANGE_PRIORWEEK)
    self:addSale(sellerName, MM_DATERANGE_PRIORWEEK, amount, stack, wasKiosk, sort, searchText)
  end
  if (timestamp >= self.sixStart) then
    self:MarkDirty(MM_DATERANGE_7DAY)
    self:addSale(sellerName, MM_DATERANGE_7DAY, amount, stack, wasKiosk, sort, searchText)
  end
  if (timestamp >= self.sevenStart) then
    self:MarkDirty(MM_DATERANGE_10DAY)
    self:addSale(sellerName, MM_DATERANGE_10DAY, amount, stack, wasKiosk, sort, searchText)
  end
  if (timestamp >= self.eightStart) then
    self:MarkDirty(MM_DATERANGE_30DAY)
    self:addSale(sellerName, MM_DATERANGE_30DAY, amount, stack, wasKiosk, sort, searchText)
  end
  if (timestamp >= self.nineStart and timestamp < self.nineEnd) then
    self:MarkDirty(MM_DATERANGE_CUSTOM)
    self:addSale(sellerName, MM_DATERANGE_CUSTOM, amount, stack, wasKiosk, sort, searchText)
  end
end

function MMGuild:removeSaleByDate(sellerName, timestamp, amount, stack)
  if sellerName == nil then return end
  if (timestamp >= self.oneStart) then self:removeSale(sellerName, MM_DATERANGE_TODAY, amount) end
  if (timestamp >= self.twoStart and timestamp < self.twoEnd) then self:removeSale(sellerName, MM_DATERANGE_YESTERDAY, amount, stack) end
  if (timestamp >= self.threeStart and timestamp < self.threeEnd) then self:removeSale(sellerName, MM_DATERANGE_THISWEEK, amount, stack) end
  if (timestamp >= self.fourStart and timestamp < self.fourEnd) then self:removeSale(sellerName, MM_DATERANGE_LASTWEEK, amount, stack) end
  if (timestamp >= self.fiveStart and timestamp < self.fiveEnd) then self:removeSale(sellerName, MM_DATERANGE_PRIORWEEK, amount, stack) end
  if (timestamp >= self.sixStart) then self:removeSale(sellerName, MM_DATERANGE_7DAY, amount, stack) end
  if (timestamp >= self.sevenStart) then self:removeSale(sellerName, MM_DATERANGE_10DAY, amount, stack) end
  if (timestamp >= self.eightStart) then self:removeSale(sellerName, MM_DATERANGE_30DAY, amount, stack) end
  if (timestamp >= self.nineStart and timestamp < self.nineEnd) then self:removeSale(sellerName, MM_DATERANGE_CUSTOM, amount, stack) end
end

function MMGuild:SortByRankIndex(rankIndex)
  --if self:IsDirty(rankIndex) then
  MasterMerchant.shellSort(self.ranks[rankIndex] or {}, function(sortA, sortB)
    return (sortA.sales[rankIndex] or 0) > (sortB.sales[rankIndex] or 0)
  end)

  local i = 1
  while self.ranks[rankIndex] and self.ranks[rankIndex][i] do
    self.ranks[rankIndex][i].rank[rankIndex] = i
    i = i + 1
  end
  self:MarkClean(rankIndex)
  --end
end

function MMGuild:SortAllRanks()
  self:SortByRankIndex(MM_DATERANGE_TODAY)
  self:SortByRankIndex(MM_DATERANGE_YESTERDAY)
  self:SortByRankIndex(MM_DATERANGE_THISWEEK)
  self:SortByRankIndex(MM_DATERANGE_LASTWEEK)
  self:SortByRankIndex(MM_DATERANGE_PRIORWEEK)
  self:SortByRankIndex(MM_DATERANGE_7DAY)
  self:SortByRankIndex(MM_DATERANGE_10DAY)
  self:SortByRankIndex(MM_DATERANGE_30DAY)
  self:SortByRankIndex(MM_DATERANGE_CUSTOM)
end

function MMGuild:addPurchaseByDate(sellerName, timestamp, amount, stack, wasKiosk, sort, searchText)
  if sellerName == nil then return end
  if timestamp == nil then return end
  if type(timestamp) ~= 'number' then return end
  self:addSale(sellerName, MM_DATERANGE_TODAY, amount, stack, wasKiosk, sort, searchText)
end