json.rs 76.2 KB
Newer Older
1
// Copyright 2012-2013 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)];
15
#[allow(missing_doc)];
E
Elly Jones 已提交
16

M
musitdev 已提交
17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32
/*!
JSON parsing and serialization

# What is JSON?

JSON (JavaScript Object Notation) is a way to write data in Javascript.
Like XML it allows one to encode structured data in a text format that can be read by humans easily.
Its native compatibility with JavaScript and its simple syntax make it used widely.

Json data are encoded in a form of "key":"value".
Data types that can be encoded are JavaScript types :
boolean (`true` or `false`), number (`f64`), string, array, object, 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:

A
Alex Crichton 已提交
33
```ignore
M
musitdev 已提交
34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53
{
    "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.
54 55
To be able to encode a piece of data, it must implement the `serialize::Encodable` trait.
To be able to decode a piece of data, it must implement the `serialize::Decodable` trait.
M
musitdev 已提交
56 57 58 59 60 61 62
The Rust compiler provides an annotation to automatically generate
the code for these traits: `#[deriving(Decodable, Encodable)]`

To encode using Encodable :

```rust
use std::io;
A
Alex Crichton 已提交
63
use serialize::{json, Encodable};
M
musitdev 已提交
64 65 66 67 68 69 70 71

 #[deriving(Encodable)]
 pub struct TestStruct   {
    data_str: ~str,
 }

fn main() {
    let to_encode_object = TestStruct{data_str:~"example of string to encode"};
72
    let mut m = io::MemWriter::new();
M
musitdev 已提交
73 74 75 76 77 78 79 80 81 82 83
    {
        let mut encoder = json::Encoder::new(&mut m as &mut std::io::Writer);
        to_encode_object.encode(&mut encoder);
    }
}
```

Two wrapper functions are provided to encode a Encodable object
into a string (~str) or buffer (~[u8]): `str_encode(&m)` and `buffer_encode(&m)`.

```rust
A
Alex Crichton 已提交
84
use serialize::json;
M
musitdev 已提交
85 86 87 88 89 90 91 92 93 94 95 96 97 98
let to_encode_object = ~"example of string to encode";
let encoded_str: ~str = json::Encoder::str_encode(&to_encode_object);
```

JSON API provide an enum `json::Json` and a trait `ToJson` to encode object.
The trait `ToJson` encode object into a container `json::Json` and the API provide writer
to encode them into a stream or a string ...

When using `ToJson` the `Encodable` trait implementation is not mandatory.

A basic `ToJson` example using a TreeMap of attribute name / attribute value:


```rust
A
Alex Crichton 已提交
99
extern crate collections;
A
Alex Crichton 已提交
100
extern crate serialize;
101

A
Alex Crichton 已提交
102 103
use serialize::json;
use serialize::json::ToJson;
104
use collections::TreeMap;
M
musitdev 已提交
105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126

pub struct MyStruct  {
    attr1: u8,
    attr2: ~str,
}

impl ToJson for MyStruct {
    fn to_json( &self ) -> json::Json {
        let mut d = ~TreeMap::new();
        d.insert(~"attr1", self.attr1.to_json());
        d.insert(~"attr2", self.attr2.to_json());
        json::Object(d)
    }
}

fn main() {
    let test2: MyStruct = MyStruct {attr1: 1, attr2:~"test"};
    let tjson: json::Json = test2.to_json();
    let json_str: ~str = tjson.to_str();
}
```

H
Huon Wilson 已提交
127
To decode a JSON string using `Decodable` trait :
M
musitdev 已提交
128 129

```rust
A
Alex Crichton 已提交
130
extern crate serialize;
A
Alex Crichton 已提交
131
use serialize::{json, Decodable};
M
musitdev 已提交
132 133 134 135 136 137 138 139 140 141

#[deriving(Decodable)]
pub struct MyStruct  {
     attr1: u8,
     attr2: ~str,
}

fn main() {
    let json_str_to_decode: ~str =
            ~"{\"attr1\":1,\"attr2\":\"toto\"}";
A
Alex Crichton 已提交
142 143
    let json_object = json::from_str(json_str_to_decode);
    let mut decoder = json::Decoder::new(json_object.unwrap());
M
musitdev 已提交
144 145 146 147 148 149 150 151 152 153 154 155
    let decoded_object: MyStruct = Decodable::decode(&mut decoder); // create the final object
}
```

# Examples of use

## Using Autoserialization

Create a struct called TestStruct1 and serialize and deserialize it to and from JSON
using the serialization API, using the derived serialization code.

```rust
A
Alex Crichton 已提交
156
extern crate serialize;
A
Alex Crichton 已提交
157
use serialize::{json, Encodable, Decodable};
M
musitdev 已提交
158 159 160 161 162 163 164 165 166 167 168 169 170 171 172

 #[deriving(Decodable, Encodable)] //generate Decodable, Encodable impl.
 pub struct TestStruct1  {
    data_int: u8,
    data_str: ~str,
    data_vector: ~[u8],
 }

// To serialize use the `json::str_encode` to encode an object in a string.
// It calls the generated `Encodable` impl.
fn main() {
    let to_encode_object = TestStruct1
         {data_int: 1, data_str:~"toto", data_vector:~[2,3,4,5]};
    let encoded_str: ~str = json::Encoder::str_encode(&to_encode_object);

A
Alex Crichton 已提交
173
    // To deserialize use the `json::from_str` and `json::Decoder`
M
musitdev 已提交
174

A
Alex Crichton 已提交
175
    let json_object = json::from_str(encoded_str);
M
musitdev 已提交
176 177 178 179 180 181 182
    let mut decoder = json::Decoder::new(json_object.unwrap());
    let decoded1: TestStruct1 = Decodable::decode(&mut decoder); // create the final object
}
```

## Using `ToJson`

H
Huon Wilson 已提交
183
This example use the ToJson impl to deserialize the JSON string.
M
musitdev 已提交
184 185 186
Example of `ToJson` trait implementation for TestStruct1.

```rust
A
Alex Crichton 已提交
187 188
extern crate serialize;
extern crate collections;
189

A
Alex Crichton 已提交
190 191
use serialize::json::ToJson;
use serialize::{json, Encodable, Decodable};
192
use collections::TreeMap;
M
musitdev 已提交
193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211

#[deriving(Decodable, Encodable)] // generate Decodable, Encodable impl.
pub struct TestStruct1  {
    data_int: u8,
    data_str: ~str,
    data_vector: ~[u8],
}

impl ToJson for TestStruct1 {
    fn to_json( &self ) -> json::Json {
        let mut d = ~TreeMap::new();
        d.insert(~"data_int", self.data_int.to_json());
        d.insert(~"data_str", self.data_str.to_json());
        d.insert(~"data_vector", self.data_vector.to_json());
        json::Object(d)
    }
}

fn main() {
H
Huon Wilson 已提交
212
    // Serialization using our impl of to_json
M
musitdev 已提交
213 214 215 216 217

    let test2: TestStruct1 = TestStruct1 {data_int: 1, data_str:~"toto", data_vector:~[2,3,4,5]};
    let tjson: json::Json = test2.to_json();
    let json_str: ~str = tjson.to_str();

H
Huon Wilson 已提交
218
    // Deserialize like before.
M
musitdev 已提交
219 220 221 222 223 224 225 226

    let mut decoder = json::Decoder::new(json::from_str(json_str).unwrap());
    // create the final object
    let decoded2: TestStruct1 = Decodable::decode(&mut decoder);
}
```

*/
B
Brian Anderson 已提交
227

228
use std::char;
D
Daniel Micay 已提交
229
use std::f64;
230
use collections::HashMap;
A
Alex Crichton 已提交
231
use std::io;
A
Alex Crichton 已提交
232
use std::io::MemWriter;
D
Daniel Micay 已提交
233
use std::num;
234
use std::str;
235
use std::fmt;
236

A
Alex Crichton 已提交
237
use Encodable;
238
use collections::TreeMap;
E
Elly Jones 已提交
239

A
Alex Crichton 已提交
240
macro_rules! try( ($e:expr) => (
A
Alex Crichton 已提交
241 242 243
    match $e { Ok(e) => e, Err(e) => { self.error = Err(e); return } }
) )

244
/// Represents a json value
245
#[deriving(Clone, Eq)]
246
pub enum Json {
D
Daniel Micay 已提交
247
    Number(f64),
248
    String(~str),
B
Ben Striegel 已提交
249
    Boolean(bool),
250 251
    List(List),
    Object(~Object),
B
Ben Striegel 已提交
252
    Null,
E
Elly Jones 已提交
253 254
}

255
pub type List = ~[Json];
256
pub type Object = TreeMap<~str, Json>;
257

258
#[deriving(Eq)]
259 260
/// If an error occurs while parsing some JSON, this is the structure which is
/// returned
261
pub struct Error {
262
    /// The line number at which the error occurred
263
    priv line: uint,
264
    /// The column number at which the error occurred
265
    priv col: uint,
266
    /// A message describing the type of the error
E
Erick Tryzelaar 已提交
267
    priv msg: ~str,
268
}
269

A
Alex Crichton 已提交
270 271 272 273 274 275 276 277
fn io_error_to_error(io: io::IoError) -> Error {
    Error {
        line: 0,
        col: 0,
        msg: format!("io error: {}", io)
    }
}

278 279
fn escape_str(s: &str) -> ~str {
    let mut escaped = ~"\"";
280
    for c in s.chars() {
281
        match c {
282 283 284 285 286 287 288 289
          '"' => escaped.push_str("\\\""),
          '\\' => escaped.push_str("\\\\"),
          '\x08' => escaped.push_str("\\b"),
          '\x0c' => escaped.push_str("\\f"),
          '\n' => escaped.push_str("\\n"),
          '\r' => escaped.push_str("\\r"),
          '\t' => escaped.push_str("\\t"),
          _ => escaped.push_char(c),
E
Elly Jones 已提交
290
        }
291
    };
292

293
    escaped.push_char('"');
294 295

    escaped
E
Elly Jones 已提交
296 297
}

298 299
fn spaces(n: uint) -> ~str {
    let mut ss = ~"";
B
Brendan Zabarauskas 已提交
300
    for _ in range(0, n) { ss.push_str(" "); }
301 302 303
    return ss;
}

304
/// A structure for implementing serialization to JSON.
E
Erik Price 已提交
305 306
pub struct Encoder<'a> {
    priv wr: &'a mut io::Writer,
A
Alex Crichton 已提交
307
    priv error: io::IoResult<()>,
308 309
}

E
Erik Price 已提交
310
impl<'a> Encoder<'a> {
311 312
    /// Creates a new JSON encoder whose output will be written to the writer
    /// specified.
313
    pub fn new<'a>(wr: &'a mut io::Writer) -> Encoder<'a> {
A
Alex Crichton 已提交
314
        Encoder { wr: wr, error: Ok(()) }
315
    }
M
musitdev 已提交
316 317

    /// Encode the specified struct into a json [u8]
A
Alex Crichton 已提交
318
    pub fn buffer_encode<T:Encodable<Encoder<'a>>>(to_encode_object: &T) -> ~[u8]  {
M
musitdev 已提交
319 320 321 322 323 324
       //Serialize the object in a string using a writer
        let mut m = MemWriter::new();
        {
            let mut encoder = Encoder::new(&mut m as &mut io::Writer);
            to_encode_object.encode(&mut encoder);
        }
325
        m.unwrap()
M
musitdev 已提交
326 327 328
    }

    /// Encode the specified struct into a json str
A
Alex Crichton 已提交
329
    pub fn str_encode<T:Encodable<Encoder<'a>>>(to_encode_object: &T) -> ~str  {
M
musitdev 已提交
330
        let buff:~[u8] = Encoder::buffer_encode(to_encode_object);
331
        str::from_utf8_owned(buff).unwrap()
M
musitdev 已提交
332
    }
333 334
}

A
Alex Crichton 已提交
335
impl<'a> ::Encoder for Encoder<'a> {
A
Alex Crichton 已提交
336
    fn emit_nil(&mut self) { try!(write!(self.wr, "null")) }
337

D
Daniel Micay 已提交
338 339 340 341 342
    fn emit_uint(&mut self, v: uint) { self.emit_f64(v as f64); }
    fn emit_u64(&mut self, v: u64) { self.emit_f64(v as f64); }
    fn emit_u32(&mut self, v: u32) { self.emit_f64(v as f64); }
    fn emit_u16(&mut self, v: u16) { self.emit_f64(v as f64); }
    fn emit_u8(&mut self, v: u8)   { self.emit_f64(v as f64); }
343

D
Daniel Micay 已提交
344 345 346 347 348
    fn emit_int(&mut self, v: int) { self.emit_f64(v as f64); }
    fn emit_i64(&mut self, v: i64) { self.emit_f64(v as f64); }
    fn emit_i32(&mut self, v: i32) { self.emit_f64(v as f64); }
    fn emit_i16(&mut self, v: i16) { self.emit_f64(v as f64); }
    fn emit_i8(&mut self, v: i8)   { self.emit_f64(v as f64); }
349 350 351

