agent_xmpp 0.1.0

data/lib/agent_xmpp/models/subscription.rb

@@ -0,0 +1,61 @@
+##############################################################################################################
+module AgentXmpp
+  
+  #####-------------------------------------------------------------------------------------------------------
+  class Subscription
+
+    #####-------------------------------------------------------------------------------------------------------
+    class << self
+
+      #.........................................................................................................
+      def subscriptions(renew=false)
+        @subscriptions ||= AgentXmpp.in_memory_db[:subscriptions]
+      end
+
+      #.........................................................................................................
+      def find_all
+        subscriptions.all
+      end
+
+      #.........................................................................................................
+      def update(msg, node, serv)
+        case msg
+          when AgentXmpp::Xmpp::Subscription then update_with_subscription(msg, node, serv)
+          when AgentXmpp::Xmpp::Iq then update_with_subscription(msg.pubsub.subscription, node, serv)
+        end                 
+      end
+ 
+      #.........................................................................................................
+      def destroy_by_node(node)
+        subscriptions.filter(:node => node).delete
+      end 
+
+      #.........................................................................................................
+      def stats_by_node
+        find_all.map do |s|
+          node = s[:node].split('/')
+          Message.stats_by_node(s[:node]).update(:node=>node.last, :jid=>"#{node[3]}@#{node[2]}")
+        end
+      end
+      
+      #.........................................................................................................
+      # private
+      #.........................................................................................................
+      def update_with_subscription(msg, node, serv)
+        begin
+          subscriptions << {:node => node, :subscription => msg.subscription, :service => serv}
+        rescue 
+        end
+      end
+ 
+      #.........................................................................................................
+      private :update_with_subscription
+
+    #### self
+    end
+
+  #### Subscription
+  end
+
+#### AgentXmpp
+end

data/lib/agent_xmpp/models/table_definitions.rb

@@ -0,0 +1,107 @@
+##############################################################################################################
+module AgentXmpp
+  
+  #####-------------------------------------------------------------------------------------------------------
+  class << self
+
+    #####-----------------------------------------------------------------------------------------------------
+    def create_in_memory_db
+      in_memory_db.create_table :roster do
+      	primary_key :id
+      	column :contact_id, :integer
+        column :jid, :text, :unique=>true
+      	column :status, :text
+        column :client_name, :text
+        column :client_version, :text
+        column :client_os, :text
+      end
+      in_memory_db.create_table :services do
+      	primary_key :id
+        column :jid, :text
+        column :name, :text
+        column :category, :text
+        column :type, :text
+        column :node, :text
+        unique [:node, :jid]
+      end
+      in_memory_db.create_table :service_items do
+      	primary_key :id
+        column :parent_node, :text
+        column :service, :text
+        column :node, :text
+        column :jid, :text
+        column :name, :text
+        unique [:node, :service]
+      end
+      in_memory_db.create_table :service_features do
+      	primary_key :id
+        column :node, :text
+        column :service, :text
+        column :var, :text
+        unique [:node, :service, :var]
+      end
+      in_memory_db.create_table :publications do
+      	primary_key :id
+        column :node, :text, :unique=>true
+        column :status, :text
+        column :title, :text
+        column :access_model, :text
+        column :max_items, :integer
+        column :deliver_notifications, :integer
+        column :deliver_payloads, :integer
+        column :persist_items, :integer
+        column :subscribe, :integer
+        column :notify_config, :integer
+        column :notify_delete, :integer
+        column :notify_retract, :integer
+      end
+      in_memory_db.create_table :subscriptions do
+      	primary_key :id
+        column :node, :text, :unique=>true
+        column :service, :text
+        column :subscription, :text
+      end
+    end
+
+    #####-----------------------------------------------------------------------------------------------------
+    def create_agent_xmpp_db
+      unless agent_xmpp_db.table_exists? :version
+        agent_xmpp_db.create_table :version do
+        	primary_key :id
+          integer :number
+        end
+      end
+      version << {:number=>1} if version.count.eql?(0)
+      unless agent_xmpp_db.table_exists? :contacts
+        agent_xmpp_db.create_table :contacts do
+        	primary_key :id
+        	column :jid, :text, :unique=>true
+          column :ask, :text
+          column :subscription, :text
+        	column :groups, :text
+        end
+      end
+      unless agent_xmpp_db.table_exists? :messages
+        agent_xmpp_db.create_table :messages do
+        	primary_key :id
+        	column :message_text, :text
+        	column :content_type, :text
+        	column :message_type, :text
+        	column :to_jid, :text
+        	column :from_jid, :text
+        	column :node, :text
+        	column :item_id, :text
+          Time :created_at
+        end
+      end
+    end
+
+    #####-----------------------------------------------------------------------------------------------------
+    def upgrade_agent_xmpp_db
+    end
+
+  #### self
+  end
+    
+#### AgentXmpp 
+end

data/lib/agent_xmpp/patches.rb

