Contents

require 'csv'
require 'set'

module PactBroker

  module Api

    module Decorators

      class RelationshipsCsvDecorator

        def initialize pacts
          @pacts = pacts
          @relationships = pacts.collect{|pact| PactBroker::Models::Relationship.new(pact.consumer,pact.provider)}
        end

        def to_csv
          hash = {}
          pacticipants = @relationships.collect{|r| r.pacticipants}.flatten.uniq

          @relationships.each do | relationship |
            hash[relationship.consumer.id] ||= pacticipant_array(relationship.consumer, hash.size + 1)
            hash[relationship.provider.id] ||= pacticipant_array(relationship.provider, hash.size + 1)
            hash[relationship.consumer.id] << relationship.provider.id
          end

          max_length = hash.values.collect{|array| array.size}.max

          hash.values.each do | array |
            while array.size < max_length
              array << 0
            end
          end

          CSV.generate do |csv|
            hash.values.each do | array |
              csv << array
            end
          end

        end

        def pacticipant_array pacticipant, order
          [pacticipant.id, pacticipant.name, 1, 1, 0, order]
        end

        private

        attr_accessor :pacts

      end
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
pact_broker-1.3.1	lib/pact_broker/api/decorators/relationships_csv_decorator.rb
pact_broker-1.3.0	lib/pact_broker/api/decorators/relationships_csv_decorator.rb
pact_broker-1.2.0	lib/pact_broker/api/decorators/relationships_csv_decorator.rb
pact_broker-1.1.0	lib/pact_broker/api/decorators/relationships_csv_decorator.rb