src/main.rs


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

mod fmt;

use std::fs::File;
use std::io::Read;
use std::io::Write; // Temporary

fn main() {
    let filename = "/tmp/DUKE3D.GRP";

    let mut file = match File::open(filename) {
        Ok(file) => file,
        Err(err) => {
            panic!("Couldn't open {}", filename);
        }
    };

    let mut bytes: Vec<u8> = Vec::new();

    match file.read_to_end(&mut bytes) {
        Ok(count) => println!("Read {} bytes", count),
        Err(err) => panic!("Couldn't read."),
    }

    let grp = fmt::Group::new(&bytes).unwrap();

    println!("Files: {}", grp.file_count);

    for entry in grp {
        println!("{}", entry.name);
        if entry.name.contains("MID") {
            let mut filename = String::from("/tmp/");
            filename.push_str(&entry.name);
            filename = filename[..filename.find('\x00').unwrap()].to_string();
            
            let mut file = File::create(filename).unwrap();
            file.write_all(&entry.data).unwrap();
        }
    }
}