mongo-ejson 0.2.0

data/lib/ejson.rb

@@ -0,0 +1,36 @@
+require_relative '../src/ejson.rb'
+
+require 'ejson/parser_action/abstract'
+require 'ejson/parser_action/wrap'
+require 'ejson/parser_action/bson'
+
+require 'ejson/version'
+
+class EJSON
+  class << self
+    def validate(json_string)
+      ExtendedJSON.parse(json_string, actions: ParserAction::Abstract)
+    end
+
+    def parse_bson(json_string)
+      boot_bson
+
+      ExtendedJSON.parse(json_string, actions: ParserAction::BSON.new)
+    end
+
+    def parse_wrap(json_string)
+      ExtendedJSON.parse(json_string, actions: ParserAction::Wrap.new)
+    end
+
+    private
+    def boot_bson
+      return if defined? ::BSON
+      require 'bson' rescue nil
+
+      unless defined? ::BSON
+        raise "Can't find constant ::BSON. Seems like gem 'bson' is not " \
+              "loaded. Make sure it's available in your bundle."
+      end
+    end
+  end
+end

data/lib/ejson/parser_action/abstract.rb

@@ -0,0 +1,9 @@
+require 'ejson/parser_action/abstract_json'
+require 'ejson/parser_action/abstract_mongo'
+
+module ParserAction
+  class Abstract
+    include ParserAction::AbstractJSON
+    include ParserAction::AbstractMongo
+  end
+end

data/lib/ejson/parser_action/abstract_json.rb

@@ -0,0 +1,37 @@
+module ParserAction
+  module AbstractJSON
+    # JSON Entities
+    def make_root(input, start, _end, elements)
+    end
+
+    def make_object(input, start, _end, elements)
+    end
+
+    def make_pair(input, start, _end, elements)
+    end
+
+    def make_empty_object(input, start, _end, elements)
+    end
+
+    def make_string(input, start, _end, elements)
+    end
+
+    def make_array(input, start, _end, elements)
+    end
+
+    def make_empty_array(input, start, _end, elements)
+    end
+
+    def make_number(input, start, _end, elements)
+    end
+
+    def make_null(input, start, _end)
+    end
+
+    def make_true(input, start, _end)
+    end
+
+    def make_false(input, start, _end)
+    end
+  end
+end

data/lib/ejson/parser_action/abstract_mongo.rb

@@ -0,0 +1,36 @@
+module ParserAction
+  module AbstractMongo
+    def make_object_id(input, start, _end, elements)
+    end
+
+    def make_bin_data(input, start, _end, elements)
+    end
+
+    def make_timestamp(input, start, _end, elements)
+    end
+
+    def make_number_long(input, start, _end, elements)
+    end
+
+    def make_number_decimal(input, start, _end, elements)
+    end
+
+    def make_date(input, start, _end, elements)
+    end
+
+    def make_regexp(input, start, _end, elements)
+    end
+
+    def make_db_ref(input, start, _end, elements)
+    end
+
+    def make_min_key(input, start, _end)
+    end
+
+    def make_max_key(input, start, _end)
+    end
+
+    def make_undefined(input, start, _end)
+    end
+  end
+end

data/lib/ejson/parser_action/bson.rb

