new-vdf-parser 0.1.0

A parser for Steam's VDF file formats including appinfo.vdf and shortcuts.vdf
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97

use std::{ path::PathBuf, fs };
use std::io::{Read, Write};

use serde_json::{ Value, Map };

use crate::reader::Reader;
use crate::writer::Writer;
use crate::vdf_reader::read_entry_map;

/// Opens the shortcuts.vdf file and returns the values as JSON.
pub fn open_shortcuts_vdf(path: &PathBuf) -> Value {
  let mut file = fs::File::open(path).expect("Path should have existed.");

  let metadata = fs::metadata(path).expect("unable to read metadata");
  let mut buffer = vec![0; metadata.len() as usize];
  file.read(&mut buffer).expect("buffer overflow");
  
  let buf_slice = buffer.as_slice();
  let mut reader = Reader::new(buf_slice);

  return read(&mut reader);
}

/// Reads the shortcuts.vdf file and returns the values as JSON.
fn read(reader: &mut Reader) -> Value {
  reader.seek(1, 0);

  let fake_header = reader.read_string(None);

  if fake_header != "shortcuts".to_owned() {
    panic!("Invalid Shortcuts File! File started with {} instead of \"shortcuts\"", fake_header);
  }

  return Value::Object(read_entry_map(reader, None, &None));
}

/// Writes the shortcuts.vdf file from JSON.
pub fn write_shortcuts_vdf(path: &PathBuf, data: Value) -> bool {
  if data.is_object() {
    let shortcuts = data.as_object().expect("Should have been able to convert to an object.");
    
    let mut buffer: Vec<u8> = vec![0; 1000];
    let mut writer: Writer = Writer::new(&mut buffer);
    
    write_entry_map(&mut writer, shortcuts);

    writer.trim();

    let mut file = fs::File::create(path).expect("Should have been able to create file or truncate if existed.");
    let write_res = file.write_all(&buffer[..]);

    if write_res.is_err() {
      let err = write_res.err().unwrap();
      panic!("Error writing shortcuts: {}", err);
    }

    return true;
  } else {
    panic!("Error writing shortcuts: data was not an object!");
  }
}

/// Writes a shortcuts.vdf entry map from JSON.
fn write_entry_map(writer: &mut Writer, map: &Map<String, Value>) {
  for (key, val) in map.into_iter() {
    write_entry_field(writer, key, val);
  }
  
  writer.write_uint8(0x08, true);
}

/// Writes a shortcuts.vdf entry field from JSON.
fn write_entry_field(writer: &mut Writer, key: &String, field: &Value) {
  let key_owned: String = key.to_owned();

  if field.is_number() {
    writer.write_uint8(0x02, true);
    writer.write_string(key_owned, false, true);

    let number: u64 = field.as_u64().expect("Should have been able to convert to a number.");
    writer.write_uint32(number as u32, true);
  } else if field.is_string() {
    writer.write_uint8(0x01, true);
    writer.write_string(key_owned, false, true);

    let string: &str = field.as_str().expect("Should have been able to convert to a string.");
    writer.write_string(string.to_owned(), false, true);
  } else if field.is_object() {
    writer.write_uint8(0x00, true);
    writer.write_string(key_owned, false, true);

    let field_map = field.as_object().expect("Should have been able to convert to an object.");
    write_entry_map(writer, field_map);
  } else {
    panic!("Value was not an object, number or string!");
  }
}