ahoy 0.0.1

data/lib/ahoy.rb

@@ -0,0 +1,18 @@
+base = File.expand_path("#{File.dirname(__FILE__)}/ahoy")
+libs = %W{chat contact contact_list errors user}
+
+libs.each do |lib|
+  require "#{base}/#{lib}"
+end
+
+module Ahoy
+  SERVICE_TYPE = "_presence._tcp"
+  
+  def self.more_coming?(reply)
+    reply.flags.to_i & DNSSD::Flags::MoreComing > 0
+  end
+  
+  def self.add?(reply)
+    reply.flags.to_i & DNSSD::Flags::Add > 0
+  end
+end

data/lib/ahoy/broadcast.rb

@@ -0,0 +1,26 @@
+require File.expand_path("#{File.dirname(__FILE__)}/../ahoy")
+
+module Ahoy
+  class Broadcast
+    def initialize(name, location="nowhere", domain="local")
+      user = Ahoy::User.new(name, location, domain)
+      user.sign_in
+      sleep 1
+      @chats = user.contacts.map {|contact| user.chat(contact)}
+    end
+    
+    def send(message)
+      @chats.each do |chat|
+        begin
+          chat.send(message)
+        rescue
+          @chats.delete(chat)
+        end
+      end
+    end
+    
+    def close
+      @chats.each {|chat| chat.close}
+    end
+  end
+end

data/lib/ahoy/chat.rb

@@ -0,0 +1,58 @@
+require 'rubygems'
+require 'xmpp4r'
+require File.expand_path("#{File.dirname(__FILE__)}/xmpp4r_hack")
+
+module Ahoy
+  class Chat
+    attr_reader :user, :contact, :client
+    private :client
+    
+    def initialize(user, contact)
+      @user = user
+      @contact = contact
+      @client = nil
+    end
+    
+    # May raise Ahoy::ContactOfflineError
+    # 
+    def start
+      user.contact.resolve.getaddrinfo
+      connect
+    end
+    
+    # May raise Ahoy::ContactOfflineError
+    # 
+    def send(message)
+      start unless client
+      
+      message = Jabber::Message.new(contact.name, message)
+      message.type = :chat
+      begin
+        @client.send(message)
+      rescue IOError
+        connect
+        retry
+      end
+      message
+    end
+    
+    def close
+      @client.close
+      @client = nil
+    end
+    
+    private
+    def connect
+      contact.resolve
+      
+      @client = Jabber::Client.new(Jabber::JID.new(user.name))
+      sleep 0.5
+      begin
+        @client.connect(contact.target, contact.port, user.contact.ip)
+      rescue Errno::ECONNREFUSED
+        raise Ahoy::ContactOfflineError.new("Contact Offline")
+      end
+    end
+    
+  end
+end

data/lib/ahoy/contact.rb

@@ -0,0 +1,59 @@
+require 'rubygems'
+require 'dnssd'
+
+module Ahoy
+  class Contact
+    attr_reader :name, :domain, :target, :ip, :port, :interface
+    attr_accessor :online
+    
+    def initialize(name, domain="local")
+      @name = name
+      @domain = domain
+      @target = nil
+      @ip = nil
+      @port = nil
+      @interface = nil
+      @online = true
+    end
+    
+    def fullname
+      [name, Ahoy::SERVICE_TYPE, domain].join(".")
+    end
+    
+    def ==(other)
+      other.is_a?(self.class) && other.fullname == fullname
+    end
+    
+    def online?
+      online
+    end
+    
+    def resolve
+      service = DNSSD::Service.new
+      main = Thread.current
+      service.resolve(name, Ahoy::SERVICE_TYPE, domain) do |resolved|
+        next if Ahoy::more_coming?(resolved)
+        service.stop unless service.stopped?
+        @target = resolved.target
+        @port = resolved.port
+        @interface = resolved.interface
+        main.run
+      end
+      Thread.stop unless service.stopped?
+      self
+    end
+    
+    def getaddrinfo
+      service = DNSSD::Service.new
+      main = Thread.current
+      service.getaddrinfo(target, DNSSD::Service::IPv4, 0, interface) do |addressed|
+        service.stop unless service.stopped?
+        @ip = addressed.address
+        main.run
+      end
+      Thread.stop unless service.stopped?
+      self
+    end
+    
+  end
+end

data/lib/ahoy/contact_list.rb

