tp-blather 0.8.2

data/examples/echo.rb

@@ -0,0 +1,19 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'blather/client'
+
+when_ready { puts "Connected ! send messages to #{jid.stripped}." }
+
+subscription :request? do |s|
+  write_to_stream s.approve!
+end
+
+message :chat?, :body => 'exit' do |m|
+  say m.from, 'Exiting ...'
+  shutdown
+end
+
+message :chat?, :body do |m|
+  say m.from, "You sent: #{m.body}"
+end

data/examples/execute.rb

@@ -0,0 +1,17 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'blather/client'
+
+message :chat?, :body => 'exit' do |m|
+  say m.from, 'Exiting ...'
+  shutdown
+end
+
+message :chat?, :body do |m|
+  begin
+    say m.from, eval(m.body)
+  rescue => e
+    say m.from, e.inspect
+  end
+end

data/examples/ping_pong.rb

@@ -0,0 +1,38 @@
+require 'rubygems'
+require 'blather/client/dsl'
+$stdout.sync = true
+
+module Ping
+  extend Blather::DSL
+  def self.run; client.run; end
+
+  setup 'echo@jabber.local/ping', 'echo'
+
+  status :from => Blather::JID.new('echo@jabber.local/pong') do |s|
+    puts "serve!"
+    say s.from, 'ping'
+  end
+
+  message :chat?, :body => 'pong' do |m|
+    puts "ping!"
+    say m.from, 'ping'
+  end
+end
+
+module Pong
+  extend Blather::DSL
+  def self.run; client.run; end
+
+  setup 'echo@jabber.local/pong', 'echo'
+  message :chat?, :body => 'ping' do |m|
+    puts "pong!"
+    say m.from, 'pong'
+  end
+end
+
+trap(:INT) { EM.stop }
+trap(:TERM) { EM.stop }
+EM.run do
+  Ping.run
+  Pong.run
+end

data/examples/print_hierarchy.rb

@@ -0,0 +1,77 @@
+require 'rubygems'
+require 'blather'
+
+class Object
+  begin
+    ObjectSpace.each_object(Class.new) {}
+
+    # Exclude this class unless it's a subclass of our supers and is defined.
+    # We check defined? in case we find a removed class that has yet to be
+    # garbage collected. This also fails for anonymous classes -- please
+    # submit a patch if you have a workaround.
+    def subclasses_of(*superclasses)
+      subclasses = []
+
+      superclasses.each do |sup|
+        ObjectSpace.each_object(class << sup; self; end) do |k|
+          if k != sup && (k.name.blank? || eval("defined?(::#{k}) && ::#{k}.object_id == k.object_id"))
+            subclasses << k
+          end
+        end
+      end
+
+      subclasses
+    end
+  rescue RuntimeError
+    # JRuby and any implementations which cannot handle the objectspace traversal
+    # above fall back to this implementation
+    def subclasses_of(*superclasses)
+      subclasses = []
+
+      superclasses.each do |sup|
+        ObjectSpace.each_object(Class) do |k|
+          if superclasses.any? { |superclass| k < superclass } &&
+            (k.name.blank? || eval("defined?(::#{k}) && ::#{k}.object_id == k.object_id"))
+            subclasses << k
+          end
+        end
+        subclasses.uniq!
+      end
+      subclasses
+    end
+  end
+end
+
+class Hash
+  def deep_merge(second)
+    # From: http://www.ruby-forum.com/topic/142809
+    # Author: Stefan Rusterholz
+    merger = proc { |key,v1,v2| Hash === v1 && Hash === v2 ? v1.merge(v2, &merger) : v2 }
+    self.merge(second, &merger)
+  end
+end
+
+handlers = {}
+(Object.subclasses_of(Blather::Stanza) + Object.subclasses_of(Blather::BlatherError)).each do |klass|
+  handlers = handlers.deep_merge klass.handler_hierarchy.inject('klass' => klass.to_s.gsub('Blather::', '')) { |h,k| {k.to_s => h} }
+end
+
+level = 0
+runner = proc do |k,v|
+  next if k == 'klass'
+
+  str = ''
+  if level > 0
+    (level - 1).times { str << '|  ' }
+    str << '|- '
+  end
+
+  puts str+k
+  if Hash === v
+    level += 1
+    v.sort.each &runner
+    level -= 1
+  end
+end
+
+handlers.sort.each &runner

