4 files changed, 60 insertions, 7 deletions

@@ -22,6 +22,36 @@ module Pokeviewer
           end
         end
 
+      if args.key? "marineRibbon"
+        game.marine_ribbon = GiftRibbon.find_by_id(args["marineRibbon"])
+      end
+
+      if args.key? "landRibbon"
+        game.land_ribbon = GiftRibbon.find_by_id(args["landRibbon"])
+      end
+
+      if args.key? "skyRibbon"
+        game.sky_ribbon = GiftRibbon.find_by_id(args["skyRibbon"])
+      end
+
+      if args.key? "countryRibbon"
+        game.country_ribbon = GiftRibbon.find_by_id(args["countryRibbon"])
+      end
+
+      if args.key? "nationalRibbon"
+        game.national_ribbon = GiftRibbon.find_by_id(args["nationalRibbon"])
+      end
+
+      if args.key? "earthRibbon"
+        game.earth_ribbon = GiftRibbon.find_by_id(args["earthRibbon"])
+      end
+
+      if args.key? "worldRibbon"
+        game.world_ribbon = GiftRibbon.find_by_id(args["worldRibbon"])
+      end
+
+      game.save! if game.changed?
+
       args["boxes"].each_with_index do |box_name,index|
         box = Box.find_or_initialize_by(trainer: game, number: index)
         box.name = box_name
@@ -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