use clap::*; use is_terminal::IsTerminal; use std::path::PathBuf; use std::str::FromStr; use crate::modes::common::{get_format_box_chars_no_border_line_separator, OutputFormat}; use crate::config; use prettytable::color; use serde_json; use serde_yaml; use prettytable::row; use prettytable::{Attr, Cell, Row, Table}; use prettytable::format::consts::FORMAT_NO_BORDER_LINE_SEPARATOR; use crate::meta_plugin::{MetaPluginType, get_meta_plugin}; use crate::common::status::{generate_status_info, PathInfo, CompressionInfo, MetaPluginInfo}; use crate::compression_engine::CompressionType; fn build_path_table(path_info: &PathInfo) -> Table { let mut path_table = Table::new(); if std::io::stdout().is_terminal() { path_table.set_format(get_format_box_chars_no_border_line_separator()); } else { path_table.set_format(*FORMAT_NO_BORDER_LINE_SEPARATOR); } path_table.set_titles(Row::new(vec![ Cell::new("Type").with_style(Attr::Bold), Cell::new("Path").with_style(Attr::Bold), ])); path_table.add_row(Row::new(vec![ Cell::new("Data"), Cell::new(&path_info.data), ])); path_table.add_row(Row::new(vec![ Cell::new("Database"), Cell::new(&path_info.database), ])); path_table } fn build_compression_table(compression_info: &Vec) -> Table { let mut compression_table = Table::new(); if std::io::stdout().is_terminal() { compression_table.set_format(get_format_box_chars_no_border_line_separator()); } else { compression_table.set_format(*FORMAT_NO_BORDER_LINE_SEPARATOR); } compression_table.set_titles(row!( b->"Type", b->"Found", b->"Enabled", b->"Binary", b->"Compress", b->"Decompress")); for info in compression_info { compression_table.add_row(Row::new(vec![ Cell::new(&info.compression_type), match info.found { true => Cell::new("Yes").with_style(Attr::ForegroundColor(color::GREEN)), false => Cell::new("No").with_style(Attr::ForegroundColor(color::RED)), }, match info.default { true => Cell::new("Yes").with_style(Attr::ForegroundColor(color::GREEN)), false => Cell::new("No"), }, match info.binary.as_str() { "" => Cell::new(&info.binary).with_style(Attr::ForegroundColor(color::BRIGHT_BLACK)), _ => Cell::new(&info.binary), }, Cell::new(&info.compress), Cell::new(&info.decompress), ])); } compression_table } fn build_config_table(settings: &config::Settings) -> Table { let mut config_table = Table::new(); if std::io::stdout().is_terminal() { config_table.set_format(get_format_box_chars_no_border_line_separator()); } else { config_table.set_format(*FORMAT_NO_BORDER_LINE_SEPARATOR); } config_table.set_titles(Row::new(vec![ Cell::new("Setting").with_style(Attr::Bold), Cell::new("Value").with_style(Attr::Bold), ])); // Add relevant configuration settings config_table.add_row(Row::new(vec![ Cell::new("Directory"), Cell::new(&settings.dir.to_string_lossy()), ])); config_table.add_row(Row::new(vec![ Cell::new("Human Readable"), Cell::new(&settings.human_readable.to_string()), ])); config_table.add_row(Row::new(vec![ Cell::new("Quiet"), Cell::new(&settings.quiet.to_string()), ])); if let Some(output_format) = &settings.output_format { config_table.add_row(Row::new(vec![ Cell::new("Output Format"), Cell::new(output_format), ])); } if let Some(compression) = settings.compression() { config_table.add_row(Row::new(vec![ Cell::new("Compression"), Cell::new(&compression), ])); } config_table } fn build_meta_plugins_configured_table(settings: &config::Settings) -> Option { let meta_plugins = settings.meta_plugins.as_ref()?; if meta_plugins.is_empty() { return None; } let mut table = Table::new(); if std::io::stdout().is_terminal() { table.set_format(get_format_box_chars_no_border_line_separator()); } else { table.set_format(*FORMAT_NO_BORDER_LINE_SEPARATOR); } table.set_titles(Row::new(vec![ Cell::new("Plugin Name").with_style(Attr::Bold), Cell::new("Options").with_style(Attr::Bold), Cell::new("Outputs").with_style(Attr::Bold), ])); for plugin_config in meta_plugins { // Create the plugin to get its default options let meta_plugin_type = match MetaPluginType::from_str(&plugin_config.name) { Ok(plugin_type) => plugin_type, Err(_) => continue, }; // First, create a default plugin to get its default options let default_plugin = get_meta_plugin( meta_plugin_type.clone(), None, None, ); // Start with the default options let mut effective_options = default_plugin.options().clone(); // Merge with the configured options for (key, value) in &plugin_config.options { effective_options.insert(key.clone(), value.clone()); } // Convert outputs from HashMap to HashMap let outputs_converted: std::collections::HashMap = plugin_config.outputs .iter() .map(|(k, v)| (k.clone(), serde_yaml::Value::String(v.clone()))) .collect(); // Create the actual plugin with merged options let actual_plugin = get_meta_plugin( meta_plugin_type.clone(), Some(effective_options.clone()), Some(outputs_converted), ); // Get the default plugin to see its default options let default_plugin = get_meta_plugin( meta_plugin_type.clone(), None, None, ); // Start with the default options let mut all_options = default_plugin.options().clone(); // Merge with the configured options for (key, value) in &effective_options { all_options.insert(key.clone(), value.clone()); } // Convert options to a YAML string, but first make sure to show the actual values used // This includes any conversions we made (like "false" string to Bool(false)) let options_str = if all_options.is_empty() { "{}".to_string() } else { serde_yaml::to_string(&all_options) .unwrap_or_else(|_| "Unable to serialize options".to_string()) .trim() .to_string() }; // Show only non-null outputs let mut enabled_output_pairs = Vec::new(); for (key, value) in actual_plugin.outputs() { // Skip null values (disabled outputs) if value.is_null() { continue; } // Convert serde_yaml::Value to a string representation let value_str = match value { serde_yaml::Value::String(s) => s.clone(), serde_yaml::Value::Number(n) => n.to_string(), serde_yaml::Value::Bool(b) => b.to_string(), serde_yaml::Value::Null => "null".to_string(), serde_yaml::Value::Sequence(_) => { serde_yaml::to_string(value).unwrap_or_else(|_| "[]".to_string()) } serde_yaml::Value::Mapping(_) => { serde_yaml::to_string(value).unwrap_or_else(|_| "{}".to_string()) } serde_yaml::Value::Tagged(_) => { serde_yaml::to_string(value).unwrap_or_else(|_| "tagged".to_string()) } }; // Trim any extra whitespace from the serialized values let value_str = value_str.trim().to_string(); if key == &value_str { enabled_output_pairs.push(key.clone()); } else { enabled_output_pairs.push(format!("{}->{}", key, value_str)); } } let outputs_str = if enabled_output_pairs.is_empty() { "{}".to_string() } else { enabled_output_pairs.join(", ") }; table.add_row(Row::new(vec![ Cell::new(&plugin_config.name), Cell::new(&options_str), Cell::new(&outputs_str), ])); } Some(table) } fn build_meta_plugin_table(meta_plugin_info: &Vec) -> Table { let mut meta_plugin_table = Table::new(); if std::io::stdout().is_terminal() { meta_plugin_table.set_format(get_format_box_chars_no_border_line_separator()); } else { meta_plugin_table.set_format(*FORMAT_NO_BORDER_LINE_SEPARATOR); } meta_plugin_table.set_titles(row!( b->"Meta Name", b->"Found", b->"Binary", b->"Args", b->"Outputs")); for info in meta_plugin_info { // Extract just the keys from the outputs let outputs_keys: Vec<&str> = info.outputs.keys().map(|s| s.as_str()).collect(); let outputs_display = if outputs_keys.is_empty() { "".to_string() } else { outputs_keys.join(", ") }; meta_plugin_table.add_row(Row::new(vec![ Cell::new(&info.meta_name), match info.found { true => Cell::new("Yes").with_style(Attr::ForegroundColor(color::GREEN)), false => Cell::new("No").with_style(Attr::ForegroundColor(color::RED)), }, match info.binary.as_str() { "" => Cell::new(&info.binary).with_style(Attr::ForegroundColor(color::BRIGHT_BLACK)), "" => Cell::new(&info.binary).with_style(Attr::ForegroundColor(color::RED)), _ => Cell::new(&info.binary), }, Cell::new(&info.args), Cell::new(&outputs_display), ])); } meta_plugin_table } pub fn mode_status( _cmd: &mut Command, settings: &config::Settings, data_path: PathBuf, db_path: PathBuf, ) -> Result<(), anyhow::Error> { log::debug!("STATUS: Starting mode_status function"); // Determine which meta plugins would be enabled for a save operation log::debug!("STATUS: Getting meta plugin types from settings"); let mut meta_plugin_types: Vec = crate::modes::common::settings_meta_plugin_types(_cmd, settings); log::debug!("STATUS: Got {} meta plugin types", meta_plugin_types.len()); // Always add the Digest plugin if not present if !meta_plugin_types.contains(&MetaPluginType::Digest) { meta_plugin_types.push(MetaPluginType::Digest); } // Determine which compression type would be enabled for a save operation let enabled_compression_type = if let Some(compression_name) = &settings.compression() { CompressionType::from_str(compression_name).ok() } else { Some(crate::compression_engine::default_compression_type()) }; let output_format = crate::modes::common::settings_output_format(settings); log::debug!("STATUS: About to call generate_status_info"); let status_info = generate_status_info(data_path, db_path, &meta_plugin_types, enabled_compression_type); log::debug!("STATUS: generate_status_info completed successfully"); match output_format { OutputFormat::Table => { println!("CONFIG:"); build_config_table(settings).printstd(); println!(); // Print META PLUGINS CONFIGURED if they exist if let Some(meta_plugins_table) = build_meta_plugins_configured_table(settings) { println!("META PLUGINS CONFIGURED:"); meta_plugins_table.printstd(); println!(); } println!("PATHS:"); build_path_table(&status_info.paths).printstd(); println!(); println!("COMPRESSION:"); build_compression_table(&status_info.compression).printstd(); println!(); println!("META PLUGINS AVAILABLE:"); build_meta_plugin_table(&status_info.meta_plugins).printstd(); Ok(()) }, OutputFormat::Json => { // For JSON and YAML, we need to include config info in the status_info // Since we can't modify generate_status_info, we'll need to think of another approach // For now, just print the original status_info println!("{}", serde_json::to_string_pretty(&status_info)?); Ok(()) }, OutputFormat::Yaml => { println!("{}", serde_yaml::to_string(&status_info)?); Ok(()) } } }