troystribling-agent_xmpp 0.0.0 → 0.0.1

data/lib/agent_xmpp/config.rb

@@ -0,0 +1,45 @@
+module AgentXmpp
+
+  #.........................................................................................................
+  VERSION = "0.0.0"
+  AGENT_XMPP_NAME = 'AgentXMPP'
+  OS_VERSION = IO.popen('uname -sr').readlines.to_s.strip
+  SUBSCRIBE_RETRY_PERIOD = 60
+  IDENTITY = {:category => 'client', :name => AGENT_XMPP_NAME, :type => 'bot'}
+  FEATURES = ['http://jabber.org/protocol/disco#info', 
+              'http://jabber.org/protocol/disco#items',
+              'jabber:iq:version',
+              'jabber:x:data',
+              'http://jabber.org/protocol/commands',
+              'http://jabber.org/protocol/muc']
+  DEFAULT_PUBSUB_CONFIG = {
+    :title                    => 'event',
+    :access_model             => 'presence',
+    :publish_model            => 'publishers',
+    :max_items                => 20,
+    :deliver_notifications    => 1,
+    :deliver_payloads         => 1,
+    :persist_items            => 1,
+    :subscribe                => 1,
+    :notify_config            => 0,
+    :notify_delete            => 0,
+    :notify_retract           => 0,
+  }
+
+  #.........................................................................................................
+  @config_file = "agent_xmpp.yml"
+  @app_path = File.dirname($0)
+  @log_file = STDOUT
+  
+  #.........................................................................................................
+  class << self
+    attr_accessor :config_file, :app_path, :log_file
+    def logger; @logger ||= Logger.new(STDOUT); end
+    def logger=(logger); @logger = logger; end
+  end
+  
+  #.........................................................................................................
+  class AgentXmppError < Exception; end
+  
+end
+

data/lib/agent_xmpp/main.rb

@@ -0,0 +1,126 @@
+##############################################################################################################
+OptionParser.new do |opts|
+  opts.banner = 'Usage: agent_xmpp.rb [options]'
+  opts.separator ''
+  opts.on('-c', '--config config.yml', 'YAML agent configuration file relative to application path') {|f| AgentXmpp.config_file = f}
+  opts.on('-l', '--logfile file.log', 'name of logfile') {|f| AgentXmpp.log_file = f}
+  opts.on_tail('-h', '--help', 'Show this message') {
+    puts opts
+    exit
+  }
+  opts.parse!(ARGV)
+end
+
+##############################################################################################################
+module AgentXmpp
+  
+  #...........................................................................................................
+  @config = {}
+
+  #####-------------------------------------------------------------------------------------------------------
+  class << self
+
+    #.........................................................................................................
+    attr_accessor :config
+    
+    #.........................................................................................................
+    def roster
+      @roster ||= RosterModel.new(jid, config['roster'])
+    end
+
+    #.........................................................................................................
+    def pubsub_root
+      @pubsub_root ||= "/home/#{AgentXmpp.jid.domain}"  
+    end     
+    
+    #.........................................................................................................
+    def user_pubsub_root
+      @user_pubsub_root ||= "#{@pubsub_root}/#{AgentXmpp.jid.node}" 
+    end
+
+    #.........................................................................................................
+    def services
+      @services ||= ServicesModel.new
+    end
+    
+    #.........................................................................................................
+    def published
+      @published ||= PublishModel.new(config['publish'])
+    end
+        
+    #.........................................................................................................
+    def jid
+      @jid ||= Xmpp::Jid.new("#{config['jid']}/#{resource}")
+    end
+      
+    #.........................................................................................................
+    def jid=(jid)
+      @jid = jid
+    end
+          
+    #.........................................................................................................
+    def resource
+      config['resource'] || Socket.gethostname
+    end
+         
+    #.........................................................................................................
+    def port
+      config['port'] || 5222
+    end
+
+    #.........................................................................................................
+    def password
+      config['password']
+    end
+            
+    #.........................................................................................................
+    def priority
+      @priority ||= if config['priority']
+                    if config['priority'] < -127
+                      -127
+                    elsif config['priority'] > 128
+                      128
+                    else
+                      config['priority']
+                    end
+                  else; 1; end
+    end
+            
+  #### self
+  end
+  
+  #####-------------------------------------------------------------------------------------------------------
+  module Delegator 
+
+    #####-------------------------------------------------------------------------------------------------------
+    class << self
+
+      #.........................................................................................................
+      def delegate(del, *methods)
+        methods.each do |method_name|
+          class_eval <<-RUBY
+            def #{method_name.to_s}(*args, &blk)
+              ::#{del}.send(#{method_name.inspect}, *args, &blk)
+            end
+          RUBY
+        end
+      end
+
+    #### self
+    end
+
+    delegate AgentXmpp::BaseController, :execute, :chat, :event
+    delegate AgentXmpp::Boot, :before_start, :after_connected, :restarting_client, :discovered_pubsub_node, 
+                              :discovered_command_nodes, :received_presence
+
+  #### Delegator 
+  end
+    
+#### AgentXmpp 
+end
+
+##############################################################################################################
+include AgentXmpp::Delegator
+
+##############################################################################################################
+at_exit {AgentXmpp::Boot.boot}

