path: root/lib/action_cable/channel/callbacks.rb

module ActionCable
  module Channel
    module Callbacks
      extend ActiveSupport::Concern

      included do
        class_attribute :on_subscribe_callbacks, :on_unsubscribe_callbacks, instance_reader: false

        self.on_subscribe_callbacks = []
        self.on_unsubscribe_callbacks = []
      end

      module ClassMethods
        # Name methods that should be called when the channel is subscribed to.
        # (These methods should be private, so they're not callable by the user).
        def on_subscribe(*methods)
          self.on_subscribe_callbacks += methods
        end

        # Name methods that should be called when the channel is unsubscribed from.
        # (These methods should be private, so they're not callable by the user).
        def on_unsubscribe(*methods)
          self.on_unsubscribe_callbacks += methods
        end
      end
    end
  end
end