vines-mongodb 0.1.0

data/Gemfile

@@ -0,0 +1,3 @@
+source 'https://rubygems.org'
+
+gemspec

data/LICENSE

@@ -0,0 +1,19 @@
+Copyright (c) 2010-2013 Negative Code
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.md

@@ -0,0 +1,53 @@
+# Welcome to Vines
+
+Vines is a scalable XMPP chat server, using EventMachine for asynchronous IO.
+This gem provides support for storing user data in
+[MongoDB](http://www.mongodb.org/).
+
+Additional documentation can be found at [getvines.org](http://www.getvines.org/).
+
+## Usage
+
+```
+$ gem install vines vines-mongodb
+$ vines init wonderland.lit
+$ cd wonderland.lit && vines start
+```
+
+## Configuration
+
+Add the following configuration block to a virtual host definition in
+the server's `conf/config.rb` file.
+
+```ruby
+storage 'mongodb' do
+  host 'localhost', 27017
+  host 'localhost', 27018 # optional, connects to replica set
+  database 'xmpp'
+  tls true
+  username ''
+  password ''
+  pool 5
+end
+```
+
+## Dependencies
+
+Vines requires Ruby 1.9.3 or better. Instructions for installing the
+needed OS packages, as well as Ruby itself, are available at
+[getvines.org/ruby](http://www.getvines.org/ruby).
+
+## Development
+
+```
+$ script/bootstrap
+$ script/tests
+```
+
+## Contact
+
+* David Graham <david@negativecode.com>
+
+## License
+
+Vines is released under the MIT license. Check the LICENSE file for details.

data/Rakefile

@@ -0,0 +1,20 @@
+require 'rake'
+require 'rake/clean'
+require 'rake/testtask'
+
+CLOBBER.include('pkg')
+
+directory 'pkg'
+
+desc 'Build distributable packages'
+task :build => [:pkg] do
+  system 'gem build vines-mongodb.gemspec && mv vines-*.gem pkg/'
+end
+
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'spec'
+  test.pattern = 'spec/**/*_spec.rb'
+  test.warning = false
+end
+
+task :default => [:clobber, :test, :build]

data/lib/vines/storage/mongodb.rb

@@ -0,0 +1,132 @@
+require 'mongo'
+
+module Vines
+  class Storage
+    class MongoDB < Storage
+      register :mongodb
+
+      %w[database tls username password pool].each do |name|
+        define_method(name) do |*args|
+          if args.first
+            @config[name.to_sym] = args.first
+          else
+            @config[name.to_sym]
+          end
+        end
+      end
+
+      def initialize(&block)
+        @config, @hosts = {}, []
+        instance_eval(&block)
+        raise "Must provide database" unless @config[:database]
+        raise "Must provide at least one host connection" if @hosts.empty?
+      end
+
+      def host(name, port)
+        pair = [name, port]
+        raise "duplicate hosts not allowed: #{name}:#{port}" if @hosts.include?(pair)
+        @hosts << pair
+      end
+
+      def find_user(jid)
+        jid = JID.new(jid).bare.to_s
+        return if jid.empty?
+        doc = get(:users, jid)
+        return unless doc
+        User.new(jid: jid).tap do |user|
+          user.name, user.password = doc.values_at('name', 'password')
+          (doc['roster'] || {}).each_pair do |jid, props|
+            user.roster << Contact.new(
+              jid: jid,
+              name: props['name'],
+              subscription: props['subscription'],
+              ask: props['ask'],
+              groups: props['groups'] || [])
+          end
+        end
+      end
+
+      def save_user(user)
+        id = user.jid.bare.to_s
+        doc = get(:users, id) || {'_id' => id}
+        doc['name'] = user.name
+        doc['password'] = user.password
+        doc['roster'] = {}
+        user.roster.each do |contact|
+          doc['roster'][contact.jid.bare.to_s] = contact.to_h
+        end
+        save_doc(:users, doc)
+      end
+
+      def find_vcard(jid)
+        jid = JID.new(jid).bare.to_s
+        return if jid.empty?
+        doc = get(:vcards, jid)
+        return unless doc
+        Nokogiri::XML(doc['card']).root rescue nil
+      end
+
+      def save_vcard(jid, card)
+        jid = JID.new(jid).bare.to_s
+        doc = get(:vcards, jid) || {'_id' => jid}
+        doc['card'] = card.to_xml
+        save_doc(:vcards, doc)
+      end
+
+      def find_fragment(jid, node)
+        jid = JID.new(jid).bare.to_s
+        return if jid.empty?
+        doc = get(:fragments, fragment_id(jid, node))
+        return unless doc
+        Nokogiri::XML(doc['xml']).root rescue nil
+      end
+
+      def save_fragment(jid, node)
+        jid = JID.new(jid).bare.to_s
+        id = fragment_id(jid, node)
+        doc = get(:fragments, id) || {'_id' => id}
+        doc['xml'] = node.to_xml
+        save_doc(:fragments, doc)
+      end
+
+      private
+
+      def fragment_id(jid, node)
+        id = Digest::SHA1.hexdigest("#{node.name}:#{node.namespace.href}")
+        "#{jid}:#{id}"
+      end
+
+      def get(collection, id)
+        db.collection(collection).find_one({_id: id})
+      end
+      defer :get
+
+      def save_doc(collection, doc)
+        db.collection(collection).save(doc, safe: true)
+      end
+      defer :save_doc
+
+      def db
+        @db ||= connect
+      end
+
+      def connect
+        opts = {
+          pool_timeout: 5,
+          pool_size: @config[:pool] || 5,
+          ssl: @config[:tls]
+        }
+        conn = if @hosts.size == 1
+          Mongo::Connection.new(@hosts.first[0], @hosts.first[1], opts)
+        else
+          Mongo::ReplSetConnection.new(*@hosts, opts)
+        end
+        conn.db(@config[:database]).tap do |db|
+          user = @config[:username] || ''
+          pass = @config[:password] || ''
+          db.authenticate(user, pass) unless user.empty? || pass.empty?
+        end
+      end
+    end
+  end
+end

data/spec/mock_mongo.rb

@@ -0,0 +1,38 @@
+# A mock implementation of the Mongo::Connection class.
+class MockMongo
+  def initialize
+    @collections = collections
+  end
+
+  def collection(name)
+    @collections[name]
+  end
+
+  def clear
+    @collections = collections
+  end
+
+  def collections
+    {
+      users: MockCollection.new,
+      vcards: MockCollection.new,
+      fragments: MockCollection.new
+    }
+  end
+
+  class MockCollection
+    def initialize
+      @docs = {}
+    end
+
+    def find_one(doc)
+      id = doc['_id'] || doc[:_id]
+      @docs[id]
+    end
+
+    def save(doc, opts={})
+      id = doc['_id'] || doc[:_id]
+      @docs[id] = doc
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,4 @@
+require 'vines'
+require 'mock_mongo'
+require 'storage_specs'
+require 'minitest/autorun'

data/spec/storage/mongodb_spec.rb

@@ -0,0 +1,98 @@
+require 'spec_helper'
+
+describe Vines::Storage::MongoDB do
+  include StorageSpecs
+
+  MOCK_MONGO = MockMongo.new
+
+  before do
+    fibered do
+      db = MOCK_MONGO
+      db.collection(:users).save({'_id' => 'empty@wonderland.lit'})
+      db.collection(:users).save({'_id' => 'no_password@wonderland.lit', 'foo' => 'bar'})
+      db.collection(:users).save({'_id' => 'clear_password@wonderland.lit', 'password' => 'secret'})
+      db.collection(:users).save({'_id' => 'bcrypt_password@wonderland.lit', 'password' => BCrypt::Password.create('secret')})
+      db.collection(:users).save({
+        '_id'      => 'full@wonderland.lit',
+        'password' => BCrypt::Password.create('secret'),
+        'name'     => 'Tester',
+        'roster'   => {
+          'contact1@wonderland.lit' => {
+            'name'   => 'Contact1',
+            'groups' => %w[Group1 Group2]
+          },
+          'contact2@wonderland.lit' => {
+            'name'   => 'Contact2',
+            'groups' => %w[Group3 Group4]
+          }
+        }
+      })
+      db.collection(:vcards).save({'_id' => 'full@wonderland.lit', 'card' => vcard.to_xml})
+      db.collection(:fragments).save({'_id' => "full@wonderland.lit:#{fragment_id}", 'xml' => fragment.to_xml})
+    end
+  end
+
+  after do
+    MOCK_MONGO.clear
+  end
+
+  def storage
+    storage = Vines::Storage::MongoDB.new do
+      host 'localhost', 27017
+      database 'xmpp_testcase'
+    end
+    def storage.db
+      MOCK_MONGO
+    end
+    storage
+  end
+
+  describe 'creating a new instance' do
+    it 'raises with no database' do
+      fibered do
+        -> { Vines::Storage::MongoDB.new {} }.must_raise RuntimeError
+      end
+    end
+
+    it 'raises with no host' do
+      fibered do
+        -> { Vines::Storage::MongoDB.new { database 'test' } }.must_raise RuntimeError
+      end
+    end
+
+    it 'raises with duplicate hosts' do
+      fibered do
+        proc do
+          Vines::Storage::MongoDB.new do
+            host 'localhost', 27017
+            host 'localhost', 27017
+            database 'test'
+          end
+        end.must_raise RuntimeError
+      end
+    end
+
+    it 'does not raise with host and database' do
+      fibered do
+        obj =
+          Vines::Storage::MongoDB.new do
+            host 'localhost', 27017
+            database 'test'
+          end
+        obj.wont_be_nil
+      end
+    end
+
+    it 'does not raise with two hosts' do
+      fibered do
+        obj =
+          Vines::Storage::MongoDB.new do
+            host 'localhost', 27017
+            host 'localhost', 27018
+            database 'test'
+          end
+        obj.wont_be_nil
+      end
+    end
+  end
+end

data/spec/storage_specs.rb

@@ -0,0 +1,182 @@
+# Mixin methods for storage implementation test classes. The behavioral
+# tests are the same regardless of implementation so share those methods
+# here.
+module StorageSpecs
+  def fragment_id
+    Digest::SHA1.hexdigest("characters:urn:wonderland")
+  end
+
+  def fragment
+    Nokogiri::XML(%q{
+      <characters xmlns="urn:wonderland">
+        <character>Alice</character>
+      </characters>
+    }.strip).root
+  end
+
+  def vcard
+    Nokogiri::XML(%q{
+      <vCard xmlns="vcard-temp">
+        <FN>Alice in Wonderland</FN>
+      </vCard>
+    }.strip).root
+  end
+
+  def fibered
+    EM.run do
+      Fiber.new do
+        yield
+        EM.stop
+      end.resume
+    end
+  end
+
+  def test_authenticate
+    fibered do
+      db = storage
+      db.authenticate(nil, nil).must_be_nil
+      db.authenticate(nil, 'secret').must_be_nil
+      db.authenticate('bogus', nil).must_be_nil
+      db.authenticate('bogus', 'secret').must_be_nil
+      db.authenticate('empty@wonderland.lit', 'secret').must_be_nil
+      db.authenticate('no_password@wonderland.lit', 'secret').must_be_nil
+      db.authenticate('clear_password@wonderland.lit', 'secret').must_be_nil
+
+      user = db.authenticate('bcrypt_password@wonderland.lit', 'secret')
+      user.wont_be_nil
+      user.jid.to_s.must_equal 'bcrypt_password@wonderland.lit'
+
+      user = db.authenticate('full@wonderland.lit', 'secret')
+      user.wont_be_nil
+      user.name.must_equal 'Tester'
+      user.jid.to_s.must_equal 'full@wonderland.lit'
+
+      user.roster.length.must_equal 2
+      user.roster[0].jid.to_s.must_equal 'contact1@wonderland.lit'
+      user.roster[0].name.must_equal 'Contact1'
+      user.roster[0].groups.length.must_equal 2
+      user.roster[0].groups[0].must_equal 'Group1'
+      user.roster[0].groups[1].must_equal 'Group2'
+
+      user.roster[1].jid.to_s.must_equal 'contact2@wonderland.lit'
+      user.roster[1].name.must_equal 'Contact2'
+      user.roster[1].groups.length.must_equal 2
+      user.roster[1].groups[0].must_equal 'Group3'
+      user.roster[1].groups[1].must_equal 'Group4'
+    end
+  end
+
+  def test_find_user
+    fibered do
+      db = storage
+      user = db.find_user(nil)
+      user.must_be_nil
+
+      user = db.find_user('full@wonderland.lit')
+      user.wont_be_nil
+      user.jid.to_s.must_equal 'full@wonderland.lit'
+
+      user = db.find_user(Vines::JID.new('full@wonderland.lit'))
+      user.wont_be_nil
+      user.jid.to_s.must_equal 'full@wonderland.lit'
+
+      user = db.find_user(Vines::JID.new('full@wonderland.lit/resource'))
+      user.wont_be_nil
+      user.jid.to_s.must_equal 'full@wonderland.lit'
+    end
+  end
+
+  def test_save_user
+    fibered do
+      db = storage
+      user = Vines::User.new(
+        jid: 'save_user@domain.tld/resource1',
+        name: 'Save User',
+        password: 'secret')
+      user.roster << Vines::Contact.new(
+        jid: 'contact1@domain.tld/resource2',
+        name: 'Contact 1')
+      db.save_user(user)
+      user = db.find_user('save_user@domain.tld')
+      user.wont_be_nil
+      user.jid.to_s.must_equal 'save_user@domain.tld'
+      user.name.must_equal 'Save User'
+      user.roster.length.must_equal 1
+      user.roster[0].jid.to_s.must_equal 'contact1@domain.tld'
+      user.roster[0].name.must_equal 'Contact 1'
+    end
+  end
+
+  def test_find_vcard
+    fibered do
+      db = storage
+      card = db.find_vcard(nil)
+      card.must_be_nil
+
+      card = db.find_vcard('full@wonderland.lit')
+      card.wont_be_nil
+      card.to_s.must_equal vcard.to_s
+
+      card = db.find_vcard(Vines::JID.new('full@wonderland.lit'))
+      card.wont_be_nil
+      card.to_s.must_equal vcard.to_s
+
+      card = db.find_vcard(Vines::JID.new('full@wonderland.lit/resource'))
+      card.wont_be_nil
+      card.to_s.must_equal vcard.to_s
+    end
+  end
+
+  def test_save_vcard
+    fibered do
+      db = storage
+      db.save_user(Vines::User.new(jid: 'save_user@domain.tld'))
+      db.save_vcard('save_user@domain.tld/resource1', vcard)
+      card = db.find_vcard('save_user@domain.tld')
+      card.wont_be_nil
+      card.to_s.must_equal vcard.to_s
+    end
+  end
+
+  def test_find_fragment
+    fibered do
+      db = storage
+      root = Nokogiri::XML(%q{<characters xmlns="urn:wonderland"/>}).root
+      bad_name = Nokogiri::XML(%q{<not_characters xmlns="urn:wonderland"/>}).root
+      bad_ns = Nokogiri::XML(%q{<characters xmlns="not:wonderland"/>}).root
+
+      node = db.find_fragment(nil, nil)
+      node.must_be_nil
+
+      node = db.find_fragment('full@wonderland.lit', bad_name)
+      node.must_be_nil
+
+      node = db.find_fragment('full@wonderland.lit', bad_ns)
+      node.must_be_nil
+
+      node = db.find_fragment('full@wonderland.lit', root)
+      node.wont_be_nil
+      node.to_s.must_equal fragment.to_s
+
+      node = db.find_fragment(Vines::JID.new('full@wonderland.lit'), root)
+      node.wont_be_nil
+      node.to_s.must_equal fragment.to_s
+
+      node = db.find_fragment(Vines::JID.new('full@wonderland.lit/resource'), root)
+      node.wont_be_nil
+      node.to_s.must_equal fragment.to_s
+    end
+  end
+
+  def test_save_fragment
+    fibered do
+      db = storage
+      root = Nokogiri::XML(%q{<characters xmlns="urn:wonderland"/>}).root
+      db.save_user(Vines::User.new(jid: 'save_user@domain.tld'))
+      db.save_fragment('save_user@domain.tld/resource1', fragment)
+      node = db.find_fragment('save_user@domain.tld', root)
+      node.wont_be_nil
+      node.to_s.must_equal fragment.to_s
+    end
+  end
+end

data/vines-mongodb.gemspec

@@ -0,0 +1,24 @@
+Gem::Specification.new do |s|
+  s.name         = 'vines-mongodb'
+  s.version      = '0.1.0'
+  s.summary      = %q[Provides a MongoDB storage adapter for the Vines XMPP chat server.]
+  s.description  = %q[Stores Vines user data in MongoDB.]
+
+  s.authors      = ['David Graham']
+  s.email        = %w[david@negativecode.com]
+  s.homepage     = 'http://www.getvines.org'
+  s.license      = 'MIT'
+
+  s.files        = Dir['[A-Z]*', 'vines-mongodb.gemspec', 'lib/**/*'] - ['Gemfile.lock']
+  s.test_files   = Dir['spec/**/*']
+  s.require_path = 'lib'
+
+  s.add_dependency 'mongo', '~> 1.5.2'
+  s.add_dependency 'bson_ext', '~> 1.5.2'
+  s.add_dependency 'vines', '>= 0.4.5'
+
+  s.add_development_dependency 'minitest', '~> 4.7.4'
+  s.add_development_dependency 'rake', '~> 10.1.0'
+
+  s.required_ruby_version = '>= 1.9.3'
+end

metadata

@@ -0,0 +1,143 @@
+--- !ruby/object:Gem::Specification
+name: vines-mongodb
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- David Graham
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-06-21 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: mongo
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.5.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.5.2
+- !ruby/object:Gem::Dependency
+  name: bson_ext
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.5.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.5.2
+- !ruby/object:Gem::Dependency
+  name: vines
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.4.5
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.4.5
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 4.7.4
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 4.7.4
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 10.1.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 10.1.0
+description: Stores Vines user data in MongoDB.
+email:
+- david@negativecode.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- Gemfile
+- LICENSE
+- Rakefile
+- README.md
+- vines-mongodb.gemspec
+- lib/vines/storage/mongodb.rb
+- spec/mock_mongo.rb
+- spec/spec_helper.rb
+- spec/storage/mongodb_spec.rb
+- spec/storage_specs.rb
+homepage: http://www.getvines.org
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: 1.9.3
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: 1526150033927747490
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Provides a MongoDB storage adapter for the Vines XMPP chat server.
+test_files:
+- spec/mock_mongo.rb
+- spec/spec_helper.rb
+- spec/storage/mongodb_spec.rb
+- spec/storage_specs.rb