view_context.rb

require 'draper/view_context/build_strategy'
require 'request_store'

module Draper
  module ViewContext
    # Hooks into a controller or mailer to save the view context in {current}.
    def view_context
      super.tap do |context|
        Draper::ViewContext.current = context
      end
    end

    # Set the current controller
    def activate_draper
      Draper::ViewContext.controller = self
    end

    # Returns the current controller.
    def self.controller
      RequestStore.store[:current_controller]
    end

    # Sets the current controller. Clears view context when we are setting
    # different controller.
    def self.controller=(controller)
      clear! if RequestStore.store[:current_controller] != controller
      RequestStore.store[:current_controller] = controller
    end

    # Returns the current view context, or builds one if none is saved.
    #
    # @return [HelperProxy]
    def self.current
      RequestStore.store.fetch(:current_view_context) { build! }
    end

    # Sets the current view context.
    def self.current=(view_context)
      RequestStore.store[:current_view_context] = Draper::HelperProxy.new(view_context)
    end

    # Clears the saved controller and view context.
    def self.clear!
      RequestStore.store.delete :current_controller
      RequestStore.store.delete :current_view_context
    end

    # Builds a new view context for usage in tests. See {test_strategy} for
    # details of how the view context is built.
    def self.build
      build_strategy.call
    end

    # Builds a new view context and sets it as the current view context.
    #
    # @return [HelperProxy]
    def self.build!
      # send because we want to return the HelperProxy returned from #current=
      send :current=, build
    end

    # Configures the strategy used to build view contexts in tests, which
    # defaults to `:full` if `test_strategy` has not been called. Evaluates
    # the block, if given, in the context of the view context's class.
    #
    # @example Pass a block to add helper methods to the view context:
    #   Draper::ViewContext.test_strategy :fast do
    #     include ApplicationHelper
    #   end
    #
    # @param [:full, :fast] name
    #   the strategy to use:
    #
    #   `:full` - build a fully-working view context. Your Rails environment
    #   must be loaded, including your `ApplicationController`.
    #
    #   `:fast` - build a minimal view context in tests, with no dependencies
    #   on other components of your application.
    def self.test_strategy(name, &block)
      @build_strategy = Draper::ViewContext::BuildStrategy.new(name, &block)
    end

    # @private
    def self.build_strategy
      @build_strategy ||= Draper::ViewContext::BuildStrategy.new(:full)
    end
  end
end