data/examples/rosterprint.rb

@@ -0,0 +1,15 @@
+#!/usr/bin/env ruby
+
+# Prints out each roster entry
+
+require 'rubygems'
+require 'blather/client'
+
+when_ready do
+  roster.grouped.each do |group, items|
+    puts "#{'*'*3} #{group || 'Ungrouped'} #{'*'*3}"
+    items.each { |item| puts "- #{item.name} (#{item.jid})" }
+    puts
+  end
+  shutdown
+end

data/examples/stream_only.rb

@@ -0,0 +1,28 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'blather'
+
+trap(:INT) { EM.stop }
+trap(:TERM) { EM.stop }
+EM.run do
+  Blather::Stream::Client.start(Class.new {
+    attr_accessor :jid
+
+    def post_init(stream, jid = nil)
+      @stream = stream
+      self.jid = jid
+
+      @stream.send_data Blather::Stanza::Presence::Status.new
+      puts "Stream started!"
+    end
+
+    def receive_data(stanza)
+      @stream.send_data stanza.reply!
+    end
+
+    def unbind
+      puts "Stream ended!"
+    end
+  }.new, 'echo@jabber.local', 'echo')
+end

data/examples/trusted_echo.rb

@@ -0,0 +1,21 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'blather/client'
+
+setup 'chris@vines.local', 'test', 'vines.local', 5222, "./certs"
+
+when_ready { puts "Connected ! send messages to #{jid.stripped}." }
+
+subscription :request? do |s|
+  write_to_stream s.approve!
+end
+
+message :chat?, :body => 'exit' do |m|
+  say m.from, 'Exiting ...'
+  shutdown
+end
+
+message :chat?, :body do |m|
+  say m.from, "You sent: #{m.body}"
+end

data/examples/xmpp4r/echo.rb

@@ -0,0 +1,36 @@
+#!/usr/bin/ruby
+
+# This bot will reply to every message it receives. To end the game, send 'exit'
+
+require 'rubygems'
+require 'xmpp4r/client'
+include Jabber
+
+# settings
+if ARGV.length != 2
+  puts "Run with ./echo_thread.rb user@server/resource password"
+  exit 1
+end
+myJID = JID.new(ARGV[0])
+myPassword = ARGV[1]
+cl = Client.new(myJID)
+cl.connect
+cl.auth(myPassword)
+cl.send(Presence.new)
+puts "Connected ! send messages to #{myJID.strip.to_s}."
+mainthread = Thread.current
+cl.add_message_callback do |m|
+  if m.type != :error
+    m2 = Message.new(m.from, "You sent: #{m.body}")
+    m2.type = m.type
+    cl.send(m2)
+    if m.body == 'exit'
+      m2 = Message.new(m.from, "Exiting ...")
+      m2.type = m.type
+      cl.send(m2)
+      mainthread.wakeup
+    end
+  end
+end
+Thread.stop
+cl.close

data/lib/blather.rb

