troystribling-agent_xmpp 0.0.0 → 0.0.1

data/lib/agent_xmpp/client/message_pipe.rb

@@ -0,0 +1,176 @@
+##############################################################################################################
+module AgentXmpp
+  
+  #####-------------------------------------------------------------------------------------------------------
+  class MessagePipe
+
+    #---------------------------------------------------------------------------------------------------------
+    attr_reader   :connection_status, :delegates, :id_callbacks, :connection, :stream_features, 
+                  :stream_mechanisms
+    #---------------------------------------------------------------------------------------------------------
+    alias_method :send_to_method, :send
+    #---------------------------------------------------------------------------------------------------------
+
+    #.........................................................................................................
+    def initialize(connection)
+      @connection = connection
+      @connection_status = :offline;
+      @delegates = [MessageDelegate]
+      @id_callbacks = {}
+    end
+    
+    #.........................................................................................................
+    def add_delegate(delegate)
+      @delegates << delegate unless @delegates.include?(delegate)
+    end
+
+    #.........................................................................................................
+    def remove_delegate(delegate)
+      @delegates.delete(delegate)
+    end
+    
+    #.........................................................................................................
+    def delegates_respond_to?(method)
+     delegates.inject(0){|r,d| d.respond_to?(method) ? r + 1 : r} > 0
+    end
+    
+    #.........................................................................................................
+    def broadcast_to_delegates(method, *args)
+      delegates.inject([]){|r,d| d.respond_to?(method) ? r.push(d.send(method, *args)) : r}.smash
+    end
+
+    #.........................................................................................................
+    def responder_list
+      @id_callbacks
+    end
+    
+    #.........................................................................................................
+    def send(data, &blk)
+      raise AgentXmppError, 'not connected'  unless connected?
+      if block_given? and data.kind_of?(Xmpp::Stanza)
+        if data.id.nil?
+          data.id = Xmpp::IdGenerator.generate_id
+        end
+        @id_callbacks[data.id] = blk
+      end
+      AgentXmpp.logger.info "SEND: #{data.to_s}"
+      @connection.send_data(data.to_s)
+    end
+
+    #.........................................................................................................
+    def send_resp(resp)
+      [resp].flatten.map {|r| r.kind_of?(AgentXmpp::Response) ? send(r.message, &r.responds_with) : r}
+    end
+
+    #.........................................................................................................
+    def connected?
+      connection and !connection.error?
+    end
+    
+    #---------------------------------------------------------------------------------------------------------
+    # connection callbacks
+    #.........................................................................................................
+    def receive(stanza)
+      AgentXmpp.logger.info "RECV: #{stanza.to_s}"
+      result = if stanza.kind_of?(Xmpp::Stanza) and stanza.id and blk = id_callbacks[stanza.id]
+                 id_callbacks.delete(stanza.id)
+                 blk.call(stanza)
+               else
+                 process_stanza(stanza)
+               end
+      send_resp(result)          
+    end
+
+    #.........................................................................................................
+    def connection_completed
+      Boot.call_if_implemented(:call_after_connected, self)     
+      broadcast_to_delegates(:did_connect, self)
+      init_connection.collect{|m| send(m)}
+    end
+
+    #.........................................................................................................
+    def unbind
+      @connection_status = :off_line
+      broadcast_to_delegates(:did_disconnect, self)
+    end
+    
+  private
+
+    #.........................................................................................................
+    def process_stanza(stanza)
+      case stanza.name
+      when 'features'
+        set_stream_features_and_mechanisms(stanza)
+        if connection_status.eql?(:offline)
+          broadcast_to_delegates(:did_receive_preauthenticate_features, self)
+        elsif connection_status.eql?(:authenticated)
+          broadcast_to_delegates(:did_receive_postauthenticate_features, self)
+        end
+      when 'stream'
+      when 'success'
+        if connection_status.eql?(:offline)
+          @connection.reset_parser
+          @connection_status = :authenticated
+          broadcast_to_delegates(:did_authenticate, self)
+          init_connection(false)
+        end
+      when 'failure'
+        if connection_status.eql?(:offline)
+          @connection.reset_parser
+          broadcast_to_delegates(:did_not_authenticate, self)
+        end
+      else
+        demux_stanza(stanza)
+      end
+    end
+  
+    #.........................................................................................................
+    def demux_stanza(stanza)
+      meth = 'did_receive_' + if stanza.class.eql?(AgentXmpp::Xmpp::Iq)
+                                iqclass = if stanza.query
+                                            stanza.query.class
+                                          elsif stanza.command
+                                            stanza.command.class
+                                          else
+                                            nil
+                                          end
+                                if iqclass
+                                  /.*::Iq(.*)/.match(iqclass.to_s).to_a.last 
+                                else
+                                  'fail'
+                                end
+                              else
+                                /.*::(.*)/.match(stanza.class.to_s).to_a.last
+                              end.downcase
+      meth += '_' + stanza.type.to_s if stanza.type
+      if delegates_respond_to?(meth.to_sym) 
+        broadcast_to_delegates(meth.to_sym, self, stanza)
+      else
+        broadcast_to_delegates(:did_receive_unsupported_message, self, stanza)
+      end
+    end
+  
+    #.........................................................................................................
+    def set_stream_features_and_mechanisms(stanza)
+      @stream_features, @stream_mechanisms = {}, []
+      stanza.elements.each do |e|
+        if e.name == 'mechanisms' and e.namespace == 'urn:ietf:params:xml:ns:xmpp-sasl'
+          e.each_element('mechanism') {|mech| @stream_mechanisms.push(mech.text)}
+        else
+          @stream_features[e.name] = e.namespace
+        end
+      end
+    end
+  
+    #.........................................................................................................
+    def init_connection(starting = true)
+      msg = []
+      msg.push(Send("<?xml version='1.0' ?>")) if starting
+      msg.push(Send("<stream:stream xmlns='jabber:client' xmlns:stream='http://etherx.jabber.org/streams' version='1.0' to='#{AgentXmpp.jid.domain}'>"))
+    end
+         
+  #### MessagePipe
+  end
+
+#### AgentXmpp
+end

