plm-rs 0.1.4

Rust powered BOM management and material planning for your command line.
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
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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105

// License: Apache 2.0
// Author: Jared Wolff

use std::io::{BufRead, Write};

pub struct Prompt<R, W> {
    pub reader: R,
    pub writer: W,
}

// Returns a string response. Newline removed.
impl<R, W> Prompt<R, W>
where
    R: BufRead,
    W: Write,
{
    pub fn ask_text_entry(&mut self, question: &str) -> String {
        // Ask the question
        write!(&mut self.writer, "{}", question).expect("Unable to write");
        self.writer.flush().expect("Unable to flush");

        // Parse input
        let mut input = String::new();
        self.reader.read_line(&mut input).unwrap();
        let response = &input[..(input.len() - 1)]; // Drop the newline character
        let response = response.trim();

        // Return a copy
        response.to_string()
    }

    pub fn ask_yes_no_question(&mut self, question: &str) -> bool {
        // Ask the question
        write!(&mut self.writer, "{} (y/n) ", question).expect("Unable to write!");
        self.writer.flush().expect("Unable to flush output!");

        // Parse input
        let mut input = String::new();
        self.reader.read_line(&mut input).unwrap();
        let ch = input.chars().next().unwrap();

        // Return result
        ch == 'y'
    }
}

#[test]
fn test_text_entry_with_whitespace() {
    let input = b"I love cookies!   \n";
    let mut output = Vec::new();

    let answer = {
        let mut prompt = Prompt {
            reader: &input[..],
            writer: &mut output,
        };

        prompt.ask_text_entry("Do you like cookies?")
    };

    let output = String::from_utf8(output).expect("Not UTF-8");

    assert_eq!("Do you like cookies?", output); // output to stdout
    assert_eq!("I love cookies!", answer); // input
}

#[test]
fn test_yes_no_expect_true() {
    let input = b"y\n";
    let mut output = Vec::new();

    let answer = {
        let mut prompt = Prompt {
            reader: &input[..],
            writer: &mut output,
        };

        prompt.ask_yes_no_question("Would you like to do things?")
    };

    let output = String::from_utf8(output).expect("Not UTF-8");

    assert_eq!("Would you like to do things? (y/n) ", output);
    assert_eq!(true, answer);
}

#[test]
fn test_yes_no_expect_false() {
    let input = b"\n";
    let mut output = Vec::new();

    let answer = {
        let mut prompt = Prompt {
            reader: &input[..],
            writer: &mut output,
        };

        prompt.ask_yes_no_question("Would you like to do things?")
    };

    let output = String::from_utf8(output).expect("Not UTF-8");

    assert_eq!("Would you like to do things? (y/n) ", output);
    assert_eq!(false, answer);
}