3 files changed, 17 insertions, 0 deletions

@@ -0,0 +1,6 @@
+module Pokeviewer
+  class PokedexEntry < ApplicationRecord
+    belongs_to :trainer
+    belongs_to :species
+  end
+end
@@ -4,6 +4,8 @@ module Pokeviewer
 
     has_many :revisions, dependent: :restrict_with_exception
 
+    has_many :pokedex_entries, dependent: :destroy
+
     validates :name, presence: true, uniqueness: true
 
     validates :type_1, presence: true
@@ -13,5 +15,12 @@ module Pokeviewer
 
     belongs_to :ability_1, class_name: "Ability"
     belongs_to :ability_2, class_name: "Ability", optional: true
+
+    def current_revisions
+      revisions.
+        where("pokeviewer_pokemon.current_id = pokeviewer_revisions.id").
+        includes(:pokemon).
+        references(:pokemon)
+    end
   end
 end
@@ -4,6 +4,8 @@ module Pokeviewer
 
     has_many :pokemon, dependent: :nullify
 
+    has_many :pokedex_entries, dependent: :destroy
+
     validates :number, presence: true,
       numericality: { greater_than_or_equal_to: 0, only_integer: true }