display.rs

use crate::commands::cesium;
use crate::entities::vault_account;
use crate::entities::vault_account::AccountTreeNode;
use crate::keys::CryptoScheme;
use crate::utils::GcliError;
use comfy_table::{Cell, Table};
use serde::Serialize;
use std::cell::RefCell;
use std::rc::Rc;
use std::str;

#[deprecated(
	note = "Should be removed in a future version when db persistence of vault is present for a while"
)]
pub fn compute_vault_key_files_table(vault_key_addresses: &[String]) -> Result<Table, GcliError> {
	let mut table = Table::new();
	table.load_preset(comfy_table::presets::UTF8_BORDERS_ONLY);
	table.set_header(vec!["Key file"]);

	vault_key_addresses.iter().for_each(|address| {
		table.add_row(vec![Cell::new(address)]);
	});

	Ok(table)
}

pub fn compute_vault_accounts_table(
	account_tree_nodes: &[Rc<RefCell<AccountTreeNode>>],
) -> Result<Table, GcliError> {
	// Calling the new function with show_g1v1 = true to maintain compatibility
	compute_vault_accounts_table_with_g1v1(account_tree_nodes, true)
}

pub fn compute_vault_accounts_table_with_g1v1(
	account_tree_nodes: &[Rc<RefCell<AccountTreeNode>>],
	show_g1v1: bool,
) -> Result<Table, GcliError> {
	let mut table = Table::new();
	table.load_preset(comfy_table::presets::UTF8_BORDERS_ONLY);

	// Prepare header based on options
	table.set_header(vec![
		if show_g1v1 {
			"SS58 Address/G1v1 public key"
		} else {
			"SS58 Address"
		},
		"Crypto",
		"Path",
		"Name",
	]);

	for account_tree_node in account_tree_nodes {
		let _ = add_account_tree_node_to_table_with_g1v1(&mut table, account_tree_node, show_g1v1);
	}

	Ok(table)
}

fn add_account_tree_node_to_table_with_g1v1(
	table: &mut Table,
	account_tree_node: &Rc<RefCell<AccountTreeNode>>,
	show_g1v1: bool,
) -> Result<(), GcliError> {
	let rows = compute_vault_accounts_row_with_g1v1(account_tree_node, show_g1v1)?;
	rows.iter().for_each(|row| {
		table.add_row(row.clone());
	});

	for child in &account_tree_node.borrow().children {