himalaya 2.2.1

CLI to manage emails
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

//! # ManageSieve script argument
//!
//! The argument `put` and `check` read a Sieve source from.

use std::{
    fs,
    io::{IsTerminal, Read, stdin},
    path::PathBuf,
};

use anyhow::{Context, Result, bail};
use clap::Parser;

/// Sieve source shared by `put` and `check`.
#[derive(Debug, Parser)]
pub struct SieveScriptArg {
    /// Read the script bytes from a file.
    #[arg(long = "script-file", value_name = "PATH", conflicts_with = "script")]
    pub script_file: Option<PathBuf>,
    /// Inline script text, or omit it and pipe the script on stdin.
    #[arg(name = "script", value_name = "SCRIPT", trailing_var_arg = true)]
    pub script: Vec<String>,
}

impl SieveScriptArg {
    /// Resolves the script bytes from the file, the inline argument or
    /// piped stdin, in that order.
    pub fn read(&self) -> Result<Vec<u8>> {
        if let Some(path) = &self.script_file {
            return fs::read(path)
                .with_context(|| format!("Failed to read Sieve script `{}`", path.display()));
        }

        if !self.script.is_empty() {
            return Ok(self.script.join(" ").into_bytes());
        }

        if !stdin().is_terminal() {
            let mut script = Vec::new();
            stdin().read_to_end(&mut script)?;
            return Ok(script);
        }

        bail!("No Sieve script provided: pass SCRIPT, --script-file, or pipe stdin")
    }
}