@@ -0,0 +1,76 @@
+require 'thread'
+require 'weakref'
+
+module Ahoy
+  class ContactList
+    include Enumerable
+    
+    attr_reader :list, :weak_list, :lock, :user
+    private :list, :weak_list, :lock
+    
+    def initialize(user)
+      @user = user
+      @list = []
+      @weak_list = []
+      @lock = Mutex.new
+      
+      start_browse
+    end
+    
+    def each(&block)
+      lock.synchronize do
+        list.each(&block)
+      end
+    end
+    
+    private
+    def start_browse
+      DNSSD.browse(Ahoy::SERVICE_TYPE) do |browsed|
+        # next if Ahoy::more_coming?(browsed)
+        if Ahoy::add?(browsed) && browsed.name != user.name
+          add(Ahoy::Contact.new(browsed.name, browsed.domain))
+        elsif Ahoy::add?(browsed)
+          user.contact = Ahoy::Contact.new(browsed.name, browsed.domain)
+        else
+          remove(browsed.fullname)
+        end
+      end
+    end
+    
+    def add(contact)
+      lock.synchronize do
+        unless list.find {|in_list| contact == in_list}
+          contact ||= find_in_weak_list(contact)
+          contact.online = true
+          list.push(contact)
+        end
+      end
+    end
+    
+    def remove(fullname)
+      fullname = fullname.fullname if fullname.respond_to?(:fullname)
+      lock.synchronize do
+        contact = list.find {|c| c.fullname == fullname}
+        if contact
+          list.delete(contact)
+          contact.online = false
+          weak_list.push(WeakRef.new(contact))
+          contact
+        end
+      end
+    end
+    
+    def find_in_weak_list(contact)
+      existing_contact = nil
+      Thread.exclusive do
+        GC.disable
+        weak_list.select! {|ref| ref.weakref_alive?}
+        contact_ref = weak_list.find {|ref| contact == ref}
+        existing_contact = contact_ref.__getobj__
+        GC.enable
+      end
+      existing_contact
+    end
+    
+  end
+end

data/lib/ahoy/errors.rb

@@ -0,0 +1,3 @@
+module Ahoy
+  ContactOfflineError = Class.new(StandardError)
+end

data/lib/ahoy/user.rb

@@ -0,0 +1,57 @@
+require 'rubygems'
+require 'dnssd'
+
+module Ahoy
+  class User
+    attr_reader :short_name, :location, :domain, :contacts
+    attr_accessor :port, :flags, :interface, :contact
+    
+    def initialize(name, location="nowhere", domain="local")
+      @short_name = name
+      @location = location
+      @domain = domain
+      
+      @contacts = Ahoy::ContactList.new(self)
+      @contact = nil
+      
+      @port = 5562
+      @flags = 0
+      @interface = "en0"
+    end
+    
+    def name
+      "#{short_name}@#{location}"
+    end
+    
+    def sign_in(status="avail", msg=nil)
+      @registrar = DNSSD.register(
+        name,
+        Ahoy::SERVICE_TYPE,
+        domain,
+        port,
+        txt_record(status, msg),
+        flags.to_i,
+        interface)
+    end
+    
+    def contact
+      sleep 0.01 until @contact
+      @contact
+    end
+    
+    def chat(contact)
+      Ahoy::Chat.new(self, contact)
+    end
+    
+    private
+    def txt_record(status, msg)
+      DNSSD::TextRecord.new(
+       "txtvers" => 1,
+       "port.p2pj" => port,
+       "status" => status,
+       "msg" => msg,
+       "1st" => short_name)
+    end
+    
+  end
+end

data/lib/ahoy/xmpp4r_hack.rb

@@ -0,0 +1,41 @@
+require 'rubygems'
+require 'xmpp4r'
+
+module Jabber
+  class Connection
+    def connect(host, port, local_host=nil, local_port=nil)
+      @host = host
+      @port = port
+      # Reset is_tls?, so that it works when reconnecting
+      @tls = false
+      
+      Jabber::debuglog("CONNECTING:\n#{@host}:#{@port}, local #{local_host}:#{local_port}")
+      @socket = TCPSocket.new(@host, @port, local_host, local_port)
+      
+      # We want to use the old and deprecated SSL protocol (usually on port 5223)
+      if @use_ssl
+        ssl = OpenSSL::SSL::SSLSocket.new(@socket)
+        ssl.connect # start SSL session
+        ssl.sync_close = true
+        Jabber::debuglog("SSL connection established.")
+        @socket = ssl
+      end
+      
+      start
+      
+      accept_features
+      
+      @keepaliveThread = Thread.new do
+        Thread.current.abort_on_exception = true
+        keepalive_loop
+      end
+    end
+  end
+  
+  class Client
+    def connect(host, port, local_host=nil, local_port=nil)
+      super
+      self
+    end
+  end
+end

metadata

@@ -0,0 +1,82 @@
+--- !ruby/object:Gem::Specification 
+name: ahoy
+version: !ruby/object:Gem::Version 
+  version: 0.0.1
+platform: ruby
+authors: 
+- Mat Sadler
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-01-27 00:00:00 +00:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: dnssd
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 1.3.1
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: xmpp4r
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        version: "0.5"
+    version: 
+description: Serverless Messaging using DNSDS/mDNS, XMPP, and Ruby
+email: mat@sourcetagsandcodes.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/ahoy/broadcast.rb
+- lib/ahoy/chat.rb
+- lib/ahoy/contact.rb
+- lib/ahoy/contact_list.rb
+- lib/ahoy/errors.rb
+- lib/ahoy/user.rb
+- lib/ahoy/xmpp4r_hack.rb
+- lib/ahoy.rb
+has_rdoc: true
+homepage: http://sourcetagsandcodes.com
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --inline-source
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: Bonjour Chat for Ruby
+test_files: []
+