animehunter-mongo 0.9

data/lib/mongo/util/byte_buffer.rb

@@ -0,0 +1,167 @@
+# --
+# Copyright (C) 2008-2009 10gen Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ++
+
+# A byte buffer.
+class ByteBuffer
+
+  attr_reader :order
+
+  def initialize(initial_data=[])
+    @buf = initial_data
+    @cursor = 0
+    self.order = :little_endian
+  end
+
+  # +endianness+ should be :little_endian or :big_endian. Default is :little_endian
+  def order=(endianness)
+    @order = endianness
+    @int_pack_order = endianness == :little_endian ? 'V' : 'N'
+    @double_pack_order = endianness == :little_endian ? 'E' : 'G'
+  end
+
+  def rewind
+    @cursor = 0
+  end
+
+  def position
+    @cursor
+  end
+
+  def position=(val)
+    @cursor = val
+  end
+
+  def clear
+    @buf = []
+    rewind
+  end
+
+  def size
+    @buf.size
+  end
+  alias_method :length, :size
+
+  def put(byte, offset=nil)
+    @cursor = offset if offset
+    @buf[@cursor] = byte
+    @cursor += 1
+  end
+
+  def put_array(array, offset=nil)
+    @cursor = offset if offset
+    @buf[@cursor, array.length] = array
+    @cursor += array.length
+  end
+
+  def put_int(i, offset=nil)
+    a = []
+    [i].pack(@int_pack_order).each_byte { |b| a << b }
+    put_array(a, offset)
+  end
+
+  def put_long(i, offset=nil)
+    offset = @cursor unless offset
+    if @int_pack_order == 'N'
+      put_int(i >> 32, offset)
+      put_int(i & 0xffffffff, offset + 4)
+    else
+      put_int(i & 0xffffffff, offset)
+      put_int(i >> 32, offset + 4)
+    end
+  end
+
+  def put_double(d, offset=nil)
+    a = []
+    [d].pack(@double_pack_order).each_byte { |b| a << b }
+    put_array(a, offset)
+  end
+
+  # If +size+ == nil, returns one byte. Else returns array of bytes of length
+  # # +size+.
+  def get(len=nil)
+    one_byte = len.nil?
+    len ||= 1
+    check_read_length(len)
+    start = @cursor
+    @cursor += len
+    if one_byte
+      @buf[start]
+    else
+      if @buf.respond_to? "unpack"
+        @buf[start, len].unpack("C*")
+      else
+        @buf[start, len]
+      end
+    end
+  end
+
+  def get_int
+    check_read_length(4)
+    vals = ""
+    (@cursor..@cursor+3).each { |i| vals << @buf[i].chr }
+    @cursor += 4
+    vals.unpack(@int_pack_order)[0]
+  end
+
+  def get_long
+    i1 = get_int
+    i2 = get_int
+    if @int_pack_order == 'N'
+      (i1 << 32) + i2
+    else
+      (i2 << 32) + i1
+    end
+  end
+
+  def get_double
+    check_read_length(8)
+    vals = ""
+    (@cursor..@cursor+7).each { |i| vals << @buf[i].chr }
+    @cursor += 8
+    vals.unpack(@double_pack_order)[0]
+  end
+
+  def more?
+    @cursor < @buf.size
+  end
+
+  def to_a
+    if @buf.respond_to? "unpack"
+      @buf.unpack("C*")
+    else
+      @buf
+    end
+  end
+
+  def to_s
+    if @buf.respond_to? "pack"
+      @buf.pack("C*")
+    else
+      @buf
+    end
+  end
+
+  def dump
+    @buf.each_with_index { |c, i| $stderr.puts "#{'%04d' % i}: #{'%02x' % c} #{'%03o' % c} #{'%s' % c.chr} #{'%3d' % c}" }
+  end
+
+  private
+
+  def check_read_length(len)
+    raise "attempt to read past end of buffer" if @cursor + len > @buf.length
+  end
+
+end

data/lib/mongo/util/ordered_hash.rb

