-- User: Krakyn
-- Created: 12/15/2015 7:31 PM
--[[
--  Turok by @project-author@
-- @file-author@
-- @file-revision@:@project-revision@
-- @file-date-iso@
--
-- Visible element operations begin here
--]]
local _G = _G
local T, pairs, select, setmetatable, type, tinsert = _G.Turok, pairs, select, setmetatable, type, tinsert
local mod = T:NewModule("PowerBar")
local UnitPower, UnitPowerMax, GetTalentInfoByID, GetTalentInfo, CreateFrame = UnitPower, UnitPowerMax, GetTalentInfoByID, GetTalentInfo, CreateFrame
local bar, db, prototype -- convenience upvalues
local cType, cText, cNum, cWord, cKey, cPink, cBool = cText, cNum, cWord, cKey, cPink, cBool
  --@debug
local cType, cText, cNum, cWord, cKey, cPink, cBool = cText, cNum, cWord, cKey, cPink, cBool
local print = function(...)
  if _G.Devian and _G.DevianDB.workspace ~= 1 then
    _G.print('PowerBar', ...)
  end
end
local uprint = function(...)
  if _G.Devian and _G.DevianDB.workspace ~= 1 then
    _G.print('Update', ...)
  end
end
local addon, tg = ...
tg.what = tostring(tg.what)..'more'
print(tg.what)
--@end-debug@
mod.OnInitialize = function(self)
  self.UNIT_SPELLCAST_START = self.SpellCastEvent
  self.UNIT_SPELLCAST_STOP = self.SpellCastEvent
  self.UNIT_SPELLCAST_SUCCEEDED = self.SpellCastEvent
  self.UNIT_SPELLCAST_CHANNEL_START = self.SpellCastEvent
  self.UNIT_SPELLCAST_CHANNEL_STOP = self.SpellCastEvent
  self.SPELL_UPDATE_COOLDOWN = self.SpellCooldownEvent
  self.PLAYER_REGEN_DISABLED = self.CombatStart
  self.PLAYER_REGEN_ENABLED = self.CombatEnd
  self.focusbar = {}
  self.parserLog = {}
  self.currentParse = {}
end
local SPELL_POWER_MANA, SPELL_POWER_ENERGY, SPELL_POWER_RAGE, SPELL_POWER_FOCUS = SPELL_POWER_MANA, SPELL_POWER_ENERGY, SPELL_POWER_RAGE, SPELL_POWER_FOCUS
local SPELL_POWER_SHADOW_ORBS = SPELL_POWER_SHADOW_ORBS
local SPELL_POWER_SOUL_SHARDS, SPELL_POWER_BURNING_EMBERS, SPELL_POWER_DEMONIC_FURY = SPELL_POWER_SOUL_SHARDS, SPELL_POWER_BURNING_EMBERS, SPELL_POWER_DEMONIC_FURY
local SPELL_POWER_HOLY_POWER = SPELL_POWER_HOLY_POWER
local SPELL_POWER_CHI = SPELL_POWER_CHI
local SPELL_POWER_COMBO_POINTS = SPELL_POWER_COMBO_POINTS

-- indexes for talent_update cleanup
mod.secondary_rows = {}
mod.disabled_frames = {}

