vpk 0.3.0

Valve Pak format parser for Rust
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

extern crate vpk;

use std::env;
use std::fs;
use std::fs::File;
use std::io::{Read, Write};
use std::path::Path;
use std::vec::Vec;

fn main() -> std::io::Result<()> {
    let args: Vec<_> = env::args().collect();

    if args.len() != 3 {
        panic!("Usage: extract <path to vpk_dir.vpk> <path to export dir>");
    }

    // Check destination dir
    let path = Path::new(&args[2]);
    if !path.is_dir() {
        panic!("Given export path is not directory or doesn't exists");
    }

    let mut vpk_file = match vpk::from_path(&args[1]) {
        Err(e) => panic!("Error while open file {}, err {}", &args[1], e),
        Ok(vpk_file) => vpk_file,
    };

    for (file, vpk_entry) in vpk_file.tree.iter_mut() {
        println!(
            "Extract {}, archive index {}...",
            file, vpk_entry.dir_entry.archive_index
        );
        let file_path = Path::new(file);
        fs::create_dir_all(path.join(file_path.parent().unwrap()))?;

        let mut buf = Vec::new();
        vpk_entry.reader()?.read_to_end(&mut buf)?;

        let mut out_buf = File::create(path.join(file_path))?;
        out_buf.write_all(&buf)?;
    }

    Ok(())
}