module Katello
  class SmartProxyHelper
    attr_accessor :smart_proxy
    def initialize(smart_proxy)
      @smart_proxy = smart_proxy
    end

    def ==(other)
      other.class == self.class && other.smart_proxy == smart_proxy
    end

    def default_capsule?
      @smart_proxy.pulp_primary?
    end

    def lifecycle_environment_check(environment = nil, repository = nil)
      environment = repository.environment if repository

      if environment && !self.smart_proxy.lifecycle_environments.include?(environment)
        fail _("Lifecycle environment '%{environment}' is not attached to this capsule.") % { :environment => environment.name }
      end
    end

    def library_export_repo(repository = nil)
      cv = repository.content_view if repository
      cv.library_export?
    end

    def clear_smart_proxy_sync_histories(repo_list = [])
      if repo_list.empty?
        @smart_proxy.smart_proxy_sync_histories.delete_all
        return
      end
      repo_ids = repo_list.map(&:id)
      @smart_proxy.smart_proxy_sync_histories.where("repository_id IN (?)", repo_ids).delete_all
    end

    def combined_repos_available_to_capsule(environment = nil, content_view = nil, repository = nil)
      lifecycle_environment_check(environment, repository)
      if repository && !library_export_repo(repository)
        [repository]
      else
        repositories_available_to_capsule(environment, content_view)
      end
    end

    def repositories_available_to_capsule(environments = nil, content_view = nil)
      environments = @smart_proxy.lifecycle_environments if environments.nil?
      yum_repos = Katello::Repository.in_environment(environments)
      yum_repos = yum_repos.in_content_views([content_view]) if content_view
      yum_repos.smart_proxy_syncable
    end

    def unsyncable_content_types
      unsyncable_content_types = ::Katello::RepositoryTypeManager.defined_repository_types.keys.collect do |type_name|
        unless @smart_proxy.capabilities(::SmartProxy::PULP3_FEATURE).include?(::Katello::RepositoryTypeManager.defined_repository_types[type_name].pulp3_plugin)
          type_name
        end
      end
      unsyncable_content_types.compact & repositories_available_to_capsule.map(&:content_type)
    end
  end
end