websocketio.rb in sinatra-websocketio-0.1.0

- old
+ new
@@ -1,80 +1,83 @@
-class WebSocketIO
+module Sinatra
+  module WebSocketIO
 
-  def self.start
-    EM::defer do
-      while !EM::reactor_running? do
-        sleep 1
-      end
-      puts "WebSocketIO.start port:#{options[:port]}"
-      EM::WebSocket.run :host => "0.0.0.0", :port => options[:port] do |ws|
-        ws.onopen do |handshake|
-          params = parse_handshake_params handshake.path
-          session_id = params[:session] || (create_session Socket.unpack_sockaddr_in ws.get_peername)
+    def self.start
+      EM::defer do
+        while !EM::reactor_running? do
+          sleep 1
+        end
+        puts "Sinatra::WebSocketIO.start port:#{options[:port]}"
+        EM::WebSocket.run :host => "0.0.0.0", :port => options[:port] do |ws|
+          ws.onopen do |handshake|
+            params = parse_handshake_params handshake.path
+            session_id = params[:session] || (create_session Socket.unpack_sockaddr_in ws.get_peername)
 
-          if self.sessions.include? session_id
-            ws.send({:type => :error, :data => "invalid session_id (#{session_id})"}.to_json)
-            ws.close
-          else
-            self.sessions[session_id] = {
-              :websocket => ws
-            }
-            ws.onclose do
-              self.sessions.delete session_id
-              self.emit :disconnect, session_id
-            end
-            ws.onmessage do |msg|
-              begin
-                data = JSON.parse msg
-              rescue => e
-                self.emit :error, e
+            if self.sessions.include? session_id
+              ws.send({:type => :error, :data => "invalid session_id (#{session_id})"}.to_json)
+              ws.close
+            else
+              self.sessions[session_id] = {
+                :websocket => ws
+              }
+              ws.onclose do
+                self.sessions.delete session_id
+                self.emit :disconnect, session_id
               end
-              self.emit data['type'], data['data'], session_id if data
+              ws.onmessage do |msg|
+                begin
+                  data = ::JSON.parse msg
+                rescue => e
+                  self.emit :error, e
+                end
+                self.emit data['type'], data['data'], session_id if data
+              end
+              ws.send({:type => :__session_id, :data => session_id}.to_json)
+              self.emit :connect, session_id
             end
-            ws.send({:type => :__session_id, :data => session_id}.to_json)
-            self.emit :connect, session_id
-          end
 
+          end
         end
       end
     end
-  end
 
-  def self.push(type, data, opt={})
-    session_ids = opt[:to].to_s.empty? ? self.sessions.keys : [opt[:to]]
-    session_ids.each do |id|
-      s = sessions[id]
-      begin
-        s[:websocket].send({:type => type, :data => data}.to_json)
-      rescue
-        next
+    def self.push(type, data, opt={})
+      session_ids = opt[:to].to_s.empty? ? self.sessions.keys : [opt[:to]]
+      session_ids.each do |id|
+        s = sessions[id]
+        begin
+          s[:websocket].send({:type => type, :data => data}.to_json)
+        rescue
+          next
+        end
       end
     end
-  end
 
-  def self.sessions
-    @@sessions ||= Hash.new{|h,session_id|
-      h[session_id] = {
-        :websocket => nil
+    def self.sessions
+      @@sessions ||= Hash.new{|h,session_id|
+        h[session_id] = {
+          :websocket => nil
+        }
       }
-    }
-  end
+    end
 
-  def self.create_session(ip_addr)
-    Digest::MD5.hexdigest "#{Time.now.to_i}_#{Time.now.usec}_#{ip_addr}"
-  end
+    def self.create_session(ip_addr)
+      Digest::MD5.hexdigest "#{Time.now.to_i}_#{Time.now.usec}_#{ip_addr}"
+    end
 
-  def self.parse_handshake_params(path)
-    tmp = path.gsub(/^\//,'').split(/\=+/)
-    params = {}
-    while !tmp.empty?
-      params[tmp.shift.to_sym] = tmp.shift
+    def self.parse_handshake_params(path)
+      tmp = path.gsub(/^\//,'').split(/\=+/)
+      params = {}
+      while !tmp.empty?
+        params[tmp.shift.to_sym] = tmp.shift
+      end
+      params
     end
-    params
-  end
 
+  end
 end
-EventEmitter.apply WebSocketIO
-WebSocketIO.on :__session_id do |data, from|
-  WebSocketIO.push :__session_id, from, :to => from
+
+EventEmitter.apply Sinatra::WebSocketIO
+Sinatra::WebSocketIO.on :__session_id do |data, from|
+  push :__session_id, from, :to => from
 end
 