--[[
--  Prototype list naming all the data sources and events that need to be handled for the logged in character
--  .power_type   {[bliz const] = event token}   list of resources represented by global SPELL_POWER_* constants in the blizzard ui and the UNIT_POWER* token argument representing it
--       .frame   string                         frameXML template
--      .spells   {[spell name/id] = {events}}   list of spells tracked by the updater
--    .secondary   {[aura name] = {}}             list of auras tracked as secondary resources such as Thrill of Hunt, Anticipatin, Evangelism, etc.
--]]
mod.prototype = {
  ['HUNTER'] = {
    primary = {
      [1] = {"FOCUS", SPELL_POWER_FOCUS}, -- array of power type constants associated to event strings
    },
    frame = 'TkThinComboTemplate',           ---------------------- desired frame template
    spells = {
      ["Steady Shot"] = {'UNIT_SPELL_CAST_SUCCEEDED', 'UNIT_SPELLCAST_STOP', 'UNIT_SPELLCAST_START'} -- spell events that this frame should listen to
    },
    secondary = {},
    spec = {
      [1] = {
        secondary = {
          ['Frenzy'] = {
            type = 'aura',
            order = 1,
            scale = 5,
            filters = 'HELPFUL',
            max = 5,
            specPage = 1,
            unit = 'player',
            spellID = 19623,
          },
          ['Focus Fire'] = {
            type = 'aura',
            max = 40,
            order = 2,
            scale = 5,
            line = 3, --------------- use this subtext value instead of count field
            filters = 'HELPFUL',
            specPage = 1,
            unit = 'player',
            spellID = 19623,
          }
        }
      },
      [2] = {
        secondary = {                       ------------------------ list of buffs that act as a secondary resource
          ['Thrill of the Hunt'] = {
            order = 1,
            unit = 'player',
            type = 'aura',
            max = 3,
            scale = 5,
            filters = 'HELPFUL',
            talent   = {4,3},
            display  = 'progressbar'
          },
        },
      },
    },
  },
  ['PRIEST'] = {
    primary = {
      [1] = {'MANA', SPELL_POWER_MANA}
    },
    frame = 'TkThinComboTemplate',
    secondary = {},
    spec = {
      [1] = {
        secondary = {
          ['Evangelism'] = {
            order = 1,
            max = 5,
            scale = 5,
            type='aura',
            unit = 'player',
            filters = 'HELPFUL|PLAYER',
            spellID = 81662,
          }
        }
      },
      [3] = {
        primary = {
          [1] = {'SHADOW_ORBS', SPELL_POWER_SHADOW_ORBS},
        },
        secondary = {
          ["Surge of Darkness"] = {
            order = 2,
            type = 'aura',
            filters = 'HELPFUL|PLAYER',
            spellID = 87160,
            talentID = 21751,
            max = 3,
            scale = 5,
            unit = 'player',
          },
          ["Insanity"] = {
            order = 2,
            type = 'aura',
            binary = true,
            regress = true,
            size = 1, scale = 1, max = 1,
            filters = 'HELPFUL|PLAYER',
            spellID = 132573,
            unit = 'player',
            talentID = 21753},
        }
      },
    },
  },
  ['ROGUE'] = {
    primary = {
      [1] = {'ENERGY', SPELL_POWER_ENERGY},
      [2] = {'COMBO_POINTS', SPELL_POWER_COMBO_POINTS}
    },
    frame = 'TkThinComboTemplate',
    secondary = {
      ['Anticipation'] = {
        type = 'aura',
        order = 1,
        max = 5,
        scale = 5,
        unit = 'player',
        talentID = 19250,
      },
    },
    spec = {},
  },
  ['MAGE'] = {
    primary = {
      [1] = {'MANA', SPELL_POWER_MANA},
    },
    frame = 'TkThinComboTemplate',
    secondary = {

      ["Incanter's Flow"] = {
        type = 'aura',
        unit = 'player',
        filters = 'HELPFUL|PLAYER',
        spellID = 1463,
        max = 5,
        scale = 5,
        order = 2,
        talentID = 16033,
      },
      ["Rune of Power"] = {
        type = 'aura',
        unit = 'player',
        filters = 'HELPFUL|PLAYER',
        binary = true,
        max = 1,
        scale = 1,
        order = 2,
        talentID = 16032,
      }
    },
    spec = {
      [1] = {
        secondary = {
          ['Arcane Charge'] = {
            type  ='aura',
            unit = 'player',
            filters = 'HARMFUL|PLAYER',
            spellID = 114664,
            scale = 4,
            max = 4,
            order = 1,
          },
        }
      },
      [3] = {
        secondary = {
          ['Fingers of Frost'] = {
            type = 'aura',
            unit = 'player',
            filters = '',
            spellID = 112965,
            max = 2,
            scale = 4,
            order = 1,
          },
          ['Brain Freeze'] = {
            type = 'aura',
            unit = 'player',
            filters = '',
            spellID = 44549,
            scale = 4,
            max = 2,
            order = 1,
            mirror = true,
          }
        }
      },
    }
  }
}
local P = mod.prototype

