social_stream-presence 0.0.2 → 0.0.4

data/ejabberd/mod_sspresence/mod_sspresence.erl

@@ -0,0 +1,257 @@
+-module(mod_sspresence).
+
+-behavior(gen_mod).
+
+-include("ejabberd.hrl").
+
+-export([start/2, stop/1, on_register_connection/3, on_remove_connection/3, on_presence/4, on_unset_presence/4, isConnected/1, on_packet_send/3]).
+
+start(Host, _Opts) ->
+    ?INFO_MSG("mod_sspresence starting", []),
+    ejabberd_hooks:add(sm_register_connection_hook, 	Host, ?MODULE, on_register_connection, 50),
+    ejabberd_hooks:add(sm_remove_connection_hook, 	Host, ?MODULE, on_remove_connection, 50),
+    ejabberd_hooks:add(set_presence_hook, 		Host, ?MODULE, on_presence, 50),
+    ejabberd_hooks:add(unset_presence_hook, 		Host, ?MODULE, on_unset_presence, 50),
+    ejabberd_hooks:add(user_send_packet, 		Host, ?MODULE, on_packet_send, 50),
+    %Uncomment to reset connected users directly when module sspresence starts.
+    %reset_connections(),
+    ok.
+
+stop(Host) ->
+    ?INFO_MSG("mod_sspresence stopping", []),
+    ejabberd_hooks:delete(sm_register_connection_hook, 	Host, ?MODULE, on_register_connection, 50),
+    ejabberd_hooks:delete(sm_remove_connection_hook, 	Host, ?MODULE, on_remove_connection, 50),
+    ejabberd_hooks:delete(set_presence_hook, 		Host, ?MODULE, on_presence, 50),
+    ejabberd_hooks:delete(unset_presence_hook, 		Host, ?MODULE, on_unset_presence, 50),
+    ejabberd_hooks:delete(user_send_packet, 		Host, ?MODULE, on_packet_send, 50),
+    ok.
+
+
+
+on_register_connection(_SID, _JID, _Info) ->
+    {_A,User,_B,_C,_D,_E,_F} = _JID,
+    ?INFO_MSG("mod_sspresence: on_register_connection (~p)", [User]),
+    Login_path = string:concat(getOptionValue("scripts_path="), "/set_connection_script "),
+    os:cmd(string:join([Login_path, User ], " ")),
+    ok.
+
+on_remove_connection(_SID, _JID, _SessionInfo) ->
+    {_A,User,_B,_C,_D,_E,_F} = _JID,
+    ?INFO_MSG("mod_sspresence: on_remove_connection (~p)", [User]),
+    Connected = isConnected(User),
+    case Connected of
+	true -> ok;
+	_ -> Logout_path = string:concat(getOptionValue("scripts_path="), "/unset_connection_script "),
+             os:cmd(string:join([Logout_path, User ], " "))
+    end,
+    ok.
+
+on_presence(User, _Server, _Resource, Packet) ->
+     ?INFO_MSG("mod_sspresence: on_presence (~p)", [User]),
+     {_xmlelement, Type, _Attr, Subel} = Packet,
+
+    case Type of
+	"presence" -> Status = getStatusFromSubel(Subel),
+		      Login_path = string:concat(getOptionValue("scripts_path="), "/set_presence_script "),
+		      ?INFO_MSG("mod_sspresence: set_presence_script call with  user (~p) and status (~p)", [User,Status]),
+    		      os:cmd(string:join([Login_path, User , Status], " "));
+	_ -> ok
+    end,
+    ok.
+
+on_unset_presence(User, _Server, _Resource, _Status) ->
+    ?INFO_MSG("mod_sspresence: on_unset_presence (~p)", [User]),
+    ok.
+
+on_packet_send(From, _To, {xmlelement, Type, _Attr, Subel} = _Packet) ->
+    case Type of
+	"message" -> 
+		{_SenderJID,Sender,_SenderDomain,_A,_B,_C,_D} = From,
+		
+		SSlogin = getOptionValue("ss_login="),
+
+		case Sender of
+			SSlogin ->
+                                Message = getMessageFromSubel(Subel),
+				execute(Message);
+                   	_ -> ok
+                end;
+	_ -> ok
+    end.
+
+
+getStatusFromSubel(Subel) ->	
+	try
+            %Strophe tuples
+	    [{_A,"status",[],[{_B,_StatusStrophe}]},{_C,"show",[],[{_D,ShowStrophe}]}|_R1] = Subel,
+	    parseXmlCdata(ShowStrophe)
+	catch
+		_:Reason ->      
+			try
+				%Pidgin tuples
+				[{_E,"show",[],[{_F,ShowPidgin}]}|_R2] = Subel,
+			     	parseXmlCdata(ShowPidgin)
+			catch
+				_:Reason -> "chat" %Status when parsed failed
+			after
+				noop
+			end
+	after
+		noop
+	end. 
+
+
+getMessageFromSubel(Subel) ->
+    [{_I,_J,_K,[{_L,MessageData}]}] = Subel,
+    parseXmlCdata(MessageData).
+
+
+parseXmlCdata(Msg) ->
+	MessageString = binary_to_list(list_to_binary(io_lib:format("~p", [Msg]))),
+	lists:sublist(MessageString, 4, string:len(MessageString)-6).
+
+
+execute(Message) ->
+    %?INFO_MSG("Message vale: ~p", [Message]),
+    %?INFO_MSG("Order vale: ~p", [Order]),
+    [Order|Params] = string:tokens(Message, "&"),
+    case Order of
+	"AddItemToRoster" -> 
+	  	case length(Params) of
+			4 -> 	[UserSID,BuddySID,BuddyName,Subscription_type] = Params,
+				?INFO_MSG("Execute: ~p with params ~p", [Order,Params]),
+				[ContactName|_R] = string:tokens(BuddyName, " "),
+				[UserSlug,UserDomain] = string:tokens(UserSID, "@"),
+				[BuddySlug,BuddyDomain] = string:tokens(BuddySID, "@"),
+				addItemToRoster(UserSlug,UserDomain,BuddySlug,BuddyDomain,ContactName,"SocialStream",Subscription_type),
+				ok;
+			_ -> 	?INFO_MSG("Incorrect parameters in order ~p", [Order]),
+				ok
+	  	end,
+		ok;
+
+	"SetRosterForBidirectionalTie" -> 
+	  	case length(Params) of
+			4 -> 	[UserASID,UserBSID,UserAName,UserBName] = Params,
+				?INFO_MSG("Executing: ~p with params ~p", [Order,Params]),
+				[UserASlug,UserADomain] = string:tokens(UserASID, "@"),
+				[UserBSlug,UserBDomain] = string:tokens(UserBSID, "@"),
+				addItemToRoster(UserASlug,UserADomain,UserBSlug,UserBDomain,UserBName,"SocialStream","both"),
+				addItemToRoster(UserBSlug,UserBDomain,UserASlug,UserADomain,UserAName,"SocialStream","both"),
+				ok;
+			_ -> 	?INFO_MSG("Incorrect parameters in order ~p", [Order]),
+				ok
+	  	end,
+		ok;
+
+	"Synchronize" ->
+	synchronizePresence(),
+	ok;
+
+	"SynchronizeRosters" ->
+	synchronizeRosters(),
+	ok;
+
+	_ -> ?INFO_MSG("Command not found", []),
+		ok
+    end.
+
+
+
+%%Call ejabberdctl command add_rosteritem
+	%Command Name: add_rosteritem
+	%Needs mod_admin_extra (http://www.ejabberd.im/ejabberd-modules)
+	%ejabberdctl add_rosteritem localuser localserver buddy buddyserver nick group subs
+	%subs= none, from, to or both,
+
+	%Example
+	%ejabberdctl add_rosteritem frank-williamson trapo demo trapo NickName SocialStream from
+	%frank-williamson@trapo adds demo@trapo to its roster with subscription from, in the group SocialStream with the name Nickname
+%%
+addItemToRoster(UserSlug,UserDomain,BuddySlug,BuddyDomain,ContactName,Group,Subscription_type) ->
+	Command = lists:concat(["ejabberdctl add_rosteritem ", UserSlug , " ", UserDomain, " ", BuddySlug, " ", BuddyDomain , " \\'", ContactName , "\\' ", Group , " ", 					Subscription_type]),
+	os:cmd(Command),
+	?INFO_MSG("Execute command: ~p", [Command]),
+ok.
+
+%%GETTERS CONFIG VALUES
+%%CONFIG FILE: /etc/ejabberd/ssconfig.cfg
+
+getOptionValue(Option) ->
+{_, In} = file:open("/etc/ejabberd/ssconfig.cfg", read),
+parser(In,Option).
+
+
+parser(In,Option) ->
+  L = io:get_line(In, ''),
+
+  case L of
+  	eof -> "Undefined";
+  	_ -> {_,S,_} = regexp:gsub(L, "\\n", ""),
+
+
+	%%IGNORE COMMENTS
+	case string:str(S, "#") of
+        	0 -> 
+			case string:str(S, Option) of
+				0 -> ok,
+				%% Continue with next line
+				parser(In,Option);
+		
+				%% return path
+				_ ->  lists:sublist(S, string:len(Option)+1, string:len(S))	
+			end;
+
+        	_ ->  %% Comment: Continue with next line
+		     parser(In,Option)
+	end
+  end. 
+
+
+%%Check if a user is connected (any active session with Ejabberd server)
+isConnected(User) ->
+
+Command = string:concat("ejabberdctl connected-users | grep ", User),
+Output = os:cmd(Command),
+
+case Output of
+  [] -> false;
+  _ -> Sessions = string:tokens(Output, "\n"),
+
+      catch lists:foreach(
+		fun(S) ->
+		        [Slug|_R] = string:tokens(S, "@"),
+		        %User.slug connected = Slug
+			
+			case Slug of
+			  User -> throw(true);
+			  _ -> false
+			end                
+
+		end,
+		Sessions)
+end.
+
+
+%%Send all connected users to Social Stream Rails Application
+synchronizePresence() ->
+	Synchronize_path = string:concat(getOptionValue("scripts_path="), "/synchronize_presence_script "),
+	os:cmd(Synchronize_path),
+ok.
+
+
+%%Reset all rosters and wait to received data from Social Stream Rails Application
+synchronizeRosters() ->
+	SynchronizeRosters_path = string:concat(getOptionValue("scripts_path="), "/emanagement removeAllRosters"),
+	os:cmd(SynchronizeRosters_path),
+ok.
+
+
+%Reset all connections
+%reset_connections() ->
+%	Reset_path = string:concat(getOptionValue("scripts_path="), "/reset_connection_script "),
+%	os:cmd(Reset_path),
+%ok.
+
+
+

data/lib/social_stream-presence.rb

@@ -1,7 +1,24 @@
 require 'social_stream-base'
 
-module Socialstream
-  module Presence
+module SocialStream
+  module Presence   
+
+    module Models
+      autoload :BuddyManager, 'social_stream/presence/models/buddy_manager'
+    end
+
+    mattr_accessor :domain
+    mattr_accessor :bosh_service
+    mattr_accessor :password
+    mattr_accessor :xmpp_server_password
+    mattr_accessor :social_stream_presence_username
+
+    class << self
+      def setup
+        yield self
+      end
+    end
+
   end
 end
 

data/lib/social_stream/migrations/presence.rb

@@ -0,0 +1,9 @@
+require 'social_stream/migrations/components'
+
+module SocialStream
+  module Migrations
+    # Migrate Documents
+    class Presence < Components
+    end
+  end
+end

data/lib/social_stream/presence/engine.rb

@@ -0,0 +1,62 @@
+require 'xmpp4r'
+require 'xmpp4r/muc'
+require 'xmpp4r/roster'
+require 'xmpp4r/client'
+require 'xmpp4r/message'
+
+module SocialStream
+  module Presence
+    class Engine < Rails::Engine
+      config.to_prepare do
+        
+        #Patching Tie
+        Tie.class_eval do
+          include SocialStream::Presence::Models::BuddyManager
+        end
+        
+      end 
+               
+      initializer "social_stream-presence.synchronize" do
+          #Synchronize User Presence
+          #Reset connected users when XMMP Server Down
+          Thread.start {
+            begin
+              #XMPP DOMAIN
+              domain = SocialStream::Presence.domain
+              #PASSWORD
+              password= SocialStream::Presence.password
+              #SS Username
+              ss_name = SocialStream::Presence.social_stream_presence_username
+              
+              ss_sid = ss_name + "@" + domain
+              client = Jabber::Client.new(Jabber::JID.new(ss_sid))
+              client.connect
+              client.auth(password)
+       
+              msg = Jabber::Message::new(ss_sid, "Synchronize")
+              msg.type=:chat
+              client.send(msg)
+              client.close()
+            
+            rescue Exception => e
+              case e
+                when Errno::ECONNREFUSED
+                  begin
+                    users = User.find_all_by_connected(true)
+                    users.each do |user|
+                      user.connected = false
+                      user.save!
+                    end
+                    puts "Connection to XMPP Server refused: Reset Connected Users"
+                  rescue
+                  end
+                else
+                  puts "Unknown exception: #{e.to_s}"
+              end  
+            end           
+          }
+      end
+      
+    end
+  end
+end

data/lib/social_stream/presence/models/buddy_manager.rb

@@ -0,0 +1,76 @@
+require 'xmpp4r'
+require 'xmpp4r/muc'
+require 'xmpp4r/roster'
+require 'xmpp4r/client'
+require 'xmpp4r/message'
+
+module SocialStream
+  module Presence
+    module Models
+      module BuddyManager
+        extend ActiveSupport::Concern
+        
+        included do
+          after_create :save_buddy
+        end
+        
+        module InstanceMethods
+          
+          def save_buddy
+            
+            unless self.receiver.subject_type == "User" and self.sender.subject_type == "User"
+              return
+            end
+
+            #XMPP DOMAIN
+            domain = SocialStream::Presence.domain
+            #PASSWORD
+            password= SocialStream::Presence.password
+            #SS Username
+            ss_name = SocialStream::Presence.social_stream_presence_username
+            ss_sid = ss_name + "@" + domain
+              
+            user_sid = self.sender.slug + "@" + domain
+            user_name =  self.sender.name  
+            buddy_sid = self.receiver.slug + "@" + domain
+            buddy_name =  self.receiver.name
+            
+            
+            begin
+              client = Jabber::Client.new(Jabber::JID.new(ss_sid))
+              client.connect
+              client.auth(password)
+   
+              #Check if is a positive and replied tie         
+              if self.bidirectional?
+                #SetRosterForBidirectionalTie[UserASID,UserBSID,UserAName,UserBName]
+                msg = Jabber::Message::new(ss_sid, "SetRosterForBidirectionalTie&" + user_sid + "&" + buddy_sid + "&" + buddy_name + "&" + user_name) 
+              elsif self.positive?
+                #Case: Possitive tie unidirectional
+                sType = "from"
+                #AddItemToRoster[UserSID,BuddySID,BuddyName,Subscription_type]
+                msg = Jabber::Message::new(ss_sid, "AddItemToRoster&" + user_sid + "&" + buddy_sid + "&" + buddy_name + "&" + sType)
+              else
+                return  
+              end
+
+              msg.type=:chat
+              client.send(msg)
+              client.close()
+
+            rescue Exception => e
+              case e
+                when Errno::ECONNREFUSED
+                  puts "Connection to XMPP Server refused"
+                else
+                  puts "Unknown exception: #{e.to_s}"
+              end
+            end   
+            
+          end
+          
+        end
+      end
+    end
+  end
+end

data/lib/social_stream/presence/version.rb

@@ -0,0 +1,5 @@
+module Socialstream
+  module Presence
+    VERSION = "0.0.4"
+  end
+end

data/lib/tasks/presence/synchronize.rake

@@ -0,0 +1,63 @@
+namespace :presence do
+  desc 'Synchronize the Xmpp Server database with the Social Stream database'
+  task :synchronize => [ 'presence:synchronize:connections', 'presence:synchronize:rosters' ]
+
+  namespace :synchronize do
+
+
+    desc "Synchronize user presence."
+    task :connections => :environment do
+      puts "Starting presence:synchronize:connections"
+      puts "Synchronization complete"
+    end
+
+    desc "Remove all rosters and populate rosters from Social Stream data."
+    task :rosters => :environment do
+        puts "Starting presence:synchronize:rosters"
+        
+        #XMPP DOMAIN
+        domain = SocialStream::Presence.domain
+        #PASSWORD
+        password= SocialStream::Presence.password
+        #SS Username
+        ss_name = SocialStream::Presence.social_stream_presence_username      
+        ss_sid = ss_name + "@" + domain 
+        
+        puts "Connecting to Xmpp Server"
+        client = Jabber::Client.new(Jabber::JID.new(ss_sid))
+        client.connect
+        client.auth(password)
+        puts "Connected to Xmpp Server"
+        
+        puts "Remove all rosters"
+        msg = Jabber::Message::new(ss_sid, "SynchronizeRosters")
+        msg.type=:chat
+        client.send(msg)
+        
+   
+        puts "Populate rosters"
+        users = User.all
+        checkedUsers = []
+      
+        users.each do |user|
+          checkedUsers << user.slug
+          contacts = user.contact_actors(:type=>:user)
+          contacts.each do |contact|
+            unless checkedUsers.include?(contact.slug)
+              user_sid = user.slug + "@" + domain
+              buddy_sid = contact.slug + "@" + domain  
+              msg = Jabber::Message::new(ss_sid, "SetRosterForBidirectionalTie&" + user_sid + "&" + buddy_sid + "&" + user.name + "&" + contact.name)
+              msg.type=:chat
+              client.send(msg)
+            end
+          end
+        end
+
+        puts "Synchronization complete"
+        puts "Closing connection"
+        client.close()  
+        puts "Connection closing"
+      
+    end
+  end
+end