pixterm 0.1.0

A CLI to show images in a terminal
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
65
66
67
68
69
70
71
72
73
74

use std::{path::PathBuf, process, fs::File, io::Write};
use structopt::StructOpt;

/// A CLI tool to show images in a terminal
#[derive(StructOpt, Debug)]
#[structopt(author)]
struct PixTerm {
    /// Path to image file to display
    #[structopt(short, long)]
    file: PathBuf,

    /// Maximum width in pixels of the resized image. Also see --height
    #[structopt(short = "W", long, default_value = "32")]
    width: u16,

    /// Maximum height in pixels of the resized image. Also see --width
    #[structopt(short = "H", long, default_value = "32")]
    height: u16,

    /// Minimum alpha value of a pixel for it to be shown
    #[structopt(short, long, default_value = "50")]
    threshold: u8,

    /// File to write the resulting string into. See --raw to get literal escape sequences and --silent to suppress stdout
    #[structopt(short, long)]
    outfile: Option<PathBuf>,

    /// Print escape sequences literal
    #[structopt(short, long)]
    raw: bool,

    /// Do not print to stdout. Useful with --outfile
    #[structopt(short, long)]
    silent: bool,
}

fn run(pixterm: PixTerm) {
    let img = ansipix::of_image(
        pixterm.file.clone(),
        (pixterm.width as usize, pixterm.height as usize),
        pixterm.threshold,
        pixterm.raw
    ).unwrap_or_else(|_| {
        eprintln!(
            "\x1b[1;31m{}\x1b[22m could not be opened as an image. Does it exist? Is it an image?\x1b[0m",
            pixterm.file.to_str().unwrap_or("The specified file")
        );
        process::exit(1);
    });

    if !pixterm.silent {
        println!("{}", img);
    }
    if pixterm.outfile.is_some() {
        let outfile = pixterm.outfile.unwrap();
        if outfile.exists() {
            eprintln!("\x1b[1;31m{}\x1b[22m already exists.\x1b[0m", outfile.to_str().unwrap_or("The specified output file"));
            process::exit(2);
        }
        let mut file = File::create(outfile).unwrap_or_else(|e| {
            eprintln!("\x1b[1;31mError while creating the file:\x1b[22m {}", e);
            process::exit(3);
        });
        file.write_all(img.as_bytes()).unwrap_or_else(|e| {
            eprintln!("\x1b[1;31mError while writing to the file:\x1b[22m {}", e);
            process::exit(4);
        });
    }
}

fn main() {
    let pixterm = PixTerm::from_args();
    run(pixterm);
}