    fn emit_bool(&mut self, v: bool) {
        if v {
A
Alex Crichton 已提交
352
            try!(write!(self.wr, "true"));
353
        } else {
A
Alex Crichton 已提交
354
            try!(write!(self.wr, "false"));
355 356 357
        }
    }

A
Alex Crichton 已提交
358
    fn emit_f64(&mut self, v: f64) {
A
Alex Crichton 已提交
359
        try!(write!(self.wr, "{}", f64::to_str_digits(v, 6u)))
A
Alex Crichton 已提交
360
    }
D
Daniel Micay 已提交
361
    fn emit_f32(&mut self, v: f32) { self.emit_f64(v as f64); }
362 363

    fn emit_char(&mut self, v: char) { self.emit_str(str::from_char(v)) }
A
Alex Crichton 已提交
364
    fn emit_str(&mut self, v: &str) {
A
Alex Crichton 已提交
365
        try!(write!(self.wr, "{}", escape_str(v)))
A
Alex Crichton 已提交
366
    }
367

E
Erik Price 已提交
368
    fn emit_enum(&mut self, _name: &str, f: |&mut Encoder<'a>|) { f(self) }
369 370 371 372 373

    fn emit_enum_variant(&mut self,
                         name: &str,
                         _id: uint,
                         cnt: uint,
E
Erik Price 已提交
374
                         f: |&mut Encoder<'a>|) {
375
        // enums are encoded as strings or objects
376
        // Bunny => "Bunny"
377
        // Kangaroo(34,"William") => {"variant": "Kangaroo", "fields": [34,"William"]}
378
        if cnt == 0 {
A
Alex Crichton 已提交
379
            try!(write!(self.wr, "{}", escape_str(name)));
380
        } else {
A
Alex Crichton 已提交
381 382 383
            try!(write!(self.wr, "\\{\"variant\":"));
            try!(write!(self.wr, "{}", escape_str(name)));
            try!(write!(self.wr, ",\"fields\":["));
384
            f(self);
A
Alex Crichton 已提交
385
            try!(write!(self.wr, "]\\}"));
386 387 388
        }
    }

E
Erik Price 已提交
389
    fn emit_enum_variant_arg(&mut self, idx: uint, f: |&mut Encoder<'a>|) {
390
        if idx != 0 {
A
Alex Crichton 已提交
391
            try!(write!(self.wr, ","));
392 393 394 395 396 397 398 399
        }
        f(self);
    }

    fn emit_enum_struct_variant(&mut self,
                                name: &str,
                                id: uint,
                                cnt: uint,
E
Erik Price 已提交
400
                                f: |&mut Encoder<'a>|) {
401 402 403 404 405 406
        self.emit_enum_variant(name, id, cnt, f)
    }

    fn emit_enum_struct_variant_field(&mut self,
                                      _: &str,
                                      idx: uint,
E
Erik Price 已提交
407
                                      f: |&mut Encoder<'a>|) {
408 409 410
        self.emit_enum_variant_arg(idx, f)
    }

E
Erik Price 已提交
411
    fn emit_struct(&mut self, _: &str, _: uint, f: |&mut Encoder<'a>|) {
A
Alex Crichton 已提交
412
        try!(write!(self.wr, r"\{"));
413
        f(self);
A
Alex Crichton 已提交
414
        try!(write!(self.wr, r"\}"));
415 416 417 418 419
    }

    fn emit_struct_field(&mut self,
                         name: &str,
                         idx: uint,
E
Erik Price 已提交
420
                         f: |&mut Encoder<'a>|) {
A
Alex Crichton 已提交
421 422
        if idx != 0 { try!(write!(self.wr, ",")) }
        try!(write!(self.wr, "{}:", escape_str(name)));
423 424 425
        f(self);
    }

E
Erik Price 已提交
426
    fn emit_tuple(&mut self, len: uint, f: |&mut Encoder<'a>|) {
427 428
        self.emit_seq(len, f)
    }
E
Erik Price 已提交
429
    fn emit_tuple_arg(&mut self, idx: uint, f: |&mut Encoder<'a>|) {
430 431 432 433 434 435
        self.emit_seq_elt(idx, f)
    }

    fn emit_tuple_struct(&mut self,
                         _name: &str,
                         len: uint,
E
Erik Price 已提交
436
                         f: |&mut Encoder<'a>|) {
437 438
        self.emit_seq(len, f)
    }
E
Erik Price 已提交
439
    fn emit_tuple_struct_arg(&mut self, idx: uint, f: |&mut Encoder<'a>|) {
440 441 442
        self.emit_seq_elt(idx, f)
    }

E
Erik Price 已提交
443
    fn emit_option(&mut self, f: |&mut Encoder<'a>|) { f(self); }
444
    fn emit_option_none(&mut self) { self.emit_nil(); }
E
Erik Price 已提交
445
    fn emit_option_some(&mut self, f: |&mut Encoder<'a>|) { f(self); }
446

E
Erik Price 已提交
447
    fn emit_seq(&mut self, _len: uint, f: |&mut Encoder<'a>|) {
A
Alex Crichton 已提交
448
        try!(write!(self.wr, "["));
449
        f(self);
A
Alex Crichton 已提交
450
        try!(write!(self.wr, "]"));
451 452
    }

E
Erik Price 已提交
453
    fn emit_seq_elt(&mut self, idx: uint, f: |&mut Encoder<'a>|) {
454
        if idx != 0 {
A
Alex Crichton 已提交
455
            try!(write!(self.wr, ","));
456 457 458 459
        }
        f(self)
    }

E
Erik Price 已提交
460
    fn emit_map(&mut self, _len: uint, f: |&mut Encoder<'a>|) {
A
Alex Crichton 已提交
461
        try!(write!(self.wr, r"\{"));
462
        f(self);
A
Alex Crichton 已提交
463
        try!(write!(self.wr, r"\}"));
464 465
    }

E
Erik Price 已提交
466
    fn emit_map_elt_key(&mut self, idx: uint, f: |&mut Encoder<'a>|) {
A
Alex Crichton 已提交
467
        if idx != 0 { try!(write!(self.wr, ",")) }
468 469 470
        f(self)
    }

E
Erik Price 已提交
471
    fn emit_map_elt_val(&mut self, _idx: uint, f: |&mut Encoder<'a>|) {
A
Alex Crichton 已提交
472
        try!(write!(self.wr, ":"));
473 474 475 476
        f(self)
    }
}

477 478
/// Another encoder for JSON, but prints out human-readable JSON instead of
/// compact data
E
Erik Price 已提交
479 480
pub struct PrettyEncoder<'a> {
    priv wr: &'a mut io::Writer,
481
    priv indent: uint,
A
Alex Crichton 已提交
482
    priv error: io::IoResult<()>,
483 484
}

E
Erik Price 已提交
485
impl<'a> PrettyEncoder<'a> {
486
    /// Creates a new encoder whose output will be written to the specified writer
487
    pub fn new<'a>(wr: &'a mut io::Writer) -> PrettyEncoder<'a> {
488 489 490
        PrettyEncoder {
            wr: wr,
            indent: 0,
A
Alex Crichton 已提交
491
            error: Ok(())
492
        }
493
    }
494
}
495

A
Alex Crichton 已提交
496
impl<'a> ::Encoder for PrettyEncoder<'a> {
A
Alex Crichton 已提交
497
    fn emit_nil(&mut self) { try!(write!(self.wr, "null")); }
498

D
Daniel Micay 已提交
499 500 501 502 503
    fn emit_uint(&mut self, v: uint) { self.emit_f64(v as f64); }
    fn emit_u64(&mut self, v: u64) { self.emit_f64(v as f64); }
    fn emit_u32(&mut self, v: u32) { self.emit_f64(v as f64); }
    fn emit_u16(&mut self, v: u16) { self.emit_f64(v as f64); }
    fn emit_u8(&mut self, v: u8)   { self.emit_f64(v as f64); }
504

D
Daniel Micay 已提交
505 506 507 508 509
    fn emit_int(&mut self, v: int) { self.emit_f64(v as f64); }
    fn emit_i64(&mut self, v: i64) { self.emit_f64(v as f64); }
    fn emit_i32(&mut self, v: i32) { self.emit_f64(v as f64); }
    fn emit_i16(&mut self, v: i16) { self.emit_f64(v as f64); }
    fn emit_i8(&mut self, v: i8)   { self.emit_f64(v as f64); }
510 511 512

    fn emit_bool(&mut self, v: bool) {
        if v {
A
Alex Crichton 已提交
513
            try!(write!(self.wr, "true"));
514
        } else {
A
Alex Crichton 已提交
515
            try!(write!(self.wr, "false"));
516 517 518
        }
    }

A
Alex Crichton 已提交
519
    fn emit_f64(&mut self, v: f64) {
A
Alex Crichton 已提交
520
        try!(write!(self.wr, "{}", f64::to_str_digits(v, 6u)));
A
Alex Crichton 已提交
521
    }
D
Daniel Micay 已提交
522
    fn emit_f32(&mut self, v: f32) { self.emit_f64(v as f64); }
523 524

    fn emit_char(&mut self, v: char) { self.emit_str(str::from_char(v)) }
A
Alex Crichton 已提交
525
    fn emit_str(&mut self, v: &str) {
A
Alex Crichton 已提交
526
        try!(write!(self.wr, "{}", escape_str(v)));
A
Alex Crichton 已提交
527
    }
528

E
Erik Price 已提交
529
    fn emit_enum(&mut self, _name: &str, f: |&mut PrettyEncoder<'a>|) {
530 531 532 533 534 535 536
        f(self)
    }

    fn emit_enum_variant(&mut self,
                         name: &str,
                         _: uint,
                         cnt: uint,
E
Erik Price 已提交
537
                         f: |&mut PrettyEncoder<'a>|) {
538
        if cnt == 0 {
A
Alex Crichton 已提交
539
            try!(write!(self.wr, "{}", escape_str(name)));
540 541
        } else {
            self.indent += 2;
A
Alex Crichton 已提交
542
            try!(write!(self.wr, "[\n{}{},\n", spaces(self.indent),
A
Alex Crichton 已提交
543
                          escape_str(name)));
544 545
            f(self);
            self.indent -= 2;
A
Alex Crichton 已提交
546
            try!(write!(self.wr, "\n{}]", spaces(self.indent)));
547 548 549 550 551
        }
    }

    fn emit_enum_variant_arg(&mut self,
                             idx: uint,
E
Erik Price 已提交
552
                             f: |&mut PrettyEncoder<'a>|) {
553
        if idx != 0 {
A
Alex Crichton 已提交
554
            try!(write!(self.wr, ",\n"));
555
        }
A
Alex Crichton 已提交
556
        try!(write!(self.wr, "{}", spaces(self.indent)));
557 558 559 560 561 562 563
        f(self)
    }

    fn emit_enum_struct_variant(&mut self,
                                name: &str,
                                id: uint,
                                cnt: uint,
E
Erik Price 已提交
564
                                f: |&mut PrettyEncoder<'a>|) {
565 566 567 568 569 570
        self.emit_enum_variant(name, id, cnt, f)
    }

    fn emit_enum_struct_variant_field(&mut self,
                                      _: &str,
                                      idx: uint,
E
Erik Price 已提交
571
                                      f: |&mut PrettyEncoder<'a>|) {
572 573 574 575 576 577 578
        self.emit_enum_variant_arg(idx, f)
    }


    fn emit_struct(&mut self,
                   _: &str,
                   len: uint,
E
Erik Price 已提交
579
                   f: |&mut PrettyEncoder<'a>|) {
580
        if len == 0 {
A
Alex Crichton 已提交
581
            try!(write!(self.wr, "\\{\\}"));
582
        } else {
A
Alex Crichton 已提交
583
            try!(write!(self.wr, "\\{"));
584 585 586
            self.indent += 2;
            f(self);
            self.indent -= 2;
A
Alex Crichton 已提交
587
            try!(write!(self.wr, "\n{}\\}", spaces(self.indent)));
588 589 590 591 592 593
        }
    }

    fn emit_struct_field(&mut self,
                         name: &str,
                         idx: uint,
E
Erik Price 已提交
594
                         f: |&mut PrettyEncoder<'a>|) {
595
        if idx == 0 {
A
Alex Crichton 已提交
596
            try!(write!(self.wr, "\n"));
597
        } else {
A
Alex Crichton 已提交
598
            try!(write!(self.wr, ",\n"));
599
        }
A
Alex Crichton 已提交
600
        try!(write!(self.wr, "{}{}: ", spaces(self.indent), escape_str(name)));
601 602 603
        f(self);
    }

E
Erik Price 已提交
604
    fn emit_tuple(&mut self, len: uint, f: |&mut PrettyEncoder<'a>|) {
605 606
        self.emit_seq(len, f)
    }
E
Erik Price 已提交
607
    fn emit_tuple_arg(&mut self, idx: uint, f: |&mut PrettyEncoder<'a>|) {
608 609 610 611 612 613
        self.emit_seq_elt(idx, f)
    }

