From 82a6b0d598639e464a4d191507338735d082ee55 Mon Sep 17 00:00:00 2001 From: Justin Edmund Date: Tue, 1 Mar 2022 23:33:21 -0800 Subject: [PATCH] Add WeaponKey associations to GridWeapon --- app/models/grid_weapon.rb | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/app/models/grid_weapon.rb b/app/models/grid_weapon.rb index acef6ce..9bf51a4 100644 --- a/app/models/grid_weapon.rb +++ b/app/models/grid_weapon.rb @@ -1,7 +1,20 @@ class GridWeapon < ApplicationRecord belongs_to :party + belongs_to :weapon_key1, class_name: 'WeaponKey', foreign_key: :weapon_key1_id + belongs_to :weapon_key2, class_name: 'WeaponKey', foreign_key: :weapon_key2_id + belongs_to :weapon_key3, class_name: 'WeaponKey', foreign_key: :weapon_key3_id + def weapon Weapon.find(self.weapon_id) end + + def weapon_keys + weapon_keys = [] + weapon_keys.push(self.weapon_key1) if self.weapon_key1 != nil + weapon_keys.push(self.weapon_key2) if self.weapon_key2 != nil + weapon_keys.push(self.weapon_key3) if self.weapon_key3 != nil + + weapon_keys + end end