Contents

# frozen_string_literal: true
module LiteCable
  # AnyCable extensions
  module AnyCable
    module Broadcasting # :nodoc:
      def broadcast(stream, message, coder: nil)
        coder ||= LiteCable.config.coder
        Anycable.broadcast stream, coder.encode(message)
      end
    end

    module Connection # :nodoc:
      def self.extended(base)
        base.prepend InstanceMethods
      end

      def create(socket, **options)
        new(socket, **options)
      end

      module InstanceMethods # :nodoc:
        def initialize(socket, subscriptions: nil, **hargs)
          super(socket, **hargs)
          # Initialize channels if any
          subscriptions&.each { |id| @subscriptions.add(id, false) }
        end

        def request
          @request ||= Rack::Request.new(socket.env)
        end

        def handle_channel_command(identifier, command, data)
          channel = subscriptions.add(identifier, false)
          case command
          when "subscribe"
            !subscriptions.send(:subscribe_channel, channel).nil?
          when "unsubscribe"
            subscriptions.remove(identifier)
            true
          when "message"
            subscriptions.perform_action identifier, data
            true
          else
            false
          end
        rescue LiteCable::Connection::Subscriptions::Error,
               LiteCable::Channel::Error,
               LiteCable::Channel::Registry::Error => e
          log(:error, log_fmt("Connection command failed: #{e}"))
          close
          false
        end
      end
    end
  end

  # Patch Lite Cable  with AnyCable functionality
  def self.anycable!
    LiteCable::Connection::Base.extend LiteCable::AnyCable::Connection
    LiteCable.singleton_class.prepend LiteCable::AnyCable::Broadcasting
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
litecable-0.4.2	lib/lite_cable/anycable.rb
litecable-0.4.1	lib/lite_cable/anycable.rb
litecable-0.4.0	lib/lite_cable/anycable.rb