    fn emit_tuple_struct(&mut self,
                         _: &str,
                         len: uint,
E
Erik Price 已提交
614
                         f: |&mut PrettyEncoder<'a>|) {
615 616 617 618
        self.emit_seq(len, f)
    }
    fn emit_tuple_struct_arg(&mut self,
                             idx: uint,
E
Erik Price 已提交
619
                             f: |&mut PrettyEncoder<'a>|) {
620 621 622
        self.emit_seq_elt(idx, f)
    }

E
Erik Price 已提交
623
    fn emit_option(&mut self, f: |&mut PrettyEncoder<'a>|) { f(self); }
624
    fn emit_option_none(&mut self) { self.emit_nil(); }
E
Erik Price 已提交
625
    fn emit_option_some(&mut self, f: |&mut PrettyEncoder<'a>|) { f(self); }
626

E
Erik Price 已提交
627
    fn emit_seq(&mut self, len: uint, f: |&mut PrettyEncoder<'a>|) {
628
        if len == 0 {
A
Alex Crichton 已提交
629
            try!(write!(self.wr, "[]"));
630
        } else {
A
Alex Crichton 已提交
631
            try!(write!(self.wr, "["));
632 633 634
            self.indent += 2;
            f(self);
            self.indent -= 2;
A
Alex Crichton 已提交
635
            try!(write!(self.wr, "\n{}]", spaces(self.indent)));
636 637 638
        }
    }

E
Erik Price 已提交
639
    fn emit_seq_elt(&mut self, idx: uint, f: |&mut PrettyEncoder<'a>|) {
640
        if idx == 0 {
A
Alex Crichton 已提交
641
            try!(write!(self.wr, "\n"));
642
        } else {
A
Alex Crichton 已提交
643
            try!(write!(self.wr, ",\n"));
644
        }
A
Alex Crichton 已提交
645
        try!(write!(self.wr, "{}", spaces(self.indent)));
646 647 648
        f(self)
    }

E
Erik Price 已提交
649
    fn emit_map(&mut self, len: uint, f: |&mut PrettyEncoder<'a>|) {
650
        if len == 0 {
A
Alex Crichton 已提交
651
            try!(write!(self.wr, "\\{\\}"));
652
        } else {
A
Alex Crichton 已提交
653
            try!(write!(self.wr, "\\{"));
654 655 656
            self.indent += 2;
            f(self);
            self.indent -= 2;
A
Alex Crichton 已提交
657
            try!(write!(self.wr, "\n{}\\}", spaces(self.indent)));
658 659 660
        }
    }

E
Erik Price 已提交
661
    fn emit_map_elt_key(&mut self, idx: uint, f: |&mut PrettyEncoder<'a>|) {
662
        if idx == 0 {
A
Alex Crichton 已提交
663
            try!(write!(self.wr, "\n"));
664
        } else {
A
Alex Crichton 已提交
665
            try!(write!(self.wr, ",\n"));
666
        }
A
Alex Crichton 已提交
667
        try!(write!(self.wr, "{}", spaces(self.indent)));
668 669 670
        f(self);
    }

E
Erik Price 已提交
671
    fn emit_map_elt_val(&mut self, _idx: uint, f: |&mut PrettyEncoder<'a>|) {
A
Alex Crichton 已提交
672
        try!(write!(self.wr, ": "));
673 674 675 676
        f(self);
    }
}

A
Alex Crichton 已提交
677
impl<E: ::Encoder> Encodable<E> for Json {
678 679 680 681 682 683 684 685 686 687 688 689
    fn encode(&self, e: &mut E) {
        match *self {
            Number(v) => v.encode(e),
            String(ref v) => v.encode(e),
            Boolean(v) => v.encode(e),
            List(ref v) => v.encode(e),
            Object(ref v) => v.encode(e),
            Null => e.emit_nil(),
        }
    }
}

690
impl Json {
691
    /// Encodes a json value into a io::writer.  Uses a single line.
A
Alex Crichton 已提交
692
    pub fn to_writer(&self, wr: &mut io::Writer) -> io::IoResult<()> {
693
        let mut encoder = Encoder::new(wr);
A
Alex Crichton 已提交
694 695
        self.encode(&mut encoder);
        encoder.error
696
    }
697

698 699
    /// Encodes a json value into a io::writer.
    /// Pretty-prints in a more readable format.
A
Alex Crichton 已提交
700
    pub fn to_pretty_writer(&self, wr: &mut io::Writer) -> io::IoResult<()> {
701
        let mut encoder = PrettyEncoder::new(wr);
A
Alex Crichton 已提交
702 703
        self.encode(&mut encoder);
        encoder.error
704
    }
705

706 707
    /// Encodes a json value into a string
    pub fn to_pretty_str(&self) -> ~str {
708
        let mut s = MemWriter::new();
A
Alex Crichton 已提交
709
        self.to_pretty_writer(&mut s as &mut io::Writer).unwrap();
710
        str::from_utf8_owned(s.unwrap()).unwrap()
711
    }
712 713 714 715 716 717 718 719 720 721 722

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

    /// Attempts to get a nested Json Object for each key in `keys`.
723
    /// If any key is found not to exist, find_path will return None.
724 725
    /// Otherwise, it will return the Json value associated with the final key.
    pub fn find_path<'a>(&'a self, keys: &[&~str]) -> Option<&'a Json>{
726 727 728 729 730 731 732 733
        let mut target = self;
        for key in keys.iter() {
            match target.find(*key) {
                Some(t) => { target = t; },
                None => return None
            }
        }
        Some(target)
734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761
    }

    /// 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.
    pub fn search<'a>(&'a self, key: &~str) -> Option<&'a Json> {
        match self {
            &Object(ref map) => {
                match map.find(key) {
                    Some(json_value) => Some(json_value),
                    None => {
                        let mut value : Option<&'a Json> = None;
                        for (_, v) in map.iter() {
                            value = v.search(key);
                            if value.is_some() {
                                break;
                            }
                        }
                        value
                    }
                }
            },
            _ => None
        }
    }

    /// Returns true if the Json value is an Object. Returns false otherwise.
    pub fn is_object<'a>(&'a self) -> bool {
762
        self.as_object().is_some()
763 764 765 766 767 768 769 770 771 772 773 774 775
    }

    /// If the Json value is an Object, returns the associated TreeMap.
    /// Returns None otherwise.
    pub fn as_object<'a>(&'a self) -> Option<&'a Object> {
        match self {
            &Object(ref map) => Some(&**map),
            _ => None
        }
    }

    /// Returns true if the Json value is a List. Returns false otherwise.
    pub fn is_list<'a>(&'a self) -> bool {
776
        self.as_list().is_some()
777 778 779 780 781 782 783 784 785 786 787 788
    }

    /// If the Json value is a List, returns the associated vector.
    /// Returns None otherwise.
    pub fn as_list<'a>(&'a self) -> Option<&'a List> {
        match self {
            &List(ref list) => Some(&*list),
            _ => None
        }
    }

    /// Returns true if the Json value is a String. Returns false otherwise.
789 790
    pub fn is_string<'a>(&'a self) -> bool {
        self.as_string().is_some()
791 792 793 794
    }

    /// If the Json value is a String, returns the associated str.
    /// Returns None otherwise.
795
    pub fn as_string<'a>(&'a self) -> Option<&'a str> {
796 797 798 799 800 801 802 803
        match *self {
            String(ref s) => Some(s.as_slice()),
            _ => None
        }
    }

    /// Returns true if the Json value is a Number. Returns false otherwise.
    pub fn is_number(&self) -> bool {
804
        self.as_number().is_some()
805 806 807 808 809 810 811 812 813 814 815 816 817
    }

    /// If the Json value is a Number, returns the associated f64.
    /// Returns None otherwise.
    pub fn as_number(&self) -> Option<f64> {
        match self {
            &Number(n) => Some(n),
            _ => None
        }
    }

    /// Returns true if the Json value is a Boolean. Returns false otherwise.
    pub fn is_boolean(&self) -> bool {
818
        self.as_boolean().is_some()
819 820 821 822 823 824 825 826 827 828 829 830 831
    }

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

    /// Returns true if the Json value is a Null. Returns false otherwise.
    pub fn is_null(&self) -> bool {
832
        self.as_null().is_some()
833 834 835 836 837 838 839 840 841 842
    }

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

G
Gary Linscott 已提交
845
pub struct Parser<T> {
846
    priv rdr: T,
847
    priv ch: Option<char>,
848 849
    priv line: uint,
    priv col: uint,
850 851
}

852 853
impl<T: Iterator<char>> Parser<T> {
    /// Decode a json value from an Iterator<char>
854
    pub fn new(rdr: T) -> Parser<T> {
855 856
        let mut p = Parser {
            rdr: rdr,
857
            ch: Some('\x00'),
858 859 860 861 862 863
            line: 1,
            col: 0,
        };
        p.bump();
        p
    }
864
}
E
Elly Jones 已提交
865

866
impl<T: Iterator<char>> Parser<T> {
867
    pub fn parse(&mut self) -> Result<Json, Error> {
L
Luqman Aden 已提交
868 869
        match self.parse_value() {
          Ok(value) => {
870 871 872 873
            // Skip trailing whitespaces.
            self.parse_whitespace();
            // Make sure there is no trailing characters.
            if self.eof() {
L
Luqman Aden 已提交
874
                Ok(value)
875 876 877 878
            } else {
                self.error(~"trailing characters")
            }
          }
L
Luqman Aden 已提交
879
          Err(e) => Err(e)
880 881
        }
    }
882 883
}

