nocebo/diaspora
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Websocket now uses cookie to authenticate

Browse source
This commit is contained in:
Raphael 2010-11-02 17:39:22 -07:00
parent 153265aae8
commit ae039442c3
3 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
app/views/js/_websocket_js.haml
View file

@ -8,7 +8,7 @@
$(document).ready(function(){ $(document).ready(function(){
function debug(str){ $("#debug").append("<p>" + str); }; function debug(str){ $("#debug").append("<p>" + str); };
ws = new WebSocket("ws://#{request.host}:#{APP_CONFIG[:socket_port]}/#{CGI::escape(current_user.id.to_s)}"); ws = new WebSocket("ws://#{request.host}:#{APP_CONFIG[:socket_port]}/");
//Attach onmessage to websocket //Attach onmessage to websocket
ws.onmessage = function(evt) { ws.onmessage = function(evt) {

2
lib/diaspora/websocket.rb
View file

@ -15,7 +15,7 @@ module Diaspora
def self.push_to_user(uid, data) def self.push_to_user(uid, data)
Rails.logger.debug "Websocketing to #{uid}" Rails.logger.debug "Websocketing to #{uid}"
@channels[uid.to_s][0].push(data) if @channels[uid.to_s] @channels[uid][0].push(data) if @channels[uid]
end end
def self.subscribe(uid, ws) def self.subscribe(uid, ws)

8
script/websocket_server.rb
View file

@ -48,11 +48,15 @@ begin
:debug =>APP_CONFIG[:socket_debug]) do |ws| :debug =>APP_CONFIG[:socket_debug]) do |ws|
ws.onopen { ws.onopen {
sid = Diaspora::WebSocket.subscribe(ws.request['Path'].gsub('/',''), ws) encoded_cookie = ws.request["Cookie"].gsub("_diaspora_session=","")
cookie = Marshal.load(encoded_cookie.unpack("m*").first)
user_id = cookie["warden.user.user.key"].last
sid = Diaspora::WebSocket.subscribe(user_id, ws)
ws.onmessage { |msg| SocketsController.new.incoming(msg) } ws.onmessage { |msg| SocketsController.new.incoming(msg) }
ws.onclose { Diaspora::WebSocket.unsubscribe(ws.request['Path'].gsub('/',''), sid) } ws.onclose { Diaspora::WebSocket.unsubscribe(user_id, sid) }
} }
end end
PID_FILE = APP_CONFIG[:socket_pidfile] PID_FILE = APP_CONFIG[:socket_pidfile]