module Mongoid
  module Report

    AttachProxy = Struct.new(:context, :collection, :options) do
      attr_reader :report_name

      def initialize(context, collection, options)
        # Lets remove as option because of passing to the next blocks options
        report_name = options.delete(:as)

        if collection
          report_name ||= Collections.name(collection)
        end

        options.merge!(
          report_name:  report_name,
          collection:   collection,
        )

        super(context, collection, options)
      end

      def columns(*fields)
        proxy_options = fields.extract_options!
        proxy_options.merge!(options)
        context.columns(*fields, proxy_options)
      end

      def column(*fields)
        proxy_options = fields.extract_options!
        proxy_options.merge!(options)
        context.column(*fields, proxy_options)
      end

      def mapping(*fields)
        proxy_options = fields.extract_options!
        proxy_options.merge!(options)
        context.mapping(*fields, proxy_options)
      end

      def group_by(*fields)
        proxy_options = fields.extract_options!
        proxy_options.merge!(options)
        context.group_by(*fields, proxy_options)
      end

      def match(*fields)
        proxy_options = fields.extract_options!
        proxy_options.merge!(options)
        context.match(*fields, proxy_options)
      end

      def query(*fields)
        proxy_options = fields.extract_options!
        proxy_options.merge!(options)
        context.query(*fields, proxy_options)
      end

      def batches(*fields)
        proxy_options = fields.extract_options!
        proxy_options.merge!(options)
        context.batches(*fields, proxy_options)
      end
    end

  end
end