torchbear 0.10.9

Lua programming environment in Rust
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

use rlua::prelude::*;
use std::fs;
use std::io::Read;
use std::result;
use std::path::Path;
use tar::Archive;
use error::Error;

use super::ByteBuf;

fn extract<T: Read>(archive: &mut Archive<T>, dst: &Path) -> result::Result<(), LuaError> {
    for file in archive.entries().map_err(LuaError::external)? {
        file
            .and_then(|mut file| file.unpack_in(&dst))
            .map_err(LuaError::external)?;
    }
    Ok(())
}

pub fn init(lua: &Lua) -> ::Result<()> {
    let module = lua.create_table()?;
    module.set("decompress", lua.create_function(|_, (src, dst): (String, String)| {
        let tar = fs::File::open(src).map_err(LuaError::external)?;
        let dst = Path::new(&dst);
        let mut archive = Archive::new(tar);
        extract(&mut archive, &dst)
    })?)?;

    module.set("decompress_buf", lua.create_function(|_, (data, dst): (ByteBuf, String)| {
        let dst = Path::new(&dst);
        let mut archive = Archive::new(&data.0[..]);
        extract(&mut archive, &dst)
    })?)?;

    lua.globals().set("tar", module).map_err(Error::from)?;

    Ok(())
}