module Combat
  ##
  # A module for units which can have their own bonuses. E.G., weapons, equipment, ammo.
  module BonusContainer
    def attack_bonus(type)
      @attack_bonuses[type]
    end

    def defence_bonus(type)
      @defence_bonuses[type]
    end

    def other_bonus(type)
      @other_bonuses[type]
    end

    def other_bonuses(melee_strength: 0, ranged_strength: 0, prayer: 0)
      @other_bonuses = {
        melee_strength: melee_strength,
        ranged_strength: ranged_strength,
        prayer: prayer
      }
    end

    def defence_bonuses(stab: 0, slash: 0, crush: 0, magic: 0, range: 0)
      @defence_bonuses = {
        stab: stab,
        slash: slash,
        crush: crush,
        magic: magic,
        range: range
      }
    end

    def attack_bonuses(stab: 0, slash: 0, crush: 0, magic: 0, range: 0)
      @attack_bonuses = {
        stab: stab,
        slash: slash,
        crush: crush,
        magic: magic,
        range: range
      }
    end
  end
end