884
impl<T : Iterator<char>> Parser<T> {
885 886
    fn eof(&self) -> bool { self.ch.is_none() }
    fn ch_or_null(&self) -> char { self.ch.unwrap_or('\x00') }
887
    fn bump(&mut self) {
888
        self.ch = self.rdr.next();
E
Elly Jones 已提交
889

890
        if self.ch_is('\n') {
891 892
            self.line += 1u;
            self.col = 1u;
G
Gary Linscott 已提交
893 894
        } else {
            self.col += 1u;
E
Elly Jones 已提交
895
        }
896 897
    }

898
    fn next_char(&mut self) -> Option<char> {
899 900 901
        self.bump();
        self.ch
    }
902 903 904
    fn ch_is(&self, c: char) -> bool {
        self.ch == Some(c)
    }
905

B
Ben Striegel 已提交
906
    fn error<T>(&self, msg: ~str) -> Result<T, Error> {
E
Erick Tryzelaar 已提交
907
        Err(Error { line: self.line, col: self.col, msg: msg })
E
Elly Jones 已提交
908 909
    }

910
    fn parse_value(&mut self) -> Result<Json, Error> {
911 912
        self.parse_whitespace();

B
Brian Anderson 已提交
913
        if self.eof() { return self.error(~"EOF while parsing value"); }
E
Elly Jones 已提交
914

915 916 917 918 919 920 921 922 923 924
        match self.ch_or_null() {
            'n' => self.parse_ident("ull", Null),
            't' => self.parse_ident("rue", Boolean(true)),
            'f' => self.parse_ident("alse", Boolean(false)),
            '0' .. '9' | '-' => self.parse_number(),
            '"' => {
                match self.parse_str() {
                    Ok(s) => Ok(String(s)),
                    Err(e) => Err(e),
                }
925
            },
926 927 928
            '[' => self.parse_list(),
            '{' => self.parse_object(),
            _ => self.error(~"invalid syntax"),
E
Elly Jones 已提交
929
        }
930 931
    }

932
    fn parse_whitespace(&mut self) {
933 934 935 936
        while self.ch_is(' ') ||
              self.ch_is('\n') ||
              self.ch_is('\t') ||
              self.ch_is('\r') { self.bump(); }
937 938
    }

939
    fn parse_ident(&mut self, ident: &str, value: Json) -> Result<Json, Error> {
940
        if ident.chars().all(|c| Some(c) == self.next_char()) {
941
            self.bump();
L
Luqman Aden 已提交
942
            Ok(value)
943
        } else {
944
            self.error(~"invalid syntax")
E
Elly Jones 已提交
945 946 947
        }
    }

948
    fn parse_number(&mut self) -> Result<Json, Error> {
D
Daniel Micay 已提交
949
        let mut neg = 1.0;
950

951
        if self.ch_is('-') {
952
            self.bump();
D
Daniel Micay 已提交
953
            neg = -1.0;
E
Elly Jones 已提交
954
        }
955

956
        let mut res = match self.parse_integer() {
957 958
          Ok(res) => res,
          Err(e) => return Err(e)
959 960
        };

961
        if self.ch_is('.') {
962
            match self.parse_decimal(res) {
963 964
              Ok(r) => res = r,
              Err(e) => return Err(e)
965
            }
E
Elly Jones 已提交
966
        }
967

968
        if self.ch_is('e') || self.ch_is('E') {
969
            match self.parse_exponent(res) {
970 971
              Ok(r) => res = r,
              Err(e) => return Err(e)
972
            }
E
Elly Jones 已提交
973
        }
974

975
        Ok(Number(neg * res))
E
Elly Jones 已提交
976 977
    }

D
Daniel Micay 已提交
978 979
    fn parse_integer(&mut self) -> Result<f64, Error> {
        let mut res = 0.0;
E
Elly Jones 已提交
980

981 982 983
        match self.ch_or_null() {
            '0' => {
                self.bump();
984

985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001
                // There can be only one leading '0'.
                match self.ch_or_null() {
                    '0' .. '9' => return self.error(~"invalid number"),
                    _ => ()
                }
            },
            '1' .. '9' => {
                while !self.eof() {
                    match self.ch_or_null() {
                        c @ '0' .. '9' => {
                            res *= 10.0;
                            res += ((c as int) - ('0' as int)) as f64;

                            self.bump();
                        }
                        _ => break,
                    }
1002 1003
                }
            }
1004
            _ => return self.error(~"invalid number"),
E
Elly Jones 已提交
1005
        }
1006
        Ok(res)
E
Elly Jones 已提交
1007 1008
    }

D
Daniel Micay 已提交
1009
    fn parse_decimal(&mut self, res: f64) -> Result<f64, Error> {
1010 1011 1012
        self.bump();

        // Make sure a digit follows the decimal place.
1013 1014 1015
        match self.ch_or_null() {
            '0' .. '9' => (),
             _ => return self.error(~"invalid number")
1016 1017
        }

1018
        let mut res = res;
D
Daniel Micay 已提交
1019
        let mut dec = 1.0;
1020
        while !self.eof() {
1021 1022 1023 1024
            match self.ch_or_null() {
                c @ '0' .. '9' => {
                    dec /= 10.0;
                    res += (((c as int) - ('0' as int)) as f64) * dec;
1025

1026 1027 1028
                    self.bump();
                }
                _ => break,
E
Elly Jones 已提交
1029 1030
            }
        }
1031

1032
        Ok(res)
E
Elly Jones 已提交
1033 1034
    }

D
Daniel Micay 已提交
1035
    fn parse_exponent(&mut self, mut res: f64) -> Result<f64, Error> {
1036 1037
        self.bump();

1038 1039
        let mut exp = 0u;
        let mut neg_exp = false;
1040

1041 1042 1043 1044 1045
        if self.ch_is('+') {
            self.bump();
        } else if self.ch_is('-') {
            self.bump();
            neg_exp = true;
1046 1047 1048
        }

        // Make sure a digit follows the exponent place.
1049 1050 1051
        match self.ch_or_null() {
            '0' .. '9' => (),
            _ => return self.error(~"invalid number")
1052 1053
        }
        while !self.eof() {
1054 1055 1056 1057
            match self.ch_or_null() {
                c @ '0' .. '9' => {
                    exp *= 10;
                    exp += (c as uint) - ('0' as uint);
1058

1059 1060 1061
                    self.bump();
                }
                _ => break
1062 1063 1064
            }
        }

1065
        let exp: f64 = num::pow(10u as f64, exp);
1066 1067 1068 1069 1070 1071
        if neg_exp {
            res /= exp;
        } else {
            res *= exp;
        }

1072
        Ok(res)
E
Elly Jones 已提交
1073 1074
    }

1075
    fn parse_str(&mut self) -> Result<~str, Error> {
1076
        let mut escape = false;
1077
        let mut res = ~"";
1078

G
Gary Linscott 已提交
1079
        loop {
1080
            self.bump();
G
Gary Linscott 已提交
1081 1082 1083
            if self.eof() {
                return self.error(~"EOF while parsing string");
            }
1084

H
Huon Wilson 已提交
1085
            if escape {
1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124
                match self.ch_or_null() {
                    '"' => res.push_char('"'),
                    '\\' => res.push_char('\\'),
                    '/' => res.push_char('/'),
                    'b' => res.push_char('\x08'),
                    'f' => res.push_char('\x0c'),
                    'n' => res.push_char('\n'),
                    'r' => res.push_char('\r'),
                    't' => res.push_char('\t'),
                    'u' => {
                        // Parse \u1234.
                        let mut i = 0u;
                        let mut n = 0u;
                        while i < 4u && !self.eof() {
                            self.bump();
                            n = match self.ch_or_null() {
                                c @ '0' .. '9' => n * 16u + (c as uint) - ('0' as uint),
                                'a' | 'A' => n * 16u + 10u,
                                'b' | 'B' => n * 16u + 11u,
                                'c' | 'C' => n * 16u + 12u,
                                'd' | 'D' => n * 16u + 13u,
                                'e' | 'E' => n * 16u + 14u,
                                'f' | 'F' => n * 16u + 15u,
                                _ => return self.error(
                                    ~"invalid \\u escape (unrecognized hex)")
                            };

                            i += 1u;
                        }

                        // Error out if we didn't parse 4 digits.
                        if i != 4u {
                            return self.error(
                                ~"invalid \\u escape (not four digits)");
                        }

                        res.push_char(char::from_u32(n as u32).unwrap());
                    }
                    _ => return self.error(~"invalid escape"),
1125 1126
                }
                escape = false;
1127
            } else if self.ch_is('\\') {
1128 1129
                escape = true;
            } else {
1130 1131 1132 1133
                match self.ch {
                    Some('"') => { self.bump(); return Ok(res); },
                    Some(c) => res.push_char(c),
                    None => unreachable!()
1134
                }
E
Elly Jones 已提交
1135 1136 1137 1138
            }
        }
    }

1139
    fn parse_list(&mut self) -> Result<Json, Error> {
1140 1141 1142
        self.bump();
        self.parse_whitespace();

1143
        let mut values = ~[];
1144

1145
        if self.ch_is(']') {
1146
            self.bump();
L
Luqman Aden 已提交
1147
            return Ok(List(values));
1148 1149
        }

1150
        loop {
L
Luqman Aden 已提交
1151 1152 1153
            match self.parse_value() {
              Ok(v) => values.push(v),
              Err(e) => return Err(e)
1154 1155 1156
            }

            self.parse_whitespace();
1157
            if self.eof() {
B
Brian Anderson 已提交
1158
                return self.error(~"EOF while parsing list");
1159
            }
1160

1161 1162 1163 1164 1165 1166 1167
            if self.ch_is(',') {
                self.bump();
            } else if self.ch_is(']') {
                self.bump();
                return Ok(List(values));
            } else {
                return self.error(~"expected `,` or `]`")
1168
            }
1169
        };
E
Elly Jones 已提交
1170 1171
    }

1172
    fn parse_object(&mut self) -> Result<Json, Error> {
1173 1174 1175
        self.bump();
        self.parse_whitespace();

1176
        let mut values = ~TreeMap::new();
1177

1178
        if self.ch_is('}') {
1179
          self.bump();
L
Luqman Aden 已提交
1180
          return Ok(Object(values));
1181 1182 1183 1184 1185
        }

        while !self.eof() {
            self.parse_whitespace();

1186
            if !self.ch_is('"') {
B
Brian Anderson 已提交
1187
                return self.error(~"key must be a string");
1188 1189
            }

L
Luqman Aden 已提交
1190 1191 1192
            let key = match self.parse_str() {
              Ok(key) => key,
              Err(e) => return Err(e)
1193 1194 1195 1196
            };

            self.parse_whitespace();

1197
            if !self.ch_is(':') {
1198
                if self.eof() { break; }
B
Brian Anderson 已提交
1199
                return self.error(~"expected `:`");
1200 1201 1202
            }
            self.bump();

L
Luqman Aden 已提交
1203 1204 1205
            match self.parse_value() {
              Ok(value) => { values.insert(key, value); }
              Err(e) => return Err(e)
1206 1207 1208
            }
            self.parse_whitespace();

1209 1210 1211 1212 1213 1214 1215
            match self.ch_or_null() {
                ',' => self.bump(),
                '}' => { self.bump(); return Ok(Object(values)); },
                _ => {
                    if self.eof() { break; }
                    return self.error(~"expected `,` or `}`");
                }
1216 1217 1218
            }
        }

B
Brian Anderson 已提交
1219
        return self.error(~"EOF while parsing object");
L
Lenny222 已提交
1220 1221 1222
    }
}

A
Alex Crichton 已提交
1223
/// Decodes a json value from an `&mut io::Reader`
1224
pub fn from_reader(rdr: &mut io::Reader) -> Result<Json, Error> {
A
Alex Crichton 已提交
1225 1226 1227 1228 1229 1230 1231 1232
    let contents = match rdr.read_to_end() {
        Ok(c) => c,
        Err(e) => return Err(io_error_to_error(e))
    };
    let s = match str::from_utf8_owned(contents) {
        Some(s) => s,
        None => return Err(Error { line: 0, col: 0, msg: ~"contents not utf-8" })
    };
1233
    let mut parser = Parser::new(s.chars());
1234
    parser.parse()
E
Elly Jones 已提交
1235 1236
}

1237
/// Decodes a json value from a string
1238
pub fn from_str(s: &str) -> Result<Json, Error> {
1239
    let mut parser = Parser::new(s.chars());
G
Gary Linscott 已提交
1240
    parser.parse()
1241 1242
}

1243
/// A structure to decode JSON to values in rust.
1244 1245 1246 1247
pub struct Decoder {
    priv stack: ~[Json],
}

1248 1249
impl Decoder {
    /// Creates a new decoder instance for decoding the specified JSON value.
1250
    pub fn new(json: Json) -> Decoder {
1251 1252 1253
        Decoder {
            stack: ~[json]
        }
1254
    }
1255 1256
}

1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267
impl Decoder {
    fn err(&self, msg: &str) -> ! {
        fail!("JSON decode error: {}", msg);
    }
    fn missing_field(&self, field: &str, object: ~Object) -> ! {
        self.err(format!("missing required '{}' field in object: {}",
                         field, Object(object).to_str()))
    }
    fn expected(&self, expected: &str, found: &Json) -> ! {
        let found_s = match *found {
            Null => "null",
A
Alex Crichton 已提交
1268 1269 1270 1271 1272
            List(..) => "list",
            Object(..) => "object",
            Number(..) => "number",
            String(..) => "string",
            Boolean(..) => "boolean"
1273 1274 1275 1276 1277 1278
        };
        self.err(format!("expected {expct} but found {fnd}: {val}",
                         expct=expected, fnd=found_s, val=found.to_str()))
    }
}

A
Alex Crichton 已提交
1279
impl ::Decoder for Decoder {
1280
    fn read_nil(&mut self) -> () {
1281
        debug!("read_nil");
1282
        match self.stack.pop().unwrap() {
1283
            Null => (),
1284
            value => self.expected("null", &value)
1285 1286 1287
        }
    }

D
Daniel Micay 已提交
1288 1289 1290 1291 1292
    fn read_u64(&mut self)  -> u64  { self.read_f64() as u64 }
    fn read_u32(&mut self)  -> u32  { self.read_f64() as u32 }
    fn read_u16(&mut self)  -> u16  { self.read_f64() as u16 }
    fn read_u8 (&mut self)  -> u8   { self.read_f64() as u8 }
    fn read_uint(&mut self) -> uint { self.read_f64() as uint }
1293

D
Daniel Micay 已提交
1294 1295 1296 1297 1298
    fn read_i64(&mut self) -> i64 { self.read_f64() as i64 }
    fn read_i32(&mut self) -> i32 { self.read_f64() as i32 }
    fn read_i16(&mut self) -> i16 { self.read_f64() as i16 }
    fn read_i8 (&mut self) -> i8  { self.read_f64() as i8 }
    fn read_int(&mut self) -> int { self.read_f64() as int }
1299 1300

    fn read_bool(&mut self) -> bool {
1301
        debug!("read_bool");
1302
        match self.stack.pop().unwrap() {
1303
            Boolean(b) => b,
1304
            value => self.expected("boolean", &value)
1305 1306 1307
        }
    }

D
Daniel Micay 已提交
1308
    fn read_f64(&mut self) -> f64 {
1309
        debug!("read_f64");
1310
        match self.stack.pop().unwrap() {
1311
            Number(f) => f,
1312
            value => self.expected("number", &value)
1313 1314
        }
    }
1315

D
Daniel Micay 已提交
1316
    fn read_f32(&mut self) -> f32 { self.read_f64() as f32 }
1317 1318

    fn read_char(&mut self) -> char {
1319
        let s = self.read_str();
1320
        {
1321
            let mut it = s.chars();
1322 1323 1324 1325 1326 1327 1328
            match (it.next(), it.next()) {
                // exactly one character
                (Some(c), None) => return c,
                _ => ()
            }
        }
        self.expected("single character string", &String(s))
1329 1330 1331
    }