@@ -0,0 +1,61 @@
+require 'ejson/parser_action/ruby'
+
+module ParserAction
+  class BSON < ParserAction::Ruby
+    def make_object_id(input, start, _end, elements)
+      value = elements.first
+
+      ::BSON::ObjectId.from_string(value)
+    end
+
+    NUMBER_TO_TYPE = {
+      0    =>  :generic,
+      1    =>  :function,
+      2    =>  :old,
+      3    =>  :uuid_old,
+      4    =>  :uuid,
+      5    =>  :md5,
+      128  =>  :user,
+    }
+
+    def make_bin_data(input, start, _end, elements)
+      data = elements.last
+      type = elements.first
+
+      if type.is_a? Numeric
+        type = NUMBER_TO_TYPE[type]
+      end
+
+      ::BSON::Binary.new(data, type.to_sym)
+    end
+
+    def make_timestamp(input, start, _end, elements)
+      ::BSON::Timestamp.new(elements.first, elements.last)
+    end
+
+    def make_number_decimal(input, start, _end, elements)
+      value = elements.first
+
+      ::BSON::Decimal128.new(value.to_s)
+    end
+
+    def make_db_ref(input, start, _end, elements)
+      {
+          "$ref": elements.first,
+          "$id": elements.last,
+      }
+    end
+
+    def make_min_key(input, start, _end)
+      ::BSON::MinKey
+    end
+
+    def make_max_key(input, start, _end)
+      ::BSON::MaxKey
+    end
+
+    def make_undefined(input, start, _end)
+      ::BSON::Undefined
+    end
+  end
+end

data/lib/ejson/parser_action/ruby.rb

@@ -0,0 +1,112 @@
+require 'ejson/parser_action/abstract'
+require 'date'
+
+module ParserAction
+  class Ruby < ParserAction::Abstract
+    def make_root(input, start, _end, elements)
+      elements.first
+    end
+
+    def make_object(input, start, _end, elements)
+      first_pair = elements.first
+      other_pairs = elements.last
+
+      object = {first_pair[0] => first_pair[1]}
+      other_pairs.each do |element|
+        pair = element.pair
+        object[pair.first] = pair.last
+      end
+
+      object
+    end
+
+    def make_pair(input, start, _end, elements)
+      elements
+    end
+
+    def make_empty_object(input, start, _end, elements)
+      {}
+    end
+
+    def make_string(input, start, _end, elements)
+      %Q|"#{elements.first.text}"|.undump
+    end
+
+    def make_numeric_string(input, start, _end, elements)
+      elements.first
+    end
+
+    def make_number_as_string(input, start, _end, elements)
+      string = input[start..._end]
+      string.downcase!
+
+      string
+    end
+
+    def make_array(input, start, _end, elements)
+      list = [elements[0]]
+      elements[1].each { |el| list << el.value }
+      list
+    end
+
+    def make_empty_array(input, start, _end, elements)
+      []
+    end
+
+    def make_number(input, start, _end, elements)
+      string = input[start..._end]
+      string.downcase!
+
+      _number_from_string(string)
+    end
+
+    def make_null(input, start, _end)
+      nil
+    end
+
+    def make_true(input, start, _end)
+      true
+    end
+
+    def make_false(input, start, _end)
+      false
+    end
+
+    # Makes ruby date from mongo Data type
+    def make_date(input, start, _end, elements)
+      value = elements.first
+
+      case value
+      when String
+        DateTime.parse(value)
+      when Numeric
+        Time.at(value)
+      else
+        raise ArgumentError,
+              "Internal parser error. Unknown date atom type #{value.class}"
+      end
+    end
+
+    def make_regexp(input, start, _end, elements)
+      Regexp.new(elements.first.text, elements.last.text)
+    end
+
+    def make_number_long(input, start, _end, elements)
+      value = elements.first
+    end
+
+    private
+
+    def _number_from_string(string)
+      if string.include? 'e'
+        return Float(string)
+      end
+
+      if string.include? '.'
+        string.to_f
+      else
+        string.to_i
+      end
+    end
+  end
+end

data/lib/ejson/parser_action/wrap.rb