data/lib/agent_xmpp/client/publish_model.rb

@@ -0,0 +1,65 @@
+##############################################################################################################
+module AgentXmpp
+  
+  #####-------------------------------------------------------------------------------------------------------
+  class PublishModel
+
+    #.........................................................................................................
+    def initialize(items)
+      @items = items || []
+    end
+
+    #.........................................................................................................
+    def find_all
+      @items.collect{|i| PublishNode.new(i)}
+    end
+
+    #.........................................................................................................
+    def find_by_node(node)
+      item = @items.select{|i| node.split('/').last.eql?(i['node'])}.first
+      item.nil? ? nil : PublishNode.new(item)
+    end
+
+    #.........................................................................................................
+    def delete_by_node(node)
+      @items.delete_if{|i| node.split('/').last.eql?(i['node'])}
+    end
+
+  #### PublishModel
+  end
+
+  #####-------------------------------------------------------------------------------------------------------
+  class PublishNode
+
+    #.........................................................................................................
+    attr_reader :node, :status, :title, :access_model, :publish_model, :send_last_published_item, :max_items,
+                :max_payload_size, :deliver_notifications, :deliver_payloads, :persist_items, 
+                :subscribe, :presence_based_delivery, :notify_config, :notify_delete, :notify_retract,
+                :notify_sub
+      
+    #.........................................................................................................
+    def initialize(pub)
+      @node = pub['node']
+      @status = pub['status']
+      @title = pub['title']
+      @access_model = pub['access_model']
+      @publish_model = pub['publish_model']
+      @send_last_published_item = pub['send_last_published_item']
+      @max_items = pub['max_items']
+      @max_payload_size = pub['max_payload_size']
+      @deliver_notifications = pub['deliver_notifications']
+      @deliver_payloads = pub['deliver_payloads']
+      @persist_items = pub['persist_items']
+      @subscribe = pub['subscribe']
+      @presence_based_delivery = pub['presence_based_delivery']
+      @notify_config = pub['notify_config']
+      @notify_delete = pub['notify_delete']
+      @notify_retract = pub['notify_retract']
+      @notify_sub = pub['notify_sub']
+    end
+
+  #### PublishNode
+  end
+
+#### AgentXmpp
+end

data/lib/agent_xmpp/{app/format.rb → client/response.rb}

@@ -1,40 +1,35 @@
+##############################################################################################################
+def Send(msg, &blk)
+  AgentXmpp::Response.new(msg, &blk)
+end
+
 ##############################################################################################################
 module AgentXmpp
-  
+    
   #####-------------------------------------------------------------------------------------------------------
-  class Format
-
-    #---------------------------------------------------------------------------------------------------------
-    attr_accessor :xmlns
-    #---------------------------------------------------------------------------------------------------------
+  class Response
 
     #.........................................................................................................