    fn read_str(&mut self) -> ~str {
1332
        debug!("read_str");
1333
        match self.stack.pop().unwrap() {
1334
            String(s) => s,
1335
            value => self.expected("string", &value)
1336 1337 1338
        }
    }

1339
    fn read_enum<T>(&mut self, name: &str, f: |&mut Decoder| -> T) -> T {
1340
        debug!("read_enum({})", name);
1341 1342 1343 1344 1345
        f(self)
    }

    fn read_enum_variant<T>(&mut self,
                            names: &[&str],
1346
                            f: |&mut Decoder, uint| -> T)
1347
                            -> T {
1348
        debug!("read_enum_variant(names={:?})", names);
1349
        let name = match self.stack.pop().unwrap() {
1350
            String(s) => s,
1351 1352 1353 1354 1355
            Object(mut o) => {
                let n = match o.pop(&~"variant") {
                    Some(String(s)) => s,
                    Some(val) => self.expected("string", &val),
                    None => self.missing_field("variant", o)
1356
                };
1357 1358 1359
                match o.pop(&~"fields") {
                    Some(List(l)) => {
                        for field in l.move_rev_iter() {
1360 1361 1362
                            self.stack.push(field.clone());
                        }
                    },
1363 1364 1365 1366 1367 1368 1369 1370
                    Some(val) => self.expected("list", &val),
                    None => {
                        // re-insert the variant field so we're
                        // printing the "whole" struct in the error
                        // message... ick.
                        o.insert(~"variant", String(n));
                        self.missing_field("fields", o);
                    }
1371
                }
1372
                n
1373
            }
1374
            json => self.expected("string or object", &json)
1375
        };
1376
        let idx = match names.iter().position(|n| str::eq_slice(*n, name)) {
1377
            Some(idx) => idx,
1378
            None => self.err(format!("unknown variant name: {}", name))
1379 1380 1381 1382
        };
        f(self, idx)
    }

1383
    fn read_enum_variant_arg<T>(&mut self, idx: uint, f: |&mut Decoder| -> T)
1384
                                -> T {
1385
        debug!("read_enum_variant_arg(idx={})", idx);
1386 1387 1388 1389 1390
        f(self)
    }

    fn read_enum_struct_variant<T>(&mut self,
                                   names: &[&str],
1391
                                   f: |&mut Decoder, uint| -> T)
1392
                                   -> T {
1393
        debug!("read_enum_struct_variant(names={:?})", names);
1394 1395 1396 1397 1398 1399 1400
        self.read_enum_variant(names, f)
    }


    fn read_enum_struct_variant_field<T>(&mut self,
                                         name: &str,
                                         idx: uint,
1401
                                         f: |&mut Decoder| -> T)
1402
                                         -> T {
1403
        debug!("read_enum_struct_variant_field(name={}, idx={})", name, idx);
1404 1405 1406 1407 1408 1409
        self.read_enum_variant_arg(idx, f)
    }

    fn read_struct<T>(&mut self,
                      name: &str,
                      len: uint,
1410
                      f: |&mut Decoder| -> T)
1411
                      -> T {
1412
        debug!("read_struct(name={}, len={})", name, len);
1413
        let value = f(self);
1414
        self.stack.pop().unwrap();
1415 1416 1417 1418 1419 1420
        value
    }

    fn read_struct_field<T>(&mut self,
                            name: &str,
                            idx: uint,
1421
                            f: |&mut Decoder| -> T)
1422
                            -> T {
1423
        debug!("read_struct_field(name={}, idx={})", name, idx);
1424
        match self.stack.pop().unwrap() {
1425
            Object(mut obj) => {
1426
                let value = match obj.pop(&name.to_owned()) {
1427
                    None => self.missing_field(name, obj),
1428 1429 1430 1431 1432 1433 1434 1435
                    Some(json) => {
                        self.stack.push(json);
                        f(self)
                    }
                };
                self.stack.push(Object(obj));
                value
            }
1436
            value => self.expected("object", &value)
1437 1438 1439
        }
    }

1440
    fn read_tuple<T>(&mut self, f: |&mut Decoder, uint| -> T) -> T {
1441
        debug!("read_tuple()");
1442 1443 1444
        self.read_seq(f)
    }

1445
    fn read_tuple_arg<T>(&mut self, idx: uint, f: |&mut Decoder| -> T) -> T {
1446
        debug!("read_tuple_arg(idx={})", idx);
1447 1448 1449 1450 1451
        self.read_seq_elt(idx, f)
    }

    fn read_tuple_struct<T>(&mut self,
                            name: &str,
1452
                            f: |&mut Decoder, uint| -> T)
1453
                            -> T {
1454
        debug!("read_tuple_struct(name={})", name);
1455 1456 1457 1458 1459
        self.read_tuple(f)
    }

    fn read_tuple_struct_arg<T>(&mut self,
                                idx: uint,
1460
                                f: |&mut Decoder| -> T)
1461
                                -> T {
1462
        debug!("read_tuple_struct_arg(idx={})", idx);
1463 1464 1465
        self.read_tuple_arg(idx, f)
    }

1466
    fn read_option<T>(&mut self, f: |&mut Decoder, bool| -> T) -> T {
1467
        match self.stack.pop().unwrap() {
1468 1469 1470 1471 1472
            Null => f(self, false),
            value => { self.stack.push(value); f(self, true) }
        }
    }

1473
    fn read_seq<T>(&mut self, f: |&mut Decoder, uint| -> T) -> T {
1474
        debug!("read_seq()");
1475
        let len = match self.stack.pop().unwrap() {
1476 1477
            List(list) => {
                let len = list.len();
1478
                for v in list.move_rev_iter() {
1479 1480 1481 1482
                    self.stack.push(v);
                }
                len
            }
1483
            value => self.expected("list", &value)
1484 1485 1486 1487
        };
        f(self, len)
    }

1488
    fn read_seq_elt<T>(&mut self, idx: uint, f: |&mut Decoder| -> T) -> T {
1489
        debug!("read_seq_elt(idx={})", idx);
1490 1491 1492
        f(self)
    }

1493
    fn read_map<T>(&mut self, f: |&mut Decoder, uint| -> T) -> T {
1494
        debug!("read_map()");
1495
        let len = match self.stack.pop().unwrap() {
1496 1497
            Object(obj) => {
                let len = obj.len();
1498
                for (key, value) in obj.move_iter() {
1499 1500 1501 1502 1503
                    self.stack.push(value);
                    self.stack.push(String(key));
                }
                len
            }
1504
            value => self.expected("object", &value)
1505 1506 1507 1508
        };
        f(self, len)
    }

1509
    fn read_map_elt_key<T>(&mut self, idx: uint, f: |&mut Decoder| -> T)
1510
                           -> T {
1511
        debug!("read_map_elt_key(idx={})", idx);
1512 1513 1514
        f(self)
    }

1515
    fn read_map_elt_val<T>(&mut self, idx: uint, f: |&mut Decoder| -> T)
1516
                           -> T {
1517
        debug!("read_map_elt_val(idx={})", idx);
1518 1519 1520 1521
        f(self)
    }
}

1522
/// Test if two json values are less than one another
1523
impl Ord for Json {
1524
    fn lt(&self, other: &Json) -> bool {
H
Huon Wilson 已提交
1525
        match *self {
1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549
            Number(f0) => {
                match *other {
                    Number(f1) => f0 < f1,
                    String(_) | Boolean(_) | List(_) | Object(_) |
                    Null => true
                }
            }

            String(ref s0) => {
                match *other {
                    Number(_) => false,
                    String(ref s1) => s0 < s1,
                    Boolean(_) | List(_) | Object(_) | Null => true
                }
            }

            Boolean(b0) => {
                match *other {
                    Number(_) | String(_) => false,
                    Boolean(b1) => b0 < b1,
                    List(_) | Object(_) | Null => true
                }
            }

1550
            List(ref l0) => {
1551 1552
                match *other {
                    Number(_) | String(_) | Boolean(_) => false,
1553
                    List(ref l1) => (*l0) < (*l1),
1554 1555 1556 1557 1558 1559 1560
                    Object(_) | Null => true
                }
            }

            Object(ref d0) => {
                match *other {
                    Number(_) | String(_) | Boolean(_) | List(_) => false,
1561
                    Object(ref d1) => d0 < d1,
1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575
                    Null => true
                }
            }

            Null => {
                match *other {
                    Number(_) | String(_) | Boolean(_) | List(_) |
                    Object(_) =>
                        false,
                    Null => true
                }
            }
        }
    }
1576
}
1577

1578
/// A trait for converting values to JSON
1579
pub trait ToJson {
1580 1581 1582
    /// Converts the value of `self` to an instance of JSON
    fn to_json(&self) -> Json;
}
1583

1584
impl ToJson for Json {
1585
    fn to_json(&self) -> Json { (*self).clone() }
1586 1587
}

1588
impl ToJson for int {
D
Daniel Micay 已提交
1589
    fn to_json(&self) -> Json { Number(*self as f64) }
1590 1591
}

1592
impl ToJson for i8 {
D
Daniel Micay 已提交
1593
    fn to_json(&self) -> Json { Number(*self as f64) }
1594 1595
}

1596
impl ToJson for i16 {
D
Daniel Micay 已提交
1597
    fn to_json(&self) -> Json { Number(*self as f64) }
1598 1599
}

1600
impl ToJson for i32 {
D
Daniel Micay 已提交
1601
    fn to_json(&self) -> Json { Number(*self as f64) }
1602 1603
}

1604
impl ToJson for i64 {
D
Daniel Micay 已提交
1605
    fn to_json(&self) -> Json { Number(*self as f64) }
1606 1607
}

1608
impl ToJson for uint {
D
Daniel Micay 已提交
1609
    fn to_json(&self) -> Json { Number(*self as f64) }
1610 1611
}

1612
impl ToJson for u8 {
D
Daniel Micay 已提交
1613
    fn to_json(&self) -> Json { Number(*self as f64) }
1614 1615
}

1616
impl ToJson for u16 {
D
Daniel Micay 已提交
1617
    fn to_json(&self) -> Json { Number(*self as f64) }
1618 1619
}

1620
impl ToJson for u32 {
D
Daniel Micay 已提交
1621
    fn to_json(&self) -> Json { Number(*self as f64) }
1622 1623
}

1624
impl ToJson for u64 {
D
Daniel Micay 已提交
1625
    fn to_json(&self) -> Json { Number(*self as f64) }
1626 1627
}

1628
impl ToJson for f32 {
D
Daniel Micay 已提交
1629
    fn to_json(&self) -> Json { Number(*self as f64) }
1630 1631
}

1632
impl ToJson for f64 {
D
Daniel Micay 已提交
1633
    fn to_json(&self) -> Json { Number(*self) }
1634 1635
}

1636
impl ToJson for () {
B
Ben Striegel 已提交
1637
    fn to_json(&self) -> Json { Null }
1638 1639
}

1640
impl ToJson for bool {
B
Ben Striegel 已提交
1641
    fn to_json(&self) -> Json { Boolean(*self) }
1642 1643
}

1644
impl ToJson for ~str {
1645
    fn to_json(&self) -> Json { String((*self).clone()) }
1646 1647
}

1648
impl<A:ToJson,B:ToJson> ToJson for (A, B) {
B
Ben Striegel 已提交
1649 1650
    fn to_json(&self) -> Json {
        match *self {
1651
          (ref a, ref b) => {
1652
            List(~[a.to_json(), b.to_json()])
1653 1654
          }
        }
1655 1656 1657
    }
}

1658
impl<A:ToJson,B:ToJson,C:ToJson> ToJson for (A, B, C) {
B
Ben Striegel 已提交
1659 1660
    fn to_json(&self) -> Json {
        match *self {
1661
          (ref a, ref b, ref c) => {
1662
            List(~[a.to_json(), b.to_json(), c.to_json()])
1663 1664
          }
        }
1665 1666 1667
    }
}

1668
impl<A:ToJson> ToJson for ~[A] {
B
Ben Striegel 已提交
1669
    fn to_json(&self) -> Json { List(self.map(|elt| elt.to_json())) }
1670 1671
}

1672
impl<A:ToJson> ToJson for TreeMap<~str, A> {
B
Ben Striegel 已提交
1673
    fn to_json(&self) -> Json {
1674
        let mut d = TreeMap::new();
D
Daniel Micay 已提交
1675
        for (key, value) in self.iter() {
1676
            d.insert((*key).clone(), value.to_json());
1677 1678 1679 1680 1681
        }
        Object(~d)
    }
}

1682
impl<A:ToJson> ToJson for HashMap<~str, A> {
G
Graydon Hoare 已提交
1683
    fn to_json(&self) -> Json {
1684
        let mut d = TreeMap::new();
D
Daniel Micay 已提交
1685
        for (key, value) in self.iter() {
1686
            d.insert((*key).clone(), value.to_json());
G
Graydon Hoare 已提交
1687 1688 1689 1690 1691
        }
        Object(~d)
    }
}