@@ -0,0 +1,44 @@
+require 'ejson/parser_action/ruby'
+require 'ejson/parser_action/wrap_types'
+require 'bigdecimal'
+
+module ParserAction
+  class Wrap < ParserAction::Ruby
+    def make_object_id(input, start, _end, elements)
+      value = elements.first
+
+      ::Wrap::ObjectId.new(value)
+    end
+
+    def make_bin_data(input, start, _end, elements)
+      data = elements.last
+      type = elements.first
+
+      ::Wrap::BinData.new(type, data)
+    end
+
+    def make_timestamp(input, start, _end, elements)
+      ::Wrap::Timestamp.new(elements.first, elements.last)
+    end
+
+    def make_number_decimal(input, start, _end, elements)
+      BigDecimal(elements.first)
+    end
+
+    def make_db_ref(input, start, _end, elements)
+      ::Wrap::DBRef.new(elements.first, elements.last)
+    end
+
+    def make_min_key(input, start, _end)
+      ::Wrap::MinKey
+    end
+
+    def make_max_key(input, start, _end)
+      ::Wrap::MaxKey
+    end
+
+    def make_undefined(input, start, _end)
+      ::Wrap::Undefined
+    end
+  end
+end

data/lib/ejson/parser_action/wrap_types.rb

@@ -0,0 +1,11 @@
+module Wrap
+  MinKey = Class.new
+  MaxKey = Class.new
+  Undefined = Class.new
+
+  ObjectId = Struct.new("ObjectId", :hex)
+  BinData = Struct.new("BinData", :type, :data)
+  Timestamp = Struct.new("Timestamp", :timestamp, :fraction)
+
+  DBRef = Struct.new("DBRef", :name, :id)
+end

data/lib/ejson/rails.rb

@@ -0,0 +1 @@
+require 'ejson/rails/engine'

data/lib/ejson/rails/engine.rb

@@ -0,0 +1,6 @@
+class EJSON
+  module Rails
+    class Engine < ::Rails::Engine
+    end
+  end
+end

data/lib/ejson/version.rb

@@ -0,0 +1,3 @@
+class EJSON
+  VERSION = "0.2.0"
+end

data/src/ejson.peg

