wpt 0.0.12

CLI for working with WPT data
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

use std::fs::{self, File};
use std::io::Read;
use std::os::unix::ffi::OsStrExt as _;
use std::path::Path;
use xz2::read::XzDecoder;

pub fn read_maybe_compressed_file(file_path: &Path) -> String {
    let file = File::open(file_path).unwrap();
    let extension = file_path.extension().unwrap().as_bytes();

    match extension {
        b"xz" => {
            let mut decompressed = XzDecoder::new(file);
            let mut s = String::new();
            decompressed.read_to_string(&mut s).unwrap();
            s
        }
        b"zst" => {
            let mut decompressed = zstd::Decoder::new(file).unwrap();
            let mut s = String::new();
            decompressed.read_to_string(&mut s).unwrap();
            s
        }
        _ => fs::read_to_string(file_path).unwrap(),
    }
}