agent_xmpp 0.1.0

data/lib/agent_xmpp/models.rb

@@ -0,0 +1,7 @@
+require 'agent_xmpp/models/table_definitions'
+require 'agent_xmpp/models/contact'
+require 'agent_xmpp/models/roster'
+require 'agent_xmpp/models/service'
+require 'agent_xmpp/models/publication'
+require 'agent_xmpp/models/subscription'
+require 'agent_xmpp/models/message'

data/lib/agent_xmpp/models/contact.rb

@@ -0,0 +1,85 @@
+##############################################################################################################
+module AgentXmpp
+  
+  #####-------------------------------------------------------------------------------------------------------
+  class Contact
+
+    #####-------------------------------------------------------------------------------------------------------
+    class << self
+
+      #.........................................................................................................
+      def contacts
+        @contacts ||= AgentXmpp.agent_xmpp_db[:contacts]
+      end
+
+      #.........................................................................................................
+      def load_config
+        if AgentXmpp.config['roster'].kind_of?(Array)
+          AgentXmpp.config['roster'].each {|c| update(c)}
+        end
+      end
+
+      #.........................................................................................................
+      def update(contact)
+        groups = contact['groups'].kind_of?(Array) ? contact['groups'].join(",") : contact['groups']
+        begin
+          contacts << {:jid => contact['jid'], :groups => groups, :subscription => 'new', :ask => 'new'}
+        rescue 
+          contacts.filter(:jid => contact['jid']).update(:groups => groups)
+        end
+      end
+
+      #.........................................................................................................
+      def update_with_roster_item(roster_item)
+        from_jid, subscription, ask = roster_item.jid.to_s, roster_item.subscription.to_s, roster_item.ask.to_s
+        contacts.filter(:jid => from_jid).update(:subscription => subscription, :ask => ask)
+      end
+
+      #.........................................................................................................
+      def find_all
+        contacts.all
+      end
+
+      #.........................................................................................................
+      def find_by_jid(jid)
+        contacts[:jid => AgentXmpp.bare_jid_to_s(jid)]
+      end
+
+      #.........................................................................................................
+      def find_all_by_subscription(subscription)
+        contacts.filter(:subscription => subscription.to_s).all
+      end
+
+      #.........................................................................................................
+      def has_jid?(jid)
+        contacts.filter(:jid => AgentXmpp.bare_jid_to_s(jid)).count > 0
+      end
+
+      #.........................................................................................................
+      def message_stats
+        find_all.map do |c|
+          stats = AgentXmpp.agent_xmpp_db["SELECT count(id) AS count, max(created_at) AS last FROM messages WHERE from_jid LIKE '#{c[:jid]}%'"].first
+          {:jid=>c[:jid], :count=>stats[:count], :last=>stats[:last].nil? ? 'Never' : Time.parse(stats[:last]).strftime("%y/%m/%d %H:%M")}
+        end        
+      end
+      
+      #.........................................................................................................
+      def destroy_by_jid(jid)
+        contact = contacts.filter(:jid => AgentXmpp.bare_jid_to_s(jid))
+        Roster.destroy_by_contact_id(contact.first[:id])
+        contact.delete
+      end 
+
+      #.........................................................................................................
+      def method_missing(meth, *args, &blk)
+        contacts.send(meth, *args, &blk)
+      end
+      
+    #### self
+    end
+
+  #### Contact
+  end
+
+#### AgentXmpp
+end

data/lib/agent_xmpp/models/message.rb

