23 lines
712 B
Ruby
23 lines
712 B
Ruby
# frozen_string_literal: true
|
|
|
|
class GridWeapon < ApplicationRecord
|
|
belongs_to :party,
|
|
counter_cache: :weapons_count
|
|
|
|
belongs_to :weapon_key1, class_name: 'WeaponKey', foreign_key: :weapon_key1_id, optional: true
|
|
belongs_to :weapon_key2, class_name: 'WeaponKey', foreign_key: :weapon_key2_id, optional: true
|
|
belongs_to :weapon_key3, class_name: 'WeaponKey', foreign_key: :weapon_key3_id, optional: true
|
|
|
|
def weapon
|
|
Weapon.find(weapon_id)
|
|
end
|
|
|
|
def weapon_keys
|
|
weapon_keys = []
|
|
weapon_keys.push(weapon_key1) unless weapon_key1.nil?
|
|
weapon_keys.push(weapon_key2) unless weapon_key2.nil?
|
|
weapon_keys.push(weapon_key3) unless weapon_key3.nil?
|
|
|
|
weapon_keys
|
|
end
|
|
end
|