3 files changed, 30 insertions, 7 deletions

@@ -0,0 +1,5 @@
+module Pokeviewer
+  class GiftRibbon < ApplicationRecord
+    validates :description, presence: true
+  end
+end
@@ -345,7 +345,7 @@ module Pokeviewer
         result << {
           filename: "marine-ribbon.png",
           name: "Marine Ribbon",
-          description: ""
+          description: pokemon.trainer.gift_ribbon_description(:marine_ribbon)
         }
       end
 
@@ -353,7 +353,7 @@ module Pokeviewer
         result << {
           filename: "land-ribbon.png",
           name: "Land Ribbon",
-          description: ""
+          description: pokemon.trainer.gift_ribbon_description(:land_ribbon)
         }
       end
 
@@ -361,7 +361,7 @@ module Pokeviewer
         result << {
           filename: "sky-ribbon.png",
           name: "Sky Ribbon",
-          description: ""
+          description: pokemon.trainer.gift_ribbon_description(:sky_ribbon)
         }
       end
 
@@ -369,7 +369,7 @@ module Pokeviewer
         result << {
           filename: "country-ribbon.png",
           name: "Country Ribbon",
-          description: ""
+          description: pokemon.trainer.gift_ribbon_description(:country_ribbon)
         }
       end
 
@@ -377,7 +377,7 @@ module Pokeviewer
         result << {
           filename: "national-ribbon.png",
           name: "National Ribbon",
-          description: ""
+          description: pokemon.trainer.gift_ribbon_description(:national_ribbon)
         }
       end
 
@@ -385,7 +385,7 @@ module Pokeviewer
         result << {
           filename: "earth-ribbon.png",
           name: "Earth Ribbon",
-          description: ""
+          description: pokemon.trainer.gift_ribbon_description(:earth_ribbon)
         }
       end
 
@@ -393,7 +393,7 @@ module Pokeviewer
         result << {
           filename: "world-ribbon.png",
           name: "World Ribbon",
-          description: ""
+          description: pokemon.trainer.gift_ribbon_description(:world_ribbon)
         }
       end
 
@@ -16,6 +16,14 @@ module Pokeviewer
     enumerize :game, in: [:ruby, :sapphire, :firered, :leafgreen, :emerald],
       predicates: true
 
+    belongs_to :marine_ribbon, class_name: "GiftRibbon", optional: true
+    belongs_to :land_ribbon, class_name: "GiftRibbon", optional: true
+    belongs_to :sky_ribbon, class_name: "GiftRibbon", optional: true
+    belongs_to :country_ribbon, class_name: "GiftRibbon", optional: true
+    belongs_to :national_ribbon, class_name: "GiftRibbon", optional: true
+    belongs_to :earth_ribbon, class_name: "GiftRibbon", optional: true
+    belongs_to :world_ribbon, class_name: "GiftRibbon", optional: true
+
     def display_number
       number.to_s.rjust(5, '0')
     end
@@ -23,5 +31,15 @@ module Pokeviewer
     def party
       pokemon.where(box: nil).order("slot ASC")
     end
+
+    def gift_ribbon_description(ribbon)
+      gift_ribbon = send ribbon
+
+      if gift_ribbon.nil?
+        ""
+      else
+        gift_ribbon.description
+      end
+    end
   end
 end