@@ -0,0 +1,152 @@
+##############################################################################################################
+module AgentXmpp
+  
+  #####-------------------------------------------------------------------------------------------------------
+  class Message
+
+    #####-------------------------------------------------------------------------------------------------------
+    class << self
+
+      #.........................................................................................................
+      def messages(renew=false)
+        @messages ||= AgentXmpp.agent_xmpp_db[:messages]
+      end
+
+      #.........................................................................................................
+      def update(stanza)
+        case stanza
+          when AgentXmpp::Xmpp::Message 
+            if stanza.type.eql?(:chat)
+              update_with_chat_message(stanza)
+            end
+          when AgentXmpp::Xmpp::Iq
+            if cmd = stanza.command
+              if data = cmd.x
+                update_with_command_data(stanza, data, cmd.node)
+              else
+                update_with_command_request(stanza, cmd.node)
+              end
+            elsif (pubsub = stanza.pubsub).kind_of?(AgentXmpp::Xmpp::IqPubSub) and pub = pubsub.publish and item = pub.item 
+              if data = item.x
+                update_with_published_data(stanza, data, pub.node)
+              end
+            end
+        end                 
+      end
+
+      #.........................................................................................................
+      def update_received_event_item(item, from, node)
+        (event_item =  item.entry || item.x) if item.respond_to?(:entry) and item.respond_to?(:x)
+        event_item.nil? ? false : update_with_event_item(event_item, from, node, item.id) 
+      end
+
+      #.........................................................................................................
+      def find_by_item_id(item_id)
+        item_id.nil? ? nil : messages[:item_id => item_id]        
+      end
+
+      #.........................................................................................................
+      def count_by_from_jid(from_jid)
+      end
+
+      #.........................................................................................................
+      def count_by_content_type(content_type)
+        messages.filter(:content_type => content_type).count
+      end
+
+      #.........................................................................................................
+      def stats_by_message_type
+        AgentXmpp.agent_xmpp_db['SELECT message_type AS type, count(id) AS count, max(created_at) AS last FROM messages GROUP BY message_type'].all.map do |s|
+          {:type=>s[:type], :count=>s[:count], :last=>Time.parse(s[:last]).strftime("%y/%m/%d %H:%M")}
+        end        
+      end
+        
+      #.........................................................................................................
+      def stats_by_node(node)
+        stats = AgentXmpp.agent_xmpp_db["SELECT node, count(id) AS count, max(created_at) AS last FROM messages WHERE node = ?", node].first
+        {:node=>node, :count=>stats[:count], :last=>stats[:last].nil? ? 'Never' : Time.parse(stats[:last]).strftime("%y/%m/%d %H:%M")}
+      end
+                
+      #.........................................................................................................
+      def stats_by_command_node
+        BaseController.command_nodes.map{|n| stats_by_node(n)}        
+      end
+                
+      #.........................................................................................................
+      # private
+      #.........................................................................................................
+      def update_with_chat_message(stanza)
+        from_jid = stanza.from || AgentXmpp.jid 
+        messages << {
+          :message_text => stanza.body,
+          :content_type => 'text',
+          :message_type => 'chat',
+          :to_jid       => stanza.to.to_s,
+          :from_jid     => from_jid.to_s,
+          :created_at   => Time.now}
+      end
+ 
+      #.........................................................................................................
+      def update_with_command_request(stanza, node)
+        from_jid = stanza.from || AgentXmpp.jid 
+        messages << {
+          :content_type  => 'command_request',
+          :message_type   => 'command',
+          :to_jid        => stanza.to.to_s,
+          :from_jid      => from_jid.to_s,
+          :node          => node,
+          :created_at    => Time.now}
+      end
+
+      #.........................................................................................................
+      def update_with_command_data(stanza, data, node)
+        from_jid = stanza.from || AgentXmpp.jid 
+        messages << {
+          :message_text  => data.to_s,
+          :content_type  => 'x',
+          :message_type  => 'command',
+          :to_jid        => stanza.to.to_s,
+          :from_jid      => from_jid.to_s,
+          :node          => node,
+          :created_at    => Time.now}
+      end
+ 
+      #.........................................................................................................
+      def update_with_published_data(stanza, data, node)
+        from_jid = stanza.from || AgentXmpp.jid 
+        messages << {
+          :message_text  => data.to_s,
+          :content_type  => 'x',
+          :message_type  => 'event',
+          :to_jid        => stanza.to.to_s,
+          :from_jid      => from_jid.to_s,
+          :node          => node,
+          :created_at    => Time.now}
+      end
+
+      #.........................................................................................................
+      def update_with_event_item(event_item, from, node, item_id)
+        unless find_by_item_id(item_id)
+          messages << {
+            :message_text  => event_item.to_s,
+            :content_type  => event_item.name,
+            :message_type  => 'event',
+            :to_jid        => AgentXmpp.jid.to_s,
+            :from_jid      => from.to_s,
+            :node          => node,
+            :item_id       => item_id,
+            :created_at    => Time.now}; true
+        else; false; end    
+      end
+       
+      #.........................................................................................................
+      private :update_with_chat_message, :update_with_command_data, :update_with_published_data, :update_with_command_request
+
+    #### self
+    end
+
+  #### Message
+  end
+
+#### AgentXmpp
+end

