agent_xmpp 0.1.0

data/Rakefile

@@ -0,0 +1,75 @@
+############################################################################################################
+$:.unshift('lib')
+require 'rubygems'
+require 'rake'
+require 'agent_xmpp/config'
+
+#####-------------------------------------------------------------------------------------------------------
+task :default => :test
+
+#####-------------------------------------------------------------------------------------------------------
+begin
+require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "agent_xmpp"
+    gem.summary = %Q{Agent XMPP is a ruby XMPP bot framework inspired by MVC web frameworks.}
+    gem.email = "troy.stribling@gmail.com"
+    gem.homepage = "http://github.com/troystribling/agent_xmpp"
+    gem.authors = ["Troy Stribling"]
+    gem.files.include %w(lib/jeweler/templates/.gitignore VERSION)
+    gem.add_dependency('rake',                              '>= 0.8.3')
+    gem.add_dependency('eventmachine',                      '>= 0.12.6')
+    gem.add_dependency('sequel',                            '>= 3.9.0')
+    gem.add_dependency('evma_xmlpushparser',                '>= 0.0.1')
+  end
+rescue LoadError
+  abort "jeweler is not available. In order to run test, you must: sudo gem install technicalpickles-jeweler --source=http://gems.github.com"
+end
+
+#####-------------------------------------------------------------------------------------------------------
+task :uninstall do
+  %x[gem uninstall agent_xmpp]
+end
+
+#####-------------------------------------------------------------------------------------------------------
+task :default => [:test]
+
+#####-------------------------------------------------------------------------------------------------------
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << ['test/cases', 'test/helpers', 'test/messages']
+  test.pattern = 'test/cases/**/test_*.rb'
+  test.verbose = true
+end
+
+#####-------------------------------------------------------------------------------------------------------
+Rake::TestTask.new(:test_case) do |test|
+  file = ENV["FILE"] || ''
+  test.libs << ['test/cases', 'test/helpers', 'test/messages']
+  test.test_files = ["test/cases/#{file}"]
+  test.verbose = true
+end
+
+#####-------------------------------------------------------------------------------------------------------
+begin
+  require 'rcov/rcovtask'
+  Rcov::RcovTask.new do |test|
+    test.libs << 'test'
+    test.pattern = 'test/**/test_*.rb'
+    test.verbose = true
+  end
+rescue LoadError
+  task :rcov do
+    abort "RCov is not available. In order to run rcov, you must: sudo gem install spicycode-rcov --source=http://gems.github.com"
+  end
+end
+
+#####-------------------------------------------------------------------------------------------------------
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "agent_xmpp #{AgentXmpp::VERSION}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end
+

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/agent_xmpp.gemspec

