troystribling-agent_xmpp 0.0.0

data/lib/agent_xmpp/client/parser.rb

@@ -0,0 +1,71 @@
+##############################################################################################################
+module AgentXmpp
+  
+  #####-------------------------------------------------------------------------------------------------------
+  module Parser
+
+    #---------------------------------------------------------------------------------------------------------
+    include EventMachine::XmlPushParser
+    #---------------------------------------------------------------------------------------------------------
+
+    #---------------------------------------------------------------------------------------------------------
+    attr_reader :stream_features, :stream_mechanisms
+    #---------------------------------------------------------------------------------------------------------
+
+    #---------------------------------------------------------------------------------------------------------
+    # EventMachine::XmlPushParser callbacks
+    #.........................................................................................................
+  	def start_document
+  	end
+  
+    #.........................................................................................................
+    def start_element(name, attrs)
+      e = REXML::Element.new(name)
+      e.add_attributes(attrs)      
+      @current = @current.nil? ? e : @current.add_element(e)  
+      if @current.xpath == 'stream:stream'
+        process
+        @current = nil
+      end
+    end
+  
+    #.........................................................................................................
+    def end_element(name)
+      if @current.parent
+        @current = @current.parent
+      else
+        process
+        @current = nil
+      end
+    end
+
+    #.........................................................................................................
+    def characters(text)
+      @current.text = @current.text.to_s + text if @current
+    end
+
+    #.........................................................................................................
+    def error(*args)
+      AgentXmpp.logger.error *args
+    end
+
+    #.........................................................................................................
+    def process
+      @current.add_namespace(@streamns) if @current.namespace('').to_s.eql?('')
+      begin
+        stanza = Jabber::XMPPStanza::import(@current)
+      rescue Jabber::NoNameXmlnsRegistered
+        stanza = @current
+      end
+      if @current.xpath.eql?('stream:stream')
+        @streamns = @current.namespace('') if @current.namespace('')
+      end
+      receive(stanza) if respond_to?(:receive)
+    end
+   
+  #### Parser
+  end
+
+#### AgentXmpp
+end
+

data/lib/agent_xmpp/client.rb

@@ -0,0 +1,3 @@
+require 'agent_xmpp/client/parser'
+require 'agent_xmpp/client/connection'
+require 'agent_xmpp/client/client'

data/lib/agent_xmpp/patches/standard_library_patches/array.rb

@@ -0,0 +1,40 @@
+##############################################################################################################
+module AgentXmpp  
+  module StandardLibrary
+    module ArrayPatches
+    
+      ####----------------------------------------------------------------------------------------------------
+      module InstanceMethods
+
+        #......................................................................................................
+        def to_x_data(type = 'result')
+          data = Jabber::Dataforms::XData.new(type)
+          reported = Jabber::Dataforms::XDataReported.new
+          if first.instance_of?(Hash)
+            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)}
+              data << item
+            end
+          else
+            field = Jabber::Dataforms::XDataField.new
+            field.values = map {|v| v.to_s}
+            data << field
+          end
+          data
+        end
+        
+      #### InstanceMethods
+      end  
+        
+    #### ArrayPatches
+    end
+  ##### StandardLibrary
+  end
+#### AgentXmpp
+end
+
+##############################################################################################################
+Array.send(:include, AgentXmpp::StandardLibrary::ArrayPatches::InstanceMethods)

data/lib/agent_xmpp/patches/standard_library_patches/float.rb

@@ -0,0 +1,25 @@
+##############################################################################################################
+module AgentXmpp  
+  module StandardLibrary
+    module FloatPatches
+    
+      ####----------------------------------------------------------------------------------------------------
+      module InstanceMethods
+
+        #......................................................................................................
+        def precision(p = 3)
+          (10.0**p*self).round.to_f/10.0**p
+        end
+        
+      #### InstanceMethods
+      end  
+        
+    #### ArrayPatches
+    end
+  ##### StandardLibrary
+  end
+#### AgentXmpp
+end
+
+##############################################################################################################
+Float.send(:include, AgentXmpp::StandardLibrary::FloatPatches::InstanceMethods)

