require 'sequencer/unit/import/common/model/statistics/mixin/action_diff'

class Sequencer
  class Unit
    module Import
      module Ldap
        module User
          module Statistics
            class Diff < Sequencer::Unit::Base
              include ::Sequencer::Unit::Import::Common::Model::Statistics::Mixin::ActionDiff

              uses :instance, :associations, :signup_role_ids

              def process
                state.provide(:statistics_diff) do
                  # remove :sum since it's already set via
                  # the outer count Unit
                  statistics = diff.except(:sum)

                  add_role_ids(statistics)
                end
              end

              private

              def add_role_ids(statistics)
                return statistics if instance.blank?

                # add the parent role_ids hash
                # so we can fill it
                statistics[:role_ids] = {}

                associations[:role_ids] ||= signup_role_ids

                # add the diff for each role_id the user is assigned to
                associations[:role_ids].each_with_object(statistics) do |role_id, result|
                  result[:role_ids][role_id] = diff
                end
              end
            end
          end
        end
      end
    end
  end
end