khiav223577/deep_pluck

# frozen_string_literal: true

module DeepPluck
  module DataCombiner
    class << self
      def combine_data(parent, children, primary_key, column_name, foreign_key, reverse, collection)
        source =  reverse ? parent : children
        target = !reverse ? parent : children
        data_hash = make_data_hash(collection, source, primary_key, column_name, reverse)
        assign_values_to_parent(collection, target, data_hash, column_name, foreign_key, reverse: reverse)
        return children
      end

      private

      def make_data_hash(collection, parent, primary_key, column_name, reverse)
        hash = {}

        parent.each do |model_hash|
          key = model_hash[primary_key]

          if reverse
            if collection
              array = (hash[key] ? hash[key][column_name] : []) # share the children if id is duplicated
              model_hash[column_name] = array # set the value of this association key to empty array by default
            else
              model_hash[column_name] = nil # set the value of this association key to be nil by default
            end
          end

          hash[key] = model_hash
        end

        return hash
      end

      def assign_values_to_parent(collection, parent, children_hash, column_name, foreign_key, reverse: false)
        parent.each do |data|
          id = data[foreign_key]

          left = reverse ? children_hash[id] : data
          right = !reverse ? children_hash[id] : data

          if collection
            left[column_name] << right
          else
            left[column_name] = right
          end
        end
      end
    end
  end
end