data/lib/agent_xmpp/patches/standard_library_patches/hash.rb

@@ -0,0 +1,28 @@
+##############################################################################################################
+module AgentXmpp  
+  module StandardLibrary
+    module HashPatches
+    
+      ####----------------------------------------------------------------------------------------------------
+      module InstanceMethods
+
+        #.......................................................................................................
+        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)
+          end
+        end
+
+        
+      #### InstanceMethods
+      end  
+        
+    #### HashPatches
+    end
+  ##### StandardLibrary
+  end
+#### AgentXmpp
+end
+
+##############################################################################################################
+Hash.send(:include, AgentXmpp::StandardLibrary::HashPatches::InstanceMethods)

data/lib/agent_xmpp/patches/standard_library_patches/object.rb

@@ -0,0 +1,30 @@
+##############################################################################################################
+module AgentXmpp  
+  module StandardLibrary
+    module ObjectPatches
+    
+      ####----------------------------------------------------------------------------------------------------
+      module InstanceMethods
+
+        #.......................................................................................................
+        def to_x_data(type = 'result')
+          Jabber::Dataforms::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  
+        
+    #### ObjectPatches
+    end
+  ##### StandardLibrary
+  end
+#### AgentXmpp
+end
+
+##############################################################################################################
+Object.send(:include, AgentXmpp::StandardLibrary::ObjectPatches::InstanceMethods)

data/lib/agent_xmpp/patches/standard_library_patches/string.rb

@@ -0,0 +1,25 @@
+##############################################################################################################
+module AgentXmpp  
+  module StandardLibrary
+    module StringPatches
+    
+      ####----------------------------------------------------------------------------------------------------
+      module InstanceMethods
+
+        #......................................................................................................
+        def classify
+          split('_').collect{|s| s.capitalize}.join
+        end
+        
+      #### InstanceMethods
+      end  
+        
+    #### ArrayPatches
+    end
+  ##### StandardLibrary
+  end
+#### AgentXmpp
+end
+
+##############################################################################################################
+String.send(:include, AgentXmpp::StandardLibrary::StringPatches::InstanceMethods)

data/lib/agent_xmpp/patches/standard_library_patches.rb

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

data/lib/agent_xmpp/patches/xmpp4r_patches/command.rb

@@ -0,0 +1,26 @@
+##############################################################################################################
+module AgentXmpp  
+  module XMPP4RPatches
+    module Command
+  
+      ####----------------------------------------------------------------------------------------------------
+      module InstanceMethods
+
+        #.....................................................................................................
+        def <<(child)
+          add(child)
+        end
+
+      #### InstanceMethods
+      end  
+        
+    ##### Command
+    end
+  ##### XMPP4RPatches
+  end
+#### AgentXmpp
+end
+
+##############################################################################################################
+Jabber::Command::IqCommand.send(:include, Jabber::XParent)
+Jabber::Command::IqCommand.send(:include, AgentXmpp::XMPP4RPatches::Command::InstanceMethods)

data/lib/agent_xmpp/patches/xmpp4r_patches/iq.rb

@@ -0,0 +1,26 @@
+##############################################################################################################
+module AgentXmpp  
+  module XMPP4RPatches
+    module Iq
+  
+      ####----------------------------------------------------------------------------------------------------
+      module InstanceMethods
+
+        #.....................................................................................................
+        def command=(newcommand)
+          delete_elements(newcommand.name)
+          add(newcommand)
+        end
+    
+      #### InstanceMethods
+      end 
+       
+    ##### Iq
+    end
+  ##### XMPP4RPatches
+  end
+#### AgentXmpp
+end
+
+##############################################################################################################
+Jabber::Iq.send(:include, AgentXmpp::XMPP4RPatches::Iq::InstanceMethods)

data/lib/agent_xmpp/patches/xmpp4r_patches/x_data.rb