@@ -0,0 +1,112 @@
+# Require the necessary files
+%w[
+  rubygems
+  eventmachine
+  niceogiri
+  ipaddr
+  digest/md5
+  digest/sha1
+  logger
+  openssl
+  girl_friday
+
+  active_support/core_ext/class/attribute
+  active_support/core_ext/object/blank
+
+  blather/core_ext/eventmachine
+  blather/core_ext/ipaddr
+
+  blather/cert_store
+  blather/errors
+  blather/errors/sasl_error
+  blather/errors/stanza_error
+  blather/errors/stream_error
+  blather/file_transfer
+  blather/file_transfer/ibb
+  blather/file_transfer/s5b
+  blather/jid
+  blather/roster
+  blather/roster_item
+  blather/xmpp_node
+
+  blather/stanza
+  blather/stanza/iq
+  blather/stanza/iq/command
+  blather/stanza/iq/ibb
+  blather/stanza/iq/ping
+  blather/stanza/iq/query
+  blather/stanza/iq/roster
+  blather/stanza/iq/s5b
+  blather/stanza/iq/si
+  blather/stanza/iq/vcard
+  blather/stanza/disco
+  blather/stanza/disco/disco_info
+  blather/stanza/disco/disco_items
+  blather/stanza/disco/capabilities
+  blather/stanza/message
+  blather/stanza/message/muc_user
+  blather/stanza/presence
+  blather/stanza/presence/c
+  blather/stanza/presence/status
+  blather/stanza/presence/subscription
+  blather/stanza/presence/muc
+  blather/stanza/presence/muc_user
+
+  blather/stanza/pubsub
+  blather/stanza/pubsub/affiliations
+  blather/stanza/pubsub/create
+  blather/stanza/pubsub/event
+  blather/stanza/pubsub/items
+  blather/stanza/pubsub/publish
+  blather/stanza/pubsub/retract
+  blather/stanza/pubsub/subscribe
+  blather/stanza/pubsub/subscription
+  blather/stanza/pubsub/subscriptions
+  blather/stanza/pubsub/unsubscribe
+
+  blather/stanza/pubsub_owner
+  blather/stanza/pubsub_owner/delete
+  blather/stanza/pubsub_owner/purge
+
+  blather/stanza/x
+
+  blather/stream
+  blather/stream/client
+  blather/stream/component
+  blather/stream/parser
+  blather/stream/features
+  blather/stream/features/resource
+  blather/stream/features/sasl
+  blather/stream/features/session
+  blather/stream/features/tls
+  blather/stream/features/register
+].each { |r| require r }
+
+module Blather
+  @@logger = nil
+
+  class << self
+
+    # Default logger level. Any internal call to log() will forward the log message to
+    # the default log level
+    attr_accessor :default_log_level
+
+    def logger
+      @@logger ||= Logger.new($stdout).tap {|logger| logger.level = Logger::INFO }
+    end
+
+    def logger=(logger)
+      @@logger = logger
+    end
+
+    def default_log_level
+      @default_log_level ||= :debug # by default is debug (as it used to be)
+    end
+
+    def log(message)
+      logger.send self.default_log_level, message
+    end
+
+  end
+
+end

data/lib/blather/cert_store.rb

@@ -0,0 +1,53 @@
+# encoding: UTF-8
+
+module Blather
+
+  # An X509 certificate store that validates certificate trust chains.
+  # This uses the #{cert_directory}/*.crt files as the list of trusted root
+  # CA certificates.
+  class CertStore
+    @@certs = nil
+    @cert_directory = nil
+
+    def initialize(cert_directory)
+      @cert_directory = cert_directory
+      @store = OpenSSL::X509::Store.new
+      certs.each {|c| @store.add_cert(c) }
+    end
+
+    # Return true if the certificate is signed by a CA certificate in the
+    # store. If the certificate can be trusted, it's added to the store so
+    # it can be used to trust other certs.
+    def trusted?(pem)
+      if cert = OpenSSL::X509::Certificate.new(pem) rescue nil
+        @store.verify(cert).tap do |trusted|
+          @store.add_cert(cert) if trusted rescue nil
+        end
+      end
+    end
+
+    # Return true if the domain name matches one of the names in the
+    # certificate. In other words, is the certificate provided to us really
+    # for the domain to which we think we're connected?
+    def domain?(pem, domain)
+      if cert = OpenSSL::X509::Certificate.new(pem) rescue nil
+        OpenSSL::SSL.verify_certificate_identity(cert, domain) rescue false
+      end
+    end
+
+    # Return the trusted root CA certificates installed in the @cert_directory. These
+    # certificates are used to start the trust chain needed to validate certs
+    # we receive from clients and servers.
+    def certs
+      unless @@certs
+        pattern = /-{5}BEGIN CERTIFICATE-{5}\n.*?-{5}END CERTIFICATE-{5}\n/m
+        dir = @cert_directory
+        certs = Dir[File.join(dir, '*.crt')].map {|f| File.read(f) }
+        certs = certs.map {|c| c.scan(pattern) }.flatten
+        certs.map! {|c| OpenSSL::X509::Certificate.new(c) }
+        @@certs = certs.reject {|c| c.not_after < Time.now }
+      end
+      @@certs
+    end
+  end
+end