@@ -0,0 +1,224 @@
+grammar ExtendedJSON
+  root                   <- @__ (object / array) @__
+                          %make_root
+
+  object                 <- (non_empty_object / empty_object)
+
+  pair                   <- @__ string @assignment value
+                          %make_pair
+
+  non_empty_object       <- @object_open
+                              pair (@delimiter pair)*
+                            @object_close
+                          %make_object
+
+  empty_object           <- @object_open
+                            @object_close
+                          %make_empty_object
+
+  array                  <- non_empty_array / empty_array
+
+  non_empty_array        <- @array_open
+                              value (@delimiter value)*
+                            @array_close
+                          %make_array
+
+  empty_array            <- @array_open
+                            @array_close
+                          %make_empty_array
+
+  value                  <- (json_values / mongo_literals / mongo_types)
+
+  json_values            <- object / array / number / string / boolean / null
+  mongo_literals         <- min_key / max_key / undefined / regexp_string
+  mongo_types            <- object_id / bin_data / timestamp / number_long /
+                            number_decimal / date / db_ref_type
+
+  # Values
+  boolean                <- (true / false)
+
+  true                   <- "true"
+                          %make_true
+
+  false                  <- "false"
+                          %make_false
+
+  null                   <- "null"
+                          %make_null
+
+  string                 <- double_quote_string / single_quote_string
+
+  double_quote_string    <- @double_quote
+                              ("\\" . / [^"])*
+                            @double_quote
+                          %make_string
+
+  single_quote_string    <- @single_quote
+                              ("\\" . / [^'])*
+                            @single_quote
+                          %make_string
+
+  hex_string             <- hex_single_quote / hex_double_quote
+  hex_value              <- [a-fA-F0-9]+
+
+  hex_single_quote       <- @single_quote
+                              hex_value
+                            @single_quote
+                          %make_string
+
+  hex_double_quote       <- @double_quote
+                              hex_value
+                            @double_quote
+                          %make_string
+
+  integer_string         <- integer_string_single / integer_string_double
+
+  integer_string_single   <- @single_quote
+                               integer_number
+                             @single_quote
+                          %make_numeric_string
+
+  integer_string_double   <- @double_quote
+                               integer_number
+                             @double_quote
+                          %make_numeric_string
+
+  number_string          <- number_string_single / number_string_double
+
+  number_string_single   <- @single_quote
+                              number
+                            @single_quote
+                          %make_numeric_string
+
+  number_string_double   <- @double_quote
+                              number
+                            @double_quote
+                          %make_numeric_string
+
+  regexp_string          <- @slash
+                              ("\\" . / [^\/])*
+                            @slash
+                            regexp_options?
+                          %make_regexp
+
+  regexp_options         <- [gims]+
+
+  base64_string          <- base64_single_quote / base64_dobule_quote
+
+  base64_value           <- [a-zA-Z0-9+\/]+ ("=" / "==")?
+
+  base64_single_quote    <- @single_quote
+                              base64_value
+                            @single_quote
+                          %make_string
+
+  base64_dobule_quote    <- @double_quote
+                              base64_value
+                            @double_quote
+                          %make_string
+
+  number                 <- "-"? integer fraction? exponent?
+                          %make_number
+
+  number_as_string       <- "-"? integer fraction? exponent?
+                          %make_number_as_string
+
+  integer                <- ("0" / [1-9] [0-9]*)
+  fraction               <- ("." [0-9]+)
+  exponent               <- (("e" / "E") ("+" / "-" / "") [0-9]+)
+
+  integer_number         <- integer @""
+                          %make_number
+
+  # Mongo Extensions
+
+  min_key                <- "MinKey"
+                          %make_min_key
+
+  max_key                <- "MaxKey"
+                          %make_max_key
+
+  undefined              <- "undefined"
+                          %make_undefined
+
+  object_id              <- @"ObjectId"
+                            @type_open
+                              hex_string
+                            @type_close
+                          %make_object_id
+
+  bin_data_type          <- string / integer_number
+  bin_data               <- @"BinData"
+                            @type_open
+                              bin_data_type
+                            @delimiter
+                              base64_string
+                            @type_close
+                          %make_bin_data
+
+  timestamp              <- @"Timestamp"
+                            @type_open
+                              integer_number
+                            @delimiter
+                              integer_number
+                            @type_close
+                          %make_timestamp
+
+  number_long_value      <- integer_number / integer_string
+  number_long            <- @"NumberLong"
+                            @type_open
+                              number_long_value
+                            @type_close
+                          %make_number_long
+
+
+  number_decimal_value   <- number_as_string / number_string
+  number_decimal         <- @"NumberDecimal"
+                            @type_open
+                              number_decimal_value
+                            @type_close
+                          %make_number_decimal
+
+  date_type              <- ("new" space+ "Date" / "ISODate")
+  date_value             <- integer_number / string
+  date                   <- @date_type
+                            @type_open
+                              date_value
+                            @type_close
+                          %make_date
+
+  db_ref_type            <- @"DBRef"
+                            @type_open
+                              string
+                            @delimiter
+                              string
+                            @type_close
+                          %make_db_ref
+
+  single_quote           <- "'"
+  double_quote           <- '"'
+  left_paren             <- '('
+  right_paren            <- ')'
+  left_brace             <- '{'
+  right_brace            <- '}'
+  left_bracket           <- '['
+  right_bracket          <- ']'
+  comma                  <- ','
+  colon                  <- ':'
+  slash                  <- '/'
+  space                  <- [\s]
+  new_line               <- [\n]
+
+  comment                <- "//" [^\n]*
+  __                     <- (space / new_line / comment)*
+
+  delimiter              <- @__ comma @__
+  assignment             <- @__ @colon @__
+
+  object_open            <- @__ @left_brace @__
+  object_close           <- @__ @right_brace @__
+  array_open             <- @__ @left_bracket @__
+  array_close            <- @__ @right_bracket @__
+
+  type_open              <- @space* @left_paren @space*
+  type_close             <- @space* @right_paren