json.rs 124.8 KB
Newer Older
M
mrec 已提交
1
// Copyright 2012-2014 The Rust Project Developers. See the COPYRIGHT
2 3 4 5 6 7 8 9 10
// file at the top-level directory of this distribution and at
// http://rust-lang.org/COPYRIGHT.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

E
Elly Jones 已提交
11 12
// Rust JSON serialization library
// Copyright (c) 2011 Google Inc.
13

14
#![forbid(non_camel_case_types)]
A
Aaron Turon 已提交
15
#![allow(missing_docs)]
E
Elly Jones 已提交
16

S
Steve Klabnik 已提交
17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
//! JSON parsing and serialization
//!
//! # What is JSON?
//!
//! JSON (JavaScript Object Notation) is a way to write data in Javascript.
//! Like XML, it allows to encode structured data in a text format that can be easily read by humans
//! Its simple syntax and native compatibility with JavaScript have made it a widely used format.
//!
//! Data types that can be encoded are JavaScript types (see the `Json` enum for more details):
//!
//! * `Boolean`: equivalent to rust's `bool`
//! * `Number`: equivalent to rust's `f64`
//! * `String`: equivalent to rust's `String`
//! * `Array`: equivalent to rust's `Vec<T>`, but also allowing objects of different types in the
//!   same array
B
Brian J Brennan 已提交
32
//! * `Object`: equivalent to rust's `BTreeMap<String, json::Json>`
S
Steve Klabnik 已提交
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
//! * `Null`
//!
//! An object is a series of string keys mapping to values, in `"key": value` format.
//! Arrays are enclosed in square brackets ([ ... ]) and objects in curly brackets ({ ... }).
//! A simple JSON document encoding a person, his/her age, address and phone numbers could look like
//!
//! ```ignore
//! {
//!     "FirstName": "John",
//!     "LastName": "Doe",
//!     "Age": 43,
//!     "Address": {
//!         "Street": "Downing Street 10",
//!         "City": "London",
//!         "Country": "Great Britain"
//!     },
//!     "PhoneNumbers": [
//!         "+44 1234567",
//!         "+44 2345678"
//!     ]
//! }
//! ```
//!
//! # Rust Type-based Encoding and Decoding
//!
//! Rust provides a mechanism for low boilerplate encoding & decoding of values to and from JSON via
//! the serialization API.
60 61
//! To be able to encode a piece of data, it must implement the `serialize::RustcEncodable` trait.
//! To be able to decode a piece of data, it must implement the `serialize::RustcDecodable` trait.
S
Steve Klabnik 已提交
62
//! The Rust compiler provides an annotation to automatically generate the code for these traits:
63
//! `#[deriving(RustcDecodable, RustcEncodable)]`
S
Steve Klabnik 已提交
64 65 66 67 68 69
//!
//! The JSON API provides an enum `json::Json` and a trait `ToJson` to encode objects.
//! The `ToJson` trait provides a `to_json` method to convert an object into a `json::Json` value.
//! A `json::Json` value can be encoded as a string or buffer using the functions described above.
//! You can also use the `json::Encoder` object, which implements the `Encoder` trait.
//!
70
//! When using `ToJson` the `RustcEncodable` trait implementation is not mandatory.
S
Steve Klabnik 已提交
71 72 73 74 75 76 77 78 79 80 81 82 83
//!
//! # Examples of use
//!
//! ## Using Autoserialization
//!
//! Create a struct called `TestStruct` and serialize and deserialize it to and from JSON using the
//! serialization API, using the derived serialization code.
//!
//! ```rust
//! extern crate serialize;
//! use serialize::json;
//!
//! // Automatically generate `Decodable` and `Encodable` trait implementations
84
//! #[deriving(RustcDecodable, RustcEncodable)]
S
Steve Klabnik 已提交
85 86 87 88 89 90 91 92 93
//! pub struct TestStruct  {
//!     data_int: u8,
//!     data_str: String,
//!     data_vector: Vec<u8>,
//! }
//!
//! fn main() {
//!     let object = TestStruct {
//!         data_int: 1,
B
Barosl Lee 已提交
94
//!         data_str: "homura".to_string(),
S
Steve Klabnik 已提交
95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114
//!         data_vector: vec![2,3,4,5],
//!     };
//!
//!     // Serialize using `json::encode`
//!     let encoded = json::encode(&object);
//!
//!     // Deserialize using `json::decode`
//!     let decoded: TestStruct = json::decode(encoded.as_slice()).unwrap();
//! }
//! ```
//!
//! ## Using the `ToJson` trait
//!
//! The examples above use the `ToJson` trait to generate the JSON string, which is required
//! for custom mappings.
//!
//! ### Simple example of `ToJson` usage
//!
//! ```rust
//! extern crate serialize;
115
//! use serialize::json::{mod, ToJson, Json};
S
Steve Klabnik 已提交
116 117 118 119 120 121 122 123 124
//!
//! // A custom data structure
//! struct ComplexNum {
//!     a: f64,
//!     b: f64,
//! }
//!
//! // JSON value representation
//! impl ToJson for ComplexNum {
125 126
//!     fn to_json(&self) -> Json {
//!         Json::String(format!("{}+{}i", self.a, self.b))
S
Steve Klabnik 已提交
127 128 129
//!     }
//! }
//!
130
//! // Only generate `RustcEncodable` trait implementation
S
Steve Klabnik 已提交
131 132 133 134
//! #[deriving(Encodable)]
//! pub struct ComplexNumRecord {
//!     uid: u8,
//!     dsc: String,
135
//!     val: Json,
S
Steve Klabnik 已提交
136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153
//! }
//!
//! fn main() {
//!     let num = ComplexNum { a: 0.0001, b: 12.539 };
//!     let data: String = json::encode(&ComplexNumRecord{
//!         uid: 1,
//!         dsc: "test".to_string(),
//!         val: num.to_json(),
//!     });
//!     println!("data: {}", data);
//!     // data: {"uid":1,"dsc":"test","val":"0.0001+12.539j"};
//! }
//! ```
//!
//! ### Verbose example of `ToJson` usage
//!
//! ```rust
//! extern crate serialize;
A
Alexis Beingessner 已提交
154
//! use std::collections::BTreeMap;
155
//! use serialize::json::{mod, Json, ToJson};
S
Steve Klabnik 已提交
156 157 158 159 160 161 162 163 164 165 166
//!
//! // Only generate `Decodable` trait implementation
//! #[deriving(Decodable)]
//! pub struct TestStruct {
//!     data_int: u8,
//!     data_str: String,
//!     data_vector: Vec<u8>,
//! }
//!
//! // Specify encoding method manually
//! impl ToJson for TestStruct {
167
//!     fn to_json(&self) -> Json {
A
Alexis Beingessner 已提交
168
//!         let mut d = BTreeMap::new();
S
Steve Klabnik 已提交
169 170 171 172
//!         // All standard types implement `to_json()`, so use it
//!         d.insert("data_int".to_string(), self.data_int.to_json());
//!         d.insert("data_str".to_string(), self.data_str.to_json());
//!         d.insert("data_vector".to_string(), self.data_vector.to_json());
173
//!         Json::Object(d)
S
Steve Klabnik 已提交
174 175 176 177 178 179 180
//!     }
//! }
//!
//! fn main() {
//!     // Serialize using `ToJson`
//!     let input_data = TestStruct {
//!         data_int: 1,
B
Barosl Lee 已提交
181
//!         data_str: "madoka".to_string(),
S
Steve Klabnik 已提交
182 183
//!         data_vector: vec![2,3,4,5],
//!     };
184
//!     let json_obj: Json = input_data.to_json();
S
Steve Klabnik 已提交
185 186 187 188 189 190
//!     let json_str: String = json_obj.to_string();
//!
//!     // Deserialize like before
//!     let decoded: TestStruct = json::decode(json_str.as_slice()).unwrap();
//! }
//! ```
B
Brian Anderson 已提交
191

192 193 194 195 196
use self::JsonEvent::*;
use self::StackElement::*;
use self::ErrorCode::*;
use self::ParserError::*;
use self::DecoderError::*;
S
Steven Fackler 已提交
197 198 199
use self::ParserState::*;
use self::InternalStackElement::*;

A
Adolfo Ochagavía 已提交
200
use std;
A
Alexis Beingessner 已提交
201
use std::collections::{HashMap, BTreeMap};
A
Adolfo Ochagavía 已提交
202 203
use std::{char, f64, fmt, io, num, str};
use std::mem::{swap, transmute};
T
Tobias Bucher 已提交
204 205 206
use std::num::{Float, Int};
use std::num::FpCategory as Fp;
use std::str::FromStr;
207
use std::string;
208
use std::ops;
A
Alex Crichton 已提交
209 210
use unicode::str as unicode_str;
use unicode::str::Utf16Item;
211

A
Alex Crichton 已提交
212
use Encodable;
E
Elly Jones 已提交
213

214
/// Represents a json value
215
#[deriving(Clone, PartialEq, PartialOrd)]
216
pub enum Json {
217 218 219
    I64(i64),
    U64(u64),
    F64(f64),
220
    String(string::String),
B
Ben Striegel 已提交
221
    Boolean(bool),
222 223
    Array(self::Array),
    Object(self::Object),
B
Ben Striegel 已提交
224
    Null,
E
Elly Jones 已提交
225 226
}

227
pub type Array = Vec<Json>;
A
Alexis Beingessner 已提交
228
pub type Object = BTreeMap<string::String, Json>;
229

230
/// The errors that can arise while parsing a JSON stream.
231
#[deriving(Clone, Copy, PartialEq)]
232 233 234 235
pub enum ErrorCode {
    InvalidSyntax,
    InvalidNumber,
    EOFWhileParsingObject,
C
Corey Farwell 已提交
236
    EOFWhileParsingArray,
237 238 239 240 241
    EOFWhileParsingValue,
    EOFWhileParsingString,
    KeyMustBeAString,
    ExpectedColon,
    TrailingCharacters,
242
    TrailingComma,
243 244 245 246 247 248 249 250 251
    InvalidEscape,
    InvalidUnicodeCodePoint,
    LoneLeadingSurrogateInHexEscape,
    UnexpectedEndOfHexEscape,
    UnrecognizedHex,
    NotFourDigit,
    NotUtf8,
}

252
#[deriving(Clone, Copy, PartialEq, Show)]
253
pub enum ParserError {
S
Sean McArthur 已提交
254
    /// msg, line, col
255 256 257 258 259 260 261
    SyntaxError(ErrorCode, uint, uint),
    IoError(io::IoErrorKind, &'static str),
}

// Builder and Parser have the same errors.
pub type BuilderError = ParserError;

262
#[deriving(Clone, PartialEq, Show)]
263 264
pub enum DecoderError {
    ParseError(ParserError),
265 266 267 268
    ExpectedError(string::String, string::String),
    MissingFieldError(string::String),
    UnknownVariantError(string::String),
    ApplicationError(string::String)
269 270 271 272
}

/// Returns a readable error string for a given error code.
pub fn error_str(error: ErrorCode) -> &'static str {
273
    match error {
274 275 276
        InvalidSyntax => "invalid syntax",
        InvalidNumber => "invalid number",
        EOFWhileParsingObject => "EOF While parsing object",
C
Corey Farwell 已提交
277
        EOFWhileParsingArray => "EOF While parsing array",
278 279 280 281 282
        EOFWhileParsingValue => "EOF While parsing value",
        EOFWhileParsingString => "EOF While parsing string",
        KeyMustBeAString => "key must be a string",
        ExpectedColon => "expected `:`",
        TrailingCharacters => "trailing characters",
283
        TrailingComma => "trailing comma",
284
        InvalidEscape => "invalid escape",
A
Alex Crichton 已提交
285 286
        UnrecognizedHex => "invalid \\u{ esc}ape (unrecognized hex)",
        NotFourDigit => "invalid \\u{ esc}ape (not four digits)",
287
        NotUtf8 => "contents not utf-8",
288
        InvalidUnicodeCodePoint => "invalid Unicode code point",
289 290 291 292 293
        LoneLeadingSurrogateInHexEscape => "lone leading surrogate in hex escape",
        UnexpectedEndOfHexEscape => "unexpected end of hex escape",
    }
}

294 295 296 297 298 299 300 301 302 303 304 305
/// Shortcut function to decode a JSON `&str` into an object
pub fn decode<T: ::Decodable<Decoder, DecoderError>>(s: &str) -> DecodeResult<T> {
    let json = match from_str(s) {
        Ok(x) => x,
        Err(e) => return Err(ParseError(e))
    };

    let mut decoder = Decoder::new(json);
    ::Decodable::decode(&mut decoder)
}

/// Shortcut function to encode a `T` into a JSON `String`
306
pub fn encode<'a, T: Encodable<Encoder<'a>, io::IoError>>(object: &T) -> string::String {
307
    let buff = Encoder::buffer_encode(object);
308
    string::String::from_utf8(buff).unwrap()
309 310
}

311 312 313 314 315 316 317 318
impl fmt::Show for ErrorCode {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        error_str(*self).fmt(f)
    }
}

fn io_error_to_error(io: io::IoError) -> ParserError {
    IoError(io.kind, io.desc)
319
}
320

321 322 323 324 325
impl std::error::Error for DecoderError {
    fn description(&self) -> &str { "decoder error" }
    fn detail(&self) -> Option<std::string::String> { Some(self.to_string()) }
}

S
Sean McArthur 已提交
326
pub type EncodeResult = io::IoResult<()>;
327
pub type DecodeResult<T> = Result<T, DecoderError>;
A
Alex Crichton 已提交
328

329 330 331 332 333 334 335 336 337
pub fn escape_bytes(wr: &mut io::Writer, bytes: &[u8]) -> Result<(), io::IoError> {
    try!(wr.write_str("\""));

    let mut start = 0;

    for (i, byte) in bytes.iter().enumerate() {
        let escaped = match *byte {
            b'"' => "\\\"",
            b'\\' => "\\\\",
R
Rolf Timmermans 已提交
338 339 340 341 342 343 344 345
            b'\x00' => "\\u0000",
            b'\x01' => "\\u0001",
            b'\x02' => "\\u0002",
            b'\x03' => "\\u0003",
            b'\x04' => "\\u0004",
            b'\x05' => "\\u0005",
            b'\x06' => "\\u0006",
            b'\x07' => "\\u0007",
346
            b'\x08' => "\\b",
R
Rolf Timmermans 已提交
347
            b'\t' => "\\t",
348
            b'\n' => "\\n",
R
Rolf Timmermans 已提交
349 350
            b'\x0b' => "\\u000b",
            b'\x0c' => "\\f",
351
            b'\r' => "\\r",
R
Rolf Timmermans 已提交
352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370
            b'\x0e' => "\\u000e",
            b'\x0f' => "\\u000f",
            b'\x10' => "\\u0010",
            b'\x11' => "\\u0011",
            b'\x12' => "\\u0012",
            b'\x13' => "\\u0013",
            b'\x14' => "\\u0014",
            b'\x15' => "\\u0015",
            b'\x16' => "\\u0016",
            b'\x17' => "\\u0017",
            b'\x18' => "\\u0018",
            b'\x19' => "\\u0019",
            b'\x1a' => "\\u001a",
            b'\x1b' => "\\u001b",
            b'\x1c' => "\\u001c",
            b'\x1d' => "\\u001d",
            b'\x1e' => "\\u001e",
            b'\x1f' => "\\u001f",
            b'\x7f' => "\\u007f",
371 372 373 374
            _ => { continue; }
        };

        if start < i {
375
            try!(wr.write(bytes[start..i]));
E
Elly Jones 已提交
376
        }
377 378 379 380

        try!(wr.write_str(escaped));

        start = i + 1;
381
    }
382 383

    if start != bytes.len() {
384
        try!(wr.write(bytes[start..]));
385 386 387
    }

    wr.write_str("\"")
388 389 390 391 392 393 394 395
}

fn escape_str(writer: &mut io::Writer, v: &str) -> Result<(), io::IoError> {
    escape_bytes(writer, v.as_bytes())
}

fn escape_char(writer: &mut io::Writer, v: char) -> Result<(), io::IoError> {
    let mut buf = [0, .. 4];
396 397
    let len = v.encode_utf8(&mut buf).unwrap();
    escape_bytes(writer, buf[mut ..len])
398 399
}

400
fn spaces(wr: &mut io::Writer, mut n: uint) -> Result<(), io::IoError> {
401 402 403 404
    const LEN: uint = 16;
    static BUF: [u8, ..LEN] = [b' ', ..LEN];

    while n >= LEN {
N
Nick Cameron 已提交
405
        try!(wr.write(&BUF));
406
        n -= LEN;
407 408 409
    }

    if n > 0 {
410
        wr.write(BUF[..n])
411 412
    } else {
        Ok(())
413
    }
E
Elly Jones 已提交
414 415
}

416
fn fmt_number_or_null(v: f64) -> string::String {
M
mrec 已提交
417
    match v.classify() {
T
Tobias Bucher 已提交
418
        Fp::Nan | Fp::Infinite => string::String::from_str("null"),
419 420
        _ if v.fract() != 0f64 => f64::to_str_digits(v, 6u),
        _ => f64::to_str_digits(v, 6u) + ".0",
M
mrec 已提交
421 422 423
    }
}

424
/// A structure for implementing serialization to JSON.
E
Erik Price 已提交
425
pub struct Encoder<'a> {
426
    writer: &'a mut (io::Writer+'a),
427 428
}

E
Erik Price 已提交
429
impl<'a> Encoder<'a> {
430 431
    /// Creates a new JSON encoder whose output will be written to the writer
    /// specified.
A
Adolfo Ochagavía 已提交
432 433
    pub fn new(writer: &'a mut io::Writer) -> Encoder<'a> {
        Encoder { writer: writer }
434
    }
M
musitdev 已提交
435 436

    /// Encode the specified struct into a json [u8]
437
    pub fn buffer_encode<T: Encodable<Encoder<'a>, io::IoError>>(object: &T) -> Vec<u8>  {
A
Adolfo Ochagavía 已提交
438
        //Serialize the object in a string using a writer
D
Daniel Micay 已提交
439
        let mut m = Vec::new();
N
Nick Cameron 已提交
440 441
        // FIXME(14302) remove the transmute and unsafe block.
        unsafe {
M
musitdev 已提交
442
            let mut encoder = Encoder::new(&mut m as &mut io::Writer);
D
Daniel Micay 已提交
443
            // Vec<u8> never Errs
A
Adolfo Ochagavía 已提交
444
            let _ = object.encode(transmute(&mut encoder));
M
musitdev 已提交
445
        }
D
Daniel Micay 已提交
446
        m
M
musitdev 已提交
447
    }
448 449
}

