1 files changed, 9 insertions, 0 deletions

@@ -1,6 +1,7 @@
 module Pokeviewer
   class Pokemon < ApplicationRecord
     extend Enumerize
+    extend ActiveModel::Naming
 
     belongs_to :species
 
@@ -69,6 +70,10 @@ module Pokeviewer
       :l, :m, :n, :o, :p, :q, :r, :s, :t, :u, :v, :w, :x, :y, :z,
       :question, :exclamation]
 
+    validates :pokeball, presence: true
+    enumerize :pokeball, in: [:master, :ultra, :great, :poke, :safari, :net,
+      :dive, :nest, :repeat, :timer, :luxury, :premier]
+
     def to_param
       uuid
     end
@@ -191,6 +196,10 @@ module Pokeviewer
       end
     end
 
+    def pokeball_icon_path
+      "pokeviewer/items/#{Pokemon.pokeball.values.find_index(pokeball) + 1}.png"
+    end
+
     private
 
       def set_uuid