function mod:OnEnable()
  self.disabled_freams = {
    [T.playerClass] = {
      [T.specPage] = {}
    }
  }
  self.watched_units = {}
  self.watched_auras = {}
  self.watched_spells = {}
  self.db = TurokData.powerbar
  db = self.db

  self:Prototype_Init()
end
function mod:Prototype_Init()
  -- consult prototype vars
  prototype = {}
  mod.dcopy = function(t1, t2, d)
    d = d or ''
    for k,v in pairs(t2) do
      if type(v) == 'table' then
        if type(t1[k]) ~= 'table' then
          t1[k] = {}
          print(d, 'adding table', cKey(k))
        else
          print(d, 'merging tables', cKey(k))
        end
        mod.dcopy(t1[k], v, d..'  ')
      else
        if t1[k] then
          print(d, 'clobbered', k)
        else
          print(d, k, '=', cType(v))
        end
        t1[k] = v
      end
    end
  end
  mod.dcopy(prototype, mod.prototype[T.playerClass])
  if mod.prototype[T.playerClass].spec[T.specPage] then
    mod.dcopy(prototype, mod.prototype[T.playerClass].spec[T.specPage])
  end

  mod.thisproto = prototype

  print('|cFFFF0088Template:|r', 'Frame', 'TkPowerBarFrame', UIParent, prototype.frame)
  db = self.db

  if bar and bar.GetObjectType then
    bar:Hide()
    mod.disabled_frames[bar.specPage] = bar
    print('putting away old frame')
  end

  if not bar then
    bar = CreateFrame('Frame', 'TkPowerBar', UIParent, prototype.frame)
  end
  bar.specPage = T.specPage
  bar.specID = T.specID
  bar.primary = {}   -- {current, max, token}
  bar.secondary = {} -- {current, max, token}
  bar.aura = {}      -- {name, duration, expires, unit, flags}
  bar.spell = {}     -- copy of the last T.spellevent match
  print('  setting layout', db)
  print(bar:GetName())
  T.SetFrameLayout(bar, prototype.cvars and db[prototype.cvars] or db)
    T.SetStatusTextures(bar, db)

  print('  setting methods')
  bar.Init   = mod.Bar_Init
  bar.Event  = mod.Bar_Event
  bar.Update = mod.Bar_Update


  --- loop through aura definitions and flag accordingly
  print('Primary power types:')
  for order, power_data in pairs(prototype.primary) do
    print( order, unpack(power_data))
    local  token, power_type = unpack(power_data)
    local power, max = UnitPower('player', power_type), UnitPowerMax('player', power_type)
    bar.primary[token] = {power, max, power_type, order}
    print('  ', cKey(token), '= {', power, max,  power_type, order, '}')
  end

  --- go through secondary data args and assign the appropriate source functions
  local useAura, useCooldown
  local used_rows = {}
  if prototype.secondary then
    mod.secondary = {}
    for name, c in pairs(prototype.secondary) do
      local isActive = true
      print('parsing extra handler', name)
      if c.talentID then
        print(c.talentID, T.specPage)
        isActive = (type(c.talentID) == 'table') and select(4, GetTalentInfo(unpack(c.talentID), T.specGroup)) or
            select(4, GetTalentInfoByID(c.talentID, T.specGroup))
        print('  talentID:', cNum(isActive))
      end
      if isActive then
        local sc = {}

        sc = c
        print('  enable:', cNum(isActive), cWord(c.type))
        if c.type == 'aura' then
          sc.spellName = name
          if c.binary then
            sc.Get = function(self)
              print('get: UnitAura', self.unit, self.spellName, c.filters)
              local exists = UnitAura(self.unit, self.spellName, nil, self.filters)
              return (exists) and 1 or 0
            end
          else
            sc.Get = function(self)
              print('get: UnitAura', self.unit, self.spellName, c.filters)
              local _,_,_, count = UnitAura(self.unit, self.spellName, nil, self.filters)
              return count or 0
            end
          end

          useAura = true
        elseif c.type == 'cooldown' then
          if c.inverse then
            sc.Get = function(self)
              local start, duration, enabled = GetSpellCooldown(c.spellID)
              sc[1] = (duration > 0) and (GetTime() - start) or c.max
              print('get: GetSpellCooldown (inverse)', c.spellID, '=', sc[1])
            end
          else
            sc.Get = function(self)
              local start, duration, enabled = GetSpellCooldown(c.spellID)
              sc[1] = (duration > 0) and (start + duration - GetTime()) or 0
              print('get: GetSpellCooldown', c.spellID, '=', sc[1])
            end
          end
          useCooldown = true
        end
        print('  committing', name, 'to row', sc.order)
        bar.secondary[name] = sc
        used_rows[sc.order] = true -- index the drawn rows for talent_update
      end
    end
  end


  if useAura then bar:RegisterEvent('UNIT_AURA') end
  if useCooldown then bar:RegisterEvent('UNIT_SPELLCAST_SUCCEEDED') end

  bar:SetScript('OnUpdate', nil) -- make sure any xml embeds are cleaned out
  bar:SetScript('OnEvent', mod.Bar_Event)
  bar:RegisterEvent('UNIT_POWER_FREQUENT')

  bar:Init()
  bar:Show()

  -- metrics used by data plots
  bar.width = db.width
  bar.foreground_inset = db.foreground_inset
  bar.right_edge = bar:GetRight()
  bar.fill_limit = bar.right_edge
  bar.foreground.width = bar.width + (bar.foreground_inset)
  bar.spacing = 1

  mod.powerbar = bar