@@ -0,0 +1,7 @@
+require 'agent_xmpp/patches/object'
+require 'agent_xmpp/patches/array'
+require 'agent_xmpp/patches/hash'
+require 'agent_xmpp/patches/string'
+require 'agent_xmpp/patches/float'
+require 'agent_xmpp/patches/rexml'
+

data/lib/agent_xmpp/patches/array.rb

@@ -0,0 +1,32 @@
+##############################################################################################################
+class Array
+  
+  #......................................................................................................
+  def to_x_data(type = 'result')
+    data = AgentXmpp::Xmpp::XData.new(type)
+    if first.kind_of?(Hash)
+      field_type = lambda{|v| v.kind_of?(Array) ? 'list-multi' : nil}
+      reported = AgentXmpp::Xmpp::XDataReported.new
+      first.each_key {|var| reported.add_field(var.to_s)}
+      data << reported
+      each do |fields|
+        item = AgentXmpp::Xmpp::XDataItem.new
+        fields.each{|var, val| item.add_field_with_value(var.to_s, [val].flatten.collect{|v| v.to_s}, field_type[val])}
+        data << item
+      end
+    else
+      field = AgentXmpp::Xmpp::XDataField.new
+      field.values = map{|v| v.to_s}
+      field.type ='list-multi'
+      data << field
+    end
+    data
+  end
+  
+  #......................................................................................................
+  def smash
+    self.flatten.compact
+  end
+        
+#### Array
+end

data/lib/agent_xmpp/patches/float.rb

@@ -0,0 +1,10 @@
+##############################################################################################################
+class Float  
+
+  #......................................................................................................
+  def precision(p = 3)
+    (10.0**p*self).round.to_f/10.0**p
+  end
+        
+#### Float
+end

data/lib/agent_xmpp/patches/hash.rb

@@ -0,0 +1,13 @@
+##############################################################################################################
+class Hash  
+
+  #.......................................................................................................
+  def to_x_data(type = 'result')
+    field_type = lambda{|v| v.kind_of?(Array) ? 'list-multi' : nil}
+    inject(AgentXmpp::Xmpp::XData.new(type)) do |data, (var, val)| 
+      data.add_field_with_value(var, [val].flatten.map{|v| v.to_s}, field_type[val])
+    end
+  end
+        
+#### Hash
+end

data/lib/agent_xmpp/patches/object.rb

@@ -0,0 +1,15 @@
+##############################################################################################################
+class Object  
+
+  #.......................................................................................................
+  def to_x_data(type='result')
+    AgentXmpp::Xmpp::XData.new(type).add_field_with_value(nil, to_s)
+  end
+
+  #.......................................................................................................
+  def define_meta_class_method(name, &blk)
+    (class << self; self; end).instance_eval {define_method(name, &blk)}
+  end
+
+#### Object
+end

data/lib/agent_xmpp/patches/rexml.rb

@@ -0,0 +1,69 @@
+# Original from XMPP4R - XMPP Library for Ruby Website::http://home.gna.org/xmpp4r/
+##############################################################################################################
+module REXML
+
+  ####----------------------------------------------------------------------------------------------------
+  class Element
+
+    #.......................................................................................................
+    def replace_element_text(e, t)
+      el = first_element(e)
+      if el.nil?
+        el = REXML::Element.new(e)
+        add_element(el)
+      end
+      el.text = t if t
+      self
+    end
+
+    #.......................................................................................................
+    def first_element(e)
+      elements.to_a(e).first
+    end
+
+    #.......................................................................................................
+    def first_element_text(e)
+      el = first_element(e)
+      el.nil? ? nil : el.text
+    end
+
+    #.......................................................................................................
+    def typed_add(e)
+      add(e)
+    end
+
+    #.......................................................................................................
+    def import(xmlelement)
+      if @name and @name != xmlelement.name
+        raise "Trying to import an #{xmlelement.name} to a #{@name} !"
+      end
+      add_attributes(xmlelement.attributes.clone)
+      @context = xmlelement.context
+      xmlelement.each do |e|
+        if e.kind_of? REXML::Element
+          typed_add(e.deep_clone)
+        elsif e.kind_of? REXML::Text
+          add_text(e.value)
+        else
+          add(e.clone)
+        end
+      end
+      self
+    end
+
+    #.......................................................................................................
+    def self.import(xmlelement)
+      self.new(xmlelement.name).import(xmlelement)
+    end
+
+    #.......................................................................................................
+    def delete_elements(element)
+      elements.delete_all(element)
+    end
+
+  #### Element
+  end 
+
+#### REXML
+end 
+

data/lib/agent_xmpp/patches/string.rb

@@ -0,0 +1,15 @@
+##############################################################################################################
+class String
+      
+  #......................................................................................................
+  def classify
+    split('_').collect{|s| s.capitalize}.join
+  end
+
+  #......................................................................................................
+  def humanize
+    gsub(/_/, ' ')
+  end
+
+#### String
+end

data/lib/agent_xmpp/xmpp.rb