1692
impl<A:ToJson> ToJson for Option<A> {
B
Ben Striegel 已提交
1693 1694
    fn to_json(&self) -> Json {
        match *self {
B
Ben Striegel 已提交
1695
          None => Null,
1696
          Some(ref value) => value.to_json()
1697 1698 1699 1700
        }
    }
}

1701
impl fmt::Show for Json {
1702
    /// Encodes a json value into a string
1703 1704
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        self.to_writer(f.buf)
1705
    }
1706 1707
}

1708 1709 1710
impl fmt::Show for Error {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(f.buf, "{}:{}: {}", self.line, self.col, self.msg)
1711 1712 1713
    }
}

1714 1715
#[cfg(test)]
mod tests {
A
Alex Crichton 已提交
1716 1717 1718
    use {Encodable, Decodable};
    use super::{Encoder, Decoder, Error, Boolean, Number, List, String, Null,
                PrettyEncoder, Object, Json, from_str};
A
Alex Crichton 已提交
1719
    use std::io;
1720
    use collections::TreeMap;
1721

1722
    #[deriving(Eq, Encodable, Decodable, Show)]
1723 1724 1725 1726 1727
    enum Animal {
        Dog,
        Frog(~str, int)
    }

1728
    #[deriving(Eq, Encodable, Decodable, Show)]
1729 1730 1731 1732 1733 1734
    struct Inner {
        a: (),
        b: uint,
        c: ~[~str],
    }

1735
    #[deriving(Eq, Encodable, Decodable, Show)]
1736 1737 1738 1739
    struct Outer {
        inner: ~[Inner],
    }

1740
    fn mk_object(items: &[(~str, Json)]) -> Json {
1741
        let mut d = ~TreeMap::new();
1742

D
Daniel Micay 已提交
1743
        for item in items.iter() {
1744
            match *item {
1745
                (ref key, ref value) => { d.insert((*key).clone(), (*value).clone()); },
1746
            }
1747 1748
        };

L
Luqman Aden 已提交
1749
        Object(d)
1750 1751 1752 1753
    }

    #[test]
    fn test_write_null() {
1754 1755
        assert_eq!(Null.to_str(), ~"null");
        assert_eq!(Null.to_pretty_str(), ~"null");
1756 1757
    }

1758

1759
    #[test]
1760
    fn test_write_number() {
D
Daniel Micay 已提交
1761 1762
        assert_eq!(Number(3.0).to_str(), ~"3");
        assert_eq!(Number(3.0).to_pretty_str(), ~"3");
1763

D
Daniel Micay 已提交
1764 1765
        assert_eq!(Number(3.1).to_str(), ~"3.1");
        assert_eq!(Number(3.1).to_pretty_str(), ~"3.1");
1766

D
Daniel Micay 已提交
1767 1768
        assert_eq!(Number(-1.5).to_str(), ~"-1.5");
        assert_eq!(Number(-1.5).to_pretty_str(), ~"-1.5");
1769

D
Daniel Micay 已提交
1770 1771
        assert_eq!(Number(0.5).to_str(), ~"0.5");
        assert_eq!(Number(0.5).to_pretty_str(), ~"0.5");
1772 1773 1774 1775
    }

    #[test]
    fn test_write_str() {
1776 1777
        assert_eq!(String(~"").to_str(), ~"\"\"");
        assert_eq!(String(~"").to_pretty_str(), ~"\"\"");
1778

1779 1780
        assert_eq!(String(~"foo").to_str(), ~"\"foo\"");
        assert_eq!(String(~"foo").to_pretty_str(), ~"\"foo\"");
1781 1782 1783 1784
    }

    #[test]
    fn test_write_bool() {
1785 1786
        assert_eq!(Boolean(true).to_str(), ~"true");
        assert_eq!(Boolean(true).to_pretty_str(), ~"true");
1787

1788 1789
        assert_eq!(Boolean(false).to_str(), ~"false");
        assert_eq!(Boolean(false).to_pretty_str(), ~"false");
1790 1791 1792 1793
    }

    #[test]
    fn test_write_list() {
1794 1795
        assert_eq!(List(~[]).to_str(), ~"[]");
        assert_eq!(List(~[]).to_pretty_str(), ~"[]");
1796

1797
        assert_eq!(List(~[Boolean(true)]).to_str(), ~"[true]");
1798
        assert_eq!(
1799
            List(~[Boolean(true)]).to_pretty_str(),
1800 1801 1802 1803 1804
            ~"\
            [\n  \
                true\n\
            ]"
        );
1805

1806
        let long_test_list = List(~[
1807 1808
            Boolean(false),
            Null,
D
Daniel Micay 已提交
1809
            List(~[String(~"foo\nbar"), Number(3.5)])]);
1810

1811
        assert_eq!(long_test_list.to_str(),
1812
            ~"[false,null,[\"foo\\nbar\",3.5]]");
1813
        assert_eq!(
1814
            long_test_list.to_pretty_str(),
1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826
            ~"\
            [\n  \
                false,\n  \
                null,\n  \
                [\n    \
                    \"foo\\nbar\",\n    \
                    3.5\n  \
                ]\n\
            ]"
        );
    }

1827
    #[test]
1828
    fn test_write_object() {
1829 1830
        assert_eq!(mk_object([]).to_str(), ~"{}");
        assert_eq!(mk_object([]).to_pretty_str(), ~"{}");
1831

1832
        assert_eq!(
1833
            mk_object([(~"a", Boolean(true))]).to_str(),
1834 1835
            ~"{\"a\":true}"
        );
1836
        assert_eq!(
1837
            mk_object([(~"a", Boolean(true))]).to_pretty_str(),
1838 1839 1840 1841 1842 1843
            ~"\
            {\n  \
                \"a\": true\n\
            }"
        );

1844
        let complex_obj = mk_object([
1845
                (~"b", List(~[
E
Erick Tryzelaar 已提交
1846 1847
                    mk_object([(~"c", String(~"\x0c\r"))]),
                    mk_object([(~"d", String(~""))])
1848
                ]))
1849 1850 1851
            ]);

        assert_eq!(
1852
            complex_obj.to_str(),
1853 1854 1855 1856 1857 1858 1859 1860
            ~"{\
                \"b\":[\
                    {\"c\":\"\\f\\r\"},\
                    {\"d\":\"\"}\
                ]\
            }"
        );
        assert_eq!(
1861
            complex_obj.to_pretty_str(),
1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873
            ~"\
            {\n  \
                \"b\": [\n    \
                    {\n      \
                        \"c\": \"\\f\\r\"\n    \
                    },\n    \
                    {\n      \
                        \"d\": \"\"\n    \
                    }\n  \
                ]\n\
            }"
        );
1874

E
Erick Tryzelaar 已提交
1875
        let a = mk_object([
B
Ben Striegel 已提交
1876
            (~"a", Boolean(true)),
1877
            (~"b", List(~[
E
Erick Tryzelaar 已提交
1878 1879
                mk_object([(~"c", String(~"\x0c\r"))]),
                mk_object([(~"d", String(~""))])
1880
            ]))
G
Graydon Hoare 已提交
1881
        ]);
1882

1883 1884
        // We can't compare the strings directly because the object fields be
        // printed in a different order.
1885 1886
        assert_eq!(a.clone(), from_str(a.to_str()).unwrap());
        assert_eq!(a.clone(), from_str(a.to_pretty_str()).unwrap());
1887 1888
    }

1889
    fn with_str_writer(f: |&mut io::Writer|) -> ~str {
A
Alex Crichton 已提交
1890
        use std::io::MemWriter;
A
Alex Crichton 已提交
1891 1892
        use std::str;

1893 1894
        let mut m = MemWriter::new();
        f(&mut m as &mut io::Writer);
1895
        str::from_utf8_owned(m.unwrap()).unwrap()
A
Alex Crichton 已提交
1896 1897
    }

1898
    #[test]
1899
    fn test_write_enum() {
1900
        let animal = Dog;
1901
        assert_eq!(
1902
            with_str_writer(|wr| {
1903
                let mut encoder = Encoder::new(wr);
1904
                animal.encode(&mut encoder);
1905
            }),
1906 1907 1908
            ~"\"Dog\""
        );
        assert_eq!(
1909
            with_str_writer(|wr| {
1910
                let mut encoder = PrettyEncoder::new(wr);
1911
                animal.encode(&mut encoder);
1912
            }),
1913 1914
            ~"\"Dog\""
        );
1915 1916 1917

        let animal = Frog(~"Henry", 349);
        assert_eq!(
1918
            with_str_writer(|wr| {
1919
                let mut encoder = Encoder::new(wr);
1920
                animal.encode(&mut encoder);
1921
            }),
1922
            ~"{\"variant\":\"Frog\",\"fields\":[\"Henry\",349]}"
1923 1924
        );
        assert_eq!(
1925
            with_str_writer(|wr| {
1926
                let mut encoder = PrettyEncoder::new(wr);
1927
                animal.encode(&mut encoder);
1928
            }),
1929 1930 1931
            ~"\
            [\n  \
                \"Frog\",\n  \
1932 1933
                \"Henry\",\n  \
                349\n\
1934 1935
            ]"
        );
1936 1937 1938
    }

    #[test]
1939 1940
    fn test_write_some() {
        let value = Some(~"jodhpurs");
1941
        let s = with_str_writer(|wr| {
1942
            let mut encoder = Encoder::new(wr);
1943
            value.encode(&mut encoder);
1944
        });
1945
        assert_eq!(s, ~"\"jodhpurs\"");
1946

1947
        let value = Some(~"jodhpurs");
1948
        let s = with_str_writer(|wr| {
1949
            let mut encoder = PrettyEncoder::new(wr);
1950
            value.encode(&mut encoder);
1951
        });
1952 1953 1954
        assert_eq!(s, ~"\"jodhpurs\"");
    }

1955
    #[test]
1956 1957
    fn test_write_none() {
        let value: Option<~str> = None;
1958
        let s = with_str_writer(|wr| {
1959
            let mut encoder = Encoder::new(wr);
1960
            value.encode(&mut encoder);
1961
        });
1962
        assert_eq!(s, ~"null");
J
John Clements 已提交
1963

1964
        let s = with_str_writer(|wr| {
1965
            let mut encoder = Encoder::new(wr);
1966
            value.encode(&mut encoder);
1967
        });
1968 1969 1970
        assert_eq!(s, ~"null");
    }

1971
    #[test]
1972
    fn test_trailing_characters() {
E
Erick Tryzelaar 已提交
1973
        assert_eq!(from_str("nulla"),
E
Erick Tryzelaar 已提交
1974
            Err(Error {line: 1u, col: 5u, msg: ~"trailing characters"}));
E
Erick Tryzelaar 已提交
1975
        assert_eq!(from_str("truea"),
E
Erick Tryzelaar 已提交
1976
            Err(Error {line: 1u, col: 5u, msg: ~"trailing characters"}));
E
Erick Tryzelaar 已提交
1977
        assert_eq!(from_str("falsea"),
E
Erick Tryzelaar 已提交
1978
            Err(Error {line: 1u, col: 6u, msg: ~"trailing characters"}));
E
Erick Tryzelaar 已提交
1979
        assert_eq!(from_str("1a"),
E
Erick Tryzelaar 已提交
1980
            Err(Error {line: 1u, col: 2u, msg: ~"trailing characters"}));
E
Erick Tryzelaar 已提交
1981
        assert_eq!(from_str("[]a"),
E
Erick Tryzelaar 已提交
1982
            Err(Error {line: 1u, col: 3u, msg: ~"trailing characters"}));
E
Erick Tryzelaar 已提交
1983
        assert_eq!(from_str("{}a"),
E
Erick Tryzelaar 已提交
1984
            Err(Error {line: 1u, col: 3u, msg: ~"trailing characters"}));
1985 1986 1987 1988
    }

    #[test]
    fn test_read_identifiers() {
E
Erick Tryzelaar 已提交
1989
        assert_eq!(from_str("n"),
E
Erick Tryzelaar 已提交
1990
            Err(Error {line: 1u, col: 2u, msg: ~"invalid syntax"}));
E
Erick Tryzelaar 已提交
1991
        assert_eq!(from_str("nul"),
E
Erick Tryzelaar 已提交
1992
            Err(Error {line: 1u, col: 4u, msg: ~"invalid syntax"}));
1993

E
Erick Tryzelaar 已提交
1994
        assert_eq!(from_str("t"),
E
Erick Tryzelaar 已提交
1995
            Err(Error {line: 1u, col: 2u, msg: ~"invalid syntax"}));
E
Erick Tryzelaar 已提交
1996
        assert_eq!(from_str("truz"),
E
Erick Tryzelaar 已提交
1997
            Err(Error {line: 1u, col: 4u, msg: ~"invalid syntax"}));
1998

E
Erick Tryzelaar 已提交
1999
        assert_eq!(from_str("f"),
E
Erick Tryzelaar 已提交
2000
            Err(Error {line: 1u, col: 2u, msg: ~"invalid syntax"}));
E
Erick Tryzelaar 已提交
2001
        assert_eq!(from_str("faz"),
E
Erick Tryzelaar 已提交
2002
            Err(Error {line: 1u, col: 3u, msg: ~"invalid syntax"}));
2003

E
Erick Tryzelaar 已提交
2004 2005 2006 2007 2008 2009
        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)));