@@ -0,0 +1,144 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run the gemspec command
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{agent_xmpp}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Troy Stribling"]
+  s.date = %q{2010-08-21}
+  s.email = %q{troy.stribling@gmail.com}
+  s.extra_rdoc_files = [
+    "LICENSE",
+     "README.rdoc"
+  ]
+  s.files = [
+    ".document",
+     ".gitignore",
+     "LICENSE",
+     "README.rdoc",
+     "Rakefile",
+     "VERSION",
+     "agent_xmpp.gemspec",
+     "lib/agent_xmpp.rb",
+     "lib/agent_xmpp/admin.rb",
+     "lib/agent_xmpp/client.rb",
+     "lib/agent_xmpp/client/boot.rb",
+     "lib/agent_xmpp/client/client.rb",
+     "lib/agent_xmpp/client/connection.rb",
+     "lib/agent_xmpp/client/controller.rb",
+     "lib/agent_xmpp/client/message_delegate.rb",
+     "lib/agent_xmpp/client/message_pipe.rb",
+     "lib/agent_xmpp/client/response.rb",
+     "lib/agent_xmpp/config.rb",
+     "lib/agent_xmpp/main.rb",
+     "lib/agent_xmpp/models.rb",
+     "lib/agent_xmpp/models/contact.rb",
+     "lib/agent_xmpp/models/message.rb",
+     "lib/agent_xmpp/models/publication.rb",
+     "lib/agent_xmpp/models/roster.rb",
+     "lib/agent_xmpp/models/service.rb",
+     "lib/agent_xmpp/models/subscription.rb",
+     "lib/agent_xmpp/models/table_definitions.rb",
+     "lib/agent_xmpp/patches.rb",
+     "lib/agent_xmpp/patches/array.rb",
+     "lib/agent_xmpp/patches/float.rb",
+     "lib/agent_xmpp/patches/hash.rb",
+     "lib/agent_xmpp/patches/object.rb",
+     "lib/agent_xmpp/patches/rexml.rb",
+     "lib/agent_xmpp/patches/string.rb",
+     "lib/agent_xmpp/xmpp.rb",
+     "lib/agent_xmpp/xmpp/element.rb",
+     "lib/agent_xmpp/xmpp/entry.rb",
+     "lib/agent_xmpp/xmpp/error_response.rb",
+     "lib/agent_xmpp/xmpp/iq.rb",
+     "lib/agent_xmpp/xmpp/iq_command.rb",
+     "lib/agent_xmpp/xmpp/iq_disco.rb",
+     "lib/agent_xmpp/xmpp/iq_pubsub.rb",
+     "lib/agent_xmpp/xmpp/iq_roster.rb",
+     "lib/agent_xmpp/xmpp/iq_version.rb",
+     "lib/agent_xmpp/xmpp/jid.rb",
+     "lib/agent_xmpp/xmpp/message.rb",
+     "lib/agent_xmpp/xmpp/presence.rb",
+     "lib/agent_xmpp/xmpp/sasl.rb",
+     "lib/agent_xmpp/xmpp/stanza.rb",
+     "lib/agent_xmpp/xmpp/x_data.rb",
+     "test/app/app.rb",
+     "test/cases/test_application_message_processing.rb",
+     "test/cases/test_errors.rb",
+     "test/cases/test_presence_management.rb",
+     "test/cases/test_roster_management.rb",
+     "test/cases/test_service_discovery.rb",
+     "test/cases/test_session_management.rb",
+     "test/cases/test_version_discovery.rb",
+     "test/helpers/matchers.rb",
+     "test/helpers/mocks.rb",
+     "test/helpers/test_case_extensions.rb",
+     "test/helpers/test_client.rb",
+     "test/helpers/test_delegate.rb",
+     "test/helpers/test_helper.rb",
+     "test/messages/application_messages.rb",
+     "test/messages/error_messages.rb",
+     "test/messages/presence_messages.rb",
+     "test/messages/roster_messages.rb",
+     "test/messages/service_discovery_messages.rb",
+     "test/messages/session_messages.rb",
+     "test/messages/version_discovery_messages.rb",
+     "test/peer/peer.rb"
+  ]
+  s.homepage = %q{http://github.com/troystribling/agent_xmpp}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.5}
+  s.summary = %q{Agent XMPP is a ruby XMPP bot framework inspired by MVC web frameworks.}
+  s.test_files = [
+    "test/app/app.rb",
+     "test/cases/test_application_message_processing.rb",
+     "test/cases/test_errors.rb",
+     "test/cases/test_presence_management.rb",
+     "test/cases/test_roster_management.rb",
+     "test/cases/test_service_discovery.rb",
+     "test/cases/test_session_management.rb",
+     "test/cases/test_version_discovery.rb",
+     "test/helpers/matchers.rb",
+     "test/helpers/mocks.rb",
+     "test/helpers/test_case_extensions.rb",
+     "test/helpers/test_client.rb",
+     "test/helpers/test_delegate.rb",
+     "test/helpers/test_helper.rb",
+     "test/messages/application_messages.rb",
+     "test/messages/error_messages.rb",
+     "test/messages/presence_messages.rb",
+     "test/messages/roster_messages.rb",
+     "test/messages/service_discovery_messages.rb",
+     "test/messages/session_messages.rb",
+     "test/messages/version_discovery_messages.rb",
+     "test/peer/peer.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<rake>, [">= 0.8.3"])
+      s.add_runtime_dependency(%q<eventmachine>, [">= 0.12.6"])
+      s.add_runtime_dependency(%q<sequel>, [">= 3.9.0"])
+      s.add_runtime_dependency(%q<evma_xmlpushparser>, [">= 0.0.1"])
+    else
+      s.add_dependency(%q<rake>, [">= 0.8.3"])
+      s.add_dependency(%q<eventmachine>, [">= 0.12.6"])
+      s.add_dependency(%q<sequel>, [">= 3.9.0"])
+      s.add_dependency(%q<evma_xmlpushparser>, [">= 0.0.1"])
+    end
+  else
+    s.add_dependency(%q<rake>, [">= 0.8.3"])
+    s.add_dependency(%q<eventmachine>, [">= 0.12.6"])
+    s.add_dependency(%q<sequel>, [">= 3.9.0"])
+    s.add_dependency(%q<evma_xmlpushparser>, [">= 0.0.1"])
+  end
+end
+

data/lib/agent_xmpp.rb

@@ -0,0 +1,22 @@
+$:.unshift(File.dirname(__FILE__))
+
+require 'find'
+require 'singleton'
+require 'logger'
+require 'socket'
+require 'optparse'
+require 'rexml/document'
+require 'base64'
+
+require 'eventmachine'
+require 'evma_xmlpushparser'
+
+require 'sequel'
+
+require 'agent_xmpp/patches'
+require 'agent_xmpp/client'
+require 'agent_xmpp/xmpp'
+require 'agent_xmpp/config'
+require 'agent_xmpp/models'
+require 'agent_xmpp/main'
+require 'agent_xmpp/admin'

data/lib/agent_xmpp/admin.rb

@@ -0,0 +1,113 @@
+#.........................................................................................................
+# authorize command access
+#.........................................................................................................
+before :command => :all do
+  if access = route[:opts][:access]
+    jid = params[:from]    
+    unless AgentXmpp.is_account_jid?(jid)
+      groups = AgentXmpp::Contact.find_by_jid(jid)[:groups] 
+      [access].flatten.any?{|a| groups.include?(a)}
+    else; true; end
+  else; true; end
+end
+
+#.........................................................................................................
+# admin commands
+#.........................................................................................................
+command 'admin/contacts', :access => 'admin' do
+  AgentXmpp.logger.info "ACTION: admin/contacts"
+  AgentXmpp::Contact.find_all.map{|c| {:jid => c[:jid], :sub => c[:subscription]}}
+end
+
+#.........................................................................................................
+command 'admin/on_line_users', :access => 'admin' do
+  AgentXmpp.logger.info "ACTION: admin/on_line_users"
+  AgentXmpp::Roster.find_all_by_status(:available).map do |r| 
+    jid = AgentXmpp::Xmpp::Jid.new(r[:jid]) 
+    {:jid => jid.bare, :resource => jid.resource}
+  end
+end
+
+#.........................................................................................................
+command 'admin/add_contact', :access => 'admin' do
+  AgentXmpp.logger.info "ACTION: admin/add_contact"
+  on(:execute) do |form|
+    form.add_title('Add Contact')
+    form.add_jid_single('jid', 'contact JID')
+    form.add_text_single('groups', 'groups comma seperated')
+  end
+  on(:submit) do
+    contact = params[:data]
+    if contact["jid"]
+      AgentXmpp::Contact.update(contact)
+      xmpp_msg(AgentXmpp::Xmpp::IqRoster.update(pipe, contact["jid"], contact["groups"].split(/,/))) 
+      xmpp_msg(AgentXmpp::Xmpp::Presence.subscribe(contact["jid"]))
+      delegate_to(
+        :on_update_roster_item_result => lambda do |pipe, item_jid|     
+          command_completed if item_jid.eql?(contact["jid"])
+        end,
+        :on_update_roster_item_error  => lambda do |pipe, item_jid|
+          error(:bad_request, params, 'roster updated failed') if item_jid.eql?(contact["jid"])
+        end
+      )
+    else
+      error(:bad_request, params, 'jid not specified')
+    end
+  end
+end
+
+#.........................................................................................................
+command 'admin/delete_contact', :access => 'admin' do
+  AgentXmpp.logger.info "ACTION: admin/delete_contact"
+  contacts = AgentXmpp::Contact.find_all.map{|c| c[:jid]}
+  on(:execute) do |form|
+    form.add_title('Delete Contact')
+    form.add_list_single('jid', contacts)
+  end
+  on(:submit) do
+    contact = params[:data]
+    if contact["jid"]
+      xmpp_msg(AgentXmpp::Xmpp::IqRoster.remove(pipe, contact["jid"]))  
+      delegate_to(
+        :on_remove_roster_item_result => lambda do |pipe, item_jid|           
+          command_completed if item_jid.eql?(contact["jid"])
+        end,
+        :on_remove_roster_item_error  => lambda do |pipe, item_jid|
+          error(:bad_request, params, 'roster updated failed') if item_jid.eql?(contact["jid"])
+        end
+      )
+    else
+      error(:bad_request, params, 'jid not specified')
+    end
+  end
+end
+
+#.........................................................................................................
+command 'admin/subscriptions', :access => 'admin' do
+  AgentXmpp.logger.info "ACTION: admin/subscriptions"
+  AgentXmpp::Subscription.stats_by_node    
+end
+
+#.........................................................................................................
+command 'admin/publications', :access => 'admin' do
+  AgentXmpp.logger.info "ACTION: admin/publications"
+  AgentXmpp::Publication.stats_by_node
+end
+
+#.........................................................................................................
+command 'admin/messages_by_type', :access => 'admin' do
+  AgentXmpp.logger.info "ACTION: admin/publications"
+  AgentXmpp::Message.stats_by_message_type
+end
+
+#.........................................................................................................
+command 'admin/messages_by_contact', :access => 'admin' do
+  AgentXmpp.logger.info "ACTION: admin/publications"
+  AgentXmpp::Contact.message_stats
+end
+
+#.........................................................................................................
+command 'admin/messages_by_command', :access => 'admin' do
+  AgentXmpp.logger.info "ACTION: admin/publications"
+  AgentXmpp::Message.stats_by_command_node
+end

data/lib/agent_xmpp/client.rb

@@ -0,0 +1,7 @@
+require 'agent_xmpp/client/boot'
+require 'agent_xmpp/client/controller'
+require 'agent_xmpp/client/response'
+require 'agent_xmpp/client/message_pipe'
+require 'agent_xmpp/client/message_delegate'
+require 'agent_xmpp/client/connection'
+require 'agent_xmpp/client/client'

data/lib/agent_xmpp/client/boot.rb

@@ -0,0 +1,83 @@
+##############################################################################################################
+module AgentXmpp
+
+  #####-------------------------------------------------------------------------------------------------------
+  class Boot
+    
+    ####......................................................................................................
+    class << self
+
+      ####....................................................................................................
+      # start application
+      #.......................................................................................................
+      def boot
+        
+        AgentXmpp.config_file = File.join(AgentXmpp.app_path, "agent_xmpp.yml")
+
+        AgentXmpp.log_file = add_path(AgentXmpp.log_file) if AgentXmpp.log_file.kind_of?(String)
+        AgentXmpp.logger = Logger.new(AgentXmpp.log_file, 10, 1024000)
+        AgentXmpp.logger.level = Logger::WARN 
+
+        call_if_implemented(:call_before_start)
+
+        AgentXmpp.logger.info "STARTING AgentXmpp"
+        AgentXmpp.logger.info "APPLICATION PATH: #{AgentXmpp.app_path}"
+        AgentXmpp.logger.info "LOG FILE: #{AgentXmpp.log_file.kind_of?(String) ? AgentXmpp.log_file : "STDOUT"}"
+        AgentXmpp.logger.info "CONFIGURATION FILE: #{AgentXmpp.config_file}"
+
+        raise AgentXmppError, "Configuration file #{AgentXmpp.config_file} required." unless File.exist?(AgentXmpp.config_file) 
+
+        AgentXmpp.config = File.open(AgentXmpp.config_file) {|yf| YAML::load(yf)}
+        
+        AgentXmpp.create_agent_xmpp_db  
+        AgentXmpp.create_in_memory_db        
+        AgentXmpp.upgrade_agent_xmpp_db
+        Contact.load_config 
+        Publication.load_config 
+                
+        AgentXmpp::Client.new().connect
+      
+      end
+      
+      ####....................................................................................................
+      # application deligate methods
+      #.......................................................................................................
+      def call_if_implemented(method, *args)
+        send(method, *args) if respond_to?(method)
+      end
+      
+      #.........................................................................................................
+      def callbacks(*args)
+        args.each do |meth| 
+          instance_eval <<-do_eval
+            def call_#{meth}(*args)
+              @array_#{meth}.each{|m| m.call(*args)} unless @array_#{meth}.nil?
+            end
+            def #{meth}(&blk)
+              (@array_#{meth} ||= []) << blk
+            end
+          do_eval
+        end
+      end       
+                          
+      ####......................................................................................................
+      # private
+      #.......................................................................................................
+      def add_path(dir)
+        File.join(AgentXmpp.app_path, dir)
+      end
+      
+      private :add_path
+        
+    #### self
+    end
+
+    #.........................................................................................................
+    callbacks(:before_start, :after_connected, :discovered_all_publish_nodes, :discovered_command_nodes, 
+              :discovered_pubsub_node, :received_presence, :restarting_client)
+
+  #### Boot
+  end
+  
+#### AgentXmpp
+end