zsplit-cli 0.4.0

Split text into multiple files by line
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

use error_stack::{Result, ResultExt};
use io::{BufRead, BufReader};
use std::fs::File;
use std::io;
use std::path::PathBuf;

#[cfg(test)]
#[path = "./source_test.rs"]
mod source_test;

#[derive(Debug, PartialEq, Eq)]
pub(crate) enum Source {
    PathBuf(PathBuf),
    StdIn,
}

impl Source {
    pub(crate) fn reading_buffer(&self) -> Result<Box<dyn BufRead>, io::Error> {
        match self {
            Self::PathBuf(current_file) => Ok(Box::new(BufReader::new(
                File::open(current_file).attach_printable_lazy(|| {
                    format!(
                        "Couldn't open file `{}` as readable",
                        current_file.display()
                    )
                })?,
            ))),
            Self::StdIn => {
                let stdin = Box::leak(Box::new(io::stdin()));
                Ok(Box::new(stdin.lock()))
            }
        }
    }

    pub(crate) fn from_os_str(path: &std::ffi::OsStr) -> Self {
        if path == "-" {
            return Self::StdIn;
        }
        Self::PathBuf((*path).into())
    }
}