@@ -0,0 +1,116 @@
+##############################################################################################################
+module AgentXmpp  
+  module XMPP4RPatches
+    module XData
+  
+      ####----------------------------------------------------------------------------------------------------
+      module InstanceMethods
+
+        #.....................................................................................................
+        def <<(child)
+          add(child)
+          self
+        end
+
+        #.....................................................................................................
+        def add_field_with_value(var, value)
+          field = Jabber::Dataforms::XDataField.new(var)
+          field.value = value
+          self << field
+        end
+    
+      #### InstanceMethods
+      end   
+     
+    ##### XData
+    end
+  ##### XMPP4RPatches
+  end
+#### AgentXmpp
+end
+
+##############################################################################################################
+Jabber::Dataforms::XData.send(:include, AgentXmpp::XMPP4RPatches::XData::InstanceMethods)
+
+##############################################################################################################
+module AgentXmpp  
+  module XMPP4RPatches
+      module XDataReported
+    
+        ####----------------------------------------------------------------------------------------------------
+        module InstanceMethods
+
+          #.....................................................................................................
+          def fields(including_hidden=false)
+            fields = []
+            each_element do |xe|
+              if xe.kind_of?(Jabber::Dataforms::XDataField) and (including_hidden or (xe.type != :hidden and xe.type != :fixed))
+                fields << xe
+              end
+            end
+            fields
+          end
+
+          #.....................................................................................................
+          def <<(child)
+            add(child)
+            self
+          end
+    
+          #.....................................................................................................
+          def add_field(var)
+            self << Jabber::Dataforms::XDataField.new(var)
+          end
+    
+      #### InstanceMethods
+      end  
+      
+    ##### XDataReported
+    end
+  ##### XMPP4RPatches
+  end
+#### AgentXmpp
+end
+
+##############################################################################################################
+Jabber::Dataforms::XDataReported.send(:include, AgentXmpp::XMPP4RPatches::XDataReported::InstanceMethods)
+
+##############################################################################################################
+##############################################################################################################
+module Jabber
+  module Dataforms
+    class XDataItem < XMPPElement
+
+      #.....................................................................................................
+      name_xmlns 'item', 'jabber:x:data'
+    
+      #.....................................................................................................
+      def fields(including_hidden=false)
+        fields = []
+        each_element do |xe|
+          if xe.kind_of?(Jabber::Dataforms::XDataField) and (including_hidden or (xe.type != :hidden and xe.type != :fixed))
+            fields << xe
+          end
+        end
+        fields
+      end
+    
+      #.....................................................................................................
+      def <<(child)
+        add(child)
+        self
+      end
+    
+      #.....................................................................................................
+      def add_field_with_value(var, value)
+        field = Jabber::Dataforms::XDataField.new(var)
+        field.value = value
+        self << field
+      end
+          
+    #### XDataItem
+    end 
+  #### Dataforms
+  end
+#### Jabber
+end

data/lib/agent_xmpp/patches/xmpp4r_patches.rb

@@ -0,0 +1,3 @@
+require 'agent_xmpp/patches/xmpp4r_patches/command'
+require 'agent_xmpp/patches/xmpp4r_patches/x_data'
+require 'agent_xmpp/patches/xmpp4r_patches/iq'

data/lib/agent_xmpp/patches.rb

@@ -0,0 +1,3 @@
+require 'agent_xmpp/patches/xmpp4r_patches'
+require 'agent_xmpp/patches/standard_library_patches'
+

data/lib/agent_xmpp/utils/logger.rb

@@ -0,0 +1,24 @@
+##############################################################################################################
+module AgentXmpp
+  
+  ####......................................................................................................
+  class << self
+
+    #.........................................................................................................
+    attr_accessor :log_file
+    #.........................................................................................................
+
+    #.........................................................................................................
+    def logger
+      @logger ||= Logger.new(STDOUT)
+    end
+
+    #.........................................................................................................
+    def logger=(logger)    
+      @logger = logger
+    end
+  
+  end
+
+#### AgentXmpp
+end