S
Sean McArthur 已提交
450
impl<'a> ::Encoder<io::IoError> for Encoder<'a> {
A
Adolfo Ochagavía 已提交
451
    fn emit_nil(&mut self) -> EncodeResult { write!(self.writer, "null") }
452

453 454 455 456 457
    fn emit_uint(&mut self, v: uint) -> EncodeResult { write!(self.writer, "{}", v) }
    fn emit_u64(&mut self, v: u64) -> EncodeResult { write!(self.writer, "{}", v) }
    fn emit_u32(&mut self, v: u32) -> EncodeResult { write!(self.writer, "{}", v) }
    fn emit_u16(&mut self, v: u16) -> EncodeResult { write!(self.writer, "{}", v) }
    fn emit_u8(&mut self, v: u8) -> EncodeResult { write!(self.writer, "{}", v) }
458

459 460 461 462 463
    fn emit_int(&mut self, v: int) -> EncodeResult { write!(self.writer, "{}", v) }
    fn emit_i64(&mut self, v: i64) -> EncodeResult { write!(self.writer, "{}", v) }
    fn emit_i32(&mut self, v: i32) -> EncodeResult { write!(self.writer, "{}", v) }
    fn emit_i16(&mut self, v: i16) -> EncodeResult { write!(self.writer, "{}", v) }
    fn emit_i8(&mut self, v: i8) -> EncodeResult { write!(self.writer, "{}", v) }
464

S
Sean McArthur 已提交
465
    fn emit_bool(&mut self, v: bool) -> EncodeResult {
466
        if v {
A
Adolfo Ochagavía 已提交
467
            write!(self.writer, "true")
468
        } else {
A
Adolfo Ochagavía 已提交
469
            write!(self.writer, "false")
470 471 472
        }
    }

S
Sean McArthur 已提交
473
    fn emit_f64(&mut self, v: f64) -> EncodeResult {
A
Adolfo Ochagavía 已提交
474
        write!(self.writer, "{}", fmt_number_or_null(v))
A
Alex Crichton 已提交
475
    }
B
Barosl Lee 已提交
476 477 478
    fn emit_f32(&mut self, v: f32) -> EncodeResult {
        self.emit_f64(v as f64)
    }
479

480
    fn emit_char(&mut self, v: char) -> EncodeResult {
481
        escape_char(self.writer, v)
482
    }
S
Sean McArthur 已提交
483
    fn emit_str(&mut self, v: &str) -> EncodeResult {
484
        escape_str(self.writer, v)
A
Alex Crichton 已提交
485
    }
486

487 488 489
    fn emit_enum<F>(&mut self, _name: &str, f: F) -> EncodeResult where
        F: FnOnce(&mut Encoder<'a>) -> EncodeResult,
    {
A
Adolfo Ochagavía 已提交
490 491
        f(self)
    }
492

493 494 495 496 497 498 499
    fn emit_enum_variant<F>(&mut self,
                            name: &str,
                            _id: uint,
                            cnt: uint,
                            f: F) -> EncodeResult where
        F: FnOnce(&mut Encoder<'a>) -> EncodeResult,
    {
500 501 502 503
        // enums are encoded as strings or objects
        // Bunny => "Bunny"
        // Kangaroo(34,"William") => {"variant": "Kangaroo", "fields": [34,"William"]}
        if cnt == 0 {
504
            escape_str(self.writer, name)
505
        } else {
A
Adolfo Ochagavía 已提交
506
            try!(write!(self.writer, "{{\"variant\":"));
507
            try!(escape_str(self.writer, name));
A
Adolfo Ochagavía 已提交
508
            try!(write!(self.writer, ",\"fields\":["));
509
            try!(f(self));
A
Adolfo Ochagavía 已提交
510
            write!(self.writer, "]}}")
511 512
        }
    }
513

514 515 516
    fn emit_enum_variant_arg<F>(&mut self, idx: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut Encoder<'a>) -> EncodeResult,
    {
517
        if idx != 0 {
A
Adolfo Ochagavía 已提交
518
            try!(write!(self.writer, ","));
519
        }
S
Sean McArthur 已提交
520
        f(self)
521 522
    }

523 524 525 526 527 528 529
    fn emit_enum_struct_variant<F>(&mut self,
                                   name: &str,
                                   id: uint,
                                   cnt: uint,
                                   f: F) -> EncodeResult where
        F: FnOnce(&mut Encoder<'a>) -> EncodeResult,
    {
530 531 532
        self.emit_enum_variant(name, id, cnt, f)
    }

533 534 535 536 537 538
    fn emit_enum_struct_variant_field<F>(&mut self,
                                         _: &str,
                                         idx: uint,
                                         f: F) -> EncodeResult where
        F: FnOnce(&mut Encoder<'a>) -> EncodeResult,
    {
539 540 541
        self.emit_enum_variant_arg(idx, f)
    }

542 543 544
    fn emit_struct<F>(&mut self, _: &str, _: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut Encoder<'a>) -> EncodeResult,
    {
A
Adolfo Ochagavía 已提交
545
        try!(write!(self.writer, "{{"));
546
        try!(f(self));
A
Adolfo Ochagavía 已提交
547
        write!(self.writer, "}}")
548
    }
549

550 551 552
    fn emit_struct_field<F>(&mut self, name: &str, idx: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut Encoder<'a>) -> EncodeResult,
    {
A
Adolfo Ochagavía 已提交
553
        if idx != 0 { try!(write!(self.writer, ",")); }
554 555
        try!(escape_str(self.writer, name));
        try!(write!(self.writer, ":"));
S
Sean McArthur 已提交
556
        f(self)
557 558
    }

559 560 561
    fn emit_tuple<F>(&mut self, len: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut Encoder<'a>) -> EncodeResult,
    {
562 563
        self.emit_seq(len, f)
    }
564 565 566
    fn emit_tuple_arg<F>(&mut self, idx: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut Encoder<'a>) -> EncodeResult,
    {
567 568 569
        self.emit_seq_elt(idx, f)
    }

570 571 572
    fn emit_tuple_struct<F>(&mut self, _name: &str, len: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut Encoder<'a>) -> EncodeResult,
    {
573 574
        self.emit_seq(len, f)
    }
575 576 577
    fn emit_tuple_struct_arg<F>(&mut self, idx: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut Encoder<'a>) -> EncodeResult,
    {
578 579 580
        self.emit_seq_elt(idx, f)
    }

581 582 583
    fn emit_option<F>(&mut self, f: F) -> EncodeResult where
        F: FnOnce(&mut Encoder<'a>) -> EncodeResult,
    {
S
Sean McArthur 已提交
584 585 586
        f(self)
    }
    fn emit_option_none(&mut self) -> EncodeResult { self.emit_nil() }
587 588 589
    fn emit_option_some<F>(&mut self, f: F) -> EncodeResult where
        F: FnOnce(&mut Encoder<'a>) -> EncodeResult,
    {
S
Sean McArthur 已提交
590 591
        f(self)
    }
592

593 594 595
    fn emit_seq<F>(&mut self, _len: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut Encoder<'a>) -> EncodeResult,
    {
A
Adolfo Ochagavía 已提交
596
        try!(write!(self.writer, "["));
S
Sean McArthur 已提交
597
        try!(f(self));
A
Adolfo Ochagavía 已提交
598
        write!(self.writer, "]")
599 600
    }

601 602 603
    fn emit_seq_elt<F>(&mut self, idx: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut Encoder<'a>) -> EncodeResult,
    {
604
        if idx != 0 {
A
Adolfo Ochagavía 已提交
605
            try!(write!(self.writer, ","));
606 607 608 609
        }
        f(self)
    }

610 611 612
    fn emit_map<F>(&mut self, _len: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut Encoder<'a>) -> EncodeResult,
    {
A
Adolfo Ochagavía 已提交
613
        try!(write!(self.writer, "{{"));
614
        try!(f(self));
A
Adolfo Ochagavía 已提交
615
        write!(self.writer, "}}")
616
    }
617

618 619 620
    fn emit_map_elt_key<F>(&mut self, idx: uint, mut f: F) -> EncodeResult where
        F: FnMut(&mut Encoder<'a>) -> EncodeResult,
    {
A
Adolfo Ochagavía 已提交
621
        if idx != 0 { try!(write!(self.writer, ",")) }
622 623
        // ref #12967, make sure to wrap a key in double quotes,
        // in the event that its of a type that omits them (eg numbers)
D
Daniel Micay 已提交
624
        let mut buf = Vec::new();
N
Nick Cameron 已提交
625 626 627 628 629
        // FIXME(14302) remove the transmute and unsafe block.
        unsafe {
            let mut check_encoder = Encoder::new(&mut buf);
            try!(f(transmute(&mut check_encoder)));
        }
D
Daniel Micay 已提交
630
        let out = str::from_utf8(buf[]).unwrap();
A
Adolfo Ochagavía 已提交
631 632
        let needs_wrapping = out.char_at(0) != '"' && out.char_at_reverse(out.len()) != '"';
        if needs_wrapping { try!(write!(self.writer, "\"")); }
S
Sean McArthur 已提交
633
        try!(f(self));
A
Adolfo Ochagavía 已提交
634
        if needs_wrapping { try!(write!(self.writer, "\"")); }
S
Sean McArthur 已提交
635
        Ok(())
636 637
    }

638 639 640
    fn emit_map_elt_val<F>(&mut self, _idx: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut Encoder<'a>) -> EncodeResult,
    {
A
Adolfo Ochagavía 已提交
641
        try!(write!(self.writer, ":"));
642 643 644 645
        f(self)
    }
}

646 647
/// Another encoder for JSON, but prints out human-readable JSON instead of
/// compact data
E
Erik Price 已提交
648
pub struct PrettyEncoder<'a> {
649
    writer: &'a mut (io::Writer+'a),
650
    curr_indent: uint,
651
    indent: uint,
652 653
}

E
Erik Price 已提交
654
impl<'a> PrettyEncoder<'a> {
655
    /// Creates a new encoder whose output will be written to the specified writer
N
Niko Matsakis 已提交
656
    pub fn new(writer: &'a mut io::Writer) -> PrettyEncoder<'a> {
657 658 659 660 661
        PrettyEncoder { writer: writer, curr_indent: 0, indent: 2, }
    }

    /// Set the number of spaces to indent for each level.
    /// This is safe to set during encoding.
N
Niko Matsakis 已提交
662
    pub fn set_indent(&mut self, indent: uint) {
663
        // self.indent very well could be 0 so we need to use checked division.
664
        let level = self.curr_indent.checked_div(self.indent).unwrap_or(0);
665 666
        self.indent = indent;
        self.curr_indent = level * self.indent;
667
    }
668
}
669

S
Sean McArthur 已提交
670
impl<'a> ::Encoder<io::IoError> for PrettyEncoder<'a> {
A
Adolfo Ochagavía 已提交
671
    fn emit_nil(&mut self) -> EncodeResult { write!(self.writer, "null") }
672

673 674 675 676 677
    fn emit_uint(&mut self, v: uint) -> EncodeResult { write!(self.writer, "{}", v) }
    fn emit_u64(&mut self, v: u64) -> EncodeResult { write!(self.writer, "{}", v) }
    fn emit_u32(&mut self, v: u32) -> EncodeResult { write!(self.writer, "{}", v) }
    fn emit_u16(&mut self, v: u16) -> EncodeResult { write!(self.writer, "{}", v) }
    fn emit_u8(&mut self, v: u8) -> EncodeResult { write!(self.writer, "{}", v) }
678

679 680 681 682 683
    fn emit_int(&mut self, v: int) -> EncodeResult { write!(self.writer, "{}", v) }
    fn emit_i64(&mut self, v: i64) -> EncodeResult { write!(self.writer, "{}", v) }
    fn emit_i32(&mut self, v: i32) -> EncodeResult { write!(self.writer, "{}", v) }
    fn emit_i16(&mut self, v: i16) -> EncodeResult { write!(self.writer, "{}", v) }
    fn emit_i8(&mut self, v: i8) -> EncodeResult { write!(self.writer, "{}", v) }
684

S
Sean McArthur 已提交
685
    fn emit_bool(&mut self, v: bool) -> EncodeResult {
686
        if v {
A
Adolfo Ochagavía 已提交
687
            write!(self.writer, "true")
688
        } else {
A
Adolfo Ochagavía 已提交
689
            write!(self.writer, "false")
690 691 692
        }
    }

S
Sean McArthur 已提交
693
    fn emit_f64(&mut self, v: f64) -> EncodeResult {
A
Adolfo Ochagavía 已提交
694
        write!(self.writer, "{}", fmt_number_or_null(v))
A
Alex Crichton 已提交
695
    }
696 697 698
    fn emit_f32(&mut self, v: f32) -> EncodeResult {
        self.emit_f64(v as f64)
    }
699

700
    fn emit_char(&mut self, v: char) -> EncodeResult {
701
        escape_char(self.writer, v)
702
    }
S
Sean McArthur 已提交
703
    fn emit_str(&mut self, v: &str) -> EncodeResult {
704
        escape_str(self.writer, v)
A
Alex Crichton 已提交
705
    }
706

707 708 709
    fn emit_enum<F>(&mut self, _name: &str, f: F) -> EncodeResult where
        F: FnOnce(&mut PrettyEncoder<'a>) -> EncodeResult,
    {
710 711 712
        f(self)
    }

713 714 715 716 717 718 719 720
    fn emit_enum_variant<F>(&mut self,
                            name: &str,
                            _id: uint,
                            cnt: uint,
                            f: F)
                            -> EncodeResult where
        F: FnOnce(&mut PrettyEncoder<'a>) -> EncodeResult,
    {
721
        if cnt == 0 {
722
            escape_str(self.writer, name)
723
        } else {
724
            try!(write!(self.writer, "{{\n"));
725 726
            self.curr_indent += self.indent;
            try!(spaces(self.writer, self.curr_indent));
727
            try!(write!(self.writer, "\"variant\": "));
728 729
            try!(escape_str(self.writer, name));
            try!(write!(self.writer, ",\n"));
730 731 732
            try!(spaces(self.writer, self.curr_indent));
            try!(write!(self.writer, "\"fields\": [\n"));
            self.curr_indent += self.indent;
S
Sean McArthur 已提交
733
            try!(f(self));
734
            self.curr_indent -= self.indent;
735
            try!(write!(self.writer, "\n"));
736
            try!(spaces(self.writer, self.curr_indent));
737 738 739 740
            self.curr_indent -= self.indent;
            try!(write!(self.writer, "]\n"));
            try!(spaces(self.writer, self.curr_indent));
            write!(self.writer, "}}")
741 742 743
        }
    }

744 745 746
    fn emit_enum_variant_arg<F>(&mut self, idx: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut PrettyEncoder<'a>) -> EncodeResult,
    {
747
        if idx != 0 {
A
Adolfo Ochagavía 已提交
748
            try!(write!(self.writer, ",\n"));
749
        }
750
        try!(spaces(self.writer, self.curr_indent));
751 752 753
        f(self)
    }

754 755 756 757 758 759 760
    fn emit_enum_struct_variant<F>(&mut self,
                                   name: &str,
                                   id: uint,
                                   cnt: uint,
                                   f: F) -> EncodeResult where
        F: FnOnce(&mut PrettyEncoder<'a>) -> EncodeResult,
    {
761 762 763
        self.emit_enum_variant(name, id, cnt, f)
    }

764 765 766 767 768 769
    fn emit_enum_struct_variant_field<F>(&mut self,
                                         _: &str,
                                         idx: uint,
                                         f: F) -> EncodeResult where
        F: FnOnce(&mut PrettyEncoder<'a>) -> EncodeResult,
    {
770 771 772 773
        self.emit_enum_variant_arg(idx, f)
    }


774 775 776
    fn emit_struct<F>(&mut self, _: &str, len: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut PrettyEncoder<'a>) -> EncodeResult,
    {
777
        if len == 0 {
A
Adolfo Ochagavía 已提交
778
            write!(self.writer, "{{}}")
779
        } else {
A
Adolfo Ochagavía 已提交
780
            try!(write!(self.writer, "{{"));
781
            self.curr_indent += self.indent;
782
            try!(f(self));
783
            self.curr_indent -= self.indent;
784
            try!(write!(self.writer, "\n"));
785
            try!(spaces(self.writer, self.curr_indent));
786
            write!(self.writer, "}}")
787 788
        }
    }
789

790 791 792
    fn emit_struct_field<F>(&mut self, name: &str, idx: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut PrettyEncoder<'a>) -> EncodeResult,
    {
793
        if idx == 0 {
A
Adolfo Ochagavía 已提交
794
            try!(write!(self.writer, "\n"));
795
        } else {
A
Adolfo Ochagavía 已提交
796
            try!(write!(self.writer, ",\n"));
797
        }
798
        try!(spaces(self.writer, self.curr_indent));
799 800
        try!(escape_str(self.writer, name));
        try!(write!(self.writer, ": "));
S
Sean McArthur 已提交
801
        f(self)
802 803
    }

804 805 806
    fn emit_tuple<F>(&mut self, len: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut PrettyEncoder<'a>) -> EncodeResult,
    {
807 808
        self.emit_seq(len, f)
    }
809 810 811
    fn emit_tuple_arg<F>(&mut self, idx: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut PrettyEncoder<'a>) -> EncodeResult,
    {
812 813 814
        self.emit_seq_elt(idx, f)
    }

815 816 817
    fn emit_tuple_struct<F>(&mut self, _: &str, len: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut PrettyEncoder<'a>) -> EncodeResult,
    {
818 819
        self.emit_seq(len, f)
    }
820 821 822
    fn emit_tuple_struct_arg<F>(&mut self, idx: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut PrettyEncoder<'a>) -> EncodeResult,
    {
823 824 825
        self.emit_seq_elt(idx, f)
    }

826 827 828
    fn emit_option<F>(&mut self, f: F) -> EncodeResult where
        F: FnOnce(&mut PrettyEncoder<'a>) -> EncodeResult,
    {
S
Sean McArthur 已提交
829 830 831
        f(self)
    }
    fn emit_option_none(&mut self) -> EncodeResult { self.emit_nil() }
832 833 834
    fn emit_option_some<F>(&mut self, f: F) -> EncodeResult where
        F: FnOnce(&mut PrettyEncoder<'a>) -> EncodeResult,
    {
S
Sean McArthur 已提交
835 836
        f(self)
    }
837

838 839 840
    fn emit_seq<F>(&mut self, len: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut PrettyEncoder<'a>) -> EncodeResult,
    {
841
        if len == 0 {
A
Adolfo Ochagavía 已提交
842
            write!(self.writer, "[]")
843
        } else {
A
Adolfo Ochagavía 已提交
844
            try!(write!(self.writer, "["));
845
            self.curr_indent += self.indent;
S
Sean McArthur 已提交
846
            try!(f(self));
847
            self.curr_indent -= self.indent;
848
            try!(write!(self.writer, "\n"));
849
            try!(spaces(self.writer, self.curr_indent));
850
            write!(self.writer, "]")
851 852 853
        }
    }

854 855 856
    fn emit_seq_elt<F>(&mut self, idx: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut PrettyEncoder<'a>) -> EncodeResult,
    {
857
        if idx == 0 {
A
Adolfo Ochagavía 已提交
858
            try!(write!(self.writer, "\n"));
859
        } else {
A
Adolfo Ochagavía 已提交
860
            try!(write!(self.writer, ",\n"));
861
        }
862
        try!(spaces(self.writer, self.curr_indent));
863 864 865
        f(self)
    }

866 867 868
    fn emit_map<F>(&mut self, len: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut PrettyEncoder<'a>) -> EncodeResult,
    {
869
        if len == 0 {
A
Adolfo Ochagavía 已提交
870
            write!(self.writer, "{{}}")
871
        } else {
A
Adolfo Ochagavía 已提交
872
            try!(write!(self.writer, "{{"));
873
            self.curr_indent += self.indent;
874
            try!(f(self));
875
            self.curr_indent -= self.indent;
876
            try!(write!(self.writer, "\n"));
877
            try!(spaces(self.writer, self.curr_indent));
878
            write!(self.writer, "}}")
879 880
        }
    }
881

882 883 884
    fn emit_map_elt_key<F>(&mut self, idx: uint, mut f: F) -> EncodeResult where
        F: FnMut(&mut PrettyEncoder<'a>) -> EncodeResult,
    {
885
        if idx == 0 {
A
Adolfo Ochagavía 已提交
886
            try!(write!(self.writer, "\n"));
887
        } else {
A
Adolfo Ochagavía 已提交
888
            try!(write!(self.writer, ",\n"));
889
        }
890
        try!(spaces(self.writer, self.curr_indent));
891 892
        // ref #12967, make sure to wrap a key in double quotes,
        // in the event that its of a type that omits them (eg numbers)
D
Daniel Micay 已提交
893
        let mut buf = Vec::new();
N
Nick Cameron 已提交
894 895 896 897 898
        // FIXME(14302) remove the transmute and unsafe block.
        unsafe {
            let mut check_encoder = PrettyEncoder::new(&mut buf);
            try!(f(transmute(&mut check_encoder)));
        }
D
Daniel Micay 已提交
899
        let out = str::from_utf8(buf[]).unwrap();
A
Adolfo Ochagavía 已提交
900 901
        let needs_wrapping = out.char_at(0) != '"' && out.char_at_reverse(out.len()) != '"';
        if needs_wrapping { try!(write!(self.writer, "\"")); }
S
Sean McArthur 已提交
902
        try!(f(self));
A
Adolfo Ochagavía 已提交
903
        if needs_wrapping { try!(write!(self.writer, "\"")); }
S
Sean McArthur 已提交
904
        Ok(())
905 906
    }

907 908 909
    fn emit_map_elt_val<F>(&mut self, _idx: uint, f: F) -> EncodeResult where
        F: FnOnce(&mut PrettyEncoder<'a>) -> EncodeResult,
    {
A
Adolfo Ochagavía 已提交
910
        try!(write!(self.writer, ": "));
S
Sean McArthur 已提交
911
        f(self)
912 913 914
    }
}

915 916
impl<E: ::Encoder<S>, S> Encodable<E, S> for Json {
    fn encode(&self, e: &mut E) -> Result<(), S> {
917
        match *self {
918 919 920 921 922 923 924 925
            Json::I64(v) => v.encode(e),
            Json::U64(v) => v.encode(e),
            Json::F64(v) => v.encode(e),
            Json::String(ref v) => v.encode(e),
            Json::Boolean(v) => v.encode(e),
            Json::Array(ref v) => v.encode(e),
            Json::Object(ref v) => v.encode(e),
            Json::Null => e.emit_nil(),
926 927 928 929
        }
    }
}

930
impl Json {
A
Adolfo Ochagavía 已提交
931 932 933
    /// Encodes a json value into an io::writer. Uses a single line.
    pub fn to_writer(&self, writer: &mut io::Writer) -> EncodeResult {
        let mut encoder = Encoder::new(writer);
S
Sean McArthur 已提交
934
        self.encode(&mut encoder)
935
    }
936

J
Jorge Aparicio 已提交
937
    /// Encodes a json value into an io::writer.
938
    /// Pretty-prints in a more readable format.
A
Adolfo Ochagavía 已提交
939 940
    pub fn to_pretty_writer(&self, writer: &mut io::Writer) -> EncodeResult {
        let mut encoder = PrettyEncoder::new(writer);
S
Sean McArthur 已提交
941
        self.encode(&mut encoder)
942
    }
943

944
    /// Encodes a json value into a string
945
    pub fn to_pretty_str(&self) -> string::String {
D
Daniel Micay 已提交
946
        let mut s = Vec::new();
A
Alex Crichton 已提交
947
        self.to_pretty_writer(&mut s as &mut io::Writer).unwrap();
D
Daniel Micay 已提交
948
        string::String::from_utf8(s).unwrap()
949
    }
950 951 952

     /// If the Json value is an Object, returns the value associated with the provided key.
    /// Otherwise, returns None.
953 954
    pub fn find<'a>(&'a self, key: &str) -> Option<&'a Json>{
        match self {
955
            &Json::Object(ref map) => map.get(key),
956 957 958 959
            _ => None
        }
    }

960
    /// Attempts to get a nested Json Object for each key in `keys`.
961
    /// If any key is found not to exist, find_path will return None.
962
    /// Otherwise, it will return the Json value associated with the final key.
963
    pub fn find_path<'a>(&'a self, keys: &[&str]) -> Option<&'a Json>{
964 965 966 967 968 969 970 971
        let mut target = self;
        for key in keys.iter() {
            match target.find(*key) {
                Some(t) => { target = t; },
                None => return None
            }
        }
        Some(target)
972 973 974 975 976
    }

    /// If the Json value is an Object, performs a depth-first search until
    /// a value associated with the provided key is found. If no value is found
    /// or the Json value is not an Object, returns None.
977 978
    pub fn search<'a>(&'a self, key: &str) -> Option<&'a Json> {
        match self {
979
            &Json::Object(ref map) => {
A
Aaron Turon 已提交
980
                match map.get(key) {
981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996
                    Some(json_value) => Some(json_value),
                    None => {
                        for (_, v) in map.iter() {
                            match v.search(key) {
                                x if x.is_some() => return x,
                                _ => ()
                            }
                        }
                        None
                    }
                }
            },
            _ => None
        }
    }

997 998
    /// Returns true if the Json value is an Object. Returns false otherwise.
    pub fn is_object<'a>(&'a self) -> bool {
999
        self.as_object().is_some()
1000 1001
    }

A
Alexis Beingessner 已提交
1002
    /// If the Json value is an Object, returns the associated BTreeMap.
1003
    /// Returns None otherwise.
1004
    pub fn as_object<'a>(&'a self) -> Option<&'a Object> {
1005
        match self {
1006
            &Json::Object(ref map) => Some(map),
1007 1008 1009 1010
            _ => None
        }
    }

C
Corey Farwell 已提交
1011
    /// Returns true if the Json value is an Array. Returns false otherwise.
C
Corey Farwell 已提交
1012 1013
    pub fn is_array<'a>(&'a self) -> bool {
        self.as_array().is_some()
1014 1015
    }

C
Corey Farwell 已提交
1016
    /// If the Json value is an Array, returns the associated vector.
1017
    /// Returns None otherwise.
1018
    pub fn as_array<'a>(&'a self) -> Option<&'a Array> {
1019
        match self {
1020
            &Json::Array(ref array) => Some(&*array),
1021 1022 1023 1024 1025
            _ => None
        }
    }

    /// Returns true if the Json value is a String. Returns false otherwise.
1026 1027
    pub fn is_string<'a>(&'a self) -> bool {
        self.as_string().is_some()
1028 1029 1030 1031
    }

    /// If the Json value is a String, returns the associated str.
    /// Returns None otherwise.
1032
    pub fn as_string<'a>(&'a self) -> Option<&'a str> {
1033
        match *self {
A
Alex Crichton 已提交
1034
            Json::String(ref s) => Some(s[]),
1035 1036 1037 1038 1039 1040
            _ => None
        }
    }

    /// Returns true if the Json value is a Number. Returns false otherwise.
    pub fn is_number(&self) -> bool {
1041
        match *self {
1042
            Json::I64(_) | Json::U64(_) | Json::F64(_) => true,
1043 1044 1045 1046 1047 1048 1049
            _ => false,
        }
    }

    /// Returns true if the Json value is a i64. Returns false otherwise.
    pub fn is_i64(&self) -> bool {
        match *self {
1050
            Json::I64(_) => true,
1051 1052 1053 1054 1055 1056 1057
            _ => false,
        }
    }

    /// Returns true if the Json value is a u64. Returns false otherwise.
    pub fn is_u64(&self) -> bool {
        match *self {
1058
            Json::U64(_) => true,
1059 1060
            _ => false,
        }
1061 1062
    }

1063 1064 1065
    /// Returns true if the Json value is a f64. Returns false otherwise.
    pub fn is_f64(&self) -> bool {
        match *self {
1066
            Json::F64(_) => true,
1067 1068 1069 1070
            _ => false,
        }
    }

1071
    /// If the Json value is a number, return or cast it to a i64.
1072
    /// Returns None otherwise.
1073 1074
    pub fn as_i64(&self) -> Option<i64> {
        match *self {
1075 1076
            Json::I64(n) => Some(n),
            Json::U64(n) => num::cast(n),
1077 1078 1079 1080 1081 1082 1083 1084
            _ => None
        }
    }

    /// If the Json value is a number, return or cast it to a u64.
    /// Returns None otherwise.
    pub fn as_u64(&self) -> Option<u64> {
        match *self {
1085 1086
            Json::I64(n) => num::cast(n),
            Json::U64(n) => Some(n),
1087 1088 1089 1090
            _ => None
        }
    }

1091
    /// If the Json value is a number, return or cast it to a f64.
1092 1093 1094
    /// Returns None otherwise.
    pub fn as_f64(&self) -> Option<f64> {
        match *self {
1095 1096 1097
            Json::I64(n) => num::cast(n),
            Json::U64(n) => num::cast(n),
            Json::F64(n) => Some(n),
1098 1099 1100 1101 1102 1103
            _ => None
        }
    }

    /// Returns true if the Json value is a Boolean. Returns false otherwise.
    pub fn is_boolean(&self) -> bool {
1104
        self.as_boolean().is_some()
1105 1106 1107 1108 1109 1110
    }

    /// If the Json value is a Boolean, returns the associated bool.
    /// Returns None otherwise.
    pub fn as_boolean(&self) -> Option<bool> {
        match self {
1111
            &Json::Boolean(b) => Some(b),
1112 1113 1114 1115 1116 1117
            _ => None
        }
    }

    /// Returns true if the Json value is a Null. Returns false otherwise.
    pub fn is_null(&self) -> bool {
1118
        self.as_null().is_some()
1119 1120 1121 1122 1123 1124
    }

    /// If the Json value is a Null, returns ().
    /// Returns None otherwise.
    pub fn as_null(&self) -> Option<()> {
        match self {
1125
            &Json::Null => Some(()),
1126 1127 1128
            _ => None
        }
    }
E
Elly Jones 已提交
1129 1130
}

1131
impl<'a> ops::Index<&'a str, Json>  for Json {
N
Niko Matsakis 已提交
1132
    fn index(&self, idx: & &str) -> &Json {
1133 1134 1135 1136 1137 1138 1139
        self.find(*idx).unwrap()
    }
}

impl ops::Index<uint, Json> for Json {
    fn index<'a>(&'a self, idx: &uint) -> &'a Json {
        match self {
1140
            &Json::Array(ref v) => v.index(idx),
C
Corey Farwell 已提交
1141
            _ => panic!("can only index Json with uint if it is an array")
1142 1143 1144 1145
        }
    }
}

1146
/// The output of the streaming parser.
1147
#[deriving(PartialEq, Clone, Show)]
1148 1149 1150
pub enum JsonEvent {
    ObjectStart,
    ObjectEnd,
C
Corey Farwell 已提交
1151 1152
    ArrayStart,
    ArrayEnd,
1153
    BooleanValue(bool),
1154 1155 1156
    I64Value(i64),
    U64Value(u64),
    F64Value(f64),
1157
    StringValue(string::String),
1158 1159 1160 1161
    NullValue,
    Error(ParserError),
}

1162
#[deriving(PartialEq, Show)]
1163
enum ParserState {
C
Corey Farwell 已提交
1164
    // Parse a value in an array, true means first element.
1165
    ParseArray(bool),
C
Corey Farwell 已提交
1166
    // Parse ',' or ']' after an element in an array.
C
Corey Farwell 已提交
1167
    ParseArrayComma,
1168 1169 1170 1171
    // Parse a key:value in an object, true means first element.
    ParseObject(bool),
    // Parse ',' or ']' after an element in an object.
    ParseObjectComma,
J
Joseph Crail 已提交
1172
    // Initial state.
1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190
    ParseStart,
    // Expecting the stream to end.
    ParseBeforeFinish,
    // Parsing can't continue.
    ParseFinished,
}

/// A Stack represents the current position of the parser in the logical
/// structure of the JSON stream.
/// For example foo.bar[3].x
pub struct Stack {
    stack: Vec<InternalStackElement>,
    str_buffer: Vec<u8>,
}

/// StackElements compose a Stack.
/// For example, Key("foo"), Key("bar"), Index(3) and Key("x") are the
/// StackElements compositing the stack that represents foo.bar[3].x
1191
#[deriving(PartialEq, Clone, Show)]
1192 1193 1194 1195 1196 1197 1198
pub enum StackElement<'l> {
    Index(u32),
    Key(&'l str),
}

// Internally, Key elements are stored as indices in a buffer to avoid
// allocating a string for every member of an object.
1199
#[deriving(PartialEq, Clone, Show)]
1200 1201 1202 1203 1204 1205 1206
enum InternalStackElement {
    InternalIndex(u32),
    InternalKey(u16, u16), // start, size
}

impl Stack {
    pub fn new() -> Stack {
A
Adolfo Ochagavía 已提交
1207
        Stack { stack: Vec::new(), str_buffer: Vec::new() }
1208 1209 1210 1211 1212
    }

    /// Returns The number of elements in the Stack.
    pub fn len(&self) -> uint { self.stack.len() }

A
Adolfo Ochagavía 已提交
1213 1214
    /// Returns true if the stack is empty.
    pub fn is_empty(&self) -> bool { self.stack.is_empty() }
1215 1216 1217 1218 1219

    /// Provides access to the StackElement at a given index.
    /// lower indices are at the bottom of the stack while higher indices are
    /// at the top.
    pub fn get<'l>(&'l self, idx: uint) -> StackElement<'l> {
N
Nick Cameron 已提交
1220
        match self.stack[idx] {
1221
            InternalIndex(i) => Index(i),
A
Adolfo Ochagavía 已提交
1222
            InternalKey(start, size) => {
A
Adolfo Ochagavía 已提交
1223
                Key(str::from_utf8(
1224
                    self.str_buffer[start as uint .. start as uint + size as uint]).unwrap())
A
Adolfo Ochagavía 已提交
1225
            }
1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265
        }
    }

    /// Compares this stack with an array of StackElements.
    pub fn is_equal_to(&self, rhs: &[StackElement]) -> bool {
        if self.stack.len() != rhs.len() { return false; }
        for i in range(0, rhs.len()) {
            if self.get(i) != rhs[i] { return false; }
        }
        return true;
    }

    /// Returns true if the bottom-most elements of this stack are the same as
    /// the ones passed as parameter.
    pub fn starts_with(&self, rhs: &[StackElement]) -> bool {
        if self.stack.len() < rhs.len() { return false; }
        for i in range(0, rhs.len()) {
            if self.get(i) != rhs[i] { return false; }
        }
        return true;
    }

    /// Returns true if the top-most elements of this stack are the same as
    /// the ones passed as parameter.
    pub fn ends_with(&self, rhs: &[StackElement]) -> bool {
        if self.stack.len() < rhs.len() { return false; }
        let offset = self.stack.len() - rhs.len();
        for i in range(0, rhs.len()) {
            if self.get(i + offset) != rhs[i] { return false; }
        }
        return true;
    }

    /// Returns the top-most element (if any).
    pub fn top<'l>(&'l self) -> Option<StackElement<'l>> {
        return match self.stack.last() {
            None => None,
            Some(&InternalIndex(i)) => Some(Index(i)),
            Some(&InternalKey(start, size)) => {
                Some(Key(str::from_utf8(
1266
                    self.str_buffer[start as uint .. (start+size) as uint]
1267 1268 1269 1270 1271 1272
                ).unwrap()))
            }
        }
    }

    // Used by Parser to insert Key elements at the top of the stack.
1273
    fn push_key(&mut self, key: string::String) {
1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290
        self.stack.push(InternalKey(self.str_buffer.len() as u16, key.len() as u16));
        for c in key.as_bytes().iter() {
            self.str_buffer.push(*c);
        }
    }

    // Used by Parser to insert Index elements at the top of the stack.
    fn push_index(&mut self, index: u32) {
        self.stack.push(InternalIndex(index));
    }

    // Used by Parser to remove the top-most element of the stack.
    fn pop(&mut self) {
        assert!(!self.is_empty());
        match *self.stack.last().unwrap() {
            InternalKey(_, sz) => {
                let new_size = self.str_buffer.len() - sz as uint;
A
Adolfo Ochagavía 已提交
1291
                self.str_buffer.truncate(new_size);
1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310
            }
            InternalIndex(_) => {}
        }
        self.stack.pop();
    }

    // Used by Parser to test whether the top-most element is an index.
    fn last_is_index(&self) -> bool {
        if self.is_empty() { return false; }
        return match *self.stack.last().unwrap() {
            InternalIndex(_) => true,
            _ => false,
        }
    }

    // Used by Parser to increment the index of the top-most element.
    fn bump_index(&mut self) {
        let len = self.stack.len();
        let idx = match *self.stack.last().unwrap() {
A
Adolfo Ochagavía 已提交
1311
            InternalIndex(i) => { i + 1 }
S
Steve Klabnik 已提交
1312
            _ => { panic!(); }
1313
        };
1314
        self.stack[len - 1] = InternalIndex(idx);
1315 1316 1317 1318 1319
    }
}

/// A streaming JSON parser implemented as an iterator of JsonEvent, consuming
/// an iterator of char.
G
Gary Linscott 已提交
1320
pub struct Parser<T> {
1321 1322 1323 1324
    rdr: T,
    ch: Option<char>,
    line: uint,
    col: uint,
1325 1326 1327
    // We maintain a stack representing where we are in the logical structure
    // of the JSON stream.
    stack: Stack,
J
Joseph Crail 已提交
1328
    // A state machine is kept to make it possible to interrupt and resume parsing.
1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350
    state: ParserState,
}

impl<T: Iterator<char>> Iterator<JsonEvent> for Parser<T> {
    fn next(&mut self) -> Option<JsonEvent> {
        if self.state == ParseFinished {
            return None;
        }

        if self.state == ParseBeforeFinish {
            self.parse_whitespace();
            // Make sure there is no trailing characters.
            if self.eof() {
                self.state = ParseFinished;
                return None;
            } else {
                return Some(self.error_event(TrailingCharacters));
            }
        }

        return Some(self.parse());
    }
1351 1352
}

1353
impl<T: Iterator<char>> Parser<T> {
1354
    /// Creates the JSON parser.
1355
    pub fn new(rdr: T) -> Parser<T> {
1356 1357
        let mut p = Parser {
            rdr: rdr,
1358
            ch: Some('\x00'),
1359 1360
            line: 1,
            col: 0,
1361 1362
            stack: Stack::new(),
            state: ParseStart,
1363 1364
        };
        p.bump();
1365
        return p;
1366
    }
E
Elly Jones 已提交
1367

1368 1369 1370
    /// Provides access to the current position in the logical structure of the
    /// JSON stream.
    pub fn stack<'l>(&'l self) -> &'l Stack {
1371
        return &self.stack;
1372
    }
1373

1374 1375
    fn eof(&self) -> bool { self.ch.is_none() }
    fn ch_or_null(&self) -> char { self.ch.unwrap_or('\x00') }
1376
    fn bump(&mut self) {
1377
        self.ch = self.rdr.next();
E
Elly Jones 已提交
1378

1379
        if self.ch_is('\n') {
1380 1381
            self.line += 1u;
            self.col = 1u;
G
Gary Linscott 已提交
1382 1383
        } else {
            self.col += 1u;
E
Elly Jones 已提交
1384
        }
1385 1386
    }

1387
    fn next_char(&mut self) -> Option<char> {
1388 1389 1390
        self.bump();
        self.ch
    }
1391 1392 1393
    fn ch_is(&self, c: char) -> bool {
        self.ch == Some(c)
    }
1394

1395 1396
    fn error<T>(&self, reason: ErrorCode) -> Result<T, ParserError> {
        Err(SyntaxError(reason, self.line, self.col))
1397 1398
    }

1399
    fn parse_whitespace(&mut self) {
1400 1401 1402 1403
        while self.ch_is(' ') ||
              self.ch_is('\n') ||
              self.ch_is('\t') ||
              self.ch_is('\r') { self.bump(); }
1404 1405
    }

1406
    fn parse_number(&mut self) -> JsonEvent {
1407
        let mut neg = false;
1408

1409
        if self.ch_is('-') {
1410
            self.bump();
1411
            neg = true;
E
Elly Jones 已提交
1412
        }
1413

1414
        let res = match self.parse_u64() {
1415 1416 1417
            Ok(res) => res,
            Err(e) => { return Error(e); }
        };
1418

1419 1420
        if self.ch_is('.') || self.ch_is('e') || self.ch_is('E') {
            let mut res = res as f64;
1421

1422 1423 1424 1425 1426 1427
            if self.ch_is('.') {
                res = match self.parse_decimal(res) {
                    Ok(res) => res,
                    Err(e) => { return Error(e); }
                };
            }
1428

1429 1430 1431 1432 1433 1434 1435
            if self.ch_is('e') || self.ch_is('E') {
                res = match self.parse_exponent(res) {
                    Ok(res) => res,
                    Err(e) => { return Error(e); }
                };
            }

1436 1437 1438 1439 1440
            if neg {
                res *= -1.0;
            }

            F64Value(res)
1441
        } else {
1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453
            if neg {
                let res = -(res as i64);

                // Make sure we didn't underflow.
                if res > 0 {
                    Error(SyntaxError(InvalidNumber, self.line, self.col))
                } else {
                    I64Value(res)
                }
            } else {
                U64Value(res)
            }
1454
        }
E
Elly Jones 已提交
1455 1456
    }

1457 1458 1459
    fn parse_u64(&mut self) -> Result<u64, ParserError> {
        let mut accum = 0;
        let last_accum = 0; // necessary to detect overflow.
E
Elly Jones 已提交
1460

1461 1462 1463
        match self.ch_or_null() {
            '0' => {
                self.bump();
1464

M
mrec 已提交
1465
                // A leading '0' must be the only digit before the decimal point.
1466
                match self.ch_or_null() {
1467
                    '0' ... '9' => return self.error(InvalidNumber),
1468 1469 1470
                    _ => ()
                }
            },
1471
            '1' ... '9' => {
1472 1473
                while !self.eof() {
                    match self.ch_or_null() {
1474
                        c @ '0' ... '9' => {
1475 1476 1477 1478 1479 1480
                            accum *= 10;
                            accum += (c as u64) - ('0' as u64);

                            // Detect overflow by comparing to the last value.
                            if accum <= last_accum { return self.error(InvalidNumber); }

1481 1482 1483 1484
                            self.bump();
                        }
                        _ => break,
                    }
1485 1486
                }
            }
1487
            _ => return self.error(InvalidNumber),
E
Elly Jones 已提交
1488
        }
1489 1490

        Ok(accum)
E
Elly Jones 已提交
1491 1492
    }

A
Adolfo Ochagavía 已提交
1493
    fn parse_decimal(&mut self, mut res: f64) -> Result<f64, ParserError> {
1494 1495 1496
        self.bump();

        // Make sure a digit follows the decimal place.
1497
        match self.ch_or_null() {
1498
            '0' ... '9' => (),
1499
             _ => return self.error(InvalidNumber)
1500 1501
        }

D
Daniel Micay 已提交
1502
        let mut dec = 1.0;
1503
        while !self.eof() {
1504
            match self.ch_or_null() {
1505
                c @ '0' ... '9' => {
1506 1507 1508 1509 1510
                    dec /= 10.0;
                    res += (((c as int) - ('0' as int)) as f64) * dec;
                    self.bump();
                }
                _ => break,
E
Elly Jones 已提交
1511 1512
            }
        }
1513

1514
        Ok(res)
E
Elly Jones 已提交
1515 1516
    }

1517
    fn parse_exponent(&mut self, mut res: f64) -> Result<f64, ParserError> {
1518 1519
        self.bump();

1520 1521
        let mut exp = 0u;
        let mut neg_exp = false;
1522

1523 1524 1525 1526 1527
        if self.ch_is('+') {
            self.bump();
        } else if self.ch_is('-') {
            self.bump();
            neg_exp = true;
1528 1529 1530
        }

        // Make sure a digit follows the exponent place.
1531
        match self.ch_or_null() {
1532
            '0' ... '9' => (),
1533
            _ => return self.error(InvalidNumber)
1534 1535
        }
        while !self.eof() {
1536
            match self.ch_or_null() {
1537
                c @ '0' ... '9' => {
1538 1539
                    exp *= 10;
                    exp += (c as uint) - ('0' as uint);
1540

1541 1542 1543
                    self.bump();
                }
                _ => break
1544 1545 1546
            }
        }

1547
        let exp = 10_f64.powi(exp as i32);
1548 1549 1550 1551 1552 1553
        if neg_exp {
            res /= exp;
        } else {
            res *= exp;
        }

1554
        Ok(res)
E
Elly Jones 已提交
1555 1556
    }

1557
    fn decode_hex_escape(&mut self) -> Result<u16, ParserError> {
1558 1559
        let mut i = 0u;
        let mut n = 0u16;
A
Adolfo Ochagavía 已提交
1560
        while i < 4 && !self.eof() {
1561 1562
            self.bump();
            n = match self.ch_or_null() {
1563
                c @ '0' ... '9' => n * 16 + ((c as u16) - ('0' as u16)),
A
Adolfo Ochagavía 已提交
1564 1565 1566 1567 1568 1569
                'a' | 'A' => n * 16 + 10,
                'b' | 'B' => n * 16 + 11,
                'c' | 'C' => n * 16 + 12,
                'd' | 'D' => n * 16 + 13,
                'e' | 'E' => n * 16 + 14,
                'f' | 'F' => n * 16 + 15,
1570
                _ => return self.error(InvalidEscape)
1571 1572 1573 1574 1575 1576
            };

            i += 1u;
        }

        // Error out if we didn't parse 4 digits.
A
Adolfo Ochagavía 已提交
1577
        if i != 4 {
1578
            return self.error(InvalidEscape);
1579 1580 1581 1582 1583
        }

        Ok(n)
    }

1584
    fn parse_str(&mut self) -> Result<string::String, ParserError> {
1585
        let mut escape = false;
1586
        let mut res = string::String::new();
1587

G
Gary Linscott 已提交
1588
        loop {
1589
            self.bump();
G
Gary Linscott 已提交
1590
            if self.eof() {
1591
                return self.error(EOFWhileParsingString);
G
Gary Linscott 已提交
1592
            }
1593

H
Huon Wilson 已提交
1594
            if escape {
1595
                match self.ch_or_null() {
1596 1597 1598 1599 1600 1601 1602 1603
                    '"' => res.push('"'),
                    '\\' => res.push('\\'),
                    '/' => res.push('/'),
                    'b' => res.push('\x08'),
                    'f' => res.push('\x0c'),
                    'n' => res.push('\n'),
                    'r' => res.push('\r'),
                    't' => res.push('\t'),
1604
                    'u' => match try!(self.decode_hex_escape()) {
1605 1606 1607
                        0xDC00 ... 0xDFFF => {
                            return self.error(LoneLeadingSurrogateInHexEscape)
                        }
1608 1609 1610

                        // Non-BMP characters are encoded as a sequence of
                        // two hex escapes, representing UTF-16 surrogates.
1611
                        n1 @ 0xD800 ... 0xDBFF => {
A
Adolfo Ochagavía 已提交
1612
                            match (self.next_char(), self.next_char()) {
1613
                                (Some('\\'), Some('u')) => (),
1614
                                _ => return self.error(UnexpectedEndOfHexEscape),
1615
                            }
1616

1617
                            let buf = [n1, try!(self.decode_hex_escape())];
A
Alex Crichton 已提交
1618 1619
                            match unicode_str::utf16_items(&buf).next() {
                                Some(Utf16Item::ScalarValue(c)) => res.push(c),
1620
                                _ => return self.error(LoneLeadingSurrogateInHexEscape),
1621
                            }
1622 1623
                        }

1624
                        n => match char::from_u32(n as u32) {
1625
                            Some(c) => res.push(c),
1626
                            None => return self.error(InvalidUnicodeCodePoint),
1627 1628
                        },
                    },
1629
                    _ => return self.error(InvalidEscape),
1630 1631
                }
                escape = false;
1632
            } else if self.ch_is('\\') {
1633 1634
                escape = true;
            } else {
1635
                match self.ch {
1636 1637
                    Some('"') => {
                        self.bump();
1638
                        return Ok(res);
1639
                    },
1640
                    Some(c) => res.push(c),
1641
                    None => unreachable!()
1642
                }
E
Elly Jones 已提交
1643 1644 1645 1646
            }
        }
    }

1647 1648 1649 1650
    // Invoked at each iteration, consumes the stream until it has enough
    // information to return a JsonEvent.
    // Manages an internal state so that parsing can be interrupted and resumed.
    // Also keeps track of the position in the logical structure of the json
J
Joseph Crail 已提交
1651
    // stream int the form of a stack that can be queried by the user using the
1652 1653 1654 1655
    // stack() method.
    fn parse(&mut self) -> JsonEvent {
        loop {
            // The only paths where the loop can spin a new iteration
C
Corey Farwell 已提交
1656
            // are in the cases ParseArrayComma and ParseObjectComma if ','
1657
            // is parsed. In these cases the state is set to (respectively)
1658
            // ParseArray(false) and ParseObject(false), which always return,
1659 1660 1661
            // so there is no risk of getting stuck in an infinite loop.
            // All other paths return before the end of the loop's iteration.
            self.parse_whitespace();
1662

1663 1664 1665 1666
            match self.state {
                ParseStart => {
                    return self.parse_start();
                }
1667
                ParseArray(first) => {
C
Corey Farwell 已提交
1668
                    return self.parse_array(first);
1669
                }
C
Corey Farwell 已提交
1670 1671
                ParseArrayComma => {
                    match self.parse_array_comma_or_end() {
1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697
                        Some(evt) => { return evt; }
                        None => {}
                    }
                }
                ParseObject(first) => {
                    return self.parse_object(first);
                }
                ParseObjectComma => {
                    self.stack.pop();
                    if self.ch_is(',') {
                        self.state = ParseObject(false);
                        self.bump();
                    } else {
                        return self.parse_object_end();
                    }
                }
                _ => {
                    return self.error_event(InvalidSyntax);
                }
            }
        }
    }

    fn parse_start(&mut self) -> JsonEvent {
        let val = self.parse_value();
        self.state = match val {
1698 1699 1700 1701
            Error(_) => ParseFinished,
            ArrayStart => ParseArray(true),
            ObjectStart => ParseObject(true),
            _ => ParseBeforeFinish,
1702 1703 1704
        };
        return val;
    }
1705

C
Corey Farwell 已提交
1706
    fn parse_array(&mut self, first: bool) -> JsonEvent {
1707
        if self.ch_is(']') {
1708
            if !first {
1709
                self.error_event(InvalidSyntax)
1710
            } else {
1711 1712 1713
                self.state = if self.stack.is_empty() {
                    ParseBeforeFinish
                } else if self.stack.last_is_index() {
C
Corey Farwell 已提交
1714
                    ParseArrayComma
1715 1716
                } else {
                    ParseObjectComma
1717 1718 1719
                };
                self.bump();
                ArrayEnd
1720
            }
1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732
        } else {
            if first {
                self.stack.push_index(0);
            }
            let val = self.parse_value();
            self.state = match val {
                Error(_) => ParseFinished,
                ArrayStart => ParseArray(true),
                ObjectStart => ParseObject(true),
                _ => ParseArrayComma,
            };
            val
1733
        }
1734
    }
1735

C
Corey Farwell 已提交
1736
    fn parse_array_comma_or_end(&mut self) -> Option<JsonEvent> {
1737 1738
        if self.ch_is(',') {
            self.stack.bump_index();
1739
            self.state = ParseArray(false);
1740
            self.bump();
1741
            None
1742 1743
        } else if self.ch_is(']') {
            self.stack.pop();
1744 1745 1746 1747
            self.state = if self.stack.is_empty() {
                ParseBeforeFinish
            } else if self.stack.last_is_index() {
                ParseArrayComma
1748
            } else {
1749 1750
                ParseObjectComma
            };
1751
            self.bump();
1752
            Some(ArrayEnd)
1753
        } else if self.eof() {
1754
            Some(self.error_event(EOFWhileParsingArray))
1755
        } else {
1756
            Some(self.error_event(InvalidSyntax))
1757
        }
E
Elly Jones 已提交
1758 1759
    }

1760 1761 1762
    fn parse_object(&mut self, first: bool) -> JsonEvent {
        if self.ch_is('}') {
            if !first {
1763 1764 1765 1766 1767
                if self.stack.is_empty() {
                    return self.error_event(TrailingComma);
                } else {
                    self.stack.pop();
                }
1768
            }
1769 1770 1771 1772
            self.state = if self.stack.is_empty() {
                ParseBeforeFinish
            } else if self.stack.last_is_index() {
                ParseArrayComma
1773
            } else {
1774 1775
                ParseObjectComma
            };
1776 1777 1778 1779 1780 1781 1782 1783 1784 1785
            self.bump();
            return ObjectEnd;
        }
        if self.eof() {
            return self.error_event(EOFWhileParsingObject);
        }
        if !self.ch_is('"') {
            return self.error_event(KeyMustBeAString);
        }
        let s = match self.parse_str() {
1786
            Ok(s) => s,
1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798
            Err(e) => {
                self.state = ParseFinished;
                return Error(e);
            }
        };
        self.parse_whitespace();
        if self.eof() {
            return self.error_event(EOFWhileParsingObject);
        } else if self.ch_or_null() != ':' {
            return self.error_event(ExpectedColon);
        }
        self.stack.push_key(s);
1799 1800 1801
        self.bump();
        self.parse_whitespace();

1802
        let val = self.parse_value();
1803

1804
        self.state = match val {
1805 1806 1807 1808
            Error(_) => ParseFinished,
            ArrayStart => ParseArray(true),
            ObjectStart => ParseObject(true),
            _ => ParseObjectComma,
1809 1810 1811 1812 1813
        };
        return val;
    }

    fn parse_object_end(&mut self) -> JsonEvent {
1814
        if self.ch_is('}') {
1815 1816 1817 1818
            self.state = if self.stack.is_empty() {
                ParseBeforeFinish
            } else if self.stack.last_is_index() {
                ParseArrayComma
1819
            } else {
1820 1821
                ParseObjectComma
            };
1822
            self.bump();
A
Adolfo Ochagavía 已提交
1823
            ObjectEnd
1824
        } else if self.eof() {
A
Adolfo Ochagavía 已提交
1825
            self.error_event(EOFWhileParsingObject)
1826
        } else {
A
Adolfo Ochagavía 已提交
1827
            self.error_event(InvalidSyntax)
1828
        }
1829
    }
1830

1831 1832 1833
    fn parse_value(&mut self) -> JsonEvent {
        if self.eof() { return self.error_event(EOFWhileParsingValue); }
        match self.ch_or_null() {
A
Adolfo Ochagavía 已提交
1834 1835 1836
            'n' => { self.parse_ident("ull", NullValue) }
            't' => { self.parse_ident("rue", BooleanValue(true)) }
            'f' => { self.parse_ident("alse", BooleanValue(false)) }
1837
            '0' ... '9' | '-' => self.parse_number(),
A
Adolfo Ochagavía 已提交
1838
            '"' => match self.parse_str() {
1839 1840 1841 1842 1843
                Ok(s) => StringValue(s),
                Err(e) => Error(e),
            },
            '[' => {
                self.bump();
C
Corey Farwell 已提交
1844
                ArrayStart
1845 1846 1847
            }
            '{' => {
                self.bump();
A
Adolfo Ochagavía 已提交
1848
                ObjectStart
1849
            }
A
Adolfo Ochagavía 已提交
1850
            _ => { self.error_event(InvalidSyntax) }
1851 1852
        }
    }
1853

1854 1855 1856 1857 1858 1859 1860 1861
    fn parse_ident(&mut self, ident: &str, value: JsonEvent) -> JsonEvent {
        if ident.chars().all(|c| Some(c) == self.next_char()) {
            self.bump();
            value
        } else {
            Error(SyntaxError(InvalidSyntax, self.line, self.col))
        }
    }
1862

1863 1864 1865 1866 1867
    fn error_event(&mut self, reason: ErrorCode) -> JsonEvent {
        self.state = ParseFinished;
        Error(SyntaxError(reason, self.line, self.col))
    }
}
1868

1869 1870 1871 1872 1873
/// A Builder consumes a json::Parser to create a generic Json structure.
pub struct Builder<T> {
    parser: Parser<T>,
    token: Option<JsonEvent>,
}
1874

1875 1876 1877
impl<T: Iterator<char>> Builder<T> {
    /// Create a JSON Builder.
    pub fn new(src: T) -> Builder<T> {
A
Adolfo Ochagavía 已提交
1878
        Builder { parser: Parser::new(src), token: None, }
1879 1880 1881 1882 1883 1884 1885 1886 1887 1888
    }

    // Decode a Json value from a Parser.
    pub fn build(&mut self) -> Result<Json, BuilderError> {
        self.bump();
        let result = self.build_value();
        self.bump();
        match self.token {
            None => {}
            Some(Error(e)) => { return Err(e); }
S
Steve Klabnik 已提交
1889
            ref tok => { panic!("unexpected token {}", tok.clone()); }
1890
        }
A
Adolfo Ochagavía 已提交
1891
        result
1892 1893 1894 1895 1896 1897 1898
    }

    fn bump(&mut self) {
        self.token = self.parser.next();
    }

    fn build_value(&mut self) -> Result<Json, BuilderError> {
1899 1900 1901 1902 1903 1904
        return match self.token {
            Some(NullValue) => Ok(Json::Null),
            Some(I64Value(n)) => Ok(Json::I64(n)),
            Some(U64Value(n)) => Ok(Json::U64(n)),
            Some(F64Value(n)) => Ok(Json::F64(n)),
            Some(BooleanValue(b)) => Ok(Json::Boolean(b)),
1905
            Some(StringValue(ref mut s)) => {
1906
                let mut temp = string::String::new();
1907
                swap(s, &mut temp);
1908
                Ok(Json::String(temp))
1909
            }
1910 1911 1912 1913 1914 1915
            Some(Error(e)) => Err(e),
            Some(ArrayStart) => self.build_array(),
            Some(ObjectStart) => self.build_object(),
            Some(ObjectEnd) => self.parser.error(InvalidSyntax),
            Some(ArrayEnd) => self.parser.error(InvalidSyntax),
            None => self.parser.error(EOFWhileParsingValue),
1916 1917
        }
    }
1918

C
Corey Farwell 已提交
1919
    fn build_array(&mut self) -> Result<Json, BuilderError> {
1920 1921 1922 1923
        self.bump();
        let mut values = Vec::new();

        loop {
C
Corey Farwell 已提交
1924
            if self.token == Some(ArrayEnd) {
1925
                return Ok(Json::Array(values.into_iter().collect()));
1926 1927 1928 1929
            }
            match self.build_value() {
                Ok(v) => values.push(v),
                Err(e) => { return Err(e) }
1930
            }
1931
            self.bump();
1932
        }
1933
    }
1934

1935 1936 1937
    fn build_object(&mut self) -> Result<Json, BuilderError> {
        self.bump();

A
Alexis Beingessner 已提交
1938
        let mut values = BTreeMap::new();
1939

A
Adolfo Ochagavía 已提交
1940
        loop {
1941
            match self.token {
1942
                Some(ObjectEnd) => { return Ok(Json::Object(values)); }
1943 1944 1945 1946 1947
                Some(Error(e)) => { return Err(e); }
                None => { break; }
                _ => {}
            }
            let key = match self.parser.stack().top() {
1948
                Some(Key(k)) => { k.to_string() }
S
Steve Klabnik 已提交
1949
                _ => { panic!("invalid state"); }
1950 1951 1952 1953 1954 1955 1956 1957
            };
            match self.build_value() {
                Ok(value) => { values.insert(key, value); }
                Err(e) => { return Err(e); }
            }
            self.bump();
        }
        return self.parser.error(EOFWhileParsingObject);
L
Lenny222 已提交
1958 1959 1960
    }
}

A
Alex Crichton 已提交
1961
/// Decodes a json value from an `&mut io::Reader`
1962
pub fn from_reader(rdr: &mut io::Reader) -> Result<Json, BuilderError> {
A
Alex Crichton 已提交
1963
    let contents = match rdr.read_to_end() {
1964
        Ok(c)  => c,
1965
        Err(e) => return Err(io_error_to_error(e))
A
Alex Crichton 已提交
1966
    };
A
Alex Crichton 已提交
1967
    let s = match str::from_utf8(contents.as_slice()).ok() {
1968 1969
        Some(s) => s,
        _       => return Err(SyntaxError(NotUtf8, 0, 0))
A
Alex Crichton 已提交
1970
    };
1971
    let mut builder = Builder::new(s.chars());
1972
    builder.build()
E
Elly Jones 已提交
1973 1974
}

1975
/// Decodes a json value from a string
1976 1977
pub fn from_str(s: &str) -> Result<Json, BuilderError> {
    let mut builder = Builder::new(s.chars());
A
Adolfo Ochagavía 已提交
1978
    builder.build()
1979 1980
}

1981
/// A structure to decode JSON to values in rust.
1982
pub struct Decoder {
1983
    stack: Vec<Json>,
1984 1985
}

1986 1987
impl Decoder {
    /// Creates a new decoder instance for decoding the specified JSON value.
1988
    pub fn new(json: Json) -> Decoder {
A
Adolfo Ochagavía 已提交
1989
        Decoder { stack: vec![json] }
1990
    }
1991 1992
}

1993
impl Decoder {
S
Sean McArthur 已提交
1994 1995
    fn pop(&mut self) -> Json {
        self.stack.pop().unwrap()
1996 1997 1998
    }
}

1999
macro_rules! expect {
S
Sean McArthur 已提交
2000 2001
    ($e:expr, Null) => ({
        match $e {
2002
            Json::Null => Ok(()),
A
Alex Crichton 已提交
2003
            other => Err(ExpectedError("Null".to_string(),
A
Alex Crichton 已提交
2004
                                       format!("{}", other)))
S
Sean McArthur 已提交
2005 2006 2007 2008
        }
    });
    ($e:expr, $t:ident) => ({
        match $e {
2009
            Json::$t(v) => Ok(v),
2010
            other => {
2011
                Err(ExpectedError(stringify!($t).to_string(),
A
Alex Crichton 已提交
2012
                                  format!("{}", other)))
2013
            }
2014
        }
S
Sean McArthur 已提交
2015
    })
2016
}
S
Sean McArthur 已提交
2017

2018 2019 2020 2021
macro_rules! read_primitive {
    ($name:ident, $ty:ty) => {
        fn $name(&mut self) -> DecodeResult<$ty> {
            match self.pop() {
B
Barosl Lee 已提交
2022 2023
                Json::I64(f) => match num::cast(f) {
                    Some(f) => Ok(f),
A
Alex Crichton 已提交
2024
                    None => Err(ExpectedError("Number".to_string(), format!("{}", f))),
2025
                },
B
Barosl Lee 已提交
2026 2027
                Json::U64(f) => match num::cast(f) {
                    Some(f) => Ok(f),
A
Alex Crichton 已提交
2028
                    None => Err(ExpectedError("Number".to_string(), format!("{}", f))),
B
Barosl Lee 已提交
2029
                },
A
Alex Crichton 已提交
2030
                Json::F64(f) => Err(ExpectedError("Integer".to_string(), format!("{}", f))),
B
Barosl Lee 已提交
2031 2032 2033 2034
                // re: #12967.. a type w/ numeric keys (ie HashMap<uint, V> etc)
                // is going to have a string here, as per JSON spec.
                Json::String(s) => match std::str::from_str(s.as_slice()) {
                    Some(f) => Ok(f),
A
Alex Crichton 已提交
2035
                    None => Err(ExpectedError("Number".to_string(), s)),
2036
                },
A
Alex Crichton 已提交
2037
                value => Err(ExpectedError("Number".to_string(), format!("{}", value))),
2038 2039 2040 2041 2042
            }
        }
    }
}

2043
impl ::Decoder<DecoderError> for Decoder {
S
Sean McArthur 已提交
2044
    fn read_nil(&mut self) -> DecodeResult<()> {
A
Adolfo Ochagavía 已提交
2045
        expect!(self.pop(), Null)
2046 2047
    }

2048 2049 2050 2051 2052 2053 2054 2055 2056 2057
    read_primitive! { read_uint, uint }
    read_primitive! { read_u8, u8 }
    read_primitive! { read_u16, u16 }
    read_primitive! { read_u32, u32 }
    read_primitive! { read_u64, u64 }
    read_primitive! { read_int, int }
    read_primitive! { read_i8, i8 }
    read_primitive! { read_i16, i16 }
    read_primitive! { read_i32, i32 }
    read_primitive! { read_i64, i64 }
2058

2059
    fn read_f32(&mut self) -> DecodeResult<f32> { self.read_f64().map(|x| x as f32) }
2060

S
Sean McArthur 已提交
2061 2062
    fn read_f64(&mut self) -> DecodeResult<f64> {
        match self.pop() {
2063 2064 2065 2066
            Json::I64(f) => Ok(f as f64),
            Json::U64(f) => Ok(f as f64),
            Json::F64(f) => Ok(f),
            Json::String(s) => {
2067
                // re: #12967.. a type w/ numeric keys (ie HashMap<uint, V> etc)
A
Adolfo Ochagavía 已提交
2068
                // is going to have a string here, as per JSON spec.
A
Alex Crichton 已提交
2069
                match s.parse() {
2070
                    Some(f) => Ok(f),
A
Alex Crichton 已提交
2071
                    None => Err(ExpectedError("Number".to_string(), s)),
2072
                }
2073
            },
2074
            Json::Null => Ok(f64::NAN),
A
Alex Crichton 已提交
2075
            value => Err(ExpectedError("Number".to_string(), format!("{}", value)))
2076 2077
        }
    }
2078

2079 2080 2081
    fn read_bool(&mut self) -> DecodeResult<bool> {
        expect!(self.pop(), Boolean)
    }
2082

S
Sean McArthur 已提交
2083 2084
    fn read_char(&mut self) -> DecodeResult<char> {
        let s = try!(self.read_str());
2085
        {
2086
            let mut it = s.chars();
2087 2088
            match (it.next(), it.next()) {
                // exactly one character
S
Sean McArthur 已提交
2089
                (Some(c), None) => return Ok(c),
2090 2091 2092
                _ => ()
            }
        }
A
Alex Crichton 已提交
2093
        Err(ExpectedError("single character string".to_string(), format!("{}", s)))
2094 2095
    }

2096
    fn read_str(&mut self) -> DecodeResult<string::String> {
A
Adolfo Ochagavía 已提交
2097
        expect!(self.pop(), String)
2098 2099
    }

E
Erick Tryzelaar 已提交
2100
    fn read_enum<T, F>(&mut self, _name: &str, f: F) -> DecodeResult<T> where
2101 2102
        F: FnOnce(&mut Decoder) -> DecodeResult<T>,
    {
2103 2104 2105
        f(self)
    }

2106 2107 2108
    fn read_enum_variant<T, F>(&mut self, names: &[&str],
                               mut f: F) -> DecodeResult<T>
        where F: FnMut(&mut Decoder, uint) -> DecodeResult<T>,
2109
    {
S
Sean McArthur 已提交
2110
        let name = match self.pop() {
2111 2112
            Json::String(s) => s,
            Json::Object(mut o) => {
A
Alex Crichton 已提交
2113
                let n = match o.remove(&"variant".to_string()) {
2114
                    Some(Json::String(s)) => s,
2115
                    Some(val) => {
A
Alex Crichton 已提交
2116
                        return Err(ExpectedError("String".to_string(), format!("{}", val)))
2117 2118
                    }
                    None => {
A
Alex Crichton 已提交
2119
                        return Err(MissingFieldError("variant".to_string()))
2120
                    }
2121
                };
A
Alex Crichton 已提交
2122
                match o.remove(&"fields".to_string()) {
2123
                    Some(Json::Array(l)) => {
A
Aaron Turon 已提交
2124
                        for field in l.into_iter().rev() {
A
Adolfo Ochagavía 已提交
2125
                            self.stack.push(field);
2126 2127
                        }
                    },
2128
                    Some(val) => {
A
Alex Crichton 已提交
2129
                        return Err(ExpectedError("Array".to_string(), format!("{}", val)))
2130 2131
                    }
                    None => {
A
Alex Crichton 已提交
2132
                        return Err(MissingFieldError("fields".to_string()))
2133
                    }
2134
                }
2135
                n
2136
            }
2137
            json => {
A
Alex Crichton 已提交
2138
                return Err(ExpectedError("String or Object".to_string(), format!("{}", json)))
2139
            }
2140
        };
A
Alex Crichton 已提交
2141
        let idx = match names.iter().position(|n| *n == name[]) {
2142
            Some(idx) => idx,
S
Sean McArthur 已提交
2143
            None => return Err(UnknownVariantError(name))
2144 2145 2146 2147
        };
        f(self, idx)
    }

E
Erick Tryzelaar 已提交
2148
    fn read_enum_variant_arg<T, F>(&mut self, _idx: uint, f: F) -> DecodeResult<T> where
2149 2150
        F: FnOnce(&mut Decoder) -> DecodeResult<T>,
    {
2151 2152 2153
        f(self)
    }

2154
    fn read_enum_struct_variant<T, F>(&mut self, names: &[&str], f: F) -> DecodeResult<T> where
2155
        F: FnMut(&mut Decoder, uint) -> DecodeResult<T>,
2156
    {
2157 2158 2159 2160
        self.read_enum_variant(names, f)
    }


2161
    fn read_enum_struct_variant_field<T, F>(&mut self,
E
Erick Tryzelaar 已提交
2162
                                         _name: &str,
2163
                                         idx: uint,
2164 2165 2166 2167
                                         f: F)
                                         -> DecodeResult<T> where
        F: FnOnce(&mut Decoder) -> DecodeResult<T>,
    {
2168 2169 2170
        self.read_enum_variant_arg(idx, f)
    }

E
Erick Tryzelaar 已提交
2171
    fn read_struct<T, F>(&mut self, _name: &str, _len: uint, f: F) -> DecodeResult<T> where
2172 2173
        F: FnOnce(&mut Decoder) -> DecodeResult<T>,
    {
S
Sean McArthur 已提交
2174 2175 2176
        let value = try!(f(self));
        self.pop();
        Ok(value)
2177 2178
    }

2179 2180
    fn read_struct_field<T, F>(&mut self,
                               name: &str,
E
Erick Tryzelaar 已提交
2181
                               _idx: uint,
2182 2183 2184 2185
                               f: F)
                               -> DecodeResult<T> where
        F: FnOnce(&mut Decoder) -> DecodeResult<T>,
    {
S
Sean McArthur 已提交
2186 2187
        let mut obj = try!(expect!(self.pop(), Object));

2188
        let value = match obj.remove(&name.to_string()) {
2189 2190 2191
            None => {
                // Add a Null and try to parse it as an Option<_>
                // to get None as a default value.
2192
                self.stack.push(Json::Null);
2193 2194 2195 2196 2197
                match f(self) {
                    Ok(x) => x,
                    Err(_) => return Err(MissingFieldError(name.to_string())),
                }
            },
S
Sean McArthur 已提交
2198 2199 2200
            Some(json) => {
                self.stack.push(json);
                try!(f(self))
2201
            }
S
Sean McArthur 已提交
2202
        };
2203
        self.stack.push(Json::Object(obj));
S
Sean McArthur 已提交
2204
        Ok(value)
2205 2206
    }

2207 2208 2209 2210
    fn read_tuple<T, F>(&mut self, tuple_len: uint, f: F) -> DecodeResult<T> where
        F: FnOnce(&mut Decoder) -> DecodeResult<T>,
    {
        self.read_seq(move |d, len| {
2211 2212 2213 2214 2215
            if len == tuple_len {
                f(d)
            } else {
                Err(ExpectedError(format!("Tuple{}", tuple_len), format!("Tuple{}", len)))
            }
2216
        })
2217 2218
    }

2219 2220 2221
    fn read_tuple_arg<T, F>(&mut self, idx: uint, f: F) -> DecodeResult<T> where
        F: FnOnce(&mut Decoder) -> DecodeResult<T>,
    {
2222 2223 2224
        self.read_seq_elt(idx, f)
    }

2225
    fn read_tuple_struct<T, F>(&mut self,
E
Erick Tryzelaar 已提交
2226
                               _name: &str,
2227 2228 2229 2230 2231
                               len: uint,
                               f: F)
                               -> DecodeResult<T> where
        F: FnOnce(&mut Decoder) -> DecodeResult<T>,
    {
2232
        self.read_tuple(len, f)
2233 2234
    }

2235 2236 2237 2238 2239 2240
    fn read_tuple_struct_arg<T, F>(&mut self,
                                   idx: uint,
                                   f: F)
                                   -> DecodeResult<T> where
        F: FnOnce(&mut Decoder) -> DecodeResult<T>,
    {
2241 2242 2243
        self.read_tuple_arg(idx, f)
    }

2244 2245
    fn read_option<T, F>(&mut self, mut f: F) -> DecodeResult<T> where
        F: FnMut(&mut Decoder, bool) -> DecodeResult<T>,
2246
    {
S
Sean McArthur 已提交
2247
        match self.pop() {
2248
            Json::Null => f(self, false),
2249 2250 2251 2252
            value => { self.stack.push(value); f(self, true) }
        }
    }

2253 2254 2255
    fn read_seq<T, F>(&mut self, f: F) -> DecodeResult<T> where
        F: FnOnce(&mut Decoder, uint) -> DecodeResult<T>,
    {
C
Corey Farwell 已提交
2256 2257 2258
        let array = try!(expect!(self.pop(), Array));
        let len = array.len();
        for v in array.into_iter().rev() {
S
Sean McArthur 已提交
2259 2260
            self.stack.push(v);
        }
2261 2262 2263
        f(self, len)
    }

E
Erick Tryzelaar 已提交
2264
    fn read_seq_elt<T, F>(&mut self, _idx: uint, f: F) -> DecodeResult<T> where
2265 2266
        F: FnOnce(&mut Decoder) -> DecodeResult<T>,
    {
2267 2268 2269
        f(self)
    }

2270 2271 2272
    fn read_map<T, F>(&mut self, f: F) -> DecodeResult<T> where
        F: FnOnce(&mut Decoder, uint) -> DecodeResult<T>,
    {
S
Sean McArthur 已提交
2273 2274
        let obj = try!(expect!(self.pop(), Object));
        let len = obj.len();
A
Aaron Turon 已提交
2275
        for (key, value) in obj.into_iter() {
S
Sean McArthur 已提交
2276
            self.stack.push(value);
2277
            self.stack.push(Json::String(key));
S
Sean McArthur 已提交
2278
        }
2279 2280 2281
        f(self, len)
    }

E
Erick Tryzelaar 已提交
2282
    fn read_map_elt_key<T, F>(&mut self, _idx: uint, f: F) -> DecodeResult<T> where
2283 2284
       F: FnOnce(&mut Decoder) -> DecodeResult<T>,
    {
2285 2286 2287
        f(self)
    }

E
Erick Tryzelaar 已提交
2288
    fn read_map_elt_val<T, F>(&mut self, _idx: uint, f: F) -> DecodeResult<T> where
2289 2290
       F: FnOnce(&mut Decoder) -> DecodeResult<T>,
    {
2291 2292
        f(self)
    }
2293 2294 2295 2296

    fn error(&mut self, err: &str) -> DecoderError {
        ApplicationError(err.to_string())
    }
2297 2298
}

2299
/// A trait for converting values to JSON
J
Jorge Aparicio 已提交
2300
pub trait ToJson for Sized? {
2301 2302 2303
    /// Converts the value of `self` to an instance of JSON
    fn to_json(&self) -> Json;
}
2304

2305
macro_rules! to_json_impl_i64 {
2306 2307
    ($($t:ty), +) => (
        $(impl ToJson for $t {
2308
            fn to_json(&self) -> Json { Json::I64(*self as i64) }
2309 2310
        })+
    )
2311
}
2312

2313
to_json_impl_i64! { int, i8, i16, i32, i64 }
2314

2315
macro_rules! to_json_impl_u64 {
A
Adolfo Ochagavía 已提交
2316 2317
    ($($t:ty), +) => (
        $(impl ToJson for $t {
2318
            fn to_json(&self) -> Json { Json::U64(*self as u64) }
A
Adolfo Ochagavía 已提交
2319 2320
        })+
    )
2321
}
2322

2323
to_json_impl_u64! { uint, u8, u16, u32, u64 }
2324

A
Adolfo Ochagavía 已提交
2325 2326
impl ToJson for Json {
    fn to_json(&self) -> Json { self.clone() }
2327 2328
}

2329
impl ToJson for f32 {
M
mrec 已提交
2330
    fn to_json(&self) -> Json { (*self as f64).to_json() }
2331 2332
}

2333
impl ToJson for f64 {
M
mrec 已提交
2334 2335
    fn to_json(&self) -> Json {
        match self.classify() {
T
Tobias Bucher 已提交
2336
            Fp::Nan | Fp::Infinite => Json::Null,
2337
            _                  => Json::F64(*self)
M
mrec 已提交
2338 2339
        }
    }
2340 2341
}

2342
impl ToJson for () {
2343
    fn to_json(&self) -> Json { Json::Null }
2344 2345
}

2346
impl ToJson for bool {
2347
    fn to_json(&self) -> Json { Json::Boolean(*self) }
2348 2349
}

2350
impl ToJson for str {
A
Alex Crichton 已提交
2351
    fn to_json(&self) -> Json { Json::String(self.to_string()) }
2352 2353
}

2354
impl ToJson for string::String {
2355
    fn to_json(&self) -> Json { Json::String((*self).clone()) }
2356 2357
}

2358 2359 2360 2361 2362 2363 2364 2365 2366
macro_rules! tuple_impl {
    // use variables to indicate the arity of the tuple
    ($($tyvar:ident),* ) => {
        // the trailing commas are for the 1 tuple
        impl<
            $( $tyvar : ToJson ),*
            > ToJson for ( $( $tyvar ),* , ) {

            #[inline]
2367
            #[allow(non_snake_case)]
2368 2369
            fn to_json(&self) -> Json {
                match *self {
2370
                    ($(ref $tyvar),*,) => Json::Array(vec![$($tyvar.to_json()),*])
2371
                }
A
Adolfo Ochagavía 已提交
2372
            }
2373
        }
2374 2375 2376
    }
}

2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388
tuple_impl!{A}
tuple_impl!{A, B}
tuple_impl!{A, B, C}
tuple_impl!{A, B, C, D}
tuple_impl!{A, B, C, D, E}
tuple_impl!{A, B, C, D, E, F}
tuple_impl!{A, B, C, D, E, F, G}
tuple_impl!{A, B, C, D, E, F, G, H}
tuple_impl!{A, B, C, D, E, F, G, H, I}
tuple_impl!{A, B, C, D, E, F, G, H, I, J}
tuple_impl!{A, B, C, D, E, F, G, H, I, J, K}
tuple_impl!{A, B, C, D, E, F, G, H, I, J, K, L}
2389

J
Jorge Aparicio 已提交
2390
impl<A: ToJson> ToJson for [A] {
2391
    fn to_json(&self) -> Json { Json::Array(self.iter().map(|elt| elt.to_json()).collect()) }
2392 2393
}

A
Adolfo Ochagavía 已提交
2394
impl<A: ToJson> ToJson for Vec<A> {
2395
    fn to_json(&self) -> Json { Json::Array(self.iter().map(|elt| elt.to_json()).collect()) }
2396 2397
}

A
Alexis Beingessner 已提交
2398
impl<A: ToJson> ToJson for BTreeMap<string::String, A> {
B
Ben Striegel 已提交
2399
    fn to_json(&self) -> Json {
A
Alexis Beingessner 已提交
2400
        let mut d = BTreeMap::new();
D
Daniel Micay 已提交
2401
        for (key, value) in self.iter() {
2402
            d.insert((*key).clone(), value.to_json());
2403
        }
2404
        Json::Object(d)
2405 2406 2407
    }
}

2408
impl<A: ToJson> ToJson for HashMap<string::String, A> {
G
Graydon Hoare 已提交
2409
    fn to_json(&self) -> Json {
A
Alexis Beingessner 已提交
2410
        let mut d = BTreeMap::new();
D
Daniel Micay 已提交
2411
        for (key, value) in self.iter() {
2412
            d.insert((*key).clone(), value.to_json());
G
Graydon Hoare 已提交
2413
        }
2414
        Json::Object(d)
G
Graydon Hoare 已提交
2415 2416 2417
    }
}

2418
impl<A:ToJson> ToJson for Option<A> {
B
Ben Striegel 已提交
2419 2420
    fn to_json(&self) -> Json {
        match *self {
2421
            None => Json::Null,
A
Adolfo Ochagavía 已提交
2422
            Some(ref value) => value.to_json()
2423 2424 2425 2426
        }
    }
}

2427
impl fmt::Show for Json {
2428
    /// Encodes a json value into a string
2429
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
A
Alex Crichton 已提交
2430
        self.to_writer(f).map_err(|_| fmt::Error)
2431
    }
2432 2433
}

B
Brendan Zabarauskas 已提交
2434
impl FromStr for Json {
A
Adolfo Ochagavía 已提交
2435 2436 2437 2438 2439
    fn from_str(s: &str) -> Option<Json> {
        from_str(s).ok()
    }
}

2440 2441
#[cfg(test)]
mod tests {
2442
    extern crate test;
S
Steven Fackler 已提交
2443 2444
    use self::Animal::*;
    use self::DecodeEnum::*;
2445
    use self::test::Bencher;
A
Alex Crichton 已提交
2446
    use {Encodable, Decodable};
2447 2448 2449 2450 2451 2452 2453 2454
    use super::Json::*;
    use super::ErrorCode::*;
    use super::ParserError::*;
    use super::DecoderError::*;
    use super::JsonEvent::*;
    use super::StackElement::*;
    use super::{PrettyEncoder, Json, from_str, DecodeResult, DecoderError, JsonEvent, Parser,
                StackElement, Stack, Encoder, Decoder};
2455
    use std::{i64, u64, f32, f64, io};
A
Alexis Beingessner 已提交
2456
    use std::collections::BTreeMap;
2457
    use std::num::Float;
2458
    use std::string;
2459

2460
    #[deriving(RustcDecodable, Eq, PartialEq, Show)]
2461 2462 2463 2464 2465 2466 2467 2468 2469 2470 2471 2472 2473 2474 2475 2476 2477 2478 2479 2480 2481
    struct OptionData {
        opt: Option<uint>,
    }

    #[test]
    fn test_decode_option_none() {
        let s ="{}";
        let obj: OptionData = super::decode(s).unwrap();
        assert_eq!(obj, OptionData { opt: None });
    }

    #[test]
    fn test_decode_option_some() {
        let s = "{ \"opt\": 10 }";
        let obj: OptionData = super::decode(s).unwrap();
        assert_eq!(obj, OptionData { opt: Some(10u) });
    }

    #[test]
    fn test_decode_option_malformed() {
        check_err::<OptionData>("{ \"opt\": [] }",
A
Alex Crichton 已提交
2482
                                ExpectedError("Number".to_string(), "[]".to_string()));
2483
        check_err::<OptionData>("{ \"opt\": false }",
A
Alex Crichton 已提交
2484
                                ExpectedError("Number".to_string(), "false".to_string()));
2485 2486
    }

2487
    #[deriving(PartialEq, RustcEncodable, RustcDecodable, Show)]
2488 2489
    enum Animal {
        Dog,
2490
        Frog(string::String, int)
2491 2492
    }

2493
    #[deriving(PartialEq, RustcEncodable, RustcDecodable, Show)]
2494 2495 2496
    struct Inner {
        a: (),
        b: uint,
2497
        c: Vec<string::String>,
2498 2499
    }

2500
    #[deriving(PartialEq, RustcEncodable, RustcDecodable, Show)]
2501
    struct Outer {
K
Kevin Ballard 已提交
2502
        inner: Vec<Inner>,
2503 2504
    }

2505
    fn mk_object(items: &[(string::String, Json)]) -> Json {
A
Alexis Beingessner 已提交
2506
        let mut d = BTreeMap::new();
2507

D
Daniel Micay 已提交
2508
        for item in items.iter() {
2509
            match *item {
2510
                (ref key, ref value) => { d.insert((*key).clone(), (*value).clone()); },
2511
            }
2512 2513
        };

L
Luqman Aden 已提交
2514
        Object(d)
2515 2516
    }

A
Adolfo Ochagavía 已提交
2517 2518 2519
    #[test]
    fn test_from_str_trait() {
        let s = "null";
2520
        assert!(s.parse::<Json>().unwrap() == s.parse().unwrap());
A
Adolfo Ochagavía 已提交
2521 2522
    }

2523 2524
    #[test]
    fn test_write_null() {
B
Barosl Lee 已提交
2525 2526
        assert_eq!(Null.to_string(), "null");
        assert_eq!(Null.to_pretty_str(), "null");
2527 2528
    }

2529 2530
    #[test]
    fn test_write_i64() {
B
Barosl Lee 已提交
2531 2532
        assert_eq!(U64(0).to_string(), "0");
        assert_eq!(U64(0).to_pretty_str(), "0");
2533

B
Barosl Lee 已提交
2534 2535
        assert_eq!(U64(1234).to_string(), "1234");
        assert_eq!(U64(1234).to_pretty_str(), "1234");
2536

B
Barosl Lee 已提交
2537 2538
        assert_eq!(I64(-5678).to_string(), "-5678");
        assert_eq!(I64(-5678).to_pretty_str(), "-5678");
2539 2540 2541

        assert_eq!(U64(7650007200025252000).to_string(), "7650007200025252000");
        assert_eq!(U64(7650007200025252000).to_pretty_str(), "7650007200025252000");
2542
    }
2543

2544
    #[test]
2545
    fn test_write_f64() {
B
Barosl Lee 已提交
2546 2547
        assert_eq!(F64(3.0).to_string(), "3.0");
        assert_eq!(F64(3.0).to_pretty_str(), "3.0");
2548

B
Barosl Lee 已提交
2549 2550
        assert_eq!(F64(3.1).to_string(), "3.1");
        assert_eq!(F64(3.1).to_pretty_str(), "3.1");
2551

B
Barosl Lee 已提交
2552 2553
        assert_eq!(F64(-1.5).to_string(), "-1.5");
        assert_eq!(F64(-1.5).to_pretty_str(), "-1.5");
2554

B
Barosl Lee 已提交
2555 2556
        assert_eq!(F64(0.5).to_string(), "0.5");
        assert_eq!(F64(0.5).to_pretty_str(), "0.5");
M
mrec 已提交
2557

B
Barosl Lee 已提交
2558 2559
        assert_eq!(F64(f64::NAN).to_string(), "null");
        assert_eq!(F64(f64::NAN).to_pretty_str(), "null");
M
mrec 已提交
2560

B
Barosl Lee 已提交
2561 2562
        assert_eq!(F64(f64::INFINITY).to_string(), "null");
        assert_eq!(F64(f64::INFINITY).to_pretty_str(), "null");
M
mrec 已提交
2563

B
Barosl Lee 已提交
2564 2565
        assert_eq!(F64(f64::NEG_INFINITY).to_string(), "null");
        assert_eq!(F64(f64::NEG_INFINITY).to_pretty_str(), "null");
2566 2567 2568 2569
    }

    #[test]
    fn test_write_str() {
A
Alex Crichton 已提交
2570 2571
        assert_eq!(String("".to_string()).to_string(), "\"\"");
        assert_eq!(String("".to_string()).to_pretty_str(), "\"\"");
2572

A
Alex Crichton 已提交
2573 2574
        assert_eq!(String("homura".to_string()).to_string(), "\"homura\"");
        assert_eq!(String("madoka".to_string()).to_pretty_str(), "\"madoka\"");
2575 2576 2577 2578
    }

    #[test]
    fn test_write_bool() {
B
Barosl Lee 已提交
2579 2580
        assert_eq!(Boolean(true).to_string(), "true");
        assert_eq!(Boolean(true).to_pretty_str(), "true");
2581

B
Barosl Lee 已提交
2582 2583
        assert_eq!(Boolean(false).to_string(), "false");
        assert_eq!(Boolean(false).to_pretty_str(), "false");
2584 2585 2586
    }

    #[test]
C
Corey Farwell 已提交
2587
    fn test_write_array() {
B
Barosl Lee 已提交
2588 2589
        assert_eq!(Array(vec![]).to_string(), "[]");
        assert_eq!(Array(vec![]).to_pretty_str(), "[]");
2590

B
Barosl Lee 已提交
2591
        assert_eq!(Array(vec![Boolean(true)]).to_string(), "[true]");
2592
        assert_eq!(
B
Barosl Lee 已提交
2593
            Array(vec![Boolean(true)]).to_pretty_str(),
2594
            "\
2595 2596
            [\n  \
                true\n\
2597
            ]"
2598
        );
2599

C
Corey Farwell 已提交
2600
        let long_test_array = Array(vec![
2601 2602
            Boolean(false),
            Null,
A
Alex Crichton 已提交
2603
            Array(vec![String("foo\nbar".to_string()), F64(3.5)])]);
2604

B
Barosl Lee 已提交
2605
        assert_eq!(long_test_array.to_string(),
2606
            "[false,null,[\"foo\\nbar\",3.5]]");
2607
        assert_eq!(
B
Barosl Lee 已提交
2608
            long_test_array.to_pretty_str(),
2609
            "\
2610 2611 2612 2613 2614 2615 2616
            [\n  \
                false,\n  \
                null,\n  \
                [\n    \
                    \"foo\\nbar\",\n    \
                    3.5\n  \
                ]\n\
2617
            ]"
2618 2619 2620
        );
    }

2621
    #[test]
2622
    fn test_write_object() {
B
Barosl Lee 已提交
2623 2624
        assert_eq!(mk_object(&[]).to_string(), "{}");
        assert_eq!(mk_object(&[]).to_pretty_str(), "{}");
2625

2626
        assert_eq!(
N
Nick Cameron 已提交
2627
            mk_object(&[
A
Alex Crichton 已提交
2628
                ("a".to_string(), Boolean(true))
B
Barosl Lee 已提交
2629
            ]).to_string(),
2630
            "{\"a\":true}"
2631
        );
2632
        assert_eq!(
A
Alex Crichton 已提交
2633
            mk_object(&[("a".to_string(), Boolean(true))]).to_pretty_str(),
2634
            "\
2635 2636
            {\n  \
                \"a\": true\n\
2637
            }"
2638 2639
        );

N
Nick Cameron 已提交
2640
        let complex_obj = mk_object(&[
A
Alex Crichton 已提交
2641 2642 2643
                ("b".to_string(), Array(vec![
                    mk_object(&[("c".to_string(), String("\x0c\r".to_string()))]),
                    mk_object(&[("d".to_string(), String("".to_string()))])
2644
                ]))
2645 2646 2647
            ]);

        assert_eq!(
B
Barosl Lee 已提交
2648
            complex_obj.to_string(),
2649
            "{\
2650 2651 2652 2653
                \"b\":[\
                    {\"c\":\"\\f\\r\"},\
                    {\"d\":\"\"}\
                ]\
2654
            }"
2655 2656
        );
        assert_eq!(
B
Barosl Lee 已提交
2657
            complex_obj.to_pretty_str(),
2658
            "\
2659 2660 2661 2662 2663 2664 2665 2666 2667
            {\n  \
                \"b\": [\n    \
                    {\n      \
                        \"c\": \"\\f\\r\"\n    \
                    },\n    \
                    {\n      \
                        \"d\": \"\"\n    \
                    }\n  \
                ]\n\
2668
            }"
2669
        );
2670

N
Nick Cameron 已提交
2671
        let a = mk_object(&[
A
Alex Crichton 已提交
2672 2673 2674 2675
            ("a".to_string(), Boolean(true)),
            ("b".to_string(), Array(vec![
                mk_object(&[("c".to_string(), String("\x0c\r".to_string()))]),
                mk_object(&[("d".to_string(), String("".to_string()))])
2676
            ]))
G
Graydon Hoare 已提交
2677
        ]);
2678

2679 2680
        // We can't compare the strings directly because the object fields be
        // printed in a different order.
2681
        assert_eq!(a.clone(), from_str(a.to_string().as_slice()).unwrap());
2682 2683
        assert_eq!(a.clone(),
                   from_str(a.to_pretty_str().as_slice()).unwrap());
2684 2685
    }

2686
    fn with_str_writer<F>(f: F) -> string::String where F: FnOnce(&mut io::Writer){
D
Daniel Micay 已提交
2687
        let mut m = Vec::new();
2688
        f(&mut m as &mut io::Writer);
D
Daniel Micay 已提交
2689
        string::String::from_utf8(m).unwrap()
A
Alex Crichton 已提交
2690 2691
    }

2692
    #[test]
2693
    fn test_write_enum() {
2694
        let animal = Dog;
2695
        assert_eq!(
A
Adolfo Ochagavía 已提交
2696 2697
            with_str_writer(|writer| {
                let mut encoder = Encoder::new(writer);
S
Sean McArthur 已提交
2698
                animal.encode(&mut encoder).unwrap();
2699
            }),
2700
            "\"Dog\""
2701 2702
        );
        assert_eq!(
A
Adolfo Ochagavía 已提交
2703 2704
            with_str_writer(|writer| {
                let mut encoder = PrettyEncoder::new(writer);
S
Sean McArthur 已提交
2705
                animal.encode(&mut encoder).unwrap();
2706
            }),
2707
            "\"Dog\""
2708
        );
2709

A
Alex Crichton 已提交
2710
        let animal = Frog("Henry".to_string(), 349);
2711
        assert_eq!(
A
Adolfo Ochagavía 已提交
2712 2713
            with_str_writer(|writer| {
                let mut encoder = Encoder::new(writer);
S
Sean McArthur 已提交
2714
                animal.encode(&mut encoder).unwrap();
2715
            }),
2716
            "{\"variant\":\"Frog\",\"fields\":[\"Henry\",349]}"
2717 2718
        );
        assert_eq!(
A
Adolfo Ochagavía 已提交
2719 2720
            with_str_writer(|writer| {
                let mut encoder = PrettyEncoder::new(writer);
S
Sean McArthur 已提交
2721
                animal.encode(&mut encoder).unwrap();
2722
            }),
2723 2724 2725 2726 2727 2728
            "{\n  \
               \"variant\": \"Frog\",\n  \
               \"fields\": [\n    \
                 \"Henry\",\n    \
                 349\n  \
               ]\n\
2729
             }"
2730
        );
2731 2732
    }

2733
    macro_rules! check_encoder_for_simple {
2734 2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745 2746
        ($value:expr, $expected:expr) => ({
            let s = with_str_writer(|writer| {
                let mut encoder = Encoder::new(writer);
                $value.encode(&mut encoder).unwrap();
            });
            assert_eq!(s, $expected);

            let s = with_str_writer(|writer| {
                let mut encoder = PrettyEncoder::new(writer);
                $value.encode(&mut encoder).unwrap();
            });
            assert_eq!(s, $expected);
        })
2747
    }
2748

2749
    #[test]
2750
    fn test_write_some() {
2751
        check_encoder_for_simple!(Some("jodhpurs".to_string()), "\"jodhpurs\"");
2752 2753
    }

2754
    #[test]
2755
    fn test_write_none() {
2756 2757 2758 2759 2760 2761 2762
        check_encoder_for_simple!(None::<string::String>, "null");
    }

    #[test]
    fn test_write_char() {
        check_encoder_for_simple!('a', "\"a\"");
        check_encoder_for_simple!('\t', "\"\\t\"");
2763 2764 2765
        check_encoder_for_simple!('\u{0000}', "\"\\u0000\"");
        check_encoder_for_simple!('\u{001b}', "\"\\u001b\"");
        check_encoder_for_simple!('\u{007f}', "\"\\u007f\"");
2766 2767 2768
        check_encoder_for_simple!('\u{00a0}', "\"\u{00a0}\"");
        check_encoder_for_simple!('\u{abcd}', "\"\u{abcd}\"");
        check_encoder_for_simple!('\u{10ffff}', "\"\u{10ffff}\"");
2769 2770
    }

2771
    #[test]
2772
    fn test_trailing_characters() {
2773 2774 2775 2776 2777 2778
        assert_eq!(from_str("nulla"),  Err(SyntaxError(TrailingCharacters, 1, 5)));
        assert_eq!(from_str("truea"),  Err(SyntaxError(TrailingCharacters, 1, 5)));
        assert_eq!(from_str("falsea"), Err(SyntaxError(TrailingCharacters, 1, 6)));
        assert_eq!(from_str("1a"),     Err(SyntaxError(TrailingCharacters, 1, 2)));
        assert_eq!(from_str("[]a"),    Err(SyntaxError(TrailingCharacters, 1, 3)));
        assert_eq!(from_str("{}a"),    Err(SyntaxError(TrailingCharacters, 1, 3)));
2779 2780 2781 2782
    }

    #[test]
    fn test_read_identifiers() {
2783 2784 2785 2786 2787 2788
        assert_eq!(from_str("n"),    Err(SyntaxError(InvalidSyntax, 1, 2)));
        assert_eq!(from_str("nul"),  Err(SyntaxError(InvalidSyntax, 1, 4)));
        assert_eq!(from_str("t"),    Err(SyntaxError(InvalidSyntax, 1, 2)));
        assert_eq!(from_str("truz"), Err(SyntaxError(InvalidSyntax, 1, 4)));
        assert_eq!(from_str("f"),    Err(SyntaxError(InvalidSyntax, 1, 2)));
        assert_eq!(from_str("faz"),  Err(SyntaxError(InvalidSyntax, 1, 3)));
2789

E
Erick Tryzelaar 已提交
2790 2791 2792 2793 2794 2795
        assert_eq!(from_str("null"), Ok(Null));
        assert_eq!(from_str("true"), Ok(Boolean(true)));
        assert_eq!(from_str("false"), Ok(Boolean(false)));
        assert_eq!(from_str(" null "), Ok(Null));
        assert_eq!(from_str(" true "), Ok(Boolean(true)));
        assert_eq!(from_str(" false "), Ok(Boolean(false)));
2796 2797
    }

2798 2799
    #[test]
    fn test_decode_identifiers() {
2800
        let v: () = super::decode("null").unwrap();
2801 2802
        assert_eq!(v, ());

2803
        let v: bool = super::decode("true").unwrap();
2804 2805
        assert_eq!(v, true);

2806
        let v: bool = super::decode("false").unwrap();
2807 2808 2809
        assert_eq!(v, false);
    }

2810
    #[test]
2811
    fn test_read_number() {
2812 2813
        assert_eq!(from_str("+"),   Err(SyntaxError(InvalidSyntax, 1, 1)));
        assert_eq!(from_str("."),   Err(SyntaxError(InvalidSyntax, 1, 1)));
M
mrec 已提交
2814
        assert_eq!(from_str("NaN"), Err(SyntaxError(InvalidSyntax, 1, 1)));
2815 2816 2817 2818 2819
        assert_eq!(from_str("-"),   Err(SyntaxError(InvalidNumber, 1, 2)));
        assert_eq!(from_str("00"),  Err(SyntaxError(InvalidNumber, 1, 2)));
        assert_eq!(from_str("1."),  Err(SyntaxError(InvalidNumber, 1, 3)));
        assert_eq!(from_str("1e"),  Err(SyntaxError(InvalidNumber, 1, 3)));
        assert_eq!(from_str("1e+"), Err(SyntaxError(InvalidNumber, 1, 4)));
2820

2821 2822 2823 2824 2825 2826 2827 2828 2829 2830 2831 2832 2833 2834 2835
        assert_eq!(from_str("18446744073709551616"), Err(SyntaxError(InvalidNumber, 1, 20)));
        assert_eq!(from_str("-9223372036854775809"), Err(SyntaxError(InvalidNumber, 1, 21)));

        assert_eq!(from_str("3"), Ok(U64(3)));
        assert_eq!(from_str("3.1"), Ok(F64(3.1)));
        assert_eq!(from_str("-1.2"), Ok(F64(-1.2)));
        assert_eq!(from_str("0.4"), Ok(F64(0.4)));
        assert_eq!(from_str("0.4e5"), Ok(F64(0.4e5)));
        assert_eq!(from_str("0.4e+15"), Ok(F64(0.4e15)));
        assert_eq!(from_str("0.4e-01"), Ok(F64(0.4e-01)));
        assert_eq!(from_str(" 3 "), Ok(U64(3)));

        assert_eq!(from_str("-9223372036854775808"), Ok(I64(i64::MIN)));
        assert_eq!(from_str("9223372036854775807"), Ok(U64(i64::MAX as u64)));
        assert_eq!(from_str("18446744073709551615"), Ok(U64(u64::MAX)));
2836 2837
    }

2838 2839
    #[test]
    fn test_decode_numbers() {
2840
        let v: f64 = super::decode("3").unwrap();
D
Daniel Micay 已提交
2841
        assert_eq!(v, 3.0);
2842

2843
        let v: f64 = super::decode("3.1").unwrap();
D
Daniel Micay 已提交
2844
        assert_eq!(v, 3.1);
2845

2846
        let v: f64 = super::decode("-1.2").unwrap();
D
Daniel Micay 已提交
2847
        assert_eq!(v, -1.2);
2848

2849
        let v: f64 = super::decode("0.4").unwrap();
D
Daniel Micay 已提交
2850
        assert_eq!(v, 0.4);
2851

2852
        let v: f64 = super::decode("0.4e5").unwrap();
D
Daniel Micay 已提交
2853
        assert_eq!(v, 0.4e5);
2854

2855
        let v: f64 = super::decode("0.4e15").unwrap();
D
Daniel Micay 已提交
2856
        assert_eq!(v, 0.4e15);
2857

2858
        let v: f64 = super::decode("0.4e-01").unwrap();
D
Daniel Micay 已提交
2859
        assert_eq!(v, 0.4e-01);
2860 2861 2862 2863 2864 2865 2866 2867 2868 2869 2870 2871

        let v: u64 = super::decode("0").unwrap();
        assert_eq!(v, 0);

        let v: u64 = super::decode("18446744073709551615").unwrap();
        assert_eq!(v, u64::MAX);

        let v: i64 = super::decode("-9223372036854775808").unwrap();
        assert_eq!(v, i64::MIN);

        let v: i64 = super::decode("9223372036854775807").unwrap();
        assert_eq!(v, i64::MAX);
2872 2873

        let res: DecodeResult<i64> = super::decode("765.25252");
A
Alex Crichton 已提交
2874
        assert_eq!(res, Err(ExpectedError("Integer".to_string(), "765.25252".to_string())));
2875 2876
    }

G
Gary Linscott 已提交
2877
    #[test]
2878
    fn test_read_str() {
2879 2880 2881
        assert_eq!(from_str("\""),    Err(SyntaxError(EOFWhileParsingString, 1, 2)));
        assert_eq!(from_str("\"lol"), Err(SyntaxError(EOFWhileParsingString, 1, 5)));

A
Alex Crichton 已提交
2882 2883 2884 2885 2886 2887 2888 2889 2890 2891
        assert_eq!(from_str("\"\""), Ok(String("".to_string())));
        assert_eq!(from_str("\"foo\""), Ok(String("foo".to_string())));
        assert_eq!(from_str("\"\\\"\""), Ok(String("\"".to_string())));
        assert_eq!(from_str("\"\\b\""), Ok(String("\x08".to_string())));
        assert_eq!(from_str("\"\\n\""), Ok(String("\n".to_string())));
        assert_eq!(from_str("\"\\r\""), Ok(String("\r".to_string())));
        assert_eq!(from_str("\"\\t\""), Ok(String("\t".to_string())));
        assert_eq!(from_str(" \"foo\" "), Ok(String("foo".to_string())));
        assert_eq!(from_str("\"\\u12ab\""), Ok(String("\u{12ab}".to_string())));
        assert_eq!(from_str("\"\\uAB12\""), Ok(String("\u{AB12}".to_string())));
2892 2893
    }

2894
    #[test]
2895
    fn test_decode_str() {
2896 2897 2898 2899 2900 2901 2902
        let s = [("\"\"", ""),
                 ("\"foo\"", "foo"),
                 ("\"\\\"\"", "\""),
                 ("\"\\b\"", "\x08"),
                 ("\"\\n\"", "\n"),
                 ("\"\\r\"", "\r"),
                 ("\"\\t\"", "\t"),
A
Alex Crichton 已提交
2903 2904
                 ("\"\\u12ab\"", "\u{12ab}"),
                 ("\"\\uAB12\"", "\u{AB12}")];
2905 2906

        for &(i, o) in s.iter() {
2907
            let v: string::String = super::decode(i).unwrap();
2908
            assert_eq!(v, o);
2909
        }
2910 2911
    }

2912
    #[test]
C
Corey Farwell 已提交
2913
    fn test_read_array() {
2914
        assert_eq!(from_str("["),     Err(SyntaxError(EOFWhileParsingValue, 1, 2)));
C
Corey Farwell 已提交
2915
        assert_eq!(from_str("[1"),    Err(SyntaxError(EOFWhileParsingArray, 1, 3)));
2916 2917 2918
        assert_eq!(from_str("[1,"),   Err(SyntaxError(EOFWhileParsingValue, 1, 4)));
        assert_eq!(from_str("[1,]"),  Err(SyntaxError(InvalidSyntax,        1, 4)));
        assert_eq!(from_str("[6 7]"), Err(SyntaxError(InvalidSyntax,        1, 4)));
2919

C
Corey Farwell 已提交
2920 2921 2922 2923 2924
        assert_eq!(from_str("[]"), Ok(Array(vec![])));
        assert_eq!(from_str("[ ]"), Ok(Array(vec![])));
        assert_eq!(from_str("[true]"), Ok(Array(vec![Boolean(true)])));
        assert_eq!(from_str("[ false ]"), Ok(Array(vec![Boolean(false)])));
        assert_eq!(from_str("[null]"), Ok(Array(vec![Null])));
E
Erick Tryzelaar 已提交
2925
        assert_eq!(from_str("[3, 1]"),
C
Corey Farwell 已提交
2926
                     Ok(Array(vec![U64(3), U64(1)])));
E
Erick Tryzelaar 已提交
2927
        assert_eq!(from_str("\n[3, 2]\n"),
C
Corey Farwell 已提交
2928
                     Ok(Array(vec![U64(3), U64(2)])));
E
Erick Tryzelaar 已提交
2929
        assert_eq!(from_str("[2, [4, 1]]"),
C
Corey Farwell 已提交
2930
               Ok(Array(vec![U64(2), Array(vec![U64(4), U64(1)])])));
2931 2932
    }

2933
    #[test]
C
Corey Farwell 已提交
2934
    fn test_decode_array() {
2935
        let v: Vec<()> = super::decode("[]").unwrap();
K
Kevin Ballard 已提交
2936
        assert_eq!(v, vec![]);
2937

2938
        let v: Vec<()> = super::decode("[null]").unwrap();
K
Kevin Ballard 已提交
2939
        assert_eq!(v, vec![()]);
2940

2941
        let v: Vec<bool> = super::decode("[true]").unwrap();
K
Kevin Ballard 已提交
2942
        assert_eq!(v, vec![true]);
2943

2944
        let v: Vec<int> = super::decode("[3, 1]").unwrap();
K
Kevin Ballard 已提交
2945
        assert_eq!(v, vec![3, 1]);
2946

2947
        let v: Vec<Vec<uint>> = super::decode("[[3], [1, 2]]").unwrap();
K
Kevin Ballard 已提交
2948
        assert_eq!(v, vec![vec![3], vec![1, 2]]);
2949 2950
    }

2951 2952 2953
    #[test]
    fn test_decode_tuple() {
        let t: (uint, uint, uint) = super::decode("[1, 2, 3]").unwrap();
2954
        assert_eq!(t, (1u, 2, 3));
2955 2956

        let t: (uint, string::String) = super::decode("[1, \"two\"]").unwrap();
A
Alex Crichton 已提交
2957
        assert_eq!(t, (1u, "two".to_string()));
2958 2959 2960 2961 2962 2963 2964 2965 2966
    }

    #[test]
    fn test_decode_tuple_malformed_types() {
        assert!(super::decode::<(uint, string::String)>("[1, 2]").is_err());
    }

    #[test]
    fn test_decode_tuple_malformed_length() {
2967
        assert!(super::decode::<(uint, uint)>("[1, 2, 3]").is_err());
2968 2969
    }

2970
    #[test]
2971
    fn test_read_object() {
2972 2973 2974 2975 2976 2977 2978 2979 2980 2981 2982 2983
        assert_eq!(from_str("{"),       Err(SyntaxError(EOFWhileParsingObject, 1, 2)));
        assert_eq!(from_str("{ "),      Err(SyntaxError(EOFWhileParsingObject, 1, 3)));
        assert_eq!(from_str("{1"),      Err(SyntaxError(KeyMustBeAString,      1, 2)));
        assert_eq!(from_str("{ \"a\""), Err(SyntaxError(EOFWhileParsingObject, 1, 6)));
        assert_eq!(from_str("{\"a\""),  Err(SyntaxError(EOFWhileParsingObject, 1, 5)));
        assert_eq!(from_str("{\"a\" "), Err(SyntaxError(EOFWhileParsingObject, 1, 6)));

        assert_eq!(from_str("{\"a\" 1"),   Err(SyntaxError(ExpectedColon,         1, 6)));
        assert_eq!(from_str("{\"a\":"),    Err(SyntaxError(EOFWhileParsingValue,  1, 6)));
        assert_eq!(from_str("{\"a\":1"),   Err(SyntaxError(EOFWhileParsingObject, 1, 7)));
        assert_eq!(from_str("{\"a\":1 1"), Err(SyntaxError(InvalidSyntax,         1, 8)));
        assert_eq!(from_str("{\"a\":1,"),  Err(SyntaxError(EOFWhileParsingObject, 1, 8)));
2984

N
Nick Cameron 已提交
2985
        assert_eq!(from_str("{}").unwrap(), mk_object(&[]));
E
Erick Tryzelaar 已提交
2986
        assert_eq!(from_str("{\"a\": 3}").unwrap(),
A
Alex Crichton 已提交
2987
                  mk_object(&[("a".to_string(), U64(3))]));
2988

E
Erick Tryzelaar 已提交
2989 2990
        assert_eq!(from_str(
                      "{ \"a\": null, \"b\" : true }").unwrap(),
N
Nick Cameron 已提交
2991
                  mk_object(&[
A
Alex Crichton 已提交
2992 2993
                      ("a".to_string(), Null),
                      ("b".to_string(), Boolean(true))]));
E
Erick Tryzelaar 已提交
2994
        assert_eq!(from_str("\n{ \"a\": null, \"b\" : true }\n").unwrap(),
N
Nick Cameron 已提交
2995
                  mk_object(&[
A
Alex Crichton 已提交
2996 2997
                      ("a".to_string(), Null),
                      ("b".to_string(), Boolean(true))]));
E
Erick Tryzelaar 已提交
2998 2999
        assert_eq!(from_str(
                      "{\"a\" : 1.0 ,\"b\": [ true ]}").unwrap(),
N
Nick Cameron 已提交
3000
                  mk_object(&[
A
Alex Crichton 已提交
3001 3002
                      ("a".to_string(), F64(1.0)),
                      ("b".to_string(), Array(vec![Boolean(true)]))
3003
                  ]));
E
Erick Tryzelaar 已提交
3004
        assert_eq!(from_str(
3005 3006 3007 3008 3009 3010 3011 3012
                      "{\
                          \"a\": 1.0, \
                          \"b\": [\
                              true,\
                              \"foo\\nbar\", \
                              { \"c\": {\"d\": null} } \
                          ]\
                      }").unwrap(),
N
Nick Cameron 已提交
3013
                  mk_object(&[
A
Alex Crichton 已提交
3014 3015
                      ("a".to_string(), F64(1.0)),
                      ("b".to_string(), Array(vec![
B
Ben Striegel 已提交
3016
                          Boolean(true),
A
Alex Crichton 已提交
3017
                          String("foo\nbar".to_string()),
N
Nick Cameron 已提交
3018
                          mk_object(&[
A
Alex Crichton 已提交
3019
                              ("c".to_string(), mk_object(&[("d".to_string(), Null)]))
3020 3021
                          ])
                      ]))
3022
                  ]));
3023 3024
    }

3025
    #[test]
3026
    fn test_decode_struct() {
3027
        let s = "{
3028 3029 3030
            \"inner\": [
                { \"a\": null, \"b\": 2, \"c\": [\"abc\", \"xyz\"] }
            ]
3031
        }";
3032 3033

        let v: Outer = super::decode(s).unwrap();
3034 3035 3036
        assert_eq!(
            v,
            Outer {
K
Kevin Ballard 已提交
3037
                inner: vec![
A
Alex Crichton 已提交
3038
                    Inner { a: (), b: 2, c: vec!["abc".to_string(), "xyz".to_string()] }
3039 3040 3041 3042 3043
                ]
            }
        );
    }

3044
    #[deriving(RustcDecodable)]
M
mrec 已提交
3045 3046 3047 3048 3049 3050
    struct FloatStruct {
        f: f64,
        a: Vec<f64>
    }
    #[test]
    fn test_decode_struct_with_nan() {
3051 3052 3053
        let s = "{\"f\":null,\"a\":[null,123]}";
        let obj: FloatStruct = super::decode(s).unwrap();
        assert!(obj.f.is_nan());
N
NODA, Kai 已提交
3054 3055
        assert!(obj.a[0].is_nan());
        assert_eq!(obj.a[1], 123f64);
M
mrec 已提交
3056 3057
    }

3058 3059
    #[test]
    fn test_decode_option() {
3060
        let value: Option<string::String> = super::decode("null").unwrap();
3061 3062
        assert_eq!(value, None);

3063
        let value: Option<string::String> = super::decode("\"jodhpurs\"").unwrap();
A
Alex Crichton 已提交
3064
        assert_eq!(value, Some("jodhpurs".to_string()));
3065 3066
    }

3067
    #[test]
3068
    fn test_decode_enum() {
3069
        let value: Animal = super::decode("\"Dog\"").unwrap();
3070 3071
        assert_eq!(value, Dog);

3072
        let s = "{\"variant\":\"Frog\",\"fields\":[\"Henry\",349]}";
3073
        let value: Animal = super::decode(s).unwrap();
A
Alex Crichton 已提交
3074
        assert_eq!(value, Frog("Henry".to_string(), 349));
3075 3076
    }

3077
    #[test]
3078
    fn test_decode_map() {
3079
        let s = "{\"a\": \"Dog\", \"b\": {\"variant\":\"Frog\",\
3080
                  \"fields\":[\"Henry\", 349]}}";
A
Alexis Beingessner 已提交
3081
        let mut map: BTreeMap<string::String, Animal> = super::decode(s).unwrap();
3082

A
Alex Crichton 已提交
3083 3084
        assert_eq!(map.remove(&"a".to_string()), Some(Dog));
        assert_eq!(map.remove(&"b".to_string()), Some(Frog("Henry".to_string(), 349)));
3085 3086
    }

3087
    #[test]
3088
    fn test_multiline_errors() {
E
Erick Tryzelaar 已提交
3089
        assert_eq!(from_str("{\n  \"foo\":\n \"bar\""),
3090
            Err(SyntaxError(EOFWhileParsingObject, 3u, 8u)));
3091
    }
3092

3093
    #[deriving(RustcDecodable)]
M
mrec 已提交
3094
    #[allow(dead_code)]
3095 3096 3097
    struct DecodeStruct {
        x: f64,
        y: bool,
3098
        z: string::String,
K
Kevin Ballard 已提交
3099
        w: Vec<DecodeStruct>
3100
    }
3101
    #[deriving(RustcDecodable)]
3102 3103
    enum DecodeEnum {
        A(f64),
3104
        B(string::String)
3105
    }
3106 3107
    fn check_err<T: Decodable<Decoder, DecoderError>>(to_parse: &'static str,
                                                      expected: DecoderError) {
S
Sean McArthur 已提交
3108
        let res: DecodeResult<T> = match from_str(to_parse) {
3109
            Err(e) => Err(ParseError(e)),
S
Sean McArthur 已提交
3110 3111
            Ok(json) => Decodable::decode(&mut Decoder::new(json))
        };
3112
        match res {
S
Steve Klabnik 已提交
3113
            Ok(_) => panic!("`{}` parsed & decoded ok, expecting error `{}`",
S
Sean McArthur 已提交
3114
                              to_parse, expected),
S
Steve Klabnik 已提交
3115
            Err(ParseError(e)) => panic!("`{}` is not valid json: {}",
S
Sean McArthur 已提交
3116
                                           to_parse, e),
3117
            Err(e) => {
S
Sean McArthur 已提交
3118
                assert_eq!(e, expected);
3119 3120 3121 3122 3123
            }
        }
    }
    #[test]
    fn test_decode_errors_struct() {
A
Alex Crichton 已提交
3124
        check_err::<DecodeStruct>("[]", ExpectedError("Object".to_string(), "[]".to_string()));
3125
        check_err::<DecodeStruct>("{\"x\": true, \"y\": true, \"z\": \"\", \"w\": []}",
A
Alex Crichton 已提交
3126
                                  ExpectedError("Number".to_string(), "true".to_string()));
3127
        check_err::<DecodeStruct>("{\"x\": 1, \"y\": [], \"z\": \"\", \"w\": []}",
A
Alex Crichton 已提交
3128
                                  ExpectedError("Boolean".to_string(), "[]".to_string()));
3129
        check_err::<DecodeStruct>("{\"x\": 1, \"y\": true, \"z\": {}, \"w\": []}",
A
Alex Crichton 已提交
3130
                                  ExpectedError("String".to_string(), "{}".to_string()));
3131
        check_err::<DecodeStruct>("{\"x\": 1, \"y\": true, \"z\": \"\", \"w\": null}",
A
Alex Crichton 已提交
3132
                                  ExpectedError("Array".to_string(), "null".to_string()));
3133
        check_err::<DecodeStruct>("{\"x\": 1, \"y\": true, \"z\": \"\"}",
A
Alex Crichton 已提交
3134
                                  MissingFieldError("w".to_string()));
3135 3136 3137 3138
    }
    #[test]
    fn test_decode_errors_enum() {
        check_err::<DecodeEnum>("{}",
A
Alex Crichton 已提交
3139
                                MissingFieldError("variant".to_string()));
3140
        check_err::<DecodeEnum>("{\"variant\": 1}",
A
Alex Crichton 已提交
3141
                                ExpectedError("String".to_string(), "1".to_string()));
3142
        check_err::<DecodeEnum>("{\"variant\": \"A\"}",
A
Alex Crichton 已提交
3143
                                MissingFieldError("fields".to_string()));
3144
        check_err::<DecodeEnum>("{\"variant\": \"A\", \"fields\": null}",
A
Alex Crichton 已提交
3145
                                ExpectedError("Array".to_string(), "null".to_string()));
3146
        check_err::<DecodeEnum>("{\"variant\": \"C\", \"fields\": []}",
A
Alex Crichton 已提交
3147
                                UnknownVariantError("C".to_string()));
3148
    }
3149 3150 3151 3152

    #[test]
    fn test_find(){
        let json_value = from_str("{\"dog\" : \"cat\"}").unwrap();
3153 3154
        let found_str = json_value.find("dog");
        assert!(found_str.unwrap().as_string().unwrap() == "cat");
3155 3156 3157 3158 3159
    }

    #[test]
    fn test_find_path(){
        let json_value = from_str("{\"dog\":{\"cat\": {\"mouse\" : \"cheese\"}}}").unwrap();
3160 3161
        let found_str = json_value.find_path(&["dog", "cat", "mouse"]);
        assert!(found_str.unwrap().as_string().unwrap() == "cheese");
3162 3163 3164 3165 3166
    }

    #[test]
    fn test_search(){
        let json_value = from_str("{\"dog\":{\"cat\": {\"mouse\" : \"cheese\"}}}").unwrap();
3167
        let found_str = json_value.search("mouse").and_then(|j| j.as_string());
3168
        assert!(found_str.unwrap() == "cheese");
3169 3170
    }

3171 3172 3173
    #[test]
    fn test_index(){
        let json_value = from_str("{\"animals\":[\"dog\",\"cat\",\"mouse\"]}").unwrap();
C
Corey Farwell 已提交
3174 3175 3176 3177
        let ref array = json_value["animals"];
        assert_eq!(array[0].as_string().unwrap(), "dog");
        assert_eq!(array[1].as_string().unwrap(), "cat");
        assert_eq!(array[2].as_string().unwrap(), "mouse");
3178 3179
    }

3180 3181 3182 3183 3184 3185 3186 3187 3188 3189 3190 3191 3192 3193
    #[test]
    fn test_is_object(){
        let json_value = from_str("{}").unwrap();
        assert!(json_value.is_object());
    }

    #[test]
    fn test_as_object(){
        let json_value = from_str("{}").unwrap();
        let json_object = json_value.as_object();
        assert!(json_object.is_some());
    }

    #[test]
C
Corey Farwell 已提交
3194
    fn test_is_array(){
3195
        let json_value = from_str("[1, 2, 3]").unwrap();
C
Corey Farwell 已提交
3196
        assert!(json_value.is_array());
3197 3198 3199
    }

    #[test]
C
Corey Farwell 已提交
3200
    fn test_as_array(){
3201
        let json_value = from_str("[1, 2, 3]").unwrap();
C
Corey Farwell 已提交
3202
        let json_array = json_value.as_array();
3203
        let expected_length = 3;
C
Corey Farwell 已提交
3204
        assert!(json_array.is_some() && json_array.unwrap().len() == expected_length);
3205 3206 3207
    }

    #[test]
3208
    fn test_is_string(){
3209
        let json_value = from_str("\"dog\"").unwrap();
3210
        assert!(json_value.is_string());
3211 3212 3213
    }

    #[test]
3214
    fn test_as_string(){
3215
        let json_value = from_str("\"dog\"").unwrap();
3216
        let json_str = json_value.as_string();
3217
        let expected_str = "dog";
3218 3219 3220 3221 3222 3223 3224 3225 3226 3227
        assert_eq!(json_str, Some(expected_str));
    }

    #[test]
    fn test_is_number(){
        let json_value = from_str("12").unwrap();
        assert!(json_value.is_number());
    }

    #[test]
3228
    fn test_is_i64(){
3229
        let json_value = from_str("-12").unwrap();
3230 3231
        assert!(json_value.is_i64());

3232 3233 3234
        let json_value = from_str("12").unwrap();
        assert!(!json_value.is_i64());

3235 3236 3237 3238
        let json_value = from_str("12.0").unwrap();
        assert!(!json_value.is_i64());
    }

3239 3240 3241 3242 3243 3244 3245 3246 3247 3248 3249 3250
    #[test]
    fn test_is_u64(){
        let json_value = from_str("12").unwrap();
        assert!(json_value.is_u64());

        let json_value = from_str("-12").unwrap();
        assert!(!json_value.is_u64());

        let json_value = from_str("12.0").unwrap();
        assert!(!json_value.is_u64());
    }

3251 3252
    #[test]
    fn test_is_f64(){
3253
        let json_value = from_str("12").unwrap();
3254 3255
        assert!(!json_value.is_f64());

3256 3257 3258
        let json_value = from_str("-12").unwrap();
        assert!(!json_value.is_f64());

3259 3260
        let json_value = from_str("12.0").unwrap();
        assert!(json_value.is_f64());
3261 3262 3263

        let json_value = from_str("-12.0").unwrap();
        assert!(json_value.is_f64());
3264 3265 3266 3267
    }

    #[test]
    fn test_as_i64(){
3268
        let json_value = from_str("-12").unwrap();
3269
        let json_num = json_value.as_i64();
3270 3271 3272 3273 3274 3275 3276
        assert_eq!(json_num, Some(-12));
    }

    #[test]
    fn test_as_u64(){
        let json_value = from_str("12").unwrap();
        let json_num = json_value.as_u64();
3277 3278 3279 3280 3281 3282 3283 3284
        assert_eq!(json_num, Some(12));
    }

    #[test]
    fn test_as_f64(){
        let json_value = from_str("12.0").unwrap();
        let json_num = json_value.as_f64();
        assert_eq!(json_num, Some(12f64));
3285 3286 3287 3288 3289 3290 3291 3292 3293 3294 3295 3296 3297 3298 3299 3300 3301 3302 3303 3304 3305 3306 3307 3308 3309 3310 3311 3312 3313
    }

    #[test]
    fn test_is_boolean(){
        let json_value = from_str("false").unwrap();
        assert!(json_value.is_boolean());
    }

    #[test]
    fn test_as_boolean(){
        let json_value = from_str("false").unwrap();
        let json_bool = json_value.as_boolean();
        let expected_bool = false;
        assert!(json_bool.is_some() && json_bool.unwrap() == expected_bool);
    }

    #[test]
    fn test_is_null(){
        let json_value = from_str("null").unwrap();
        assert!(json_value.is_null());
    }

    #[test]
    fn test_as_null(){
        let json_value = from_str("null").unwrap();
        let json_null = json_value.as_null();
        let expected_null = ();
        assert!(json_null.is_some() && json_null.unwrap() == expected_null);
    }
3314 3315 3316 3317 3318

    #[test]
    fn test_encode_hashmap_with_numeric_key() {
        use std::str::from_utf8;
        use std::io::Writer;
3319
        use std::collections::HashMap;
3320 3321
        let mut hm: HashMap<uint, bool> = HashMap::new();
        hm.insert(1, true);
D
Daniel Micay 已提交
3322
        let mut mem_buf = Vec::new();
3323 3324
        {
            let mut encoder = Encoder::new(&mut mem_buf as &mut io::Writer);
S
Sean McArthur 已提交
3325
            hm.encode(&mut encoder).unwrap();
3326
        }
D
Daniel Micay 已提交
3327
        let json_str = from_utf8(mem_buf[]).unwrap();
3328
        match from_str(json_str) {
S
Steve Klabnik 已提交
3329
            Err(_) => panic!("Unable to parse json_str: {}", json_str),
3330 3331 3332
            _ => {} // it parsed and we are good to go
        }
    }
3333

3334 3335 3336 3337
    #[test]
    fn test_prettyencode_hashmap_with_numeric_key() {
        use std::str::from_utf8;
        use std::io::Writer;
3338
        use std::collections::HashMap;
3339 3340
        let mut hm: HashMap<uint, bool> = HashMap::new();
        hm.insert(1, true);
D
Daniel Micay 已提交
3341
        let mut mem_buf = Vec::new();
3342 3343
        {
            let mut encoder = PrettyEncoder::new(&mut mem_buf as &mut io::Writer);
3344
            hm.encode(&mut encoder).unwrap()
3345
        }
D
Daniel Micay 已提交
3346
        let json_str = from_utf8(mem_buf[]).unwrap();
3347
        match from_str(json_str) {
S
Steve Klabnik 已提交
3348
            Err(_) => panic!("Unable to parse json_str: {}", json_str),
3349 3350 3351
            _ => {} // it parsed and we are good to go
        }
    }
3352

3353 3354 3355
    #[test]
    fn test_prettyencoder_indent_level_param() {
        use std::str::from_utf8;
A
Alexis Beingessner 已提交
3356
        use std::collections::BTreeMap;
3357

A
Alexis Beingessner 已提交
3358
        let mut tree = BTreeMap::new();
3359

A
Alex Crichton 已提交
3360 3361
        tree.insert("hello".to_string(), String("guten tag".to_string()));
        tree.insert("goodbye".to_string(), String("sayonara".to_string()));
3362

C
Corey Farwell 已提交
3363
        let json = Array(
3364 3365 3366 3367
            // The following layout below should look a lot like
            // the pretty-printed JSON (indent * x)
            vec!
            ( // 0x
A
Alex Crichton 已提交
3368
                String("greetings".to_string()), // 1x
3369 3370
                Object(tree), // 1x + 2x + 2x + 1x
            ) // 0x
C
Corey Farwell 已提交
3371
            // End JSON array (7 lines)
3372 3373 3374 3375 3376 3377 3378 3379 3380 3381
        );

        // Helper function for counting indents
        fn indents(source: &str) -> uint {
            let trimmed = source.trim_left_chars(' ');
            source.len() - trimmed.len()
        }

        // Test up to 4 spaces of indents (more?)
        for i in range(0, 4u) {
D
Daniel Micay 已提交
3382
            let mut writer = Vec::new();
3383 3384 3385 3386 3387 3388
            {
                let ref mut encoder = PrettyEncoder::new(&mut writer);
                encoder.set_indent(i);
                json.encode(encoder).unwrap();
            }

D
Daniel Micay 已提交
3389
            let printed = from_utf8(writer[]).unwrap();
3390 3391 3392 3393 3394 3395 3396 3397 3398 3399 3400 3401 3402 3403 3404 3405 3406 3407

            // Check for indents at each line
            let lines: Vec<&str> = printed.lines().collect();
            assert_eq!(lines.len(), 7); // JSON should be 7 lines

            assert_eq!(indents(lines[0]), 0 * i); // [
            assert_eq!(indents(lines[1]), 1 * i); //   "greetings",
            assert_eq!(indents(lines[2]), 1 * i); //   {
            assert_eq!(indents(lines[3]), 2 * i); //     "hello": "guten tag",
            assert_eq!(indents(lines[4]), 2 * i); //     "goodbye": "sayonara"
            assert_eq!(indents(lines[5]), 1 * i); //   },
            assert_eq!(indents(lines[6]), 0 * i); // ]

            // Finally, test that the pretty-printed JSON is valid
            from_str(printed).ok().expect("Pretty-printed JSON is invalid!");
        }
    }

3408 3409
    #[test]
    fn test_hashmap_with_numeric_key_can_handle_double_quote_delimited_key() {
3410
        use std::collections::HashMap;
3411 3412 3413
        use Decodable;
        let json_str = "{\"1\":true}";
        let json_obj = match from_str(json_str) {
S
Steve Klabnik 已提交
3414
            Err(_) => panic!("Unable to parse json_str: {}", json_str),
3415 3416 3417
            Ok(o) => o
        };
        let mut decoder = Decoder::new(json_obj);
S
Sean McArthur 已提交
3418
        let _hm: HashMap<uint, bool> = Decodable::decode(&mut decoder).unwrap();
3419
    }
3420

3421 3422 3423 3424 3425 3426
    #[test]
    fn test_hashmap_with_numeric_key_will_error_with_string_keys() {
        use std::collections::HashMap;
        use Decodable;
        let json_str = "{\"a\":true}";
        let json_obj = match from_str(json_str) {
S
Steve Klabnik 已提交
3427
            Err(_) => panic!("Unable to parse json_str: {}", json_str),
3428 3429 3430 3431
            Ok(o) => o
        };
        let mut decoder = Decoder::new(json_obj);
        let result: Result<HashMap<uint, bool>, DecoderError> = Decodable::decode(&mut decoder);
A
Alex Crichton 已提交
3432
        assert_eq!(result, Err(ExpectedError("Number".to_string(), "a".to_string())));
3433 3434
    }

3435 3436
    fn assert_stream_equal(src: &str,
                           expected: Vec<(JsonEvent, Vec<StackElement>)>) {
3437 3438 3439 3440 3441 3442 3443
        let mut parser = Parser::new(src.chars());
        let mut i = 0;
        loop {
            let evt = match parser.next() {
                Some(e) => e,
                None => { break; }
            };
E
Erick Tryzelaar 已提交
3444
            let (ref expected_evt, ref expected_stack) = expected[i];
3445
            if !parser.stack().is_equal_to(expected_stack.as_slice()) {
S
Steve Klabnik 已提交
3446
                panic!("Parser stack is not equal to {}", expected_stack);
3447 3448 3449 3450 3451 3452
            }
            assert_eq!(&evt, expected_evt);
            i+=1;
        }
    }
    #[test]
S
Steven Fackler 已提交
3453
    #[cfg_attr(target_word_size = "32", ignore)] // FIXME(#14064)
3454 3455
    fn test_streaming_parser() {
        assert_stream_equal(
3456
            r#"{ "foo":"bar", "array" : [0, 1, 2, 3, 4, 5], "idents":[null,true,false]}"#,
3457 3458
            vec![
                (ObjectStart,             vec![]),
A
Alex Crichton 已提交
3459
                  (StringValue("bar".to_string()),   vec![Key("foo")]),
C
Corey Farwell 已提交
3460
                  (ArrayStart,            vec![Key("array")]),
3461 3462 3463 3464 3465 3466
                    (U64Value(0),         vec![Key("array"), Index(0)]),
                    (U64Value(1),         vec![Key("array"), Index(1)]),
                    (U64Value(2),         vec![Key("array"), Index(2)]),
                    (U64Value(3),         vec![Key("array"), Index(3)]),
                    (U64Value(4),         vec![Key("array"), Index(4)]),
                    (U64Value(5),         vec![Key("array"), Index(5)]),
C
Corey Farwell 已提交
3467 3468
                  (ArrayEnd,              vec![Key("array")]),
                  (ArrayStart,            vec![Key("idents")]),
3469 3470 3471
                    (NullValue,           vec![Key("idents"), Index(0)]),
                    (BooleanValue(true),  vec![Key("idents"), Index(1)]),
                    (BooleanValue(false), vec![Key("idents"), Index(2)]),
C
Corey Farwell 已提交
3472
                  (ArrayEnd,              vec![Key("idents")]),
3473
                (ObjectEnd,               vec![]),
3474 3475 3476 3477 3478 3479 3480 3481 3482 3483 3484 3485 3486
            ]
        );
    }
    fn last_event(src: &str) -> JsonEvent {
        let mut parser = Parser::new(src.chars());
        let mut evt = NullValue;
        loop {
            evt = match parser.next() {
                Some(e) => e,
                None => return evt,
            }
        }
    }
3487

3488
    #[test]
S
Steven Fackler 已提交
3489
    #[cfg_attr(target_word_size = "32", ignore)] // FIXME(#14064)
3490 3491 3492 3493 3494 3495 3496 3497 3498 3499 3500 3501
    fn test_read_object_streaming() {
        assert_eq!(last_event("{ "),      Error(SyntaxError(EOFWhileParsingObject, 1, 3)));
        assert_eq!(last_event("{1"),      Error(SyntaxError(KeyMustBeAString,      1, 2)));
        assert_eq!(last_event("{ \"a\""), Error(SyntaxError(EOFWhileParsingObject, 1, 6)));
        assert_eq!(last_event("{\"a\""),  Error(SyntaxError(EOFWhileParsingObject, 1, 5)));
        assert_eq!(last_event("{\"a\" "), Error(SyntaxError(EOFWhileParsingObject, 1, 6)));

        assert_eq!(last_event("{\"a\" 1"),   Error(SyntaxError(ExpectedColon,         1, 6)));
        assert_eq!(last_event("{\"a\":"),    Error(SyntaxError(EOFWhileParsingValue,  1, 6)));
        assert_eq!(last_event("{\"a\":1"),   Error(SyntaxError(EOFWhileParsingObject, 1, 7)));
        assert_eq!(last_event("{\"a\":1 1"), Error(SyntaxError(InvalidSyntax,         1, 8)));
        assert_eq!(last_event("{\"a\":1,"),  Error(SyntaxError(EOFWhileParsingObject, 1, 8)));
3502
        assert_eq!(last_event("{\"a\":1,}"), Error(SyntaxError(TrailingComma, 1, 8)));
3503 3504 3505

        assert_stream_equal(
            "{}",
3506
            vec![(ObjectStart, vec![]), (ObjectEnd, vec![])]
3507 3508 3509
        );
        assert_stream_equal(
            "{\"a\": 3}",
3510 3511
            vec![
                (ObjectStart,        vec![]),
3512
                  (U64Value(3),      vec![Key("a")]),
3513
                (ObjectEnd,          vec![]),
3514 3515 3516 3517
            ]
        );
        assert_stream_equal(
            "{ \"a\": null, \"b\" : true }",
3518 3519 3520 3521 3522
            vec![
                (ObjectStart,           vec![]),
                  (NullValue,           vec![Key("a")]),
                  (BooleanValue(true),  vec![Key("b")]),
                (ObjectEnd,             vec![]),
3523 3524 3525 3526
            ]
        );
        assert_stream_equal(
            "{\"a\" : 1.0 ,\"b\": [ true ]}",
3527 3528
            vec![
                (ObjectStart,           vec![]),
3529
                  (F64Value(1.0),       vec![Key("a")]),
C
Corey Farwell 已提交
3530
                  (ArrayStart,          vec![Key("b")]),
3531
                    (BooleanValue(true),vec![Key("b"), Index(0)]),
C
Corey Farwell 已提交
3532
                  (ArrayEnd,            vec![Key("b")]),
3533
                (ObjectEnd,             vec![]),
3534 3535 3536 3537 3538 3539 3540 3541 3542 3543 3544
            ]
        );
        assert_stream_equal(
            r#"{
                "a": 1.0,
                "b": [
                    true,
                    "foo\nbar",
                    { "c": {"d": null} }
                ]
            }"#,
3545 3546
            vec![
                (ObjectStart,                   vec![]),
3547
                  (F64Value(1.0),               vec![Key("a")]),
C
Corey Farwell 已提交
3548
                  (ArrayStart,                  vec![Key("b")]),
3549
                    (BooleanValue(true),        vec![Key("b"), Index(0)]),
A
Alex Crichton 已提交
3550
                    (StringValue("foo\nbar".to_string()),  vec![Key("b"), Index(1)]),
3551 3552 3553 3554 3555
                    (ObjectStart,               vec![Key("b"), Index(2)]),
                      (ObjectStart,             vec![Key("b"), Index(2), Key("c")]),
                        (NullValue,             vec![Key("b"), Index(2), Key("c"), Key("d")]),
                      (ObjectEnd,               vec![Key("b"), Index(2), Key("c")]),
                    (ObjectEnd,                 vec![Key("b"), Index(2)]),
C
Corey Farwell 已提交
3556
                  (ArrayEnd,                    vec![Key("b")]),
3557
                (ObjectEnd,                     vec![]),
3558 3559 3560 3561
            ]
        );
    }
    #[test]
S
Steven Fackler 已提交
3562
    #[cfg_attr(target_word_size = "32", ignore)] // FIXME(#14064)
C
Corey Farwell 已提交
3563
    fn test_read_array_streaming() {
3564 3565
        assert_stream_equal(
            "[]",
3566
            vec![
C
Corey Farwell 已提交
3567 3568
                (ArrayStart, vec![]),
                (ArrayEnd,   vec![]),
3569 3570 3571 3572
            ]
        );
        assert_stream_equal(
            "[ ]",
3573
            vec![
C
Corey Farwell 已提交
3574 3575
                (ArrayStart, vec![]),
                (ArrayEnd,   vec![]),
3576 3577 3578 3579
            ]
        );
        assert_stream_equal(
            "[true]",
3580
            vec![
C
Corey Farwell 已提交
3581
                (ArrayStart,             vec![]),
3582
                    (BooleanValue(true), vec![Index(0)]),
C
Corey Farwell 已提交
3583
                (ArrayEnd,               vec![]),
3584 3585 3586 3587
            ]
        );
        assert_stream_equal(
            "[ false ]",
3588
            vec![
C
Corey Farwell 已提交
3589
                (ArrayStart,              vec![]),
3590
                    (BooleanValue(false), vec![Index(0)]),
C
Corey Farwell 已提交
3591
                (ArrayEnd,                vec![]),
3592 3593 3594 3595
            ]
        );
        assert_stream_equal(
            "[null]",
3596
            vec![
C
Corey Farwell 已提交
3597
                (ArrayStart,    vec![]),
3598
                    (NullValue, vec![Index(0)]),
C
Corey Farwell 已提交
3599
                (ArrayEnd,      vec![]),
3600 3601 3602 3603
            ]
        );
        assert_stream_equal(
            "[3, 1]",
3604
            vec![
C
Corey Farwell 已提交
3605
                (ArrayStart,      vec![]),
3606 3607
                    (U64Value(3), vec![Index(0)]),
                    (U64Value(1), vec![Index(1)]),
C
Corey Farwell 已提交
3608
                (ArrayEnd,        vec![]),
3609 3610 3611 3612
            ]
        );
        assert_stream_equal(
            "\n[3, 2]\n",
3613
            vec![
C
Corey Farwell 已提交
3614
                (ArrayStart,      vec![]),
3615 3616
                    (U64Value(3), vec![Index(0)]),
                    (U64Value(2), vec![Index(1)]),
C
Corey Farwell 已提交
3617
                (ArrayEnd,        vec![]),
3618 3619 3620 3621
            ]
        );
        assert_stream_equal(
            "[2, [4, 1]]",
3622
            vec![
C
Corey Farwell 已提交
3623
                (ArrayStart,           vec![]),
3624
                    (U64Value(2),      vec![Index(0)]),
C
Corey Farwell 已提交
3625
                    (ArrayStart,       vec![Index(1)]),
3626 3627
                        (U64Value(4),  vec![Index(1), Index(0)]),
                        (U64Value(1),  vec![Index(1), Index(1)]),
C
Corey Farwell 已提交
3628 3629
                    (ArrayEnd,         vec![Index(1)]),
                (ArrayEnd,             vec![]),
3630 3631 3632 3633 3634 3635
            ]
        );

        assert_eq!(last_event("["), Error(SyntaxError(EOFWhileParsingValue, 1,  2)));

        assert_eq!(from_str("["),     Err(SyntaxError(EOFWhileParsingValue, 1, 2)));
C
Corey Farwell 已提交
3636
        assert_eq!(from_str("[1"),    Err(SyntaxError(EOFWhileParsingArray, 1, 3)));
3637 3638 3639 3640 3641 3642 3643 3644 3645 3646 3647 3648 3649 3650 3651 3652 3653 3654 3655 3656 3657 3658 3659 3660 3661 3662 3663 3664 3665 3666 3667 3668 3669 3670 3671 3672 3673 3674 3675 3676
        assert_eq!(from_str("[1,"),   Err(SyntaxError(EOFWhileParsingValue, 1, 4)));
        assert_eq!(from_str("[1,]"),  Err(SyntaxError(InvalidSyntax,        1, 4)));
        assert_eq!(from_str("[6 7]"), Err(SyntaxError(InvalidSyntax,        1, 4)));

    }
    #[test]
    fn test_trailing_characters_streaming() {
        assert_eq!(last_event("nulla"),  Error(SyntaxError(TrailingCharacters, 1, 5)));
        assert_eq!(last_event("truea"),  Error(SyntaxError(TrailingCharacters, 1, 5)));
        assert_eq!(last_event("falsea"), Error(SyntaxError(TrailingCharacters, 1, 6)));
        assert_eq!(last_event("1a"),     Error(SyntaxError(TrailingCharacters, 1, 2)));
        assert_eq!(last_event("[]a"),    Error(SyntaxError(TrailingCharacters, 1, 3)));
        assert_eq!(last_event("{}a"),    Error(SyntaxError(TrailingCharacters, 1, 3)));
    }
    #[test]
    fn test_read_identifiers_streaming() {
        assert_eq!(Parser::new("null".chars()).next(), Some(NullValue));
        assert_eq!(Parser::new("true".chars()).next(), Some(BooleanValue(true)));
        assert_eq!(Parser::new("false".chars()).next(), Some(BooleanValue(false)));

        assert_eq!(last_event("n"),    Error(SyntaxError(InvalidSyntax, 1, 2)));
        assert_eq!(last_event("nul"),  Error(SyntaxError(InvalidSyntax, 1, 4)));
        assert_eq!(last_event("t"),    Error(SyntaxError(InvalidSyntax, 1, 2)));
        assert_eq!(last_event("truz"), Error(SyntaxError(InvalidSyntax, 1, 4)));
        assert_eq!(last_event("f"),    Error(SyntaxError(InvalidSyntax, 1, 2)));
        assert_eq!(last_event("faz"),  Error(SyntaxError(InvalidSyntax, 1, 3)));
    }

    #[test]
    fn test_stack() {
        let mut stack = Stack::new();

        assert!(stack.is_empty());
        assert!(stack.len() == 0);
        assert!(!stack.last_is_index());

        stack.push_index(0);
        stack.bump_index();

        assert!(stack.len() == 1);
N
Nick Cameron 已提交
3677 3678 3679
        assert!(stack.is_equal_to(&[Index(1)]));
        assert!(stack.starts_with(&[Index(1)]));
        assert!(stack.ends_with(&[Index(1)]));
3680 3681 3682
        assert!(stack.last_is_index());
        assert!(stack.get(0) == Index(1));

A
Alex Crichton 已提交
3683
        stack.push_key("foo".to_string());
3684 3685

        assert!(stack.len() == 2);
N
Nick Cameron 已提交
3686 3687 3688 3689 3690
        assert!(stack.is_equal_to(&[Index(1), Key("foo")]));
        assert!(stack.starts_with(&[Index(1), Key("foo")]));
        assert!(stack.starts_with(&[Index(1)]));
        assert!(stack.ends_with(&[Index(1), Key("foo")]));
        assert!(stack.ends_with(&[Key("foo")]));
3691 3692 3693 3694
        assert!(!stack.last_is_index());
        assert!(stack.get(0) == Index(1));
        assert!(stack.get(1) == Key("foo"));

A
Alex Crichton 已提交
3695
        stack.push_key("bar".to_string());
3696 3697

        assert!(stack.len() == 3);
N
Nick Cameron 已提交
3698 3699 3700 3701 3702 3703 3704
        assert!(stack.is_equal_to(&[Index(1), Key("foo"), Key("bar")]));
        assert!(stack.starts_with(&[Index(1)]));
        assert!(stack.starts_with(&[Index(1), Key("foo")]));
        assert!(stack.starts_with(&[Index(1), Key("foo"), Key("bar")]));
        assert!(stack.ends_with(&[Key("bar")]));
        assert!(stack.ends_with(&[Key("foo"), Key("bar")]));
        assert!(stack.ends_with(&[Index(1), Key("foo"), Key("bar")]));
3705 3706 3707 3708 3709 3710 3711 3712
        assert!(!stack.last_is_index());
        assert!(stack.get(0) == Index(1));
        assert!(stack.get(1) == Key("foo"));
        assert!(stack.get(2) == Key("bar"));

        stack.pop();

        assert!(stack.len() == 2);
N
Nick Cameron 已提交
3713 3714 3715 3716 3717
        assert!(stack.is_equal_to(&[Index(1), Key("foo")]));
        assert!(stack.starts_with(&[Index(1), Key("foo")]));
        assert!(stack.starts_with(&[Index(1)]));
        assert!(stack.ends_with(&[Index(1), Key("foo")]));
        assert!(stack.ends_with(&[Key("foo")]));
3718 3719 3720 3721 3722
        assert!(!stack.last_is_index());
        assert!(stack.get(0) == Index(1));
        assert!(stack.get(1) == Key("foo"));
    }

3723 3724
    #[test]
    fn test_to_json() {
A
Alexis Beingessner 已提交
3725
        use std::collections::{HashMap,BTreeMap};
3726 3727
        use super::ToJson;

C
Corey Farwell 已提交
3728 3729
        let array2 = Array(vec!(U64(1), U64(2)));
        let array3 = Array(vec!(U64(1), U64(2), U64(3)));
3730
        let object = {
A
Alexis Beingessner 已提交
3731
            let mut tree_map = BTreeMap::new();
3732 3733
            tree_map.insert("a".to_string(), U64(1));
            tree_map.insert("b".to_string(), U64(2));
A
Adolfo Ochagavía 已提交
3734
            Object(tree_map)
3735 3736
        };

C
Corey Farwell 已提交
3737
        assert_eq!(array2.to_json(), array2);
3738
        assert_eq!(object.to_json(), object);
3739 3740 3741 3742 3743 3744 3745 3746 3747 3748 3749 3750
        assert_eq!(3_i.to_json(), I64(3));
        assert_eq!(4_i8.to_json(), I64(4));
        assert_eq!(5_i16.to_json(), I64(5));
        assert_eq!(6_i32.to_json(), I64(6));
        assert_eq!(7_i64.to_json(), I64(7));
        assert_eq!(8_u.to_json(), U64(8));
        assert_eq!(9_u8.to_json(), U64(9));
        assert_eq!(10_u16.to_json(), U64(10));
        assert_eq!(11_u32.to_json(), U64(11));
        assert_eq!(12_u64.to_json(), U64(12));
        assert_eq!(13.0_f32.to_json(), F64(13.0_f64));
        assert_eq!(14.0_f64.to_json(), F64(14.0_f64));
3751
        assert_eq!(().to_json(), Null);
M
mrec 已提交
3752 3753
        assert_eq!(f32::INFINITY.to_json(), Null);
        assert_eq!(f64::NAN.to_json(), Null);
3754 3755
        assert_eq!(true.to_json(), Boolean(true));
        assert_eq!(false.to_json(), Boolean(false));
A
Alex Crichton 已提交
3756 3757
        assert_eq!("abc".to_json(), String("abc".to_string()));
        assert_eq!("abc".to_string().to_json(), String("abc".to_string()));
C
Corey Farwell 已提交
3758 3759 3760 3761 3762 3763
        assert_eq!((1u, 2u).to_json(), array2);
        assert_eq!((1u, 2u, 3u).to_json(), array3);
        assert_eq!([1u, 2].to_json(), array2);
        assert_eq!((&[1u, 2, 3]).to_json(), array3);
        assert_eq!((vec![1u, 2]).to_json(), array2);
        assert_eq!(vec!(1u, 2, 3).to_json(), array3);
A
Alexis Beingessner 已提交
3764
        let mut tree_map = BTreeMap::new();
3765 3766
        tree_map.insert("a".to_string(), 1u);
        tree_map.insert("b".to_string(), 2);
3767 3768
        assert_eq!(tree_map.to_json(), object);
        let mut hash_map = HashMap::new();
A
Alex Crichton 已提交
3769 3770
        hash_map.insert("a".to_string(), 1u);
        hash_map.insert("b".to_string(), 2);
3771
        assert_eq!(hash_map.to_json(), object);
3772 3773
        assert_eq!(Some(15i).to_json(), I64(15));
        assert_eq!(Some(15u).to_json(), U64(15));
3774 3775 3776
        assert_eq!(None::<int>.to_json(), Null);
    }

3777 3778 3779 3780 3781 3782 3783 3784 3785 3786 3787 3788 3789 3790 3791 3792 3793 3794 3795 3796 3797 3798 3799 3800 3801 3802 3803 3804 3805 3806 3807 3808 3809 3810 3811
    #[bench]
    fn bench_streaming_small(b: &mut Bencher) {
        b.iter( || {
            let mut parser = Parser::new(
                r#"{
                    "a": 1.0,
                    "b": [
                        true,
                        "foo\nbar",
                        { "c": {"d": null} }
                    ]
                }"#.chars()
            );
            loop {
                match parser.next() {
                    None => return,
                    _ => {}
                }
            }
        });
    }
    #[bench]
    fn bench_small(b: &mut Bencher) {
        b.iter( || {
            let _ = from_str(r#"{
                "a": 1.0,
                "b": [
                    true,
                    "foo\nbar",
                    { "c": {"d": null} }
                ]
            }"#);
        });
    }

3812
    fn big_json() -> string::String {
A
Alex Crichton 已提交
3813
        let mut src = "[\n".to_string();
3814
        for _ in range(0i, 500) {
3815 3816
            src.push_str(r#"{ "a": true, "b": null, "c":3.1415, "d": "Hello world", "e": \
                            [1,2,3]},"#);
3817
        }
3818
        src.push_str("{}]");
3819 3820 3821 3822 3823 3824 3825
        return src;
    }

    #[bench]
    fn bench_streaming_large(b: &mut Bencher) {
        let src = big_json();
        b.iter( || {
3826
            let mut parser = Parser::new(src.chars());
3827 3828 3829 3830 3831 3832 3833 3834 3835 3836 3837
            loop {
                match parser.next() {
                    None => return,
                    _ => {}
                }
            }
        });
    }
    #[bench]
    fn bench_large(b: &mut Bencher) {
        let src = big_json();
3838
        b.iter( || { let _ = from_str(src.as_slice()); });
3839
    }
3840
}