diaspora-vines 0.1.2

data/lib/vines.rb

@@ -0,0 +1,203 @@
+# encoding: UTF-8
+
+module Vines
+  NAMESPACES = {
+    :stream           => 'http://etherx.jabber.org/streams'.freeze,
+    :client           => 'jabber:client'.freeze,
+    :server           => 'jabber:server'.freeze,
+    :component        => 'jabber:component:accept'.freeze,
+    :roster           => 'jabber:iq:roster'.freeze,
+    :non_sasl         => 'jabber:iq:auth'.freeze,
+    :storage          => 'jabber:iq:private'.freeze,
+    :version          => 'jabber:iq:version'.freeze,
+    :sasl             => 'urn:ietf:params:xml:ns:xmpp-sasl'.freeze,
+    :tls              => 'urn:ietf:params:xml:ns:xmpp-tls'.freeze,
+    :bind             => 'urn:ietf:params:xml:ns:xmpp-bind'.freeze,
+    :session          => 'urn:ietf:params:xml:ns:xmpp-session'.freeze,
+    :ping             => 'urn:xmpp:ping'.freeze,
+    :pubsub           => 'http://jabber.org/protocol/pubsub'.freeze,
+    :pubsub_event     => 'http://jabber.org/protocol/pubsub#event'.freeze,
+    :pubsub_create    => 'http://jabber.org/protocol/pubsub#create-nodes'.freeze,
+    :pubsub_delete    => 'http://jabber.org/protocol/pubsub#delete-nodes'.freeze,
+    :pubsub_instant   => 'http://jabber.org/protocol/pubsub#instant-nodes'.freeze,
+    :pubsub_item_ids  => 'http://jabber.org/protocol/pubsub#item-ids'.freeze,
+    :pubsub_publish   => 'http://jabber.org/protocol/pubsub#publish'.freeze,
+    :pubsub_subscribe => 'http://jabber.org/protocol/pubsub#subscribe'.freeze,
+    :disco_items      => 'http://jabber.org/protocol/disco#items'.freeze,
+    :disco_info       => 'http://jabber.org/protocol/disco#info'.freeze,
+    :http_bind        => 'http://jabber.org/protocol/httpbind'.freeze,
+    :bosh             => 'urn:xmpp:xbosh'.freeze,
+    :vcard            => 'vcard-temp'.freeze,
+    :si               => 'http://jabber.org/protocol/si'.freeze,
+    :byte_streams     => 'http://jabber.org/protocol/bytestreams'.freeze
+  }.freeze
+
+  module Log
+    @@logger = nil
+    def log
+      unless @@logger
+        @@logger = Logger.new(STDOUT)
+        @@logger.level = Logger::INFO
+        @@logger.progname = 'vines'
+        @@logger.formatter = Class.new(Logger::Formatter) do
+          def initialize
+            @time = "%Y-%m-%dT%H:%M:%SZ".freeze
+            @fmt  = "[%s] %5s -- %s: %s\n".freeze
+          end
+          def call(severity, time, program, msg)
+            @fmt % [time.utc.strftime(@time), severity, program, msg2str(msg)]
+          end
+        end.new
+      end
+      @@logger
+    end
+  end
+end
+
+begin
+  # try to initialize diaspora AppConfig
+  require "#{Dir.pwd}/config/load_config.rb"
+rescue LoadError
+  puts "Cannot find Diaspora environment! Fallback to vines configuration."
+end
+
+%w[
+  base64
+  bcrypt
+  digest/sha1
+  em-hiredis
+  eventmachine
+  fiber
+  fileutils
+  http/parser
+  json
+  logger
+  net/ldap
+  nokogiri
+  openssl
+  optparse
+  resolv
+  set
+  socket
+  yaml
+
+  vines/cli
+  vines/log
+  vines/jid
+
+  vines/stanza
+  vines/stanza/iq
+  vines/stanza/iq/query
+  vines/stanza/iq/auth
+  vines/stanza/iq/disco_info
+  vines/stanza/iq/disco_items
+  vines/stanza/iq/error
+  vines/stanza/iq/ping
+  vines/stanza/iq/private_storage
+  vines/stanza/iq/result
+  vines/stanza/iq/roster
+  vines/stanza/iq/session
+  vines/stanza/iq/vcard
+  vines/stanza/iq/version
+  vines/stanza/message
+  vines/stanza/presence
+  vines/stanza/presence/error
+  vines/stanza/presence/probe
+  vines/stanza/presence/subscribe
+  vines/stanza/presence/subscribed
+  vines/stanza/presence/unavailable
+  vines/stanza/presence/unsubscribe
+  vines/stanza/presence/unsubscribed
+  vines/stanza/pubsub
+  vines/stanza/pubsub/create
+  vines/stanza/pubsub/delete
+  vines/stanza/pubsub/publish
+  vines/stanza/pubsub/subscribe
+  vines/stanza/pubsub/unsubscribe
+
+  vines/storage
+  vines/storage/ldap
+  vines/storage/local
+  vines/storage/sql
+  vines/storage/null
+
+  vines/config
+  vines/config/host
+  vines/config/port
+  vines/config/pubsub
+
+  vines/store
+  vines/contact
+  vines/daemon
+  vines/error
+  vines/kit
+  vines/router
+  vines/token_bucket
+  vines/user
+  vines/version
+  vines/xmpp_server
+
+  vines/cluster
+  vines/cluster/connection
+  vines/cluster/publisher
+  vines/cluster/pubsub
+  vines/cluster/sessions
+  vines/cluster/subscriber
+
+  vines/stream
+  vines/stream/sasl
+  vines/stream/state
+  vines/stream/parser
+
+  vines/stream/client
+  vines/stream/client/session
+  vines/stream/client/start
+  vines/stream/client/tls
+  vines/stream/client/auth_restart
+  vines/stream/client/auth
+  vines/stream/client/bind_restart
+  vines/stream/client/bind
+  vines/stream/client/ready
+  vines/stream/client/closed
+
+  vines/stream/component
+  vines/stream/component/start
+  vines/stream/component/handshake
+  vines/stream/component/ready
+
+  vines/stream/http
+  vines/stream/http/session
+  vines/stream/http/sessions
+  vines/stream/http/request
+  vines/stream/http/start
+  vines/stream/http/auth
+  vines/stream/http/bind_restart
+  vines/stream/http/bind
+  vines/stream/http/ready
+
+  vines/stream/server
+  vines/stream/server/start
+  vines/stream/server/tls
+  vines/stream/server/auth_restart
+  vines/stream/server/auth
+  vines/stream/server/final_restart
+  vines/stream/server/ready
+
+  vines/stream/server/outbound/start
+  vines/stream/server/outbound/tls
+  vines/stream/server/outbound/tls_result
+  vines/stream/server/outbound/auth_restart
+  vines/stream/server/outbound/auth
+  vines/stream/server/outbound/auth_result
+  vines/stream/server/outbound/final_restart
+  vines/stream/server/outbound/final_features
+
+  vines/command/bcrypt
+  vines/command/cert
+  vines/command/init
+  vines/command/ldap
+  vines/command/restart
+  vines/command/schema
+  vines/command/start
+  vines/command/stop
+].each {|f| require f }

