Katello/katello

module Katello
  module Concerns
    module HttpProxyExtensions
      extend ActiveSupport::Concern

      included do
        has_many :root_repositories, :class_name => "::Katello::RootRepository", :foreign_key => :http_proxy_id,
          :inverse_of => :http_proxy, :dependent => :nullify
        # A smart proxy's HTTP proxy is used for all related alternate content sources.
        has_many :smart_proxies, :class_name => "::SmartProxy", :foreign_key => :http_proxy_id,
          :inverse_of => :http_proxy, :dependent => :nullify
        after_update :update_default_proxy_setting
        after_commit :update_repository_proxy_details
        before_destroy :remove_references_to_proxy

        def self.default_global_content_proxy
          if Setting[:content_default_http_proxy]
            HttpProxy.unscoped.find_by(name: Setting[:content_default_http_proxy])
          end
        end
      end

      def repositories_with_proxy(proxy_id)
        root_repos = RootRepository.with_selected_proxy(proxy_id)

        if self == HttpProxy.default_global_content_proxy
          root_repos += RootRepository.with_global_proxy
        end

        root_repos
      end

      def remove_references_to_proxy
        root_repos = repositories_with_proxy(nil).uniq.sort

        setting = Setting.find_by(name: 'content_default_http_proxy')
        if setting&.value && setting.value == self.name
          setting.update_attribute(:value, '')
        end

        unless root_repos.empty?
          ForemanTasks.async_task(
            ::Actions::BulkAction,
            ::Actions::Katello::Repository::Update,
            root_repos,
            http_proxy_policy: Katello::RootRepository::GLOBAL_DEFAULT_HTTP_PROXY,
            http_proxy_id: nil)
        end

        unless smart_proxies.empty?
          smart_proxies.each do |smart_proxy|
            smart_proxy.update(http_proxy_id: nil)
          end
        end
      end

      def update_default_proxy_setting
        changes = self.previous_changes
        if changes.key?(:name)
          previous_name = changes[:name].first
          if !previous_name.blank? && Setting['content_default_http_proxy'] == previous_name
            Setting['content_default_http_proxy'] = self.name
          end
        end
      end

      def update_repository_proxy_details
        changes = self.previous_changes
        if changes.key?(:url) || changes.key?(:username) || changes.key?(:password)
          repos = ::Katello::Repository.where(root_id: repositories_with_proxy(id).pluck(:id)).where.not(remote_href: nil).where(library_instance_id: nil)

          unless repos.empty?
            ForemanTasks.async_task(
              ::Actions::BulkAction,
              ::Actions::Katello::Repository::UpdateHttpProxyDetails,
              repos.sort_by(&:pulp_id))
          end
        end
      end

      def name_and_url
        uri = URI(url)
        uri.password = nil
        uri.user = nil
        "#{name} (#{uri})"
      end
    end
  end
end