end

mod.Bar_Init = function(self)
  local mainPower, comboPower
  for token, power in pairs(self.primary) do
    if power[4] == 1 then
      mainPower = power
      elseif power[4] == 2 then
      comboPower = power
    end
  end

  if mainPower then
    local power, max, type, token = unpack(mainPower)
    if power and max then
      self.powerText:SetText(power)
      self:SetProgress(power/max)
    end
  end

  if comboPower then
    local power, max, type, token = unpack(comboPower)
    local px = (self.width-db.secondary.spacing* (max -1)-db.secondary.padding*2) / max
    self.combo = {}
    for i = 1, max do
        if not self.combo[i] then
        self.combo[i] = self:CreateTexture('TkPrimaryResourcePellet'..i, 'OVERLAY')
        end

        local k = i - 1
        local cx = db.secondary.padding + px * k + db.secondary.spacing * k
        local cy = db.secondary.padding
        self.combo[i]:ClearAllPoints()
        self.combo[i]:SetSize(px, db.secondary.height)
        self.combo[i]:SetPoint(db.secondary.anchor, self, db.secondary.anchorTo, cx, cy)
        --print('    ', self.combo[i]:GetName(), self.pointsize1, cx, cy, self.combo[i]:GetDrawLayer())

        self.combo[i]:Show()
    end
  end


  if self.secondary then
    if not self.resources then
      print('|cFFFF0000creating resources block')
      self.resources = {}
    else
      local hidecount = 0
      for i, row in pairs(self.resources) do
        for j, col in pairs(row) do
          col:Hide()
          hidecount = hidecount + 1
        end
      end
      print('hiding', hidecount, 'regions')
    end
    for name, secondary in pairs(self.secondary) do
      local n = secondary.order
      local sid = 'secondary'..n
      local c = db[sid] or db
      if not self.resources[n] then
        print('  |cFFFF8800creating resource row')
        self.resources[n] = {}
      end
      local row = self.resources[n]

      print('secondary resource', cText(name), 'max= '..cNum(secondary.max), 'scale= '..cNum(secondary.scale))
      local px = c.padding
      local pw = (self.width - c.padding*2 - c.spacing * (secondary.scale - 1)) / secondary.scale
      for i = 1, (secondary.max or 1) do
        if not row[i] then
          row[i] = bar:CreateTexture('TkResourcePellet.'..tostring(secondary.order)..'.'..tostring(i))
        end
        row[i]:Show()
        row[i]:SetDrawLayer('OVERLAY', sid)
        row[i]:SetPoint('BOTTOMLEFT', self, 'TOPLEFT', px, c.y)
        row[i]:SetSize(pw, db[sid].height or db.height)

        print('  *', cNum(i), cKey(sid), cNum(px), cNum(c.padding))
        px = px + pw + c.spacing
      end
    end
  end

  mod.Bar_Event(self, nil, 'player')
end

-- we only want to update at specific points
mod.Bar_Event = function(self, event, ...)
  local unit, token =  ...
  uprint( event, unit, token)
  if token and unit == 'player' then
    mod.Bar_Power(self, token)
  end
  --print(unit, token, ...)
  mod.Bar_Aura(bar, event, unit, token, ...)
end

mod.Bar_Aura = function (self, event, unit)
  uprint('bar updating function called', event, unit)


  if event == 'UNIT_AURA' or event == nil then
    for token, info in pairs(self.secondary) do
      local row = self.resources[info.order]
      if info.unit == unit then
        local count = info.Get(info)
        local db = db['secondary'..info.order] or db
        for i = 1, info.max do
          local palette = (i > count) and ('background_color') or ('foreground_color')

          print(token, i, count, (i > count), palette, unpack(db[palette]))
          row[i]:SetTexture(unpack(db[palette]))
        end
      end
    end
  end
end

function mod:Bar_Power(token)
  if not self.primary[token] then
    return
  end

  local p = self.primary[token]
  -- 1=cur, 2=max, 3=type, 4=token
  p[1] = UnitPower('player', p[3])
  p[2] = UnitPowerMax('player', p[3])
  uprint(' ', table.concat(self.primary[token],', '))

  if p[4] == 1 then
    uprint( 'progress:', p[1]/p[2])
    --print(unpack(p))
    self.powerText:SetText(p[1])
    self:SetProgress(p[1]/p[2])
  elseif p[4] == 2 then
    --print('update on', token, 'c:', p[1], 'm:', p[2])
    self.secondaryText:SetText(p[1])
    for i = 1, p[2] do
      local palette = (i > p[1]) and 'background_color' or 'foreground_color'
      self.combo[i]:SetTexture(unpack(db.secondary[palette]))

    end
  end
end

--- Spell parsing
function mod:SpellCastEvent(e, u, spellName, rank, castID, spellID)
  if u ~= 'player' then
    return true
  end
  if e == 'UNIT_SPELLCAST_DELAYED' then
  elseif e == 'UNIT_SPELLCAST_START' then
    bar.casting = true
    bar.spellevent = T.spellevent[u]
    bar.spell = T.casting[u]
  elseif e == 'UNIT_SPELLCAST_CHANNEL_START' then
    bar.channeling = true
    bar.spellevent = T.spellevent[u]
    bar.spell = T.channeling[u]
  elseif e == 'UNIT_SPELLCAST_SUCCEEDED' then
  elseif e == 'UNIT_SPELLCAST_STOP' then
    bar.casting = nil
    bar.casting = nil
  elseif e == 'UNIT_SPELLCAST_CHANNEL_STOP' then
    bar.channeling = nil
    bar.channeling = nil
  end
end

function mod:PLAYER_TALENT_UPDATE(event, unit)
  print(cText('*** Talent Update'), cKey('Spec:'), cWord(T.specName), cNum(T.specPage))
  mod:Prototype_Init()
end