module ForemanSalt
  module Concerns
    module HostgroupExtensions
      extend ActiveSupport::Concern

      included do
        has_many :hostgroup_salt_modules, class_name: '::ForemanSalt::HostgroupSaltModule'
        has_many :salt_modules, through: :hostgroup_salt_modules, class_name: '::ForemanSalt::SaltModule'

        belongs_to :salt_proxy, class_name: 'SmartProxy'
        belongs_to :salt_environment, class_name: 'ForemanSalt::SaltEnvironment'

        scoped_search relation: :salt_modules, on: :name, complete_value: true, rename: :salt_state
        scoped_search relation: :salt_environment, on: :name, complete_value: true, rename: :salt_environment
        scoped_search relation: :salt_proxy, on: :name, complete_value: true, rename: :saltmaster

        validates :salt_proxy, presence: true, unless: -> { salt_proxy_id.blank? }
        validates :salt_environment, presence: true, unless: -> { salt_environment_id.blank? }
      end

      def all_salt_modules
        ForemanSalt::SaltModule.in_environment(salt_environment).where(id: salt_module_ids + inherited_salt_module_ids)
      end

      def inherited_salt_modules
        ForemanSalt::SaltModule.where(id: inherited_salt_module_ids)
      end

      def inherited_salt_module_ids
        ancestors.map(&:salt_module_ids).flatten.uniq
      end

      def salt_proxy
        if ancestry.present?
          SmartProxy.with_features('Salt').find_by(id: inherited_salt_proxy_id)
        else
          super
        end
      end

      def inherited_salt_proxy_id
        self[:salt_proxy_id] || nested(:salt_proxy_id)
      end

      def salt_environment
        if ancestry.present?
          ForemanSalt::SaltEnvironment.find_by(id: inherited_salt_environment_id)
        else
          super
        end
      end

      def inherited_salt_environment_id
        self[:salt_environment_id] || nested(:salt_environment_id)
      end

      def salt_master
        salt_proxy.to_s
      end
    end
  end
end