# This class combines multiple collections with shared keys into a
# hash of collections compatible with C3.js charts
#----------------------------------------------------------------------

module Ahoy
  class DataSource
    # Adds a collection with the datasource
    # Name is the name of the collection and will be showed in the
    # chart
    def add(name, collection)
      collections.push data: collection, name: name
      shared_keys.merge(collection.keys)
    end

    def build
      data = { x: [] }
      shared_keys.each do |k|
        # Add the key with a valid date format
        data[:x].push k.strftime("%Y-%m-%d")

        # Add the value for each column, or 0 if not present
        collections.each do |col|
          data[col[:name]] ||= []
          count = col[:data][k] || 0
          data[col[:name]].push count
        end
      end

      data
    end

    private

      def collections
        @collections ||= []
      end

      def shared_keys
        @shared_keys ||= Set.new
      end
  end
end