2010 2011
    }

2012 2013
    #[test]
    fn test_decode_identifiers() {
2014
        let mut decoder = Decoder::new(from_str("null").unwrap());
2015
        let v: () = Decodable::decode(&mut decoder);
2016 2017
        assert_eq!(v, ());

2018
        let mut decoder = Decoder::new(from_str("true").unwrap());
2019
        let v: bool = Decodable::decode(&mut decoder);
2020 2021
        assert_eq!(v, true);

2022
        let mut decoder = Decoder::new(from_str("false").unwrap());
2023
        let v: bool = Decodable::decode(&mut decoder);
2024 2025 2026
        assert_eq!(v, false);
    }

2027
    #[test]
2028
    fn test_read_number() {
E
Erick Tryzelaar 已提交
2029
        assert_eq!(from_str("+"),
E
Erick Tryzelaar 已提交
2030
            Err(Error {line: 1u, col: 1u, msg: ~"invalid syntax"}));
E
Erick Tryzelaar 已提交
2031
        assert_eq!(from_str("."),
E
Erick Tryzelaar 已提交
2032
            Err(Error {line: 1u, col: 1u, msg: ~"invalid syntax"}));
2033

E
Erick Tryzelaar 已提交
2034
        assert_eq!(from_str("-"),
E
Erick Tryzelaar 已提交
2035
            Err(Error {line: 1u, col: 2u, msg: ~"invalid number"}));
E
Erick Tryzelaar 已提交
2036
        assert_eq!(from_str("00"),
E
Erick Tryzelaar 已提交
2037
            Err(Error {line: 1u, col: 2u, msg: ~"invalid number"}));
E
Erick Tryzelaar 已提交
2038
        assert_eq!(from_str("1."),
E
Erick Tryzelaar 已提交
2039
            Err(Error {line: 1u, col: 3u, msg: ~"invalid number"}));
E
Erick Tryzelaar 已提交
2040
        assert_eq!(from_str("1e"),
E
Erick Tryzelaar 已提交
2041
            Err(Error {line: 1u, col: 3u, msg: ~"invalid number"}));
E
Erick Tryzelaar 已提交
2042
        assert_eq!(from_str("1e+"),
E
Erick Tryzelaar 已提交
2043
            Err(Error {line: 1u, col: 4u, msg: ~"invalid number"}));
2044

D
Daniel Micay 已提交
2045 2046 2047 2048 2049 2050 2051 2052
        assert_eq!(from_str("3"), Ok(Number(3.0)));
        assert_eq!(from_str("3.1"), Ok(Number(3.1)));
        assert_eq!(from_str("-1.2"), Ok(Number(-1.2)));
        assert_eq!(from_str("0.4"), Ok(Number(0.4)));
        assert_eq!(from_str("0.4e5"), Ok(Number(0.4e5)));
        assert_eq!(from_str("0.4e+15"), Ok(Number(0.4e15)));
        assert_eq!(from_str("0.4e-01"), Ok(Number(0.4e-01)));
        assert_eq!(from_str(" 3 "), Ok(Number(3.0)));
2053 2054
    }

2055 2056
    #[test]
    fn test_decode_numbers() {
2057
        let mut decoder = Decoder::new(from_str("3").unwrap());
D
Daniel Micay 已提交
2058 2059
        let v: f64 = Decodable::decode(&mut decoder);
        assert_eq!(v, 3.0);
2060

2061
        let mut decoder = Decoder::new(from_str("3.1").unwrap());
D
Daniel Micay 已提交
2062 2063
        let v: f64 = Decodable::decode(&mut decoder);
        assert_eq!(v, 3.1);
2064

2065
        let mut decoder = Decoder::new(from_str("-1.2").unwrap());
D
Daniel Micay 已提交
2066 2067
        let v: f64 = Decodable::decode(&mut decoder);
        assert_eq!(v, -1.2);
2068

2069
        let mut decoder = Decoder::new(from_str("0.4").unwrap());
D
Daniel Micay 已提交
2070 2071
        let v: f64 = Decodable::decode(&mut decoder);
        assert_eq!(v, 0.4);
2072

2073
        let mut decoder = Decoder::new(from_str("0.4e5").unwrap());
D
Daniel Micay 已提交
2074 2075
        let v: f64 = Decodable::decode(&mut decoder);
        assert_eq!(v, 0.4e5);
2076

2077
        let mut decoder = Decoder::new(from_str("0.4e15").unwrap());
D
Daniel Micay 已提交
2078 2079
        let v: f64 = Decodable::decode(&mut decoder);
        assert_eq!(v, 0.4e15);
2080

2081
        let mut decoder = Decoder::new(from_str("0.4e-01").unwrap());
D
Daniel Micay 已提交
2082 2083
        let v: f64 = Decodable::decode(&mut decoder);
        assert_eq!(v, 0.4e-01);
2084 2085
    }

G
Gary Linscott 已提交
2086
    #[test]
2087
    fn test_read_str() {
E
Erick Tryzelaar 已提交
2088
        assert_eq!(from_str("\""),
E
Erick Tryzelaar 已提交
2089
            Err(Error {line: 1u, col: 2u, msg: ~"EOF while parsing string"
2090
        }));
E
Erick Tryzelaar 已提交
2091
        assert_eq!(from_str("\"lol"),
E
Erick Tryzelaar 已提交
2092
            Err(Error {line: 1u, col: 5u, msg: ~"EOF while parsing string"
2093 2094
        }));

E
Erick Tryzelaar 已提交
2095 2096 2097 2098 2099 2100 2101 2102 2103 2104
        assert_eq!(from_str("\"\""), Ok(String(~"")));
        assert_eq!(from_str("\"foo\""), Ok(String(~"foo")));
        assert_eq!(from_str("\"\\\"\""), Ok(String(~"\"")));
        assert_eq!(from_str("\"\\b\""), Ok(String(~"\x08")));
        assert_eq!(from_str("\"\\n\""), Ok(String(~"\n")));
        assert_eq!(from_str("\"\\r\""), Ok(String(~"\r")));
        assert_eq!(from_str("\"\\t\""), Ok(String(~"\t")));
        assert_eq!(from_str(" \"foo\" "), Ok(String(~"foo")));
        assert_eq!(from_str("\"\\u12ab\""), Ok(String(~"\u12ab")));
        assert_eq!(from_str("\"\\uAB12\""), Ok(String(~"\uAB12")));
2105 2106
    }

2107
    #[test]
2108
    fn test_decode_str() {
2109
        let mut decoder = Decoder::new(from_str("\"\"").unwrap());
2110
        let v: ~str = Decodable::decode(&mut decoder);
2111 2112
        assert_eq!(v, ~"");

2113
        let mut decoder = Decoder::new(from_str("\"foo\"").unwrap());
2114
        let v: ~str = Decodable::decode(&mut decoder);
2115 2116
        assert_eq!(v, ~"foo");

2117
        let mut decoder = Decoder::new(from_str("\"\\\"\"").unwrap());
2118
        let v: ~str = Decodable::decode(&mut decoder);
2119 2120
        assert_eq!(v, ~"\"");

2121
        let mut decoder = Decoder::new(from_str("\"\\b\"").unwrap());
2122
        let v: ~str = Decodable::decode(&mut decoder);
2123 2124
        assert_eq!(v, ~"\x08");

2125
        let mut decoder = Decoder::new(from_str("\"\\n\"").unwrap());
2126
        let v: ~str = Decodable::decode(&mut decoder);
2127 2128
        assert_eq!(v, ~"\n");

2129
        let mut decoder = Decoder::new(from_str("\"\\r\"").unwrap());
2130
        let v: ~str = Decodable::decode(&mut decoder);
2131 2132
        assert_eq!(v, ~"\r");

2133
        let mut decoder = Decoder::new(from_str("\"\\t\"").unwrap());
2134
        let v: ~str = Decodable::decode(&mut decoder);
2135 2136
        assert_eq!(v, ~"\t");

2137
        let mut decoder = Decoder::new(from_str("\"\\u12ab\"").unwrap());
2138
        let v: ~str = Decodable::decode(&mut decoder);
2139 2140
        assert_eq!(v, ~"\u12ab");

2141
        let mut decoder = Decoder::new(from_str("\"\\uAB12\"").unwrap());
2142
        let v: ~str = Decodable::decode(&mut decoder);
2143
        assert_eq!(v, ~"\uAB12");
2144 2145
    }

2146
    #[test]
2147
    fn test_read_list() {
E
Erick Tryzelaar 已提交
2148
        assert_eq!(from_str("["),
E
Erick Tryzelaar 已提交
2149
            Err(Error {line: 1u, col: 2u, msg: ~"EOF while parsing value"}));
E
Erick Tryzelaar 已提交
2150
        assert_eq!(from_str("[1"),
E
Erick Tryzelaar 已提交
2151
            Err(Error {line: 1u, col: 3u, msg: ~"EOF while parsing list"}));
E
Erick Tryzelaar 已提交
2152
        assert_eq!(from_str("[1,"),
E
Erick Tryzelaar 已提交
2153
            Err(Error {line: 1u, col: 4u, msg: ~"EOF while parsing value"}));
E
Erick Tryzelaar 已提交
2154
        assert_eq!(from_str("[1,]"),
E
Erick Tryzelaar 已提交
2155
            Err(Error {line: 1u, col: 4u, msg: ~"invalid syntax"}));
E
Erick Tryzelaar 已提交
2156
        assert_eq!(from_str("[6 7]"),
E
Erick Tryzelaar 已提交
2157
            Err(Error {line: 1u, col: 4u, msg: ~"expected `,` or `]`"}));
2158

E
Erick Tryzelaar 已提交
2159 2160 2161 2162 2163 2164
        assert_eq!(from_str("[]"), Ok(List(~[])));
        assert_eq!(from_str("[ ]"), Ok(List(~[])));
        assert_eq!(from_str("[true]"), Ok(List(~[Boolean(true)])));
        assert_eq!(from_str("[ false ]"), Ok(List(~[Boolean(false)])));
        assert_eq!(from_str("[null]"), Ok(List(~[Null])));
        assert_eq!(from_str("[3, 1]"),
D
Daniel Micay 已提交
2165
                     Ok(List(~[Number(3.0), Number(1.0)])));
E
Erick Tryzelaar 已提交
2166
        assert_eq!(from_str("\n[3, 2]\n"),
D
Daniel Micay 已提交
2167
                     Ok(List(~[Number(3.0), Number(2.0)])));
E
Erick Tryzelaar 已提交
2168
        assert_eq!(from_str("[2, [4, 1]]"),
D
Daniel Micay 已提交
2169
               Ok(List(~[Number(2.0), List(~[Number(4.0), Number(1.0)])])));
2170 2171
    }

2172 2173
    #[test]
    fn test_decode_list() {
2174
        let mut decoder = Decoder::new(from_str("[]").unwrap());
2175
        let v: ~[()] = Decodable::decode(&mut decoder);
2176 2177
        assert_eq!(v, ~[]);

2178
        let mut decoder = Decoder::new(from_str("[null]").unwrap());
2179
        let v: ~[()] = Decodable::decode(&mut decoder);
2180 2181
        assert_eq!(v, ~[()]);

2182
        let mut decoder = Decoder::new(from_str("[true]").unwrap());
2183
        let v: ~[bool] = Decodable::decode(&mut decoder);
2184 2185
        assert_eq!(v, ~[true]);

2186
        let mut decoder = Decoder::new(from_str("[true]").unwrap());
2187
        let v: ~[bool] = Decodable::decode(&mut decoder);
2188 2189
        assert_eq!(v, ~[true]);

2190
        let mut decoder = Decoder::new(from_str("[3, 1]").unwrap());
2191
        let v: ~[int] = Decodable::decode(&mut decoder);
2192 2193
        assert_eq!(v, ~[3, 1]);

2194
        let mut decoder = Decoder::new(from_str("[[3], [1, 2]]").unwrap());
2195
        let v: ~[~[uint]] = Decodable::decode(&mut decoder);
2196 2197 2198
        assert_eq!(v, ~[~[3], ~[1, 2]]);
    }

2199
    #[test]
