xrc 0.0.1

data/lib/xrc/connection.rb

@@ -0,0 +1,40 @@
+module Xrc
+  class Connection
+    attr_reader :block, :domain, :port, :socket
+
+    def initialize(options, &block)
+      @domain = options[:domain]
+      @port = options[:port]
+      @block = block
+    end
+
+    def connect
+      @socket = Connector.connect(domain: domain, port: port)
+      start
+    end
+
+    def encrypt
+      @socket = TslConnector.connect(socket: socket)
+      start
+    end
+
+    def open
+      write Elements::Stream.new(domain: domain)
+    end
+
+    def write(object)
+      socket << object.to_s
+    end
+
+    private
+
+    def start
+      open
+      parse
+    end
+
+    def parse
+      Parser.new(socket: socket, &block).parse
+    end
+  end
+end

data/lib/xrc/connection/connector.rb

@@ -0,0 +1,42 @@
+module Xrc
+  class Connection
+    class Connector
+      def self.connect(*args)
+        new(*args).connect
+      end
+
+      attr_reader :options
+
+      def initialize(options)
+        @options = options
+      end
+
+      def connect
+        hosts.find do |host|
+          if socket = connect_to(host)
+            break socket
+          end
+        end
+      end
+
+      private
+
+      def domain
+        options[:domain]
+      end
+
+      def port
+        options[:port]
+      end
+
+      def hosts
+        HostsResolver.call(domain)
+      end
+
+      def connect_to(host)
+        TCPSocket.new(host, port)
+      rescue SocketError, Errno::ECONNREFUSED => exception
+      end
+    end
+  end
+end

data/lib/xrc/connection/tsl_connector.rb

@@ -0,0 +1,37 @@
+module Xrc
+  class Connection
+    class TslConnector
+      def self.connect(*args)
+        new(*args).connect
+      end
+
+      attr_reader :options
+
+      def initialize(options = {})
+        @options = options
+      end
+
+      def connect
+        ssl_socket.connect
+      end
+
+      private
+
+      def ssl_socket
+        ssl_socket = Socket.new(socket, context)
+        ssl_socket.sync_close = true
+        ssl_socket
+      end
+
+      def socket
+        options[:socket]
+      end
+
+      def context
+        context = OpenSSL::SSL::SSLContext.new
+        context.verify_mode = OpenSSL::SSL::VERIFY_NONE
+        context
+      end
+    end
+  end
+end

data/lib/xrc/elements/auth.rb

@@ -0,0 +1,41 @@
+module Xrc
+  module Elements
+    class Auth < REXML::Element
+      attr_reader :options
+
+      def initialize(options = {})
+        super("auth")
+        @options = options
+        add_sasl_namespace
+        add_plain_mechanism_attribute
+        add_credentials_text
+      end
+
+      private
+
+      def jid
+        options[:jid]
+      end
+
+      def password
+        options[:password]
+      end
+
+      def plain_credentials
+        "#{jid.strip}\x00#{jid.node}\x00#{password}"
+      end
+
+      def add_sasl_namespace
+        add_namespace(Namespaces::SASL)
+      end
+
+      def add_plain_mechanism_attribute
+        attributes["mechanism"] = "PLAIN"
+      end
+
+      def add_credentials_text
+        self.text = Base64.strict_encode64(plain_credentials)
+      end
+    end
+  end
+end

data/lib/xrc/elements/bind.rb

@@ -0,0 +1,18 @@
+module Xrc
+  module Elements
+    class Bind < REXML::Element
+      def initialize(options = {})
+        super("iq")
+        bind = REXML::Element.new("bind")
+        bind.add_namespace(Namespaces::BIND)
+        if options[:resource]
+          resource = REXML::Element.new("resource")
+          resource.text = options[:resource]
+          bind.add(resource)
+        end
+        add_attributes("type" => "set")
+        add(bind)
+      end
+    end
+  end
+end

data/lib/xrc/elements/join.rb

@@ -0,0 +1,20 @@
+module Xrc
+  module Elements
+    class Join < Presence
+      def initialize(options)
+        super()
+        attributes["from"] = options[:from]
+        attributes["to"] = options[:to]
+        add(x)
+      end
+
+      private
+
+      def x
+        element = REXML::Element.new("x")
+        element.add_namespace(Namespaces::MUC)
+        element
+      end
+    end
+  end
+end

data/lib/xrc/elements/message.rb

@@ -0,0 +1,20 @@
+module Xrc
+  module Elements
+    class Message < REXML::Element
+      def initialize(options = {})
+        super("message")
+        attributes["from"] = options[:from]
+        attributes["to"] = options[:to]
+        attributes["type"] = options[:type]
+        body.text = options[:body]
+        add(body)
+      end
+
+      private
+
+      def body
+        @body ||= REXML::Element.new("body")
+      end
+    end
+  end
+end

data/lib/xrc/elements/ping.rb

@@ -0,0 +1,21 @@
+module Xrc
+  module Elements
+    class Ping < REXML::Element
+      def initialize(options = {})
+        super("iq")
+        attributes["from"] = options[:from]
+        attributes["to"] = options[:to]
+        attributes["type"] = "get"
+        add(ping)
+      end
+
+      private
+
+      def ping
+        element = REXML::Element.new("ping")
+        element.add_namespace(Namespaces::PING)
+        element
+      end
+    end
+  end
+end

