sui_aws_orchestrator/
display.rs

// Copyright (c) Mysten Labs, Inc.
// SPDX-License-Identifier: Apache-2.0

use std::{fmt::Display, io::stdout};

use crossterm::{
    cursor::{RestorePosition, SavePosition},
    style::{Print, PrintStyledContent, Stylize},
    terminal::{Clear, ClearType},
};
use prettytable::format::{self};

pub fn header<S: Display>(message: S) {
    crossterm::execute!(
        stdout(),
        PrintStyledContent(format!("\n{message}\n").green().bold()),
    )
    .unwrap();
}

pub fn error<S: Display>(message: S) {
    crossterm::execute!(
        stdout(),
        PrintStyledContent(format!("\n{message}\n").red().bold()),
    )
    .unwrap();
}

pub fn warn<S: Display>(message: S) {
    crossterm::execute!(
        stdout(),
        PrintStyledContent(format!("\n{message}\n").bold()),
    )
    .unwrap();
}

pub fn config<N: Display, V: Display>(name: N, value: V) {
    crossterm::execute!(
        stdout(),
        PrintStyledContent(format!("{name}: ").bold()),
        Print(format!("{value}\n"))
    )
    .unwrap();
}

pub fn action<S: Display>(message: S) {
    crossterm::execute!(stdout(), Print(format!("{message} ... ")), SavePosition).unwrap();
}

pub fn status<S: Display>(status: S) {
    crossterm::execute!(
        stdout(),
        RestorePosition,
        SavePosition,
        Clear(ClearType::UntilNewLine),
        Print(format!("[{status}]"))
    )
    .unwrap();
}

pub fn done() {
    crossterm::execute!(
        stdout(),
        RestorePosition,
        Clear(ClearType::UntilNewLine),
        Print(format!("[{}]\n", "Ok".green()))
    )
    .unwrap();
}

pub fn newline() {
    crossterm::execute!(stdout(), Print("\n")).unwrap();
}

/// Default style for tables printed to stdout.
pub fn default_table_format() -> format::TableFormat {
    format::FormatBuilder::new()
        .separators(
            &[
                format::LinePosition::Top,
                format::LinePosition::Bottom,
                format::LinePosition::Title,
            ],
            format::LineSeparator::new('-', '-', '-', '-'),
        )
        .padding(1, 1)
        .build()
}

#[cfg(test)]
mod test {
    use std::time::Duration;

    use tokio::time::sleep;

    use crate::display::status;

    use super::{action, config, done, error, header, newline, warn};

    #[tokio::test]
    #[ignore = "only used to manually check if prints work correctly"]
    async fn display() {
        header("This is a header");
        config("This is a config", 2);
        action("Running a long function");
        for i in 0..5 {
            sleep(Duration::from_secs(1)).await;
            if i == 2 {
                warn("This is a warning!");
            }
            status(format!("{}/5", i + 1));
        }
        done();
        error("This is an error!");
        warn("This is a warning!");
        newline();
    }
}
sui_aws_orchestrator/display.rs

sui_aws_orchestrator/
display.rs
