starship/src/print.rs

use clap::ArgMatches;
use std::io::{self, Write};

use crate::modules;

pub fn prompt(args: ArgMatches) {
    let default_prompt = vec!["directory", "nodejs", "line_break", "character"];

    // TODO:
    // - List files in directory
    // - Index binaries in PATH

    let stdout = io::stdout();
    let mut handle = stdout.lock();

    // Write a new line before the prompt
    writeln!(handle).unwrap();

    default_prompt
        .iter()
        .map(|module| modules::handle(module, &args))
        .map(|segment| segment.output())
        .for_each(|segment_string| write!(handle, "{}", segment_string).unwrap());
}