@@ -0,0 +1,96 @@
+# --
+# Copyright (C) 2008-2009 10gen Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ++
+
+# A hash in which the order of keys are preserved.
+#
+# Under Ruby 1.9 and greater, this class has no added methods because Ruby's
+# Hash already keeps its keys ordered by order of insertion.
+class OrderedHash < Hash
+
+  def ==(other)
+    begin
+      !other.nil? &&
+        keys == other.keys &&
+        values == other.values
+    rescue
+      false
+    end
+  end
+
+  # We only need the body of this class if the RUBY_VERSION is before 1.9
+  if RUBY_VERSION < '1.9'
+
+  attr_accessor :ordered_keys
+
+  def keys
+    @ordered_keys || []
+  end
+
+  def []=(key, value)
+    @ordered_keys ||= []
+    @ordered_keys << key unless @ordered_keys.include?(key)
+    super(key, value)
+  end
+
+  def each
+    @ordered_keys ||= []
+    @ordered_keys.each { |k| yield k, self[k] }
+  end
+
+  def values
+    collect { |k, v| v }
+  end
+
+  def merge(other)
+    oh = self.dup
+    oh.merge!(other)
+    oh
+  end
+
+  def merge!(other)
+    @ordered_keys ||= []
+    @ordered_keys += other.keys # unordered if not an OrderedHash
+    @ordered_keys.uniq!
+    super(other)
+  end
+
+  def inspect
+    str = '{'
+    str << (@ordered_keys || []).collect { |k| "\"#{k}\"=>#{self.[](k).inspect}" }.join(", ")
+    str << '}'
+  end
+
+  def delete(key, &block)
+    @ordered_keys.delete(key) if @ordered_keys
+    super
+  end
+
+  def delete_if(&block)
+    self.each { |k,v|
+      if yield k, v
+        delete(k)
+      end
+    }
+  end
+
+  def clear
+    super
+    @ordered_keys = []
+  end
+
+  end                           # Ruby before 1.9
+
+end

data/lib/mongo/util/xml_to_ruby.rb

@@ -0,0 +1,107 @@
+# --
+# Copyright (C) 2008-2009 10gen Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ++
+
+require 'rexml/document'
+require 'mongo'
+
+# Converts a .xson file (an XML file that describes a Mongo-type document) to
+# an OrderedHash.
+class XMLToRuby
+
+  include XGen::Mongo::Driver
+
+  def xml_to_ruby(io)
+    doc = REXML::Document.new(io)
+    doc_to_ruby(doc.root.elements['doc'])
+  end
+
+  protected
+
+  def element_to_ruby(e)
+    type = e.name
+    child = e.elements[1]
+    case type
+    when 'oid'
+      ObjectID.from_string(e.text)
+    when 'ref'
+      dbref_to_ruby(e.elements)
+    when 'int'
+      e.text.to_i
+    when 'number'
+      e.text.to_f
+    when 'string'
+      e.text.to_s
+    when 'code'
+      Code.new(e.text.to_s)
+    when 'binary'
+      bin = Binary.new
+      decoded = Base64.decode64(e.text.to_s)
+      decoded.each_byte { |b| bin.put(b) }
+      bin
+    when 'symbol'
+      e.text.to_s.intern
+    when 'boolean'
+      e.text.to_s == 'true'
+    when 'array'
+      array_to_ruby(e.elements)
+    when 'date'
+      Time.at(e.text.to_f / 1000.0)
+    when 'regex'
+      regex_to_ruby(e.elements)
+    when 'null'
+      nil
+    when 'undefined'
+      Undefined.new
+    when 'doc'
+      doc_to_ruby(e)
+    else
+      raise "Unknown type #{type} in element with name #{e.attributes['name']}"
+    end
+  end
+
+  def doc_to_ruby(element)
+    oh = OrderedHash.new
+    element.elements.each { |e| oh[e.attributes['name']] = element_to_ruby(e) }
+    oh
+  end
+
+  def array_to_ruby(elements)
+    a = []
+    elements.each { |e|
+      index_str = e.attributes['name']
+      a[index_str.to_i] = element_to_ruby(e)
+    }
+    a
+  end
+
+  def regex_to_ruby(elements)
+    pattern = elements['pattern'].text
+    options_str = elements['options'].text || ''
+
+    options = 0
+    options |= Regexp::IGNORECASE if options_str.include?('i')
+    options |= Regexp::MULTILINE if options_str.include?('m')
+    options |= Regexp::EXTENDED if options_str.include?('x')
+    Regexp.new(pattern, options)
+  end
+
+  def dbref_to_ruby(elements)
+    ns = elements['ns'].text
+    oid_str = elements['oid'].text
+    DBRef.new(ns, ObjectID.from_string(oid_str))
+  end
+
+end

data/mongo-ruby-driver.gemspec