-    def initialize(xmlns)
-      @xmlns = xmlns
-    end
+    attr_reader :text, :message, :responds_with
 
     #.........................................................................................................
-    def x_data(&blk)
-      call_block_for_xmlns('jabber:x:data', &blk)
+    def initialize(msg, &blk)
+      @message = msg
+      @text = msg.to_s
+      @responds_with = blk
     end
 
     #.........................................................................................................
-    def json(&blk)
-      call_block_for_xmlns('jabber:x:json', &blk)
+    def to_s
+      text
     end
-
-  private
-  
+    
     #.........................................................................................................
-    def call_block_for_xmlns(check_xmlns, &blk)
-      if xmlns.eql?(check_xmlns)
-        if blk
-          blk.call
-        end
-      end
+    def method_missing(meth, *args, &blk)
+      text.send(meth, *args, &blk)
     end
-  
-  #### Format
+
+  #### Message
   end
 
 #### AgentXmpp

data/lib/agent_xmpp/client/roster_model.rb

@@ -0,0 +1,139 @@
+##############################################################################################################
+module AgentXmpp
+  
+  #####-------------------------------------------------------------------------------------------------------
+  class RosterModel
+
+    #.........................................................................................................
+    def initialize(jid, roster)
+      @items = {}
+      roster.each{|r| @items[r['jid']] = {:status => :inactive, :resources => {},
+        :groups => r['groups'].nil? ? [] : r['groups'].uniq, :jid => r['jid']}} if roster
+      @items[jid.bare.to_s] = {:status => :both, :resources => {}, :groups => [],}
+      @items[jid.domain] = {:status => :host, :resources => {jid.domain => {}}, :groups => []}
+    end
+
+    #.........................................................................................................
+    def has_jid?(jid)
+      @items.has_key?(jid.bare.to_s) 
+    end
+
+    #.........................................................................................................
+    def find_all
+      @items.values.collect{|r| RosterItemModel.new(r)}  
+    end
+
+    #.........................................................................................................
+    def find_by_jid(jid)
+      @items[jid.bare.to_s].nil? ? nil : RosterItemModel.new(@items[jid.bare.to_s])   
+    end
+
+    #.........................................................................................................
+    def find_all_by_status(status)
+      @items.select{|j,r| r[:status].eql?(status)}.map{|j,r| RosterItemModel.new(r)}    
+    end
+
+    #.........................................................................................................
+    def find_all_by_group(group)
+      @items.select{|j,r| r[:group].include?(group)}.map{|j,r| RosterItemModel.new(r)}    
+    end
+
+    #.........................................................................................................
+    def destroy_by_jid(jid)
+      @items.delete(jid.bare.to_s)
+    end 
+    
+    #.........................................................................................................
+    def has_resource?(jid)
+      not @items[jid.bare.to_s][:resources][jid.to_s].nil?
+    end 
+
+    #.........................................................................................................
+    def resources(jid)
+      @items[jid.bare.to_s][:resources]
+    end 
+    
+    #.........................................................................................................
+    def available_resources(jid)
+      if @items[jid.bare.to_s] and @items[jid.bare.to_s][:resources]
+        @items[jid.bare.to_s][:resources].inject([]) {|r,(j,h)| h[:presence].status.nil? ? r << j : r}
+      else; []; end
+    end 
+    
+    #.........................................................................................................
+    def resource(jid)
+      if @items[jid.bare.to_s] and @items[jid.bare.to_s][:resources][jid.to_s]
+        @items[jid.bare.to_s][:resources][jid.to_s][:presence]
+      else; nil; end
+    end 
+
+    #.........................................................................................................
+    def version(jid)
+      if @items[jid.bare.to_s] and @items[jid.bare.to_s][:resources][jid.to_s]
+        @items[jid.bare.to_s][:resources][jid.to_s][:version]
+      else; nil; end
+    end 
+                    
+    #.........................................................................................................
+    def has_version?(jid)
+      if @items[jid.bare.to_s] and @items[jid.bare.to_s][:resources][jid.to_s]
+        not @items[jid.bare.to_s][:resources][jid.to_s][:version].nil?
+      else 
+        false
+      end
+    end 
+                    
+    #.........................................................................................................
+    def update_status(jid, status)
+      @items[jid.bare.to_s][:status] = status
+    end
+
+    #.........................................................................................................
+    def update_roster_item(roster_item)
+      @items[roster_item.jid.to_s][:roster_item] = roster_item 
+    end
+
+    #.........................................................................................................
+    def update_resource(presence)
+      from_jid = presence.from.to_s     
+      from_bare_jid = presence.from.bare.to_s     
+      @items[from_bare_jid][:resources][from_jid] ||= {}
+      @items[from_bare_jid][:resources][from_jid][:presence] = presence
+    end
+ 
+    #.........................................................................................................
+    def update_resource_version(version)
+      from_jid = version.from.to_s     
+      from_bare_jid = version.from.bare.to_s     
+      if @items[from_bare_jid][:resources][from_jid]   
+        @items[from_bare_jid][:resources][from_jid][:version] = version.query
+      end        
+    end
+        
+    #.........................................................................................................
+    def method_missing(meth, *args, &blk)
+      @items.send(meth, *args, &blk)
+    end
+
+  #### RosterModel
+  end
+
+  #####-------------------------------------------------------------------------------------------------------
+  class RosterItemModel
+
+    #.........................................................................................................
+    attr_reader :status, :groups, :jid, :priority
+    
+    #.........................................................................................................
+    def initialize(item)
+      @status = item[:status]
+      @groups = item[:groups]
+      @jid = item[:jid]
+      @priority = item[:priority]
+    end
+
+  #### RosterItemModel
+  end
+
+#### AgentXmpp
+end

