blob: 366ffb1da802434a8ae052c185d50f3c0ea6f716 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
module ActionCable
module Channel
class Base
include Callbacks
include PeriodicTimers
include Streams
on_subscribe :subscribed
on_unsubscribe :unsubscribed
attr_reader :params, :connection
delegate :logger, to: :connection
def initialize(connection, identifier, params = {})
@connection = connection
@identifier = identifier
@params = params
subscribe_to_channel
end
def subscribe_to_channel
logger.info "#{channel_name} subscribing"
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 unsubscribe_from_channel
run_unsubscribe_callbacks
logger.info "#{channel_name} unsubscribed"
end
protected
# Override in subclasses
def authorized?
true
end
def unauthorized
logger.error "#{channel_name}: Unauthorized access"
end
def subscribed
# Override in subclasses
end
def unsubscribed
# 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: @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
end
end
end
|