2200
    fn test_read_object() {
E
Erick Tryzelaar 已提交
2201
        assert_eq!(from_str("{"),
2202 2203 2204
            Err(Error {
                line: 1u,
                col: 2u,
E
Erick Tryzelaar 已提交
2205
                msg: ~"EOF while parsing object"}));
E
Erick Tryzelaar 已提交
2206
        assert_eq!(from_str("{ "),
2207 2208 2209
            Err(Error {
                line: 1u,
                col: 3u,
E
Erick Tryzelaar 已提交
2210
                msg: ~"EOF while parsing object"}));
E
Erick Tryzelaar 已提交
2211
        assert_eq!(from_str("{1"),
2212 2213 2214
            Err(Error {
                line: 1u,
                col: 2u,
E
Erick Tryzelaar 已提交
2215
                msg: ~"key must be a string"}));
E
Erick Tryzelaar 已提交
2216
        assert_eq!(from_str("{ \"a\""),
2217 2218 2219
            Err(Error {
                line: 1u,
                col: 6u,
E
Erick Tryzelaar 已提交
2220
                msg: ~"EOF while parsing object"}));
E
Erick Tryzelaar 已提交
2221
        assert_eq!(from_str("{\"a\""),
2222 2223 2224
            Err(Error {
                line: 1u,
                col: 5u,
E
Erick Tryzelaar 已提交
2225
                msg: ~"EOF while parsing object"}));
E
Erick Tryzelaar 已提交
2226
        assert_eq!(from_str("{\"a\" "),
2227 2228 2229
            Err(Error {
                line: 1u,
                col: 6u,
E
Erick Tryzelaar 已提交
2230
                msg: ~"EOF while parsing object"}));
2231

E
Erick Tryzelaar 已提交
2232
        assert_eq!(from_str("{\"a\" 1"),
E
Erick Tryzelaar 已提交
2233
            Err(Error {line: 1u, col: 6u, msg: ~"expected `:`"}));
E
Erick Tryzelaar 已提交
2234
        assert_eq!(from_str("{\"a\":"),
E
Erick Tryzelaar 已提交
2235
            Err(Error {line: 1u, col: 6u, msg: ~"EOF while parsing value"}));
E
Erick Tryzelaar 已提交
2236
        assert_eq!(from_str("{\"a\":1"),
2237 2238 2239
            Err(Error {
                line: 1u,
                col: 7u,
E
Erick Tryzelaar 已提交
2240
                msg: ~"EOF while parsing object"}));
E
Erick Tryzelaar 已提交
2241
        assert_eq!(from_str("{\"a\":1 1"),
E
Erick Tryzelaar 已提交
2242
            Err(Error {line: 1u, col: 8u, msg: ~"expected `,` or `}`"}));
E
Erick Tryzelaar 已提交
2243
        assert_eq!(from_str("{\"a\":1,"),
2244 2245 2246
            Err(Error {
                line: 1u,
                col: 8u,
E
Erick Tryzelaar 已提交
2247
                msg: ~"EOF while parsing object"}));
2248

E
Erick Tryzelaar 已提交
2249 2250
        assert_eq!(from_str("{}").unwrap(), mk_object([]));
        assert_eq!(from_str("{\"a\": 3}").unwrap(),
D
Daniel Micay 已提交
2251
                  mk_object([(~"a", Number(3.0))]));
2252

E
Erick Tryzelaar 已提交
2253 2254
        assert_eq!(from_str(
                      "{ \"a\": null, \"b\" : true }").unwrap(),
E
Erick Tryzelaar 已提交
2255
                  mk_object([
B
Ben Striegel 已提交
2256
                      (~"a", Null),
2257
                      (~"b", Boolean(true))]));
E
Erick Tryzelaar 已提交
2258
        assert_eq!(from_str("\n{ \"a\": null, \"b\" : true }\n").unwrap(),
E
Erick Tryzelaar 已提交
2259
                  mk_object([
B
Ben Striegel 已提交
2260
                      (~"a", Null),
2261
                      (~"b", Boolean(true))]));
E
Erick Tryzelaar 已提交
2262 2263
        assert_eq!(from_str(
                      "{\"a\" : 1.0 ,\"b\": [ true ]}").unwrap(),
E
Erick Tryzelaar 已提交
2264
                  mk_object([
2265 2266
                      (~"a", Number(1.0)),
                      (~"b", List(~[Boolean(true)]))
2267
                  ]));
E
Erick Tryzelaar 已提交
2268
        assert_eq!(from_str(
2269
                      ~"{" +
2270 2271 2272 2273 2274 2275
                          "\"a\": 1.0, " +
                          "\"b\": [" +
                              "true," +
                              "\"foo\\nbar\", " +
                              "{ \"c\": {\"d\": null} } " +
                          "]" +
E
Erick Tryzelaar 已提交
2276
                      "}").unwrap(),
E
Erick Tryzelaar 已提交
2277
                  mk_object([
D
Daniel Micay 已提交
2278
                      (~"a", Number(1.0)),
2279
                      (~"b", List(~[
B
Ben Striegel 已提交
2280
                          Boolean(true),
2281
                          String(~"foo\nbar"),
E
Erick Tryzelaar 已提交
2282 2283
                          mk_object([
                              (~"c", mk_object([(~"d", Null)]))
2284 2285
                          ])
                      ]))
2286
                  ]));
2287 2288
    }

2289
    #[test]
2290 2291 2292 2293 2294 2295
    fn test_decode_struct() {
        let s = ~"{
            \"inner\": [
                { \"a\": null, \"b\": 2, \"c\": [\"abc\", \"xyz\"] }
            ]
        }";
2296
        let mut decoder = Decoder::new(from_str(s).unwrap());
2297
        let v: Outer = Decodable::decode(&mut decoder);
2298 2299 2300 2301 2302 2303 2304 2305 2306 2307 2308 2309
        assert_eq!(
            v,
            Outer {
                inner: ~[
                    Inner { a: (), b: 2, c: ~[~"abc", ~"xyz"] }
                ]
            }
        );
    }

    #[test]
    fn test_decode_option() {
2310
        let mut decoder = Decoder::new(from_str("null").unwrap());
2311
        let value: Option<~str> = Decodable::decode(&mut decoder);
2312 2313
        assert_eq!(value, None);

2314
        let mut decoder = Decoder::new(from_str("\"jodhpurs\"").unwrap());
2315
        let value: Option<~str> = Decodable::decode(&mut decoder);
2316 2317 2318
        assert_eq!(value, Some(~"jodhpurs"));
    }

2319
    #[test]
2320
    fn test_decode_enum() {
2321
        let mut decoder = Decoder::new(from_str("\"Dog\"").unwrap());
2322
        let value: Animal = Decodable::decode(&mut decoder);
2323 2324
        assert_eq!(value, Dog);

2325
        let s = "{\"variant\":\"Frog\",\"fields\":[\"Henry\",349]}";
2326
        let mut decoder = Decoder::new(from_str(s).unwrap());
2327
        let value: Animal = Decodable::decode(&mut decoder);
2328 2329 2330
        assert_eq!(value, Frog(~"Henry", 349));
    }

2331
    #[test]
2332
    fn test_decode_map() {
2333
        let s = ~"{\"a\": \"Dog\", \"b\": {\"variant\":\"Frog\",\"fields\":[\"Henry\", 349]}}";
2334
        let mut decoder = Decoder::new(from_str(s).unwrap());
2335
        let mut map: TreeMap<~str, Animal> = Decodable::decode(&mut decoder);
2336

2337 2338
        assert_eq!(map.pop(&~"a"), Some(Dog));
        assert_eq!(map.pop(&~"b"), Some(Frog(~"Henry", 349)));
2339 2340
    }

2341
    #[test]
2342
    fn test_multiline_errors() {
E
Erick Tryzelaar 已提交
2343
        assert_eq!(from_str("{\n  \"foo\":\n \"bar\""),
2344 2345 2346
            Err(Error {
                line: 3u,
                col: 8u,
E
Erick Tryzelaar 已提交
2347
                msg: ~"EOF while parsing object"}));
2348
    }
2349 2350 2351 2352 2353 2354 2355 2356 2357 2358 2359 2360 2361 2362

    #[deriving(Decodable)]
    struct DecodeStruct {
        x: f64,
        y: bool,
        z: ~str,
        w: ~[DecodeStruct]
    }
    #[deriving(Decodable)]
    enum DecodeEnum {
        A(f64),
        B(~str)
    }
    fn check_err<T: Decodable<Decoder>>(to_parse: &'static str, expected_error: &str) {
M
Marvin Löbel 已提交
2363
        use std::any::AnyRefExt;
2364
        use std::task;
2365
        let res = task::try(proc() {
2366 2367 2368 2369 2370 2371
            // either fails in `decode` (which is what we want), or
            // returns Some(error_message)/None if the string was
            // invalid or valid JSON.
            match from_str(to_parse) {
                Err(e) => Some(e.to_str()),
                Ok(json) => {
2372
                    let _: T = Decodable::decode(&mut Decoder::new(json));
2373 2374 2375
                    None
                }
            }
2376
        });
2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416
        match res {
            Ok(Some(parse_error)) => fail!("`{}` is not valid json: {}",
                                           to_parse, parse_error),
            Ok(None) => fail!("`{}` parsed & decoded ok, expecting error `{}`",
                              to_parse, expected_error),
            Err(e) => {
                let err = e.as_ref::<~str>().unwrap();
                assert!(err.contains(expected_error),
                        "`{}` errored incorrectly, found `{}` expecting `{}`",
                        to_parse, *err, expected_error);
            }
        }
    }
    #[test]
    fn test_decode_errors_struct() {
        check_err::<DecodeStruct>("[]", "object but found list");
        check_err::<DecodeStruct>("{\"x\": true, \"y\": true, \"z\": \"\", \"w\": []}",
                                  "number but found boolean");
        check_err::<DecodeStruct>("{\"x\": 1, \"y\": [], \"z\": \"\", \"w\": []}",
                                  "boolean but found list");
        check_err::<DecodeStruct>("{\"x\": 1, \"y\": true, \"z\": {}, \"w\": []}",
                                  "string but found object");
        check_err::<DecodeStruct>("{\"x\": 1, \"y\": true, \"z\": \"\", \"w\": null}",
                                  "list but found null");
        check_err::<DecodeStruct>("{\"x\": 1, \"y\": true, \"z\": \"\"}",
                                  "'w' field in object");
    }
    #[test]
    fn test_decode_errors_enum() {
        check_err::<DecodeEnum>("{}",
                                "'variant' field in object");
        check_err::<DecodeEnum>("{\"variant\": 1}",
                                "string but found number");
        check_err::<DecodeEnum>("{\"variant\": \"A\"}",
                                "'fields' field in object");
        check_err::<DecodeEnum>("{\"variant\": \"A\", \"fields\": null}",
                                "list but found null");
        check_err::<DecodeEnum>("{\"variant\": \"C\", \"fields\": []}",
                                "unknown variant name");
    }
2417 2418 2419 2420 2421

    #[test]
    fn test_find(){
        let json_value = from_str("{\"dog\" : \"cat\"}").unwrap();
        let found_str = json_value.find(&~"dog");
2422
        assert!(found_str.is_some() && found_str.unwrap().as_string().unwrap() == &"cat");
2423 2424 2425 2426 2427 2428
    }

    #[test]
    fn test_find_path(){
        let json_value = from_str("{\"dog\":{\"cat\": {\"mouse\" : \"cheese\"}}}").unwrap();
        let found_str = json_value.find_path(&[&~"dog", &~"cat", &~"mouse"]);
2429
        assert!(found_str.is_some() && found_str.unwrap().as_string().unwrap() == &"cheese");
2430 2431 2432 2433 2434
    }

    #[test]
    fn test_search(){
        let json_value = from_str("{\"dog\":{\"cat\": {\"mouse\" : \"cheese\"}}}").unwrap();
2435
        let found_str = json_value.search(&~"mouse").and_then(|j| j.as_string());
2436 2437 2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448 2449 2450 2451 2452 2453 2454 2455 2456 2457 2458 2459 2460 2461 2462 2463 2464 2465 2466 2467
        assert!(found_str.is_some());
        assert!(found_str.unwrap() == &"cheese");
    }

    #[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]
    fn test_is_list(){
        let json_value = from_str("[1, 2, 3]").unwrap();
        assert!(json_value.is_list());
    }

    #[test]
    fn test_as_list(){
        let json_value = from_str("[1, 2, 3]").unwrap();
        let json_list = json_value.as_list();
        let expected_length = 3;
        assert!(json_list.is_some() && json_list.unwrap().len() == expected_length);
    }

    #[test]
2468
    fn test_is_string(){
2469
        let json_value = from_str("\"dog\"").unwrap();
2470
        assert!(json_value.is_string());
2471 2472 2473
    }

    #[test]
2474
    fn test_as_string(){
2475
        let json_value = from_str("\"dog\"").unwrap();
2476
        let json_str = json_value.as_string();
2477 2478 2479 2480 2481 2482 2483 2484 2485 2486 2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520 2521
        let expected_str = &"dog";
        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]
    fn test_as_number(){
        let json_value = from_str("12").unwrap();
        let json_num = json_value.as_number();
        let expected_num = 12f64;
        assert!(json_num.is_some() && json_num.unwrap() == expected_num);
    }

    #[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);
    }
2522
}