module JSON

def self.default_options
  @default_options ||= {:symbolize_keys => false}
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.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.load(input, *args)
  FFI_Yajl::Parser.parse(input, default_options)
rescue FFI_Yajl::ParseError => e
  raise JSON::ParserError, e.message
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.pretty_generate(obj, opts=nil)
  opts ||= {}
  options_map = {}
  options_map[:pretty] = true
  options_map[:indent] = opts[:indent] if opts.has_key?(:indent)
  FFI_Yajl::Encoder.encode(obj, options_map).chomp
rescue FFI_Yajl::EncodeError => e
  raise JSON::GeneratorError, e.message
end