data/test/cluster/publisher_test.rb

@@ -0,0 +1,57 @@
+# encoding: UTF-8
+
+require 'test_helper'
+
+describe Vines::Cluster::Publisher do
+  subject          { Vines::Cluster::Publisher.new(cluster) }
+  let(:connection) { MiniTest::Mock.new }
+  let(:cluster)    { MiniTest::Mock.new }
+
+  before do
+    cluster.expect :id, 'abc'
+    cluster.expect :connection, connection
+  end
+
+  describe '#broadcast' do
+    before do
+      msg = {from: 'abc', type: 'online', time: Time.now.to_i}.to_json
+      connection.expect :publish, nil, ["cluster:nodes:all", msg]
+    end
+
+    it 'publishes the message to every cluster node' do
+      subject.broadcast(:online)
+      connection.verify
+      cluster.verify
+    end
+  end
+
+  describe '#route' do
+    let(:stanza) { "<message>hello</message>" }
+
+    before do
+      msg = {from: 'abc', type: 'stanza', stanza: stanza}.to_json
+      connection.expect :publish, nil, ["cluster:nodes:node-42", msg]
+    end
+
+    it 'publishes the message to just one cluster node' do
+      subject.route(stanza, "node-42")
+      connection.verify
+      cluster.verify
+    end
+  end
+
+  describe '#update_user' do
+    let(:jid) { Vines::JID.new('alice@wonderland.lit') }
+
+    before do
+      msg = {from: 'abc', type: 'user', jid: jid.to_s}.to_json
+      connection.expect :publish, nil, ["cluster:nodes:node-42", msg]
+    end
+
+    it 'publishes the new user to just one cluster node' do
+      subject.update_user(jid, "node-42")
+      connection.verify
+      cluster.verify
+    end
+  end
+end

data/test/cluster/sessions_test.rb