@@ -0,0 +1,99 @@
+# We need to list all of the included files because we aren't allowed to use
+# Dir[...] in the github sandbox.
+PACKAGE_FILES = ['README.rdoc', 'Rakefile', 'mongo-ruby-driver.gemspec',
+                 'bin/bson_benchmark.rb',
+                 'bin/mongo_console',
+                 'bin/run_test_script',
+                 'bin/standard_benchmark',
+                 'examples/admin.rb',
+                 'examples/benchmarks.rb',
+                 'examples/blog.rb',
+                 'examples/capped.rb',
+                 'examples/cursor.rb',
+                 'examples/gridfs.rb',
+                 'examples/index_test.rb',
+                 'examples/info.rb',
+                 'examples/queries.rb',
+                 'examples/simple.rb',
+                 'examples/strict.rb',
+                 'examples/types.rb',
+                 'lib/mongo/admin.rb',
+                 'lib/mongo/collection.rb',
+                 'lib/mongo/cursor.rb',
+                 'lib/mongo/db.rb',
+                 'lib/mongo/gridfs/chunk.rb',
+                 'lib/mongo/gridfs/grid_store.rb',
+                 'lib/mongo/gridfs.rb',
+                 'lib/mongo/message/get_more_message.rb',
+                 'lib/mongo/message/insert_message.rb',
+                 'lib/mongo/message/kill_cursors_message.rb',
+                 'lib/mongo/message/message.rb',
+                 'lib/mongo/message/message_header.rb',
+                 'lib/mongo/message/msg_message.rb',
+                 'lib/mongo/message/opcodes.rb',
+                 'lib/mongo/message/query_message.rb',
+                 'lib/mongo/message/remove_message.rb',
+                 'lib/mongo/message/update_message.rb',
+                 'lib/mongo/message.rb',
+                 'lib/mongo/mongo.rb',
+                 'lib/mongo/query.rb',
+                 'lib/mongo/types/binary.rb',
+                 'lib/mongo/types/code.rb',
+                 'lib/mongo/types/dbref.rb',
+                 'lib/mongo/types/objectid.rb',
+                 'lib/mongo/types/regexp_of_holding.rb',
+                 'lib/mongo/types/undefined.rb',
+                 'lib/mongo/util/bson.rb',
+                 'lib/mongo/util/byte_buffer.rb',
+                 'lib/mongo/util/ordered_hash.rb',
+                 'lib/mongo/util/xml_to_ruby.rb',
+                 'lib/mongo.rb']
+TEST_FILES = ['tests/mongo-qa/_common.rb',
+              'tests/mongo-qa/admin',
+              'tests/mongo-qa/capped',
+              'tests/mongo-qa/count1',
+              'tests/mongo-qa/dbs',
+              'tests/mongo-qa/find',
+              'tests/mongo-qa/find1',
+              'tests/mongo-qa/gridfs_in',
+              'tests/mongo-qa/gridfs_out',
+              'tests/mongo-qa/indices',
+              'tests/mongo-qa/remove',
+              'tests/mongo-qa/stress1',
+              'tests/mongo-qa/test1',
+              'tests/mongo-qa/update',
+              'tests/test_admin.rb',
+              'tests/test_bson.rb',
+              'tests/test_byte_buffer.rb',
+              'tests/test_chunk.rb',
+              'tests/test_cursor.rb',
+              'tests/test_db.rb',
+              'tests/test_db_api.rb',
+              'tests/test_db_connection.rb',
+              'tests/test_grid_store.rb',
+              'tests/test_message.rb',
+              'tests/test_mongo.rb',
+              'tests/test_objectid.rb',
+              'tests/test_ordered_hash.rb',
+              'tests/test_round_trip.rb']
+
+Gem::Specification.new do |s|
+  s.name = 'mongo'
+  s.version = '0.9'
+  s.platform = Gem::Platform::RUBY
+  s.summary = 'Ruby driver for the 10gen Mongo DB'
+  s.description = 'A Ruby driver for the 10gen Mongo DB. For more information about Mongo, see http://www.mongodb.org.'
+
+  s.require_paths = ['lib']
+
+  s.files = PACKAGE_FILES
+  s.test_files = TEST_FILES
+
+  s.has_rdoc = true
+  s.rdoc_options = ['--main', 'README.rdoc', '--inline-source']
+  s.extra_rdoc_files = ['README.rdoc']
+
+  s.authors = ['Jim Menard', 'Mike Dirolf']
+  s.email = 'mongodb-dev@googlegroups.com'
+  s.homepage = 'http://www.mongodb.org'
+end