wasibox-core 1.0.4

A collection of core utilities optimized for WASI
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

use clap::Parser;
use std::ffi::OsString;
use std::fs::File;
use std::io::{self, BufRead, BufReader};
use std::path::PathBuf;

#[derive(Parser)]
#[command(name = "head", about = "Output the first part of files")]
pub struct Args {
    /// Files to read
    pub files: Vec<PathBuf>,

    /// Print the first K lines instead of the first 10
    #[arg(short = 'n', long, default_value_t = 10)]
    pub lines: usize,
}

pub fn execute<I, T>(args: I) -> Result<(), String>
where
    I: IntoIterator<Item = T>,
    T: Into<OsString> + Clone,
{
    let args = Args::try_parse_from(args).map_err(|e| e.to_string())?;

    if args.files.is_empty() {
        head_stream(io::stdin().lock(), args.lines)?;
    } else {
        for file_path in &args.files {
            if args.files.len() > 1 {
                println!("==> {} <==", file_path.display());
            }
            let file = File::open(file_path).map_err(|e| format!("head: {}: {}", file_path.display(), e))?;
            head_stream(BufReader::new(file), args.lines)?;
        }
    }
    Ok(())
}

fn head_stream<R: BufRead>(reader: R, lines: usize) -> Result<(), String> {
    for (i, line) in reader.lines().enumerate() {
        if i >= lines {
            break;
        }
        let line = line.map_err(|e| e.to_string())?;
        println!("{}", line);
    }
    Ok(())
}