module ForemanSalt::Concerns::HostManagedExtensions

def accept_salt_key
  begin
    Rails.logger.info("Host #{fqdn} is built, accepting Salt key")
    key = ForemanSalt::SmartProxies::SaltKeys.find(salt_proxy, fqdn)
    key&.accept
  rescue Foreman::Exception => e
    Rails.logger.warn("Unable to accept key for #{fqdn}: #{e}")
  end
end

def delete_salt_key
  begin
    key = ForemanSalt::SmartProxies::SaltKeys.find(salt_proxy, fqdn)
    key&.delete
  rescue Foreman::Exception => e
    Rails.logger.warn("Unable to delete key for #{fqdn}: #{e}")
  end
end

def salt_master
  salt_proxy.to_s
end

def salt_modules_for_enc
  return [] unless salt_environment
  modules = salt_modules + (hostgroup ? hostgroup.all_salt_modules : [])
  ForemanSalt::SaltModule.in_environment(salt_environment).where(:id => modules).pluck("salt_modules.name").uniq
end

def salt_modules_in_host_environment
  return unless self.salt_modules.any?
  if self.salt_environment
    errors.add(:base, _('Salt states must be in the environment of the host')) unless (self.salt_modules - self.salt_environment.salt_modules).empty?
  else
    errors.add(:base, _('Host must have an environment in order to set salt states'))
  end
end

def saltrun!
  unless salt_proxy.present?
    errors.add(:base, _("No Salt master defined - can't continue"))
    logger.warn 'Unable to execute salt run, no salt proxies defined'
    return false
  end
  ProxyAPI::Salt.new(:url => salt_proxy.url).highstate name
rescue => e
  errors.add(:base, _('Failed to execute state.highstate: %s') % e)
  false
end