summaryrefslogtreecommitdiff
path: root/lib/ffi_yajl/json_gem.rb
blob: 4773f7fa8613a8db0675a2e353bd093334939429 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123

# JSON compatibility layer, largely plagarized from yajl-ruby

require 'ffi_yajl' unless defined?(FFI_Yajl::Parser)

module JSON
  class JSONError < StandardError; end unless defined?(JSON::JSONError)
  class GeneratorError < JSONError; end unless defined?(JSON::GeneratorError)
  class ParserError < JSONError; end unless defined?(JSON::ParserError)

  def self.generate(obj, opts=nil)
    opts ||= {}
    options_map = {}
    if opts.has_key?(:indent)
      options_map[:pretty] = true
      options_map[:indent] = opts[:indent]
    end
    FFI_Yajl::Encoder.encode(obj, options_map)
  rescue FFI_Yajl::EncodeError => e
    raise JSON::GeneratorError, e.message
  end

  def self.pretty_generate(obj, opts=nil)
    opts ||= {}
    options_map = {}
    options_map[:pretty] = true
    options_map[:indent] = opts[:indent] if opts.has_key?(:indent)
    require 'pp'
    pp options_map
    FFI_Yajl::Encoder.encode(obj, options_map).chomp
  rescue FFI_Yajl::EncodeError => e
    raise JSON::GeneratorError, e.message
  end

  def self.dump(obj, io=nil, *args)
    FFI_Yajl::Encoder.encode(obj, io)
  rescue FFI_Yajl::EncodeError => e
    raise JSON::GeneratorError, e.message
  end

  def self.default_options
    @default_options ||= {:symbolize_keys => false}
  end

  def self.parse(str, opts=JSON.default_options)
    FFI_Yajl::Parser.parse(str, opts)
  rescue FFI_Yajl::ParseError => e
    raise JSON::ParserError, e.message
  end

  def self.load(input, *args)
    FFI_Yajl::Parser.parse(input, default_options)
  rescue FFI_Yajl::ParseError => e
    raise JSON::ParserError, e.message
  end
end

class Array
  def to_json(*opts, &block)
    FFI_Yajl::Encoder.encode(self, *opts)
  end
end

class Hash
  def to_json(*opts, &block)
    FFI_Yajl::Encoder.encode(self, *opts)
  end
end

class Fixnum
  def to_json(*opts, &block)
    FFI_Yajl::Encoder.encode(self, *opts)
  end
end

class Float
  def to_json(*opts, &block)
    FFI_Yajl::Encoder.encode(self, *opts)
  end
end

class String
  def to_json(*opts, &block)
    FFI_Yajl::Encoder.encode(self, *opts)
  end
end

class TrueClass
  def to_json(*opts, &block)
    FFI_Yajl::Encoder.encode(self, *opts)
  end
end

class FalseClass
  def to_json(*opts, &block)
    FFI_Yajl::Encoder.encode(self, *opts)
  end
end

class NilClass
  def to_json(*opts, &block)
    FFI_Yajl::Encoder.encode(self, *opts)
  end
end

module ::Kernel
  def JSON(object, opts = {})
    if object.respond_to? :to_s
      JSON.parse(object.to_s, JSON.default_options.merge(opts))
    else
      JSON.generate(object,opts)
    end
  end
end

class Object
  unless defined?(ActiveSupport)
    def to_json(*args, &block)
      "\"#{to_s}\""
    end
  end
end