# Copyright (c) 2012 Red Hat
#
# MIT License
#
# Permission is hereby granted, free of charge, to any person obtaining
# a copy of this software and associated documentation files (the
# "Software"), to deal in the Software without restriction, including
# without limitation the rights to use, copy, modify, merge, publish,
# distribute, sublicense, and/or sell copies of the Software, and to
# permit persons to whom the Software is furnished to do so, subject to
# the following conditions:
#
# The above copyright notice and this permission notice shall be
# included in all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.


module Runcible
  module Extensions
    class Repository < Runcible::Resources::Repository

      # Utility function that allows an importer to be added at repository creation time
      #
      # @param  [String]                               id       the id of the repository being created
      # @param  [Hash, Runcible::Extensions::Importer] importer either a hash representing an importer or an Importer object
      # @return [RestClient::Response]                          the created repository         
      def create_with_importer(id, importer)
        create_with_importer_and_distributors(id, importer)
      end

      # Utility function that allows distributors to be added at repository creation time
      #
      # @param  [String]                id            the id of the repository being created
      # @param  [Array]                 distributors  an array of hashes representing distributors or an array of Distributor objects
      # @return [RestClient::Response]                the created repository         
      def create_with_distributors(id, distributors)
        create_with_importer_and_distributors(id, nil, distributors)
      end

      # Utility function that allows an importer and distributors to be added at repository creation time
      #
      # @param  [String]                                id            the id of the repository being created
      # @param  [Hash, Runcible::Extensions::Importer]  importer      either a hash representing an importer or an Importer object
      # @param  [Array]                                 distributors  an array of hashes representing distributors or an array of Distributor objects
      # @param  [Hash]                                  optional      container for all optional parameters
      # @return [RestClient::Response]                                the created repository         
      def create_with_importer_and_distributors(id, importer, distributors=[], optional={})
        if importer.is_a?(Runcible::Models::Importer)
          optional[:importer_type_id] = importer.id
          optional[:importer_config] = importer.config
        else
          optional[:importer_type_id] = importer.delete('id') || importer.delete(:id)
          optional[:importer_config] = importer
        end if importer

        repo_type = if importer.methods.include?(:repo_type)
                      importer.repo_type
                    elsif importer.is_a?(Hash) && importer.has_key?(:repo_type)
                      importer[:repo_type]
                    else
                      nil
                    end

        if optional.has_key?(:importer_type_id) && repo_type
          # pulp needs _repo-type in order to determine the type of repo to create.
          optional[:notes] = { '_repo-type' => importer.repo_type }
        end

        optional[:distributors] = distributors.collect do |d|
          if d.is_a?(Runcible::Models::Distributor)
            {'distributor_type' => d.type_id,
              "distributor_config" => d.config,
              "auto_publish" => d.auto_publish,
              "distributor_id" => d.id
            }
          else
            {'distributor_type' => d['type_id'],
             "distributor_config" => d['config'],
             "auto_publish" => d['auto_publish'],
             "distributor_id" => d['id']
            }
          end
        end if !distributors.empty?
        optional[:id] = id
        create(id, optional)
      end

      # Retrieves the sync status for a repository
      #
      # @param  [String]                repo_id the repository ID
      # @return [RestClient::Response]          a task representing the sync status
      def sync_status(repo_id)
        Runcible::Resources::Task.new(self.config).list(["pulp:repository:#{repo_id}", "pulp:action:sync"])
      end

      # Retrieves the publish status for a repository
      #
      # @param  [String]                repo_id the repository ID
      # @return [RestClient::Response]          a task representing the sync status
      def publish_status(repo_id)
        Runcible::Resources::Task.new(self.config).list(["pulp:repository:#{repo_id}", "pulp:action:publish"])
      end

      # Retrieves a set of repositories by their IDs
      #
      # @param  [Array]                repository_ids the repository ID
      # @return [RestClient::Response]                the set of repositories requested
      def search_by_repository_ids(repository_ids)
        criteria = {:filters => 
                      { "id" => {"$in" => repository_ids}}
                   }
        search(criteria)
      end


      # Retrieves the RPM IDs for a single repository
      #
      # @param [String]             id the ID of the repository
      # @return [Array<String>]     the array of repository RPM IDs

      def rpm_ids(id)
        criteria = {:type_ids=>[Runcible::Extensions::Rpm.content_type],
                            :fields=>{:unit=>[], :association=>['unit_id']}}
        self.unit_search(id, criteria).collect{|i| i['unit_id']}
      rescue RestClient::RequestTimeout
        self.logger.warn("Call to rpm_ids timed out")
        # lazy evaluated iterator from zero to infinite
        pages = Enumerator.new { |y| page = 0; loop { y << page; page += 1 } }

        # TODO this is hotfix, pagination support should be added to Runcible
        pages.inject([]) do |rpm_ids, page|
          page_size = 500
          criteria  = { :type_ids => [Runcible::Extensions::Rpm.content_type],
                        :fields   => { :unit => [], :association => ['unit_id'] },
                        :limit    => page_size,
                        :skip     => 0 + page*page_size }
          result    = unit_search(id, criteria).collect { |i| i['unit_id'] }
          rpm_ids.concat(result)
          if result.empty? || result.size < 500
            break rpm_ids
          else
            rpm_ids
          end
        end
      end

      # Retrieves the RPMs for a single repository
      #
      # @param  [String]                id the ID of the repository
      # @return [RestClient::Response]     the set of repository RPMs
      def rpms(id)
        criteria = {:type_ids=>[Runcible::Extensions::Rpm.content_type]}
        unit_search(id, criteria).collect{|i| i['metadata'].with_indifferent_access}
      end

      # Retrieves the RPMs by NVRE for a single repository
      #
      # @param  [String]                id      the ID of the repository
      # @param  [String]                name    the name of the RPMs
      # @param  [String]                version the version of the RPMs
      # @param  [String]                release the release of the RPMs
      # @param  [String]                epoch   the epoch of the RPMs
      # @return [RestClient::Response]          the set of repository RPMs
      def rpms_by_nvre(id, name, version=nil, release=nil, epoch=nil)
        and_condition = []
        and_condition << {:name=>name} if name
        and_condition << {:version=>version} if version
        and_condition << {:release=>release} if release
        and_condition << {:epoch=>epoch} if epoch

        criteria = {:type_ids=>[Runcible::Extensions::Rpm.content_type],
                :filters => {
                    :unit => {
                      "$and" => and_condition
                    }
                },
                :sort => {
                    :unit => [ ['name', 'ascending'], ['version', 'descending'] ]
                }}
        unit_search(id, criteria).collect{|p| p['metadata'].with_indifferent_access}
      end

      # Retrieves the errata IDs for a single repository
      #
      # @param  [String]                id the ID of the repository
      # @return [RestClient::Response]     the set of repository errata IDs
      def errata_ids(id)
         criteria = {:type_ids=>[Runcible::Extensions::Errata.content_type],
                     :fields=>{:unit=>[], :association=>['unit_id']}}

         unit_search(id, criteria).collect{|i| i['unit_id']}
      end

      # Retrieves the errata for a single repository
      #
      # @param  [String]                id the ID of the repository
      # @return [RestClient::Response]     the set of repository errata
      def errata(id)
         criteria = {:type_ids=>[Runcible::Extensions::Errata.content_type]}
         unit_search(id, criteria).collect{|i| i['metadata'].with_indifferent_access}
      end

      # Retrieves the distributions for a single repository
      #
      # @param  [String]                id the ID of the repository
      # @return [RestClient::Response]     the set of repository distributions
      def distributions(id)
        criteria = {:type_ids=>[Runcible::Extensions::Distribution.content_type]}

        unit_search(id, criteria).collect{|i| i['metadata'].with_indifferent_access}
      end

      # Retrieves the package groups for a single repository
      #
      # @param  [String]                id the ID of the repository
      # @return [RestClient::Response]     the set of repository package groups
      def package_groups(id)
        criteria = {:type_ids=>[Runcible::Extensions::PackageGroup.content_type]}

        unit_search(id, criteria).collect{|i| i['metadata'].with_indifferent_access}
      end

      # Retrieves the package group categoriess for a single repository
      #
      # @param  [String]                id the ID of the repository
      # @return [RestClient::Response]     the set of repository package group categories
      def package_categories(id)
        criteria = {:type_ids=>[Runcible::Extensions::PackageCategory.content_type]}
        unit_search(id, criteria).collect{|i| i['metadata'].with_indifferent_access}
      end

      # Retrieves the puppet module IDs for a single repository
      #
      # @param  [String]                id the ID of the repository
      # @return [RestClient::Response]     the set of repository puppet module IDs
      def puppet_module_ids(id)
        criteria = {:type_ids=>[Runcible::Extensions::PuppetModule.content_type],
                    :fields=>{:unit=>[], :association=>['unit_id']}}

        unit_search(id, criteria).collect{|i| i['unit_id']}
      end

      # Retrieves the puppet modules for a single repository
      #
      # @param  [String]                id the ID of the repository
      # @return [RestClient::Response]     the set of repository puppet modules
      def puppet_modules(id)
        criteria = {:type_ids=>[Runcible::Extensions::PuppetModule.content_type]}
        unit_search(id, criteria).collect{|i| i['metadata'].with_indifferent_access}
      end

      # Creates or updates a sync schedule for a repository
      #
      # @param  [String]                repo_id   the ID of the repository
      # @param  [String]                type      the importer type
      # @param  [String]                schedule  the time as an iso8601 interval
      # @return [RestClient::Response]            the newly created or updated schedule
      def create_or_update_schedule(repo_id, type, schedule)
        schedules = Runcible::Resources::RepositorySchedule.new(self.config).list(repo_id, type)
        if schedules.empty?
          Runcible::Resources::RepositorySchedule.new(self.config).create(repo_id, type, schedule)
        else
          Runcible::Resources::RepositorySchedule.new(self.config).update(repo_id, type, schedules[0]['_id'], {:schedule=>schedule})
        end
      end

      # Removes a scheduled sync from a repository
      #
      # @param  [String]                repo_id   the ID of the repository
      # @param  [String]                type      the importer type
      # @return [RestClient::Response]            
      def remove_schedules(repo_id, type)
        schedules = Runcible::Resources::RepositorySchedule.new(self.config).list(repo_id, type)
        schedules.each do |schedule|
          Runcible::Resources::RepositorySchedule.new(self.config).delete(repo_id, type, schedule['_id'])
        end
      end

      # Publishes a repository for all of it's distributors
      #
      # @param  [String]                repo_id the ID of the repository
      # @return [RestClient::Response]          set of tasks representing each publish  
      def publish_all(repo_id)
        to_ret = []
        retrieve_with_details(repo_id)['distributors'].each do |d|
          to_ret << publish(repo_id, d['id'])
        end
        to_ret
      end

      # Retrieves a repository with details that include importer and distributors
      #
      # @param  [String]                repo_id the ID of the repository
      # @return [RestClient::Response]          the repository with full details
      def retrieve_with_details(repo_id)
        retrieve(repo_id, {:details => true})
      end

    end
  end
end