data/lib/agent_xmpp/models/publication.rb

@@ -0,0 +1,53 @@
+##############################################################################################################
+module AgentXmpp
+  
+  #####-------------------------------------------------------------------------------------------------------
+  class Publication
+
+    #####-------------------------------------------------------------------------------------------------------
+    class << self
+
+      #.........................................................................................................
+      def publications(renew=false)
+        @publications ||= AgentXmpp.in_memory_db[:publications]
+      end
+
+      #.........................................................................................................
+      def load_config
+        if AgentXmpp.config['publish'].kind_of?(Array)
+          AgentXmpp.config['publish'].each do |pub|
+            publications << AgentXmpp::DEFAULT_PUBSUB_CONFIG.inject({}) do |f, (a, v)| 
+              f.merge(a => pub[a.to_s] || v)
+            end.merge(:status => 'new', :node => pub['node']) 
+          end
+        end
+      end
+
+      #.........................................................................................................
+      def find_by_node(node)
+        publications[:node => node]
+      end
+
+      #.........................................................................................................
+      def find_all
+        publications.all
+      end
+
+      #.........................................................................................................
+      def update_status(node, status)
+        publications.filter(:node => node).update(:status => status.to_s)
+      end
+
+      #.........................................................................................................
+      def stats_by_node
+        find_all.map{|p| Message.stats_by_node("#{AgentXmpp.user_pubsub_root}/#{p[:node]}").update(:node=>p[:node].split('/').last)}
+      end
+      
+    #### self
+    end
+
+  #### Publication
+  end
+
+#### AgentXmpp
+end

data/lib/agent_xmpp/models/roster.rb

@@ -0,0 +1,107 @@
+##############################################################################################################
+module AgentXmpp
+    
+  #####-------------------------------------------------------------------------------------------------------
+  class Roster
+
+    #####-------------------------------------------------------------------------------------------------------
+    class << self
+
+      #.........................................................................................................
+      def roster(renew=false)
+        @roster ||= AgentXmpp.in_memory_db[:roster]
+      end
+
+      #.........................................................................................................
+      def update(msg, from=nil)
+        case msg
+          when AgentXmpp::Xmpp::Presence then update_with_presence(msg)
+          when AgentXmpp::Xmpp::IqVersion then update_with_version(msg, from)
+         end                 
+      end
+ 
+      #.........................................................................................................
+      def update_status(jid, status)
+        roster.filter(:jid => AgentXmpp.full_jid_to_s(jid)).update(:status => status.to_s)
+      end
+                
+      #.........................................................................................................
+      def find_all
+        roster.all  
+      end
+
+      #.........................................................................................................
+      def find_by_jid(jid)
+        roster[:jid => AgentXmpp.full_jid_to_s(jid)]
+      end 
+      
+      #.........................................................................................................
+      def find_all_by_status(status)
+        roster.filter(:status => status.to_s).all
+      end
+
+      #.........................................................................................................
+      def find_all_by_contact_jid(jid)
+        if contact = Contact.find_by_jid(jid)
+          roster.filter(:contact_id => contact[:contact_id]).all
+        else; []; end
+      end 
+      
+      #.........................................................................................................
+      def find_all_by_contact_jid_and_status(jid, status)
+        if contact = Contact.find_by_jid(jid)
+          roster.filter(:contact_id => contact[:id]).all
+        else; []; end
+      end 
+      
+      #.........................................................................................................
+      def destroy_by_contact_id(contact_id)
+        roster.filter(:contact_id => contact_id).delete
+      end 
+
+      #.........................................................................................................
+      def has_version?(jid)
+        if item = find_by_jid(jid)
+          not (item[:client_name].nil? or item[:client_version].nil?)
+        else; false; end
+      end 
+
+      #.........................................................................................................
+      def method_missing(meth, *args, &blk)
+        roster.send(meth, *args, &blk)
+      end
+
+      #.........................................................................................................
+      # private
+      #.........................................................................................................
+      def update_with_presence(presence)
+        from_jid = presence.from.to_s    
+        contact = Contact.find_by_jid(presence.from)
+        status = presence.type.nil? ? 'available' : presence.type.to_s 
+        if (contact)
+          begin
+            roster << {:jid => from_jid, :status => status, :contact_id => contact[:id]}
+          rescue 
+            roster.filter(:jid => from_jid).update(:status => status)
+          end
+        end
+      end
+
+      #.........................................................................................................
+      def update_with_version(vquery, from)
+        if (item = roster.filter(:jid => from.to_s))  
+          item.update(:client_name => vquery.iname, :client_version => vquery.version, :client_os => vquery.os)
+        end
+      end
+      
+      #.........................................................................................................
+      private :update_with_presence, :update_with_version
+
+    #### self
+    end
+
+  #### Roster
+  end
+
+#### AgentXmpp
+end

