pkgcraft 0.0.31

library of Gentoo functionality
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

use std::fs::File;
use std::{fs, io};

use scallop::builtins::BuiltinFn;
use scallop::{Error, ExecStatus};
use tempfile::tempdir;

use crate::io::stdin;
use crate::utils::is_single_component;

// Underlying implementation for new* builtins.
pub(super) fn new(args: &[&str], cmd: BuiltinFn) -> scallop::Result<ExecStatus> {
    let [source, name] = args[..] else {
        return Err(Error::Base(format!("requires 2 args, got {}", args.len())));
    };

    // filename can't contain a path separator
    if !is_single_component(name) {
        return Err(Error::Base(format!("invalid filename: {name}")));
    }

    // TODO: create tempdir in $T to avoid cross-fs issues as much as possible
    let tmp_dir =
        tempdir().map_err(|e| Error::Base(format!("failed creating tempdir: {e}")))?;
    let dest = tmp_dir.path().join(name);

    if source == "-" {
        let mut file = File::create(&dest)
            .map_err(|e| Error::Base(format!("failed opening file: {dest:?}: {e}")))?;
        io::copy(&mut stdin(), &mut file).map_err(|e| {
            Error::Base(format!("failed writing stdin to file: {dest:?}: {e}"))
        })?;
    } else {
        fs::copy(source, &dest).map_err(|e| {
            Error::Base(format!("failed copying file {source:?} to {dest:?}: {e}"))
        })?;
    }

    let path = dest
        .to_str()
        .ok_or_else(|| Error::Base(format!("invalid utf8 path: {dest:?}")))?;
    cmd(&[path])
}

#[cfg(test)]
mod tests {
    use crate::test::assert_err_re;

    use super::super::functions::newbin;
    use super::*;

    #[test]
    fn invalid_args() {
        // nonexistent
        let r = new(&["bin", "pkgcraft"], newbin);
        assert_err_re!(r, "^failed copying file \"bin\" .*$");

        // filename contains path separator
        for f in ["bin/pkgcraft", "bin//pkgcraft", "/bin/pkgcraft", "pkgcraft/", "/"] {
            let r = new(&["bin", f], newbin);
            assert_err_re!(r, format!("^invalid filename: {f}$"));
        }
    }
}