# 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

  def blueprint
    GridWeaponBlueprint
  end
end