Pokémon now show the Poké Ball that they're in

Also fixed a bug with viewing Pokémon that aren't in any game.

3 files changed, 12 insertions, 1 deletions

@@ -96,6 +96,7 @@ module Pokeviewer
           r.nature = Pokemon.nature.values[param["nature"]]
           r.gender = param["gender"]
           r.second_ability = param["secondAbility"]
+          r.pokeball = Pokemon.pokeball.values[param["pokeball"] - 1]
 
           # Handle Unown form
           if r.species_id == 201
@@ -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
@@ -1,4 +1,4 @@
-.pokemon{ class: "in-#{@pokemon.trainer.game}" }
+.pokemon{ class: (not @pokemon.trainer.nil?) && "in-#{@pokemon.trainer.game}" }
   .pokemon-basics
     .pokemon-species-id= "No. #{@pokemon.species_id}"
     .pokemon-species-name= @pokemon.species.name
@@ -90,6 +90,7 @@
             = ribbon[:description]
   .pokemon-etc
     .pd-details Misc
+    = image_tag(@pokemon.pokeball_icon_path)
     - unless @pokemon.revisions.last.item.nil?
       .pd-hold-item.pkv-has-hover
         = image_tag(@pokemon.revisions.last.item.icon_path)