require 'trice/controller_methods/reference_time_assignment'
require 'trice/controller_methods/stub_configuration'

module Trice
  module ControllerMethods

    extend ActiveSupport::Concern

    included do |controller|
      if controller.ancestors.include?(ActionController::Base)
        config = StubConfiguration.new(Trice.support_requested_at_stubbing)
        prepend_around_action ReferenceTimeAssignment.new(config)

        helper_method :requested_at
        hide_action   :requested_at
      end
    end

    def requested_at
      Trice.reference_time
    end
  end
end