@@ -0,0 +1,18 @@
+require 'agent_xmpp/xmpp/element'
+require 'agent_xmpp/xmpp/x_data'
+require 'agent_xmpp/xmpp/error_response'
+require 'agent_xmpp/xmpp/entry'
+
+require 'agent_xmpp/xmpp/stanza'
+require 'agent_xmpp/xmpp/message'
+require 'agent_xmpp/xmpp/presence'
+require 'agent_xmpp/xmpp/iq'
+
+require 'agent_xmpp/xmpp/jid'
+require 'agent_xmpp/xmpp/sasl'
+
+require 'agent_xmpp/xmpp/iq_command'
+require 'agent_xmpp/xmpp/iq_version'
+require 'agent_xmpp/xmpp/iq_roster'
+require 'agent_xmpp/xmpp/iq_disco'
+require 'agent_xmpp/xmpp/iq_pubsub'

data/lib/agent_xmpp/xmpp/element.rb

@@ -0,0 +1,158 @@
+# Original from XMPP4R - XMPP Library for Ruby Website::http://home.gna.org/xmpp4r/
+##############################################################################################################
+module AgentXmpp
+
+  #####-------------------------------------------------------------------------------------------------------
+  module Xmpp
+
+    #####-------------------------------------------------------------------------------------------------------
+    class Element < REXML::Element
+
+      #.......................................................................................................
+      @@name_xmlns_classes = {}
+
+      #####-------------------------------------------------------------------------------------------------------
+      class << self
+        
+        #.......................................................................................................
+        def name_xmlns(name, xmlns=nil)
+          @@name_xmlns_classes[[name, xmlns]] = self
+        end
+
+        #.......................................................................................................
+        def name_xmlns_for_class(klass)
+          klass.ancestors.each do |klass1|
+            @@name_xmlns_classes.each do |name_xmlns,k|
+              if klass1 == k
+                return name_xmlns
+              end
+            end
+          end
+          raise NoNameXmlnsRegistered.new(klass)
+        end
+
+        #.......................................................................................................
+        def class_for_name_xmlns(name, xmlns)
+          if @@name_xmlns_classes.has_key? [name, xmlns]
+            @@name_xmlns_classes[[name, xmlns]]
+          elsif @@name_xmlns_classes.has_key? [name, nil]
+            @@name_xmlns_classes[[name, nil]]
+          else
+            REXML::Element
+          end
+        end
+
+        #.......................................................................................................
+        def import(element)
+          klass = class_for_name_xmlns(element.name, element.namespace)
+          if klass != self and klass.ancestors.include?(self)
+            klass.new.import(element)
+          else
+            self.new.import(element)
+          end
+        end
+        
+        #.....................................................................................................
+        def xmpp_attribute(*args)
+          sym = (args.pop[:sym] if args.last.kind_of?(Hash)).nil? ? '' : '.to_sym'
+          args.each do |a|
+            class_eval <<-DEF
+              def #{a.to_s}
+                attr = attributes['#{a.to_s}']
+                attr.nil? ? nil : attr#{sym}
+              end
+              def #{a.to_s}=(v)
+                attributes['#{a.to_s}'] = v.to_s
+              end
+            DEF
+          end
+        end
+        
+        #.......................................................................................................
+        def xmpp_child(*args)
+          args.each do |a|
+            class_eval <<-DEF
+              def #{a.to_s}
+                first_element('#{a.to_s}')
+              end
+              def #{a.to_s}=(v)
+                delete_elements(v.name)
+                add(v)
+              end
+            DEF
+          end
+        end
+
+      #### self
+      end
+      
+      #.......................................................................................................
+      def initialize(*arg)
+        if arg.empty?
+          name, xmlns = self.class::name_xmlns_for_class(self.class)
+          super(name)
+          add_namespace(xmlns)
+        else
+          super
+        end
+      end
+
+      #.......................................................................................................
+      def typed_add(element)
+        if element.kind_of? REXML::Element
+          element_ns = (element.namespace.to_s == '') ? namespace : element.namespace
+          klass = Element::class_for_name_xmlns(element.name, element_ns)
+          if klass != element.class
+            element = klass.import(element)
+          end
+        end
+        super(element)
+      end
+
+      #.......................................................................................................
+      def parent=(new_parent)
+        if parent and parent.namespace('') == namespace('') and attributes['xmlns'].nil?
+          add_namespace parent.namespace('')
+        end
+        super
+        if new_parent and new_parent.namespace('') == namespace('')
+          delete_namespace
+        end
+      end
+
+      #.......................................................................................................
+      def clone
+        cloned = self.class.new
+        cloned.add_attributes self.attributes.clone
+        cloned.context = @context
+        cloned
+      end
+
+      #.......................................................................................................
+      def xml_lang
+        attributes['xml:lang']
+      end
+
+      #.......................................................................................................
+      def xml_lang=(l)
+        attributes['xml:lang'] = l
+      end
+
+      #.......................................................................................................
+      def set_xml_lang(l)
+        self.xml_lang = l; self
+      end
+      
+      #.....................................................................................................
+      def <<(child)
+        add(child); self
+      end
+      
+    #### Element
+    end  
+      
+  #### XMPP
+  end
+
+#### AgentXmpp
+end