data/lib/agent_xmpp/models/service.rb

@@ -0,0 +1,91 @@
+##############################################################################################################
+module AgentXmpp
+  
+  #####-------------------------------------------------------------------------------------------------------
+  class Service
+
+    #####-------------------------------------------------------------------------------------------------------
+    class << self
+
+      #.........................................................................................................
+      def services(renew=false)
+        @services ||= AgentXmpp.in_memory_db[:services]
+      end
+
+      #.........................................................................................................
+      def service_items
+        @service_items ||= AgentXmpp.in_memory_db[:service_items]
+      end
+
+      #.........................................................................................................
+      def service_features
+        @service_features ||= AgentXmpp.in_memory_db[:service_features]
+      end
+
+      #.........................................................................................................
+      def update(disco_iq)
+        disco = disco_iq.query
+        case disco
+          when AgentXmpp::Xmpp::IqDiscoInfo then update_with_disco_info(disco_iq)
+          when AgentXmpp::Xmpp::IqDiscoItems then update_with_disco_items(disco_iq)
+        end
+      end
+
+      #.........................................................................................................
+      def has_jid?(jid)
+        services.filter(:jid => jid.to_s).count > 0
+      end
+
+      #.........................................................................................................
+      def has_disco_info?(jid)
+        services.filter(:jid => jid.to_s).count > 0
+      end
+
+      #.........................................................................................................
+      def has_disco_items?(jid)
+        service_items.filter(:service => jid.to_s).count > 0
+      end
+
+      #.........................................................................................................
+      # private
+      #.........................................................................................................
+      def update_with_disco_info(disco_iq)
+        disco, service = disco_iq.query, disco_iq.from.to_s
+        node = disco.node
+        disco.identities.each do |i|
+          begin
+            services << {:node => node, :jid => service, :category => i.category, :type => i.type, :name => i.iname}
+          rescue
+          end
+        end
+        disco.features.each do |f|
+          begin
+            service_features << {:node => node, :service => service, :var => f.var}
+          rescue
+          end
+        end
+      end
+
+      #.........................................................................................................
+      def update_with_disco_items(disco_iq)
+        disco, service = disco_iq.query, disco_iq.from.to_s
+        parent_node = disco.node
+        disco.items.each do |i|
+          begin
+            service_items << {:parent_node => parent_node, :service => service, :node => i.node, :jid => i.jid.to_s, :name => i.iname}
+          rescue
+          end
+        end
+      end
+
+      #.........................................................................................................
+      private :update_with_disco_info, :update_with_disco_items
+
+    #### self
+    end
+
+  #### Service
+  end
+
+#### AgentXmpp
+end