# frozen_string_literal: true
require "active_support/ordered_options"
require "active_support/core_ext/object"
require "rails/paths"
require "rails/rack"
module Rails
module Configuration
# MiddlewareStackProxy is a proxy for the Rails middleware stack that allows
# you to configure middlewares in your application. It works basically as a
# command recorder, saving each command to be applied after initialization
# over the default middleware stack, so you can add, swap, or remove any
# middleware in Rails.
#
# You can add your own middlewares by using the +config.middleware.use+ method:
#
# config.middleware.use Magical::Unicorns
#
# This will put the Magical::Unicorns middleware on the end of the stack.
# You can use +insert_before+ if you wish to add a middleware before another:
#
# config.middleware.insert_before Rack::Head, Magical::Unicorns
#
# There's also +insert_after+ which will insert a middleware after another:
#
# config.middleware.insert_after Rack::Head, Magical::Unicorns
#
# Middlewares can also be completely swapped out and replaced with others:
#
# config.middleware.swap ActionDispatch::Flash, Magical::Unicorns
#
# Middlewares can be moved from one place to another:
#
# config.middleware.move_before ActionDispatch::Flash, Magical::Unicorns
#
# This will move the Magical::Unicorns middleware before the
# ActionDispatch::Flash. You can also move it after:
#
# config.middleware.move_after ActionDispatch::Flash, Magical::Unicorns
#
# And finally they can also be removed from the stack completely:
#
# config.middleware.delete ActionDispatch::Flash
#
class MiddlewareStackProxy
def initialize(operations = [], delete_operations = [])
@operations = operations
@delete_operations = delete_operations
end
def insert_before(*args, &block)
@operations << -> middleware { middleware.send(__method__, *args, &block) }
end
ruby2_keywords(:insert_before) if respond_to?(:ruby2_keywords, true)
alias :insert :insert_before
def insert_after(*args, &block)
@operations << -> middleware { middleware.send(__method__, *args, &block) }
end
ruby2_keywords(:insert_after) if respond_to?(:ruby2_keywords, true)
def swap(*args, &block)
@operations << -> middleware { middleware.send(__method__, *args, &block) }
end
ruby2_keywords(:swap) if respond_to?(:ruby2_keywords, true)
def use(*args, &block)
@operations << -> middleware { middleware.send(__method__, *args, &block) }
end
ruby2_keywords(:use) if respond_to?(:ruby2_keywords, true)
def delete(*args, &block)
@delete_operations << -> middleware { middleware.send(__method__, *args, &block) }
end
def move_before(*args, &block)
@delete_operations << -> middleware { middleware.send(__method__, *args, &block) }
end
alias :move :move_before
def move_after(*args, &block)
@delete_operations << -> middleware { middleware.send(__method__, *args, &block) }
end
def unshift(*args, &block)
@operations << -> middleware { middleware.send(__method__, *args, &block) }
end
ruby2_keywords(:unshift) if respond_to?(:ruby2_keywords, true)
def merge_into(other) #:nodoc:
(@operations + @delete_operations).each do |operation|
operation.call(other)
end
other
end
def +(other) # :nodoc:
MiddlewareStackProxy.new(@operations + other.operations, @delete_operations + other.delete_operations)
end
protected
attr_reader :operations, :delete_operations
end
class Generators #:nodoc:
attr_accessor :aliases, :options, :templates, :fallbacks, :colorize_logging, :api_only
attr_reader :hidden_namespaces
def initialize
@aliases = Hash.new { |h, k| h[k] = {} }
@options = Hash.new { |h, k| h[k] = {} }
@fallbacks = {}
@templates = []
@colorize_logging = true
@api_only = false
@hidden_namespaces = []
end
def initialize_copy(source)
@aliases = @aliases.deep_dup
@options = @options.deep_dup
@fallbacks = @fallbacks.deep_dup
@templates = @templates.dup
end
def hide_namespace(namespace)
@hidden_namespaces << namespace
end
def method_missing(method, *args)
method = method.to_s.sub(/=$/, "").to_sym
if args.empty?
if method == :rails
return @options[method]
else
return @options[:rails][method]
end
end
if method == :rails || args.first.is_a?(Hash)
namespace, configuration = method, args.shift
else
namespace, configuration = args.shift, args.shift
namespace = namespace.to_sym if namespace.respond_to?(:to_sym)
@options[:rails][method] = namespace
end
if configuration
aliases = configuration.delete(:aliases)
@aliases[namespace].merge!(aliases) if aliases
@options[namespace].merge!(configuration)
end
end
end
end
end