require "rails" require "action_cable" require_relative "helpers/action_cable_helper" require "active_support/core_ext/hash/indifferent_access" module ActionCable class Engine < Rails::Engine # :nodoc: config.action_cable = ActiveSupport::OrderedOptions.new config.action_cable.mount_path = ActionCable::INTERNAL[:default_mount_path] config.eager_load_namespaces << ActionCable initializer "action_cable.helpers" do ActiveSupport.on_load(:action_view) do include ActionCable::Helpers::ActionCableHelper end end initializer "action_cable.logger" do ActiveSupport.on_load(:action_cable) { self.logger ||= ::Rails.logger } end initializer "action_cable.set_configs" do |app| options = app.config.action_cable options.allowed_request_origins ||= /https?:\/\/localhost:\d+/ if ::Rails.env.development? app.paths.add "config/cable", with: "config/cable.yml" ActiveSupport.on_load(:action_cable) do if (config_path = Pathname.new(app.config.paths["config/cable"].first)).exist? self.cable = Rails.application.config_for(config_path).with_indifferent_access end previous_connection_class = connection_class self.connection_class = -> { "ApplicationCable::Connection".safe_constantize || previous_connection_class.call } options.each { |k, v| send("#{k}=", v) } end end initializer "action_cable.routes" do config.after_initialize do |app| config = app.config unless config.action_cable.mount_path.nil? app.routes.prepend do mount ActionCable.server => config.action_cable.mount_path, internal: true end end end end initializer "action_cable.set_work_hooks" do |app| ActiveSupport.on_load(:action_cable) do ActionCable::Server::Worker.set_callback :work, :around, prepend: true do |_, inner| app.executor.wrap do # If we took a while to get the lock, we may have been halted # in the meantime. As we haven't started doing any real work # yet, we should pretend that we never made it off the queue. unless stopping? inner.call end end end wrap = lambda do |_, inner| app.executor.wrap(&inner) end ActionCable::Channel::Base.set_callback :subscribe, :around, prepend: true, &wrap ActionCable::Channel::Base.set_callback :unsubscribe, :around, prepend: true, &wrap app.reloader.before_class_unload do ActionCable.server.restart end end end end end