rarity 0.3.0

Library of useful functions.
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

use super::command::execute_command;
use atty::Stream;
use std::io::{self, Write};
use std::{error::Error, fs};

/// Function to collect stdin or return None.
pub fn get_stdin() -> Option<String> {
	atty::isnt(Stream::Stdin).then(|| {
		io::stdin()
			.lines()
			.map(|l| l.unwrap())
			.collect::<Vec<_>>()
			.join("\n")
	})
}

pub fn ask<F>(question: &str, filter: Option<(F, &str)>) -> Result<String, Box<dyn Error>>
where
	F: Fn(&str) -> bool,
{
	let mut answer = String::new();
	println!("{}", question);
	io::stdout().flush()?;
	io::stdin().read_line(&mut answer)?;
	if let Some((filter_fn, filter_err)) = filter {
		if !filter_fn(answer.trim()) {
			eprintln!("{filter_err}");
			return ask(question, Some((filter_fn, filter_err)));
		}
	}
	Ok(answer.trim().to_string())
}

pub fn ask_longform(question: &str, temp_filename: &str) -> Result<String, Box<dyn Error>> {
	fs::File::create_new(temp_filename)?;
	fs::write(temp_filename, question)?;
	let mut stdout = io::stdout();
	stdout.write_all(b"\x1B[?1049h")?;
	stdout.flush()?;
	execute_command(&format!("vim {temp_filename}"))?;
	stdout.write_all(b"\x1B[?1049l")?;
	stdout.flush()?;
	let answer = fs::read_to_string(temp_filename)?;
	fs::remove_file(temp_filename)?;
	Ok(answer)
}