@@ -0,0 +1,47 @@
+# encoding: UTF-8
+
+require 'test_helper'
+require 'storage/storage_tests'
+require 'storage/mock_redis'
+
+describe Vines::Cluster::Sessions do
+  subject          { Vines::Cluster::Sessions.new(cluster) }
+  let(:connection) { MockRedis.new }
+  let(:cluster)    { OpenStruct.new(id: 'abc', connection: connection) }
+  let(:jid1)       { 'alice@wonderland.lit/tea' }
+  let(:jid2)       { 'alice@wonderland.lit/cake' }
+
+  describe 'when saving to the cluster' do
+    it 'writes to a redis hash' do
+      StorageTests::EMLoop.new do
+        subject.save(jid1, {available: true, interested: true})
+        subject.save(jid2, {available: false, interested: false})
+        EM.next_tick do
+          session1 = {node: 'abc', available: true, interested: true}
+          session2 = {node: 'abc', available: false, interested: false}
+          connection.db["sessions:alice@wonderland.lit"].size.must_equal 2
+          connection.db["sessions:alice@wonderland.lit"]['tea'].must_equal session1.to_json
+          connection.db["sessions:alice@wonderland.lit"]['cake'].must_equal session2.to_json
+          connection.db["cluster:nodes:abc"].to_a.must_equal [jid1, jid2]
+        end
+      end
+    end
+  end
+
+  describe 'when deleting from the cluster' do
+    it 'removes from a redis hash' do
+      StorageTests::EMLoop.new do
+        connection.db["sessions:alice@wonderland.lit"] = {}
+        connection.db["sessions:alice@wonderland.lit"]['tea'] = {node: 'abc', available: true}.to_json
+        connection.db["sessions:alice@wonderland.lit"]['cake'] = {node: 'abc', available: true}.to_json
+        connection.db["cluster:nodes:abc"] = Set.new([jid1, jid2])
+
+        subject.delete(jid1)
+        EM.next_tick do
+          connection.db["sessions:alice@wonderland.lit"].size.must_equal 1
+          connection.db["cluster:nodes:abc"].to_a.must_equal [jid2]
+        end
+      end
+    end
+  end
+end

data/test/cluster/subscriber_test.rb

@@ -0,0 +1,109 @@
+# encoding: UTF-8
+
+require 'test_helper'
+
+describe Vines::Cluster::Subscriber do
+  subject          { Vines::Cluster::Subscriber.new(cluster) }
+  let(:connection) { MiniTest::Mock.new }
+  let(:cluster)    { MiniTest::Mock.new }
+  let(:now)        { Time.now.to_i }
+
+  before do
+    cluster.expect :id, 'abc'
+  end
+
+  describe '#subscribe' do
+    before do
+      cluster.expect :connect, connection
+      connection.expect :subscribe, nil, ['cluster:nodes:all']
+      connection.expect :subscribe, nil, ['cluster:nodes:abc']
+      connection.expect :on, nil, [:message]
+    end
+
+    it 'subscribes to its own channel and the broadcast channel' do
+      subject.subscribe
+      connection.verify
+      cluster.verify
+    end
+  end
+
+  describe 'when receiving a heartbeat broadcast message' do
+    before do
+      cluster.expect :poke, nil, ['node-42', now]
+    end
+
+    it 'pokes the session manager for the broadcasting node' do
+      msg = {from: 'node-42', type: 'heartbeat', time: now}.to_json
+      subject.send(:on_message, 'cluster:nodes:all', msg)
+      connection.verify
+      cluster.verify
+    end
+  end
+
+  describe 'when receiving an initial online broadcast message' do
+    before do
+      cluster.expect :poke, nil, ['node-42', now]
+    end
+
+    it 'pokes the session manager for the broadcasting node' do
+      msg = {from: 'node-42', type: 'online', time: now}.to_json
+      subject.send(:on_message, 'cluster:nodes:all', msg)
+      connection.verify
+      cluster.verify
+    end
+  end
+
+  describe 'when receiving an offline broadcast message' do
+    before do
+      cluster.expect :delete_sessions, nil, ['node-42']
+    end
+
+    it 'deletes the sessions for the broadcasting node' do
+      msg = {from: 'node-42', type: 'offline', time: now}.to_json
+      subject.send(:on_message, 'cluster:nodes:all', msg)
+      connection.verify
+      cluster.verify
+    end
+  end
+
+  describe 'when receiving a stanza routed to my node' do
+    let(:stream) { MiniTest::Mock.new }
+    let(:stanza) { "<message to='alice@wonderland.lit/tea'>hello</message>" }
+    let(:xml) { Nokogiri::XML(stanza).root }
+
+    before do
+      stream.expect :write, nil, [xml]
+      cluster.expect :connected_resources, [stream], ['alice@wonderland.lit/tea']
+    end
+
+    it 'writes the stanza to the connected user streams' do
+      msg = {from: 'node-42', type: 'stanza', stanza: stanza}.to_json
+      subject.send(:on_message, 'cluster:nodes:abc', msg)
+      stream.verify
+      connection.verify
+      cluster.verify
+    end
+  end
+
+  describe 'when receiving a user update message to my node' do
+    let(:alice) { Vines::User.new(jid: 'alice@wonderland.lit/tea') }
+    let(:storage) { MiniTest::Mock.new }
+    let(:stream) { MiniTest::Mock.new }
+
+    before do
+      storage.expect :find_user, alice, [alice.jid.bare]
+      stream.expect :user, alice
+      cluster.expect :storage, storage, ['wonderland.lit']
+      cluster.expect :connected_resources, [stream], [alice.jid.bare]
+    end
+
+    it 'reloads the user from storage and updates their connected streams' do
+      msg = {from: 'node-42', type: 'user', jid: alice.jid.to_s}.to_json
+      subject.send(:on_message, 'cluster:nodes:abc', msg)
+      storage.verify
+      stream.verify
+      connection.verify
+      cluster.verify
+    end
+  end
+end