battle_pet 0.1.1 → 0.1.2

data/lib/battle_pet.rb

@@ -3,9 +3,10 @@ require 'json'
 require 'yaml'
 
 require_relative 'pet_type.rb'
+require_relative 'pet_ability.rb'
 
 class BattlePet
-  attr_accessor :id, :name, :description, :source, :type, :creature
+  attr_accessor :id, :name, :description, :source, :type, :creature, :abilities
 
   REGION_HOSTS = { us: 'us.battle.net',
                    eu: 'eu.battle.net',
@@ -25,6 +26,7 @@ class BattlePet
     @can_battle = info["canBattle"]
     @type = PetType.find info["petTypeId"]
     @creature = info["creatureId"]
+    @abilities = acquire_abilities info["abilities"]
   end
 
   def can_battle?
@@ -40,4 +42,10 @@ class BattlePet
   def find_name(locale)
     PET_NAMES[id] && PET_NAMES[id]["name"][locale.to_s]
   end
+
+  def acquire_abilities(abilities)
+    results = {}
+    abilities.each { |ability| results[ability['requiredLevel']] = PetAbility.new(ability) }
+    results
+  end
 end

data/lib/pet_ability.rb

@@ -0,0 +1,11 @@
+class PetAbility
+  attr_accessor :id, :name, :type, :cooldown, :rounds
+
+  def initialize(params)
+    @id = params["id"]
+    @name = params["name"]
+    @type = PetType.find(params["petTypeId"])
+    @cooldown = params["cooldown"]
+    @rounds = params["rounds"]
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: battle_pet
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
   prerelease: 
 platform: ruby
 authors:
@@ -19,6 +19,7 @@ extra_rdoc_files: []
 files:
 - lib/battle_pet.rb
 - lib/battle_pet.yml
+- lib/pet_ability.rb
 - lib/pet_type.rb
 - README.md
 homepage: https://github.com/hegwin/battle_pet