data/lib/xrc/elements/presence.rb

@@ -0,0 +1,9 @@
+module Xrc
+  module Elements
+    class Presence < REXML::Element
+      def initialize
+        super("presence")
+      end
+    end
+  end
+end

data/lib/xrc/elements/roster.rb

@@ -0,0 +1,13 @@
+module Xrc
+  module Elements
+    class Roster < REXML::Element
+      def initialize
+        super("iq")
+        query = REXML::Element.new("query")
+        query.add_namespace(Namespaces::ROSTER)
+        add_attributes("type" => "get")
+        add(query)
+      end
+    end
+  end
+end

data/lib/xrc/elements/session.rb

@@ -0,0 +1,13 @@
+module Xrc
+  module Elements
+    class Session < REXML::Element
+      def initialize
+        super("iq")
+        session = REXML::Element.new("session")
+        session.add_namespace(Namespaces::SESSION)
+        add_attributes("type" => "set")
+        add(session)
+      end
+    end
+  end
+end

data/lib/xrc/elements/starttls.rb

@@ -0,0 +1,10 @@
+module Xrc
+  module Elements
+    class Starttls < REXML::Element
+      def initialize
+        super("starttls")
+        add_namespace(Namespaces::TLS)
+      end
+    end
+  end
+end

data/lib/xrc/elements/stream.rb

@@ -0,0 +1,22 @@
+module Xrc
+  module Elements
+    class Stream
+      attr_reader :options
+
+      def initialize(options)
+        @options = options
+      end
+
+      def to_s
+        %W[
+          <stream:stream
+          xmlns:stream="http://etherx.jabber.org/streams"
+          xmlns="jabber:client"
+          to="#{options[:domain]}"
+          xml:lang="en"
+          version="1.0">
+        ].join(" ")
+      end
+    end
+  end
+end

data/lib/xrc/hosts_resolver.rb

@@ -0,0 +1,43 @@
+module Xrc
+  class HostsResolver
+    def self.call(*args)
+      new(*args).call
+    end
+
+    attr_reader :domain
+
+    def initialize(domain)
+      @domain = domain
+    end
+
+    def call
+      resolved_hosts + [domain]
+    end
+
+    private
+
+    def resolved_hosts
+      sorted_srvs.lazy.map(&:target).map(&:to_s).to_a
+    end
+
+    def sorted_srvs
+      @sorted_srvs ||= srvs.sort_by do |resource|
+        [resource.priority, -resource.weight]
+      end
+    end
+
+    def srvs
+      Resolv::DNS.open do |dns|
+        dns.getresources(name, type_class)
+      end
+    end
+
+    def name
+      "_xmpp-client._tcp.#{domain}"
+    end
+
+    def type_class
+      Resolv::DNS::Resource::IN::SRV
+    end
+  end
+end

data/lib/xrc/jid.rb

@@ -0,0 +1,44 @@
+module Xrc
+  class Jid
+    PATTERN = /\A(?:([^@]*)@)??([^@\/]*)(?:\/(.*?))?\z/
+
+    attr_reader :raw
+
+    def initialize(raw)
+      @raw = raw
+    end
+
+    def node
+      sections[0]
+    end
+
+    def domain
+      sections[1]
+    end
+
+    def resource
+      sections[2]
+    end
+
+    def strip
+      "#{node}@#{domain}"
+    end
+
+    def to_s
+      str = strip
+      str << "/#{resource}" if resource
+      str
+    end
+
+    def ==(jid)
+      jid = Jid.new(jid) unless jid.is_a?(Jid)
+      to_s == jid.to_s || strip == jid.strip
+    end
+
+    private
+
+    def sections
+      @sections ||= raw.scan(PATTERN).first
+    end
+  end
+end

data/lib/xrc/message_builder.rb

@@ -0,0 +1,62 @@
+module Xrc
+  class MessageBuilder
+    # Builds a Message object from a REXML::Element.
+    # @param [REXML::Element] element A message element
+    # @return [Xrc::Messages::Base] An ancestor of Xrc::Messages::Base instance
+    def self.build(element)
+      new(element).build
+    end
+
+    # @param [REXML::Element] element A message element
+    def initialize(element)
+      @element = element
+    end
+
+    def build
+      message_class.new(@element)
+    end
+
+    private
+
+    def message_class
+      case
+      when has_room_message?
+        Messages::RoomMessage
+      when has_private_message?
+        Messages::PrivateMessage
+      when has_subject?
+        Messages::Subject
+      else
+        Messages::Null
+      end
+    end
+
+    def has_room_message?
+      has_groupchat_type? && has_body?
+    end
+
+    def has_private_message?
+      has_chat_type? && has_body?
+    end
+
+    def has_chat_type?
+      type == "chat"
+    end
+
+    def has_groupchat_type?
+      type == "groupchat"
+    end
+
+    def has_body?
+      !!@element.elements["body"]
+    end
+
+    def has_subject?
+      !!@element.elements["subject"]
+    end
+
+    def type
+      @element.attributes["type"].to_s
+    end
+  end
+end