Class: PusherClient::Socket

Inherits:
Object
  • Object
show all
Defined in:
lib/pusher-client/socket.rb

Constant Summary collapse

CLIENT_ID = 
'pusher-ruby-client'
PROTOCOL = 
'6'

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(app_key, options = {}) ⇒ Socket

Returns a new instance of Socket.

Raises:

  • (ArgumentError) 


13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
 
# File 'lib/pusher-client/socket.rb', line 13

def initialize(app_key, options={})
  raise ArgumentError, "Missing app_key" if app_key.to_s.empty?

  @path = "#{options[:ws_path]}/app/#{app_key}?client=#{CLIENT_ID}&version=#{PusherClient::VERSION}&protocol=#{PROTOCOL}"
  @key = app_key.to_s
  @secret = options[:secret]
  @socket_id = nil
  @channels = Channels.new
  @global_channel = Channel.new('pusher_global_channel')
  @global_channel.global = true
  @connected = false
  @encrypted = options[:encrypted] || false
  @logger = options[:logger] || PusherClient.logger
  # :private_auth_method is deprecated
  @auth_method = options[:auth_method] || options[:private_auth_method]
  @cert_file = options[:cert_file]
  @ws_host = options[:ws_host] || HOST
  @ws_port = options[:ws_port] || WS_PORT
  @wss_port = options[:wss_port] || WSS_PORT
  @ssl_verify = options.fetch(:ssl_verify, true)

  if @encrypted
    @url = "wss://#{@ws_host}:#{@wss_port}#{@path}"
  else
    @url = "ws://#{@ws_host}:#{@ws_port}#{@path}"
  end

  bind('pusher:connection_established') do |data|
    socket = parser(data)
    @connected = true
    @socket_id = socket['socket_id']
    subscribe_all
  end

  bind('pusher:connection_disconnected') do |data|
    @connected = false
    @channels.channels.each { |c| c.disconnect }
  end

  bind('pusher:error') do |data|
    logger.fatal("Pusher : error : #{data.inspect}")
  end

  # Keep this in case we're using a websocket protocol that doesn't
  # implement ping/pong
  bind('pusher:ping') do
    send_event('pusher:pong', nil)
  end
end

Instance Attribute Details

#channels ⇒ Object (readonly)

Returns the value of attribute channels.



11
12
13
 
# File 'lib/pusher-client/socket.rb', line 11

def channels
  @channels
end

#connected ⇒ Object (readonly)

Returns the value of attribute connected.



11
12
13
 
# File 'lib/pusher-client/socket.rb', line 11

def connected
  @connected
end

#global_channel ⇒ Object (readonly)

Returns the value of attribute global_channel.



11
12
13
 
# File 'lib/pusher-client/socket.rb', line 11

def global_channel
  @global_channel
end

#path ⇒ Object (readonly)

Returns the value of attribute path.



11
12
13
 
# File 'lib/pusher-client/socket.rb', line 11

def path
  @path
end

#socket_id ⇒ Object (readonly)

Returns the value of attribute socket_id.



11
12
13
 
# File 'lib/pusher-client/socket.rb', line 11

def socket_id
  @socket_id
end

Instance Method Details

#[](channel_name) ⇒ Object 



124
125
126
 
# File 'lib/pusher-client/socket.rb', line 124

def [](channel_name)
  @channels[channel_name] || NullChannel.new(channel_name)
end

#authorize(channel, callback) ⇒ Object

auth for private and presence



133
134
135
136
137
138
139
140
141
 
# File 'lib/pusher-client/socket.rb', line 133

def authorize(channel, callback)
  if is_private_channel(channel.name)
    auth_data = get_private_auth(channel)
  elsif is_presence_channel(channel.name)
    auth_data = get_presence_auth(channel)
  end
  # could both be nil if didn't require auth
  callback.call(channel, auth_data, channel.user_data)
end

#authorize_callback(channel, auth_data, channel_data) ⇒ Object 



143
144
145
146
147
148
149
150
 
# File 'lib/pusher-client/socket.rb', line 143

def authorize_callback(channel, auth_data, channel_data)
  send_event('pusher:subscribe', {
    'channel' => channel.name,
    'auth' => auth_data,
    'channel_data' => channel_data
  })
  channel.acknowledge_subscription(nil)
end

#bind(event_name, &callback) ⇒ Object 



119
120
121
122
 
# File 'lib/pusher-client/socket.rb', line 119

def bind(event_name, &callback)
  @global_channel.bind(event_name, &callback)
  return self
end

#connect(async = false) ⇒ Object 



63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
 
# File 'lib/pusher-client/socket.rb', line 63

def connect(async = false)
  return if @connection
  logger.debug("Pusher : connecting : #{@url}")

  if async
    @connection_thread = Thread.new do
      begin
        connect_internal
      rescue => ex
        send_local_event "pusher:error", ex
      end
    end
  else
    connect_internal
  end
  self
end

#disconnect ⇒ Object 



81
82
83
84
85
86
87
88
89
90
91
 
# File 'lib/pusher-client/socket.rb', line 81

def disconnect
  return unless @connection
  logger.debug("Pusher : disconnecting")
  @connected = false
  @connection.close
  @connection = nil
  if @connection_thread
    @connection_thread.kill
    @connection_thread = nil
  end
end

#get_presence_auth(channel) ⇒ Object 



168
169
170
171
172
173
174
 
# File 'lib/pusher-client/socket.rb', line 168

def get_presence_auth(channel)
  return @auth_method.call(@socket_id, channel) if @auth_method

  string_to_sign = @socket_id + ':' + channel.name + ':' + channel.user_data
  signature = hmac(@secret, string_to_sign)
  "#{@key}:#{signature}"
end

#get_private_auth(channel) ⇒ Object 



160
161
162
163
164
165
166
 
# File 'lib/pusher-client/socket.rb', line 160

def get_private_auth(channel)
  return @auth_method.call(@socket_id, channel) if @auth_method

  string_to_sign = @socket_id + ':' + channel.name
  signature = hmac(@secret, string_to_sign)
  "#{@key}:#{signature}"
end

#is_presence_channel(channel_name) ⇒ Object 



156
157
158
 
# File 'lib/pusher-client/socket.rb', line 156

def is_presence_channel(channel_name)
  channel_name.match(/^presence-/)
end

#is_private_channel(channel_name) ⇒ Object 



152
153
154
 
# File 'lib/pusher-client/socket.rb', line 152

def is_private_channel(channel_name)
  channel_name.match(/^private-/)
end

#send_channel_event(channel, event_name, data) ⇒ Object 



186
187
188
189
190
 
# File 'lib/pusher-client/socket.rb', line 186

def send_channel_event(channel, event_name, data)
  payload = {'channel' => channel, 'event' => event_name, 'data' => data}.to_json
  @connection.send(payload)
  logger.debug("Pusher : sending channel event : #{payload}")
end

#send_event(event_name, data) ⇒ Object 



180
181
182
183
184
 
# File 'lib/pusher-client/socket.rb', line 180

def send_event(event_name, data)
  payload = {'event' => event_name, 'data' => data}.to_json
  @connection.send(payload)
  logger.debug("Pusher : sending event : #{payload}")
end

#subscribe(channel_name, user_data = nil) ⇒ Object 



93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
 
# File 'lib/pusher-client/socket.rb', line 93

def subscribe(channel_name, user_data = nil)
  if user_data.is_a? Hash
    user_data = user_data.to_json
  elsif user_data
    user_data = {:user_id => user_data}.to_json
  elsif is_presence_channel(channel_name)
    raise ArgumentError, "user_data is required for presence channels"
  end

  channel = @channels.add(channel_name, user_data)
  if @connected
    authorize(channel, method(:authorize_callback))
  end
  return channel
end

#subscribe_all ⇒ Object Also known as: subscribeAll 



128
129
130
 
# File 'lib/pusher-client/socket.rb', line 128

def subscribe_all
  @channels.channels.clone.each { |k,v| subscribe(v.name, v.user_data) }
end

#unsubscribe(channel_name) ⇒ Object 



109
110
111
112
113
114
115
116
117
 
# File 'lib/pusher-client/socket.rb', line 109

def unsubscribe(channel_name)
  channel = @channels.remove channel_name
  if channel && @connected
    send_event('pusher:unsubscribe', {
      'channel' => channel_name
    })
  end
  return channel
end