3 files changed, 67 insertions, 1 deletions

@@ -0,0 +1,43 @@
+module Pokeviewer
+  class Item < ApplicationRecord
+    validates :name, presence: true
+
+    belongs_to :move, optional: true
+    validates :move, presence: true, if: :tm?
+
+    validates :rs_description, presence: true, unless: :tm?
+    validates :frlg_description, presence: true, unless: :tm?
+
+    def description(game)
+      if game == :emerald
+        if not emerald_description.nil?
+          emerald_description
+        elsif not rs_description.nil?
+          rs_description
+        else
+          move.description game
+        end
+      elsif game == :firered or game == :leafgreen
+        if not frlg_description.nil?
+          frlg_description
+        else
+          move.description game
+        end
+      else
+        if not rs_description.nil?
+          rs_description
+        else
+          move.description game
+        end
+      end
+    end
+
+    def icon_path
+      if tm?
+        "pokeviewer/items/tms/#{move.move_type}.png"
+      else
+        "pokeviewer/items/#{id}.png"
+      end
+    end
+  end
+end
@@ -1,10 +1,31 @@
 module Pokeviewer
   class Move < ApplicationRecord
+    extend Enumerize
+
     has_many :revision_moves
     has_many :revisions, through: :revision_moves
 
     validates :name, presence: true, uniqueness: true
+
     validates :pp, presence: true,
       numericality: { greater_than_or_equal_to: 1, only_integer: true }
+
+    validates :move_type, presence: true
+    enumerize :move_type, in: [:normal, :fighting, :flying, :poison, :ground,
+      :rock, :bug, :ghost, :steel, :mystery, :fire, :water, :grass, :electric,
+      :psychic, :ice, :dragon, :dark], predicates: true
+
+    validates :rs_description, presence: true
+    validates :frlg_description, presence: true
+
+    def description(game)
+      if game == :emerald and not emerald_description.nil?
+        emerald_description
+      elsif game == :firered or game == :leafgreen
+        frlg_description
+      else
+        rs_description
+      end
+    end
   end
 end
@@ -6,7 +6,7 @@ module Pokeviewer
 
     diff :nickname, :level, :hp, :attack, :defense,
       :special_attack, :special_defense, :speed, :coolness, :beauty, :cuteness,
-      :smartness, :toughness, :sheen, :hold_item, :move_1_id, :move_2_id,
+      :smartness, :toughness, :sheen, :item_id, :move_1_id, :move_2_id,
       :move_3_id, :move_4_id, :move_1_pp_bonuses, :move_2_pp_bonuses,
       :move_3_pp_bonuses, :move_4_pp_bonuses, :cool_ribbons, :beauty_ribbons,
       :cute_ribbons, :smart_ribbons, :tough_ribbons, :champion_ribbon,
@@ -79,6 +79,8 @@ module Pokeviewer
         less_than_or_equal_to: 10,
         only_integer: true }
 
+    belongs_to :item, optional: true
+
     belongs_to :move_1, class_name: "Move"
     belongs_to :move_2, class_name: "Move", optional: true
     belongs_to :move_3, class_name: "Move", optional: true