wash-cli 0.34.1

use term_table::{
    row::Row,
    table_cell::{Alignment, TableCell},
    Table,
};
use wadm_types::api::{Status, VersionInfo};

use super::ModelSummary;

pub fn list_revisions_table(revisions: Vec<VersionInfo>) -> String {
    let mut table = Table::new();
    crate::util::configure_table_style(&mut table, 2);

    table.add_row(Row::new(vec![
        TableCell::new_with_alignment("Version", 1, Alignment::Left),
        TableCell::new_with_alignment("Deployed", 1, Alignment::Left),
    ]));

    revisions.iter().for_each(|r| {
        table.add_row(Row::new(vec![
            TableCell::new_with_alignment(r.version.clone(), 1, Alignment::Left),
            TableCell::new_with_alignment(r.deployed, 1, Alignment::Left),
        ]));
    });

    table.render()
}

pub fn list_models_table(models: Vec<ModelSummary>) -> String {
    let mut table = Table::new();
    crate::util::configure_table_style(&mut table, 3);
    table.add_row(Row::new(vec![
        TableCell::new_with_alignment("Name", 1, Alignment::Left),
        TableCell::new_with_alignment("Deployed Version", 1, Alignment::Left),
        TableCell::new_with_alignment("Status", 1, Alignment::Left),
    ]));
    models.iter().for_each(|m| {
        table.add_row(Row::new(vec![
            TableCell::new_with_alignment(m.name.clone(), 1, Alignment::Left),
            TableCell::new_with_alignment(
                m.deployed_version
                    .clone()
                    .unwrap_or_else(|| "N/A".to_string()),
                1,
                Alignment::Left,
            ),
            #[allow(deprecated)]
            TableCell::new_with_alignment(format!("{:?}", m.status), 1, Alignment::Left),
        ]));

        if let Some(description) = m.description.as_ref() {
            table.add_row(Row::new(vec![TableCell::new_with_alignment(
                format!("  └ {}", description),
                3,
                Alignment::Left,
            )]));
        }
    });

    table.render()
}

/// Generate a table for displaying the status of a model
///
/// This table contains a lot of information, and some potentially very long strings
/// with status messages. Because of this, there's some manual formatting of strings
/// after the table
pub fn status_table(model_name: String, status: Status) -> String {
    let mut table = Table::new();
    crate::util::configure_table_style(&mut table, 4);

    table.add_row(Row::new(vec![TableCell::new_with_alignment(
        "",
        4,
        Alignment::Center,
    )]));

    table.add_row(Row::new(vec![
        TableCell::new_with_alignment("Name", 2, Alignment::Left),
        TableCell::new_with_alignment("Kind", 1, Alignment::Left),
        TableCell::new_with_alignment("Status", 1, Alignment::Left),
    ]));

    // To better display information in the table, what we want to do here is replace
    // the generated IDs in a manifest with just the component name.
    //
    // For example, turning "rust_hello_world-http_component" into "http_component"
    // by removing the model name and the "-" character.
    let mut model_name_replacer = model_name.replace('-', "_");
    model_name_replacer.push('-');
    status.scalers.iter().for_each(|s| {
        let status = if !s.info.message.is_empty() {
            format!("{:?} (*)", s.info.status_type)
        } else {
            format!("{:?}", s.info.status_type)
        };
        table.add_row(Row::new(vec![
            TableCell::new_with_alignment(
                s.name.replace(&model_name_replacer, ""),
                2,
                Alignment::Left,
            ),
            TableCell::new_with_alignment(&s.kind, 1, Alignment::Left),
            TableCell::new_with_alignment(status, 1, Alignment::Left),
        ]))
    });

    if status.scalers.iter().any(|s| !s.info.message.is_empty()) {
        table.add_row(Row::new(vec![TableCell::new_with_alignment(
            "",
            4,
            Alignment::Center,
        )]));
        table.add_row(Row::new(vec![TableCell::new_with_alignment(
            "Status Messages",
            4,
            Alignment::Left,
        )]));
    }

    let mut table_output = table.render();

    status.scalers.iter().for_each(|s| {
        if !s.info.message.is_empty() {
            table_output.push_str(&format!(
                "  {}\n    └ {}\n\n",
                &s.name.replace(&model_name_replacer, ""),
                s.info.message
            ));
        }
    });

    // Prepend the application name, version, and status type to the table output
    #[allow(deprecated)]
    let version = status.version;
    format!(
        "{}@{} - {:?}{}",
        &model_name, version, status.info.status_type, table_output
    )
}