blob: ee22db4e0946927b0f8f2c522cc8c415b3129f69 (
plain) (
tree)
|
|
module ActionCable
module Channel
class Base
include Callbacks
include PeriodicTimers
include Streams
on_subscribe :connect
on_unsubscribe :disconnect
attr_reader :params, :connection
delegate :logger, to: :connection
class << self
def matches?(identifier)
raise "Please implement #{name}#matches? method"
end
end
def initialize(connection, channel_identifier, params = {})
@connection = connection
@channel_identifier = channel_identifier
@params = params
perform_connection
end
def perform_connection
logger.info "#{channel_name} connecting"
run_subscribe_callbacks
end
def perform_action(data)
if authorized?
action = extract_action(data)
if performable_action?(action)
logger.info action_signature(action, data)
public_send action, data
else
logger.error "Unable to process #{action_signature(action, data)}"
end
else
unauthorized
end
end
def perform_disconnection
run_unsubscribe_callbacks
logger.info "#{channel_name} disconnected"
end
protected
# Override in subclasses
def authorized?
true
end
def unauthorized
logger.error "#{channel_name}: Unauthorized access"
end
def connect
# Override in subclasses
end
def disconnect
# Override in subclasses
end
def transmit(data, via: nil)
if authorized?
logger.info "#{channel_name} transmitting #{data.inspect}".tap { |m| m << " (via #{via})" if via }
connection.transmit({ identifier: @channel_identifier, message: data }.to_json)
else
unauthorized
end
end
def channel_name
self.class.name
end
private
def extract_action(data)
(data['action'].presence || :receive).to_sym
end
def performable_action?(action)
self.class.instance_methods(false).include?(action)
end
def action_signature(action, data)
"#{channel_name}##{action}".tap do |signature|
if (arguments = data.except('action')).any?
signature << "(#{arguments.inspect})"
end
end
end
def run_subscribe_callbacks
self.class.on_subscribe_callbacks.each { |callback| send(callback) }
end
def run_unsubscribe_callbacks
self.class.on_unsubscribe_callbacks.each { |callback| send(callback) }
end
def worker_pool
connection.worker_pool
end
end
end
end
|