peppi 2.1.2

Parser for Slippi replay files
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

use std::io::Read;

use byteorder::{BigEndian, ReadBytesExt};
use serde_json::{Map, Value};

use crate::io::Result;

fn to_utf8<R: Read>(r: &mut R) -> Result<String> {
	let length = r.read_u8()?;
	let mut buf = vec![0; length as usize];
	r.read_exact(&mut buf)?;
	Ok(String::from_utf8(buf)?)
}

fn to_val<R: Read>(r: &mut R) -> Result<Value> {
	match r.read_u8()? {
		// "S": str
		0x53 => match r.read_u8()? {
			0x55 => Ok(Value::String(to_utf8(r)?)),
			c => Err(err!("Expected 0x55 for string length, but got: {}", c)),
		},
		// "l": i32
		0x6c => Ok(Value::Number(serde_json::Number::from(
			r.read_i32::<BigEndian>()?,
		))),
		// "{": map
		0x7b => Ok(Value::Object(read_map(r)?)),
		c => Err(err!("unexpected UBJSON value type: {}", c)),
	}
}

fn to_key<R: Read>(r: &mut R) -> Result<Option<String>> {
	match r.read_u8()? {
		0x55 => Ok(Some(to_utf8(r)?)),
		0x7d => Ok(None),
		c => Err(err!("unexpected UBJSON key type: {}", c)),
	}
}

pub(crate) fn read_map<R: Read>(r: &mut R) -> Result<Map<String, Value>> {
	let mut m = Map::new();
	while match to_key(r)? {
		Some(k) => {
			m.insert(k, to_val(r)?);
			true
		}
		None => false,
	} {}
	Ok(m)
}