data/lib/agent_xmpp/client/services_model.rb

@@ -0,0 +1,97 @@
+##############################################################################################################
+module AgentXmpp
+  
+  #####-------------------------------------------------------------------------------------------------------
+  class ServicesModel
+
+    #.........................................................................................................
+    def initialize
+      @service = Hash.new{|hash, key| hash[key] = {}}
+    end
+
+    #.........................................................................................................
+    def find_all_by_category(cat)
+      @service.select{|(k,v)| not v[:discoinfo].identities.select{|i| i.category.eql?(cat)}.empty?} 
+    end
+
+    #.........................................................................................................
+    def find_all_by_category_and_type(cat)
+      @service.select{|(k,v)| not v[:discoinfo].identities.select{|i| i.category.eql?(cat) and i.type.eql?(type)}.empty?} 
+    end
+
+    #.........................................................................................................
+    def create(jid)
+      @service[jid.to_s] = {} 
+    end
+
+    #.........................................................................................................
+    def has_jid?(jid)
+      @service.has_key?(jid.to_s) 
+    end
+
+    #.........................................................................................................
+    def has_feature?(jid, feature)
+       not features(jid.to_s).select{|f| f.var.eql?(feature)}.empty?
+    end 
+
+    #.........................................................................................................
+    def has_item?(jid, ijid)
+      not items(jid.to_s).select{|i| i.jid.eql?(i.jid)}.empty?
+    end 
+
+    #.........................................................................................................
+    def identities(jid, node=nil)
+      if has_jid?(jid.to_s) 
+        node.nil ? @service[from_jid][:discoinfo].identities : @service[from_jid][:node][node][:discoinfo].identities
+      else; []; end
+    end 
+
+    #.........................................................................................................
+    def features(jid, node=nil)
+      if has_jid?(jid.to_s) 
+        node.nil ? @service[from_jid][:discoinfo].features : @service[from_jid][:node][node][:discoinfo].features 
+      else; []; end
+    end 
+        
+    #.........................................................................................................
+    def items(jid, node=nil)
+      if has_jid?(jid.to_s) 
+        node.nil ? @service[from_jid][:discoitems].items : @service[from_jid][:node][node][:discoitems].items 
+      else; []; end
+    end 
+        
+    #.........................................................................................................
+    def update_with_discoinfo(disco)
+      save_item_disco(:discoinfo, disco)
+    end
+ 
+    #.........................................................................................................
+    def update_with_discoitems(disco)      
+      save_item_disco(:discoitems, disco)
+    end
+
+    #.........................................................................................................
+    def method_missing(meth, *args, &blk)
+      @service.send(meth, *args, &blk)
+    end
+
+  private
+  
+    #.........................................................................................................
+    def save_item_disco(item, disco)
+      from_jid = disco.from.to_s  
+      node = disco.query.node
+      unless node 
+        @service[from_jid][:node] = {} if @service[from_jid][:node].nil?
+        @service[from_jid][:node][node] = {} if @service[from_jid][:node][node].nil?
+        @service[from_jid][:node][node][item] = disco.query 
+      else
+        @service[from_jid][item] = disco.query 
+      end
+    end
+  
+  #### ServicesModel
+  end
+
+#### AgentXmpp
+end