data/lib/agent_xmpp/patches.rb

@@ -1,3 +1,7 @@
-require 'agent_xmpp/patches/xmpp4r_patches'
-require 'agent_xmpp/patches/standard_library_patches'
+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/{standard_library_patches/array.rb → array.rb}

@@ -8,24 +8,31 @@ module AgentXmpp
 
         #......................................................................................................
         def to_x_data(type = 'result')
-          data = Jabber::Dataforms::XData.new(type)
-          reported = Jabber::Dataforms::XDataReported.new
-          if first.instance_of?(Hash)
+          data = Xmpp::XData.new(type)
+          if first.kind_of?(Hash)
+            field_type = lambda{|v| v.kind_of?(Array) ? 'list-multi' : nil}
+            reported = Xmpp::XDataReported.new
             first.each_key {|var| reported.add_field(var.to_s)}
             data << reported
             each do |fields|
-              item = Jabber::Dataforms::XDataItem.new
-              fields.each {|var, value| item.add_field_with_value(var.to_s, value.to_s)}
+              item = 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 = Jabber::Dataforms::XDataField.new
-            field.values = map {|v| v.to_s}
+            field = 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
+        
       #### InstanceMethods
       end  
         

data/lib/agent_xmpp/patches/{standard_library_patches/hash.rb → hash.rb}

@@ -8,11 +8,11 @@ module AgentXmpp
 
         #.......................................................................................................
         def to_x_data(type = 'result')
-          inject(Jabber::Dataforms::XData.new(type)) do |data, field| 
-            data.add_field_with_value(field.first.to_s, field.last.to_s)
+          field_type = lambda{|v| v.kind_of?(Array) ? 'list-multi' : nil}
+          inject(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
-
         
       #### InstanceMethods
       end  

data/lib/agent_xmpp/patches/{standard_library_patches/object.rb → object.rb}

@@ -7,15 +7,15 @@ module AgentXmpp
       module InstanceMethods
 
         #.......................................................................................................
-        def to_x_data(type = 'result')
-          Jabber::Dataforms::XData.new(type).add_field_with_value(nil, to_s)
+        def to_x_data(type='result')
+          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
-  
+
       #### InstanceMethods
       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/{standard_library_patches/string.rb → string.rb}

@@ -10,6 +10,11 @@ module AgentXmpp
         def classify
           split('_').collect{|s| s.capitalize}.join
         end
+
+        #......................................................................................................
+        def humanize
+          gsub(/_/, ' ')
+        end
         
       #### InstanceMethods
       end  

data/lib/agent_xmpp/xmpp.rb

@@ -0,0 +1,17 @@
+require 'agent_xmpp/xmpp/element'
+require 'agent_xmpp/xmpp/stanza'
+require 'agent_xmpp/xmpp/jid'
+require 'agent_xmpp/xmpp/sasl'
+require 'agent_xmpp/xmpp/x_data'
+
+require 'agent_xmpp/xmpp/message'
+require 'agent_xmpp/xmpp/presence'
+require 'agent_xmpp/xmpp/iq'
+
+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'
+
+require 'agent_xmpp/xmpp/error_response'

data/lib/agent_xmpp/xmpp/element.rb

@@ -0,0 +1,159 @@
+# 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}
+                attributes['#{a.to_s}']#{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