Contents

module WebConsole
  module ControllerHelpers
    extend ActiveSupport::Concern

    included do
      # Flag to decide whether the console should be rendered.
      attr_internal :console_already_rendered

      # Storage of a binding the console to be rendered in.
      attr_internal :console_binding

      prepend_after_action :inject_console_into_view
    end

    # Helper for capturing a controller binding to prepare for console
    # rendering.
    def console(binding = nil)
      @_console_binding = binding || ::Kernel.binding.of_caller(1)
    end

    private

      # Attempt to inject an interactive console to a view.
      def inject_console_into_view
        return unless can_render_console?

        console_html = ActionView::Base.new(ActionController::Base.view_paths,
          console_session: REPLSession.create(binding: @_console_binding)
        ).render(partial: 'rescues/web_console')

        response.body = response.body + console_html
      end

      def can_render_console?
        console_binding && !console_already_rendered && html_format? && whitelisted_ip?
      end

      def html_format?
        content_type == Mime::HTML
      end

      def whitelisted_ip?
        request.remote_ip.in?(WebConsole.config.whitelisted_ips)
      end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
web-console-2.0.0	lib/web_console/controller_helpers.rb