lib/sinatra-websocketio/application.rb in sinatra-websocketio-0.0.1 vs lib/sinatra-websocketio/application.rb in sinatra-websocketio-0.0.2

@@ -1,14 +1,22 @@
 
 module Sinatra::WebSocketIO
 
+  def websocketio=(options)
+    WebSocketIO.options = options
+  end
+
+  def websocketio
+    WebSocketIO.options
+  end
+
   helpers do
     def websocketio_js
       "#{env['rack.url_scheme']}://#{env['HTTP_HOST']}#{env['SCRIPT_NAME']}/websocketio/websocketio.js"
     end
 
     def websocketio_url
-      "ws://#{env['SERVER_NAME']}:#{WebSocketIO.port}"
+      "ws://#{env['SERVER_NAME']}:#{WebSocketIO.options[:port]}"
     end
   end
 
   get '/websocketio/websocketio.js' do
     content_type 'application/javascript'