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
use super::WriteItem;

pub struct PrintWriteItemsOptions {
    /// The number of spaces to use when indenting when use_tabs is false,
    /// otherwise the number of columns to count a tab for when use_tabs is true.
    pub indent_width: u8,
    /// Whether to use tabs for indenting.
    pub use_tabs: bool,
    /// The newline character to use when doing a new line.
    pub new_line_text: &'static str,
}

/// Prints string based writer items.
pub fn print_write_items(write_items: impl Iterator<Item = WriteItem>, options: PrintWriteItemsOptions) -> String {
    // todo: faster string manipulation? or is this as good as it gets?
    let mut final_string = String::new();
    let indent_string = if options.use_tabs { String::from("\t") } else { " ".repeat(options.indent_width as usize) };

    for item in write_items.into_iter() {
        match item {
            WriteItem::Indent(times) => final_string.push_str(&indent_string.repeat(times as usize)),
            WriteItem::NewLine => final_string.push_str(&options.new_line_text),
            WriteItem::Tab => final_string.push_str("\t"),
            WriteItem::Space => final_string.push_str(" "),
            WriteItem::String(text) => final_string.push_str(&text.text),
        }
    }

    final_string
}