data/lib/blather/client.rb

@@ -0,0 +1,95 @@
+require 'optparse'
+require File.join(File.dirname(__FILE__), *%w[client dsl])
+
+include Blather::DSL
+
+options = {}
+optparse = OptionParser.new do |opts|
+  opts.banner = "Run with #{$0} [options] user@server/resource password [host] [port]"
+
+  opts.on('-D', '--debug', 'Run in debug mode (you will see all XMPP communication)') do
+    options[:debug] = true
+  end
+
+  opts.on('-d', '--daemonize', 'Daemonize the process') do |daemonize|
+    options[:daemonize] = daemonize
+  end
+
+  opts.on('--pid=[PID]', 'Write the PID to this file') do |pid|
+    if !File.writable?(File.dirname(pid))
+      $stderr.puts "Unable to write log file to #{pid}"
+      exit 1
+    end
+    options[:pid] = pid
+  end
+
+  opts.on('--log=[LOG]', 'Write to the [LOG] file instead of stdout/stderr') do |log|
+    if !File.writable?(File.dirname(log))
+      $stderr.puts "Unable to write log file to #{log}"
+      exit 1
+    end
+    options[:log] = log
+  end
+  
+  opts.on('--certs=[CERTS DIRECTORY]', 'The directory path where the trusted certificates are stored') do |certs|
+    if !File.directory?(certs)
+      $stderr.puts "The certs directory path (#{certs}) is no good."
+      exit 1
+    end
+    options[:certs] = certs
+  end
+  
+  opts.on_tail('-h', '--help', 'Show this message') do
+    puts opts
+    exit
+  end
+
+  opts.on_tail('-v', '--version', 'Show version') do
+    require 'yaml'
+    version = YAML.load_file File.join(File.dirname(__FILE__), %w[.. .. VERSION.yml])
+    puts "Blather v#{version[:major]}.#{version[:minor]}.#{version[:patch]}"
+    exit
+  end
+end
+optparse.parse!
+
+at_exit do
+  unless client.setup?
+    if ARGV.length < 2
+      puts optparse
+      exit 1
+    end
+    client.setup(*ARGV)
+  end
+
+  def run(options)
+    $stdin.reopen "/dev/null"
+
+    if options[:log]
+      log = File.new(options[:log], 'a')
+      log.sync = options[:debug]
+      $stdout.reopen log
+      $stderr.reopen $stdout
+    end
+
+    Blather.logger.level = Logger::DEBUG if options[:debug]
+
+    trap(:INT) { EM.stop }
+    trap(:TERM) { EM.stop }
+    EM.run { client.run }
+  end
+
+  if options[:daemonize]
+    pid = fork do
+      Process.setsid
+      exit if fork
+      File.open(options[:pid], 'w') { |f| f << Process.pid } if options[:pid]
+      run options
+      FileUtils.rm(options[:pid]) if options[:pid]
+    end
+    ::Process.detach pid
+    exit
+  else
+    run options
+  end
+end