use crate::config; use crate::services::types::ItemWithMeta; use crate::modes::common::{format_size, OutputFormat}; use anyhow::{anyhow, Result}; use serde::{Deserialize, Serialize}; use clap::Command; use clap::error::ErrorKind; use std::path::PathBuf; use crate::services::item_service::ItemService; use crate::modes::common::get_format_box_chars_no_border_line_separator; use chrono::prelude::*; use is_terminal::IsTerminal; use prettytable::format; use prettytable::{Attr, Cell, Row, Table}; pub fn mode_info( cmd: &mut Command, settings: &config::Settings, ids: &mut Vec, tags: &mut Vec, conn: &mut rusqlite::Connection, data_path: PathBuf, ) -> Result<()> { // For --info, we can use either IDs or tags, but not both if !ids.is_empty() && !tags.is_empty() { cmd.error(ErrorKind::InvalidValue, "Both ID and tags given, you must supply either IDs or tags when using --info").exit(); } else if ids.len() > 1 { cmd.error(ErrorKind::InvalidValue, "More than one ID given, you must supply exactly one ID when using --info").exit(); } // If both are empty, find_item will find the last item let item_service = ItemService::new(data_path.clone()); // Use empty metadata HashMap let item_with_meta = item_service .find_item(conn, ids, tags, &std::collections::HashMap::new()) .map_err(|e| anyhow!("Unable to find matching item in database: {}", e))?; show_item(item_with_meta, settings, data_path) } #[derive(Debug, Serialize, Deserialize)] struct ItemInfo { id: i64, timestamp: String, path: String, stream_size: Option, stream_size_formatted: String, compression: String, file_size: Option, file_size_formatted: String, tags: Vec, meta: std::collections::HashMap, } fn show_item( item_with_meta: ItemWithMeta, settings: &config::Settings, data_path: PathBuf, ) -> Result<()> { let output_format = crate::modes::common::settings_output_format(settings); if output_format != OutputFormat::Table { return show_item_structured(item_with_meta, settings, data_path, output_format); } let item = item_with_meta.item; let item_id = item.id.unwrap(); let item_tags: Vec = item_with_meta.tags.iter().map(|t| t.name.clone()).collect(); 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::consts::FORMAT_NO_BORDER_LINE_SEPARATOR); } table.add_row(Row::new(vec![ Cell::new("ID").with_style(Attr::Bold), Cell::new(&item_id.to_string()), ])); let ts_cell = Cell::new(&item.ts.with_timezone(&Local).format("%F %T %Z").to_string()); table.add_row(Row::new(vec![ Cell::new("Timestamp").with_style(Attr::Bold), ts_cell, ])); let mut item_path_buf = data_path.clone(); // Renamed to avoid conflict if item_path is used later item_path_buf.push(item.id.unwrap().to_string()); // Again, consider safer unwrap table.add_row(Row::new(vec![ Cell::new("Path").with_style(Attr::Bold), Cell::new(item_path_buf.to_str().expect("Unable to get item path")), ])); let size_cell = match item.size { Some(size) => Cell::new(format_size(size as u64, settings.human_readable).as_str()), None => Cell::new("Missing") .with_style(Attr::ForegroundColor(prettytable::color::RED)) .with_style(Attr::Bold), }; table.add_row(Row::new(vec![ Cell::new("Stream Size").with_style(Attr::Bold), size_cell, ])); table.add_row(Row::new(vec![ Cell::new("Compression").with_style(Attr::Bold), Cell::new(&item.compression), ])); let file_size_cell = match item_path_buf.metadata() { Ok(metadata) => { Cell::new(format_size(metadata.len(), settings.human_readable).as_str()) } Err(_) => Cell::new("Missing") .with_style(Attr::ForegroundColor(prettytable::color::RED)) .with_style(Attr::Bold), }; table.add_row(Row::new(vec![ Cell::new("File Size").with_style(Attr::Bold), file_size_cell, ])); table.add_row(Row::new(vec![ Cell::new("Tags").with_style(Attr::Bold), Cell::new(&item_tags.join(" ")), ])); for meta in item_with_meta.meta { let meta_name = format!("Meta: {}", &meta.name); table.add_row(Row::new(vec![ Cell::new(meta_name.as_str()).with_style(Attr::Bold), Cell::new(&meta.value), ])); } table.printstd(); Ok(()) } fn show_item_structured( item_with_meta: ItemWithMeta, settings: &config::Settings, data_path: PathBuf, output_format: OutputFormat, ) -> Result<()> { let item_tags: Vec = item_with_meta.tags.iter().map(|t| t.name.clone()).collect(); let meta_map = item_with_meta.meta_as_map(); let item = item_with_meta.item; let item_id = item.id.unwrap(); let mut item_path_buf = data_path.clone(); item_path_buf.push(item_id.to_string()); let file_size = item_path_buf.metadata().map(|m| m.len()).ok(); let file_size_formatted = match file_size { Some(size) => format_size(size, settings.human_readable), None => "Missing".to_string(), }; let stream_size_formatted = match item.size { Some(size) => format_size(size as u64, settings.human_readable), None => "Missing".to_string(), }; let item_info = ItemInfo { id: item_id, timestamp: item .ts .with_timezone(&chrono::Local) .format("%F %T %Z") .to_string(), path: item_path_buf.to_str().unwrap_or("").to_string(), stream_size: item.size.map(|s| s as u64), stream_size_formatted, compression: item.compression, file_size, file_size_formatted, tags: item_tags, meta: meta_map, }; match output_format { OutputFormat::Json => { println!("{}", serde_json::to_string_pretty(&item_info)?); } OutputFormat::Yaml => { println!("{}", serde_yaml::to_string(&item_info)?); } OutputFormat::Table => unreachable!(), } Ok(()) }