local Buffs = class()

function Buffs:init()
  self.buffs = {}
end

function Buffs:update()
  table.with(self.buffs, 'update')
end

function Buffs:add(player, code)
  local buff = self:get(player, code)
  if buff then return buff.stack and buff:stack() or buff end

  buff = new(data.buff[code])
  buff.owner = player
  f.exe(buff.activate, buff)

  table.insert(self.buffs, buff)

  return buff
end

function Buffs:remove(player, code, stack)
  local buff, i = self:get(player, code)
  if buff then
    f.exe(stack and buff.unstack or buff.deactivate, buff)
    if not stack then table.remove(self.buffs, i) end
  end
end

function Buffs:removeAll(player)
  for i = #self.buffs, 1, -1 do
    local buff = self.buffs[i]
    if buff.owner == player then
      f.exe(buff.deactivate, buff)
      table.remove(self.buffs, i)
    end
  end
end

function Buffs:get(player, code)
  for i = 1, #self.buffs do
    local buff = self.buffs[i]
    if buff.owner == player and buff.code == code then
      return buff, i
    end
  end
end

return Buffs