Class: Mattermost::WebSocketClient

Inherits:
Object
  • Object
show all
Includes:
EventEmitter
Defined in:
lib/mattermost/websocket_client.rb

Instance Method Summary collapse

Constructor Details

#initialize(url, token, option = {}) {|_self| ... } ⇒ WebSocketClient

Returns a new instance of WebSocketClient.

Yields:

  • (_self)

Yield Parameters:



9
10
11
12
13
14
15
16
17
18
 
# File 'lib/mattermost/websocket_client.rb', line 9

def initialize(url, token, option = {})
	@token = token
	@url = url
	@option = option
	@seq_mutex = Mutex.new
	@connected = false
	yield self if block_given?
	connect
	self
end

Instance Method Details

#close ⇒ Object 



70
71
72
73
74
75
76
 
# File 'lib/mattermost/websocket_client.rb', line 70

def close
	@connected = false
	@client.close if @client != nil
	@client = nil
	Thread.kill @em if @em != nil
	@em = nil
end

#connect ⇒ Object 



20
21
22
23
24
25
26
27
28
 
# File 'lib/mattermost/websocket_client.rb', line 20

def connect
	return self if connected?
	mm_ws = self
	@em = Thread.new do
		EM.run do
			mm_ws.connect_internal
		end
	end
end

#connect_internal ⇒ Object 



78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
 
# File 'lib/mattermost/websocket_client.rb', line 78

def connect_internal
	@seq = 0
	mm_ws = self
	@client = Faye::WebSocket::Client.new(@url.gsub(/^http(s?):/, 'ws\1:'), {}, { :headers => {
		'Authorization' => "Bearer #{@token}"
	}})
	@client.on :open do |e|
		mm_ws.on_open e
	end
	@client.on :message do |msg|
		mm_ws.on_message msg.data
	end
	@client.on :close do |e|
		mm_ws.on_close e
	end
	@client.on :error do |e|
		mm_ws.on_error e
	end
end

#connected? ⇒ Boolean

Returns:

  • (Boolean) 


66
67
68
 
# File 'lib/mattermost/websocket_client.rb', line 66

def connected?
	@connected && (@client != nil && @client.ready_state == Faye::WebSocket::API::OPEN)
end

#on_close(msg) ⇒ Object 



48
49
50
51
 
# File 'lib/mattermost/websocket_client.rb', line 48

def on_close(msg)
	@connected = false
	emit :close, msg
end

#on_error(err) ⇒ Object 



53
54
55
 
# File 'lib/mattermost/websocket_client.rb', line 53

def on_error(err)
	emit :error, err
end

#on_message(data) ⇒ Object 



34
35
36
37
38
39
40
41
42
43
44
45
46
 
# File 'lib/mattermost/websocket_client.rb', line 34

def on_message(data)
	json = JSON.parse data
	event = json["event"]
	#puts "on_message json:#{json}, event:#{event}"
	seq_up json
	case event
	when "hello"
		@connected = true
	else
		emit event.to_sym, json
	end
	emit :message, json
end

#on_open(e) ⇒ Object 



30
31
32
 
# File 'lib/mattermost/websocket_client.rb', line 30

def on_open(e)
	emit :open, e
end

#send_msg(action, data) ⇒ Object 



57
58
59
60
61
62
63
64
 
# File 'lib/mattermost/websocket_client.rb', line 57

def send_msg(action, data)
	payload = {
		:seq => next_seq,
		:action => action,